fix(nodes): update outdated ChatOllama import path to langchain_ollama

scrapegraphai/nodes/generate_answer_node.py

@@ -8,7 +8,7 @@
 
 from langchain_core.prompts import PromptTemplate
 from langchain_aws import ChatBedrock
-from langchain_community.chat_models import ChatOllama
+from langchain_ollama import ChatOllama
 from langchain_core.output_parsers import JsonOutputParser
 from langchain_core.runnables import RunnableParallel
 from langchain_openai import ChatOpenAI

scrapegraphai/nodes/generate_answer_node_k_level.py

@@ -5,7 +5,7 @@
 from typing import List, Optional
 
 from langchain_aws import ChatBedrock
-from langchain_community.chat_models import ChatOllama
+from langchain_ollama import ChatOllama
 from langchain_core.output_parsers import JsonOutputParser
 from langchain_core.prompts import PromptTemplate
 from langchain_core.runnables import RunnableParallel

scrapegraphai/nodes/generate_answer_omni_node.py

@@ -5,7 +5,7 @@
 from typing import List, Optional
 
 from langchain_core.prompts import PromptTemplate
-from langchain_community.chat_models import ChatOllama
+from langchain_ollama import ChatOllama
 from langchain_core.output_parsers import JsonOutputParser
 from langchain_core.runnables import RunnableParallel
 from langchain_mistralai import ChatMistralAI

scrapegraphai/nodes/generate_code_node.py

@@ -13,7 +13,7 @@
 from jsonschema import ValidationError as JSONSchemaValidationError
 from jsonschema import validate
 from langchain_classic.output_parsers import ResponseSchema, StructuredOutputParser
-from langchain_community.chat_models import ChatOllama
+from langchain_ollama import ChatOllama
 from langchain_core.output_parsers import StrOutputParser
 from langchain_core.prompts import PromptTemplate
 

scrapegraphai/nodes/html_analyzer_node.py

@@ -5,7 +5,7 @@
 from typing import List, Optional
 
 from langchain_core.prompts import PromptTemplate
-from langchain_community.chat_models import ChatOllama
+from langchain_ollama import ChatOllama
 from langchain_core.output_parsers import StrOutputParser
 
 from ..prompts import TEMPLATE_HTML_ANALYSIS, TEMPLATE_HTML_ANALYSIS_WITH_CONTEXT

scrapegraphai/nodes/merge_answers_node.py

@@ -5,7 +5,7 @@
 from typing import List, Optional
 
 from langchain_core.prompts import PromptTemplate
-from langchain_community.chat_models import ChatOllama
+from langchain_ollama import ChatOllama
 from langchain_core.output_parsers import JsonOutputParser
 from langchain_mistralai import ChatMistralAI
 from langchain_openai import ChatOpenAI

scrapegraphai/nodes/prompt_refiner_node.py

@@ -5,7 +5,7 @@
 from typing import List, Optional
 
 from langchain_core.prompts import PromptTemplate
-from langchain_community.chat_models import ChatOllama
+from langchain_ollama import ChatOllama
 from langchain_core.output_parsers import StrOutputParser
 
 from ..prompts import TEMPLATE_REFINER, TEMPLATE_REFINER_WITH_CONTEXT

scrapegraphai/nodes/reasoning_node.py

@@ -5,7 +5,7 @@
 from typing import List, Optional
 
 from langchain_core.prompts import PromptTemplate
-from langchain_community.chat_models import ChatOllama
+from langchain_ollama import ChatOllama
 from langchain_core.output_parsers import StrOutputParser
 
 from ..prompts import TEMPLATE_REASONING, TEMPLATE_REASONING_WITH_CONTEXT

scrapegraphai/nodes/search_internet_node.py

@@ -6,7 +6,7 @@
 
 from langchain_core.output_parsers import CommaSeparatedListOutputParser
 from langchain_core.prompts import PromptTemplate
-from langchain_community.chat_models import ChatOllama
+from langchain_ollama import ChatOllama
 
 from ..prompts import TEMPLATE_SEARCH_INTERNET
 from ..utils.research_web import search_on_web

tests/nodes/search_internet_node_test.py

@@ -1,6 +1,6 @@
 import unittest
 
-from langchain_community.chat_models import ChatOllama
+from langchain_ollama import ChatOllama
 
 from scrapegraphai.nodes import SearchInternetNode
 

tests/nodes/search_link_node_test.py

@@ -1,7 +1,7 @@
 from unittest.mock import patch
 
 import pytest
-from langchain_community.chat_models import ChatOllama
+from langchain_ollama import ChatOllama
 
 from scrapegraphai.nodes import SearchLinkNode