Skip to content

Commit e6054cb

Browse files
handsome-bondbond
andauthored
fix(nodes): update outdated ChatOllama import path to langchain_ollama (#1076)
Co-authored-by: bond <DELL@DESKTOP-KF74FPE>
1 parent 458d36a commit e6054cb

11 files changed

Lines changed: 11 additions & 11 deletions

scrapegraphai/nodes/generate_answer_node.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88

99
from langchain_core.prompts import PromptTemplate
1010
from langchain_aws import ChatBedrock
11-
from langchain_community.chat_models import ChatOllama
11+
from langchain_ollama import ChatOllama
1212
from langchain_core.output_parsers import JsonOutputParser
1313
from langchain_core.runnables import RunnableParallel
1414
from langchain_openai import ChatOpenAI

scrapegraphai/nodes/generate_answer_node_k_level.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
from typing import List, Optional
66

77
from langchain_aws import ChatBedrock
8-
from langchain_community.chat_models import ChatOllama
8+
from langchain_ollama import ChatOllama
99
from langchain_core.output_parsers import JsonOutputParser
1010
from langchain_core.prompts import PromptTemplate
1111
from langchain_core.runnables import RunnableParallel

scrapegraphai/nodes/generate_answer_omni_node.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
from typing import List, Optional
66

77
from langchain_core.prompts import PromptTemplate
8-
from langchain_community.chat_models import ChatOllama
8+
from langchain_ollama import ChatOllama
99
from langchain_core.output_parsers import JsonOutputParser
1010
from langchain_core.runnables import RunnableParallel
1111
from langchain_mistralai import ChatMistralAI

scrapegraphai/nodes/generate_code_node.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
from jsonschema import ValidationError as JSONSchemaValidationError
1414
from jsonschema import validate
1515
from langchain_classic.output_parsers import ResponseSchema, StructuredOutputParser
16-
from langchain_community.chat_models import ChatOllama
16+
from langchain_ollama import ChatOllama
1717
from langchain_core.output_parsers import StrOutputParser
1818
from langchain_core.prompts import PromptTemplate
1919

scrapegraphai/nodes/html_analyzer_node.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
from typing import List, Optional
66

77
from langchain_core.prompts import PromptTemplate
8-
from langchain_community.chat_models import ChatOllama
8+
from langchain_ollama import ChatOllama
99
from langchain_core.output_parsers import StrOutputParser
1010

1111
from ..prompts import TEMPLATE_HTML_ANALYSIS, TEMPLATE_HTML_ANALYSIS_WITH_CONTEXT

scrapegraphai/nodes/merge_answers_node.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
from typing import List, Optional
66

77
from langchain_core.prompts import PromptTemplate
8-
from langchain_community.chat_models import ChatOllama
8+
from langchain_ollama import ChatOllama
99
from langchain_core.output_parsers import JsonOutputParser
1010
from langchain_mistralai import ChatMistralAI
1111
from langchain_openai import ChatOpenAI

scrapegraphai/nodes/prompt_refiner_node.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
from typing import List, Optional
66

77
from langchain_core.prompts import PromptTemplate
8-
from langchain_community.chat_models import ChatOllama
8+
from langchain_ollama import ChatOllama
99
from langchain_core.output_parsers import StrOutputParser
1010

1111
from ..prompts import TEMPLATE_REFINER, TEMPLATE_REFINER_WITH_CONTEXT

scrapegraphai/nodes/reasoning_node.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
from typing import List, Optional
66

77
from langchain_core.prompts import PromptTemplate
8-
from langchain_community.chat_models import ChatOllama
8+
from langchain_ollama import ChatOllama
99
from langchain_core.output_parsers import StrOutputParser
1010

1111
from ..prompts import TEMPLATE_REASONING, TEMPLATE_REASONING_WITH_CONTEXT

scrapegraphai/nodes/search_internet_node.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
from langchain_core.output_parsers import CommaSeparatedListOutputParser
88
from langchain_core.prompts import PromptTemplate
9-
from langchain_community.chat_models import ChatOllama
9+
from langchain_ollama import ChatOllama
1010

1111
from ..prompts import TEMPLATE_SEARCH_INTERNET
1212
from ..utils.research_web import search_on_web

tests/nodes/search_internet_node_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import unittest
22

3-
from langchain_community.chat_models import ChatOllama
3+
from langchain_ollama import ChatOllama
44

55
from scrapegraphai.nodes import SearchInternetNode
66

0 commit comments

Comments
 (0)