diff --git a/pyproject.toml b/pyproject.toml index 47720cbf..197ecefe 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -30,6 +30,7 @@ dependencies = [ "undetected-playwright>=0.3.0", "google>=3.0.0", "semchunk>=1.0.1", + "langchain-ollama>=0.1.3", ] license = "MIT" diff --git a/requirements-dev.lock b/requirements-dev.lock index b816db3d..eb875abb 100644 --- a/requirements-dev.lock +++ b/requirements-dev.lock @@ -130,6 +130,7 @@ graphviz==0.20.3 # via burr greenlet==3.0.3 # via playwright + # via sqlalchemy grpcio==1.65.4 # via google-api-core # via grpcio-status @@ -147,6 +148,7 @@ httplib2==0.22.0 # via google-auth-httplib2 httpx==0.27.0 # via langchain-mistralai + # via ollama # via openai httpx-sse==0.4.0 # via langchain-mistralai @@ -194,18 +196,21 @@ langchain-aws==0.1.16 # via scrapegraphai langchain-community==0.2.11 # via scrapegraphai -langchain-core==0.2.33 +langchain-core==0.2.37 # via langchain # via langchain-aws # via langchain-community # via langchain-google-genai # via langchain-mistralai + # via langchain-ollama # via langchain-openai # via langchain-text-splitters langchain-google-genai==1.0.8 # via scrapegraphai langchain-mistralai==0.1.12 # via scrapegraphai +langchain-ollama==0.1.3 + # via scrapegraphai langchain-openai==0.1.22 # via scrapegraphai langchain-text-splitters==0.2.2 @@ -255,6 +260,8 @@ numpy==1.26.4 # via pydeck # via sf-hamilton # via streamlit +ollama==0.3.2 + # via langchain-ollama openai==1.40.3 # via burr # via langchain-openai diff --git a/requirements.lock b/requirements.lock index 30d89366..d639583d 100644 --- a/requirements.lock +++ b/requirements.lock @@ -83,6 +83,7 @@ googleapis-common-protos==1.63.2 # via grpcio-status greenlet==3.0.3 # via playwright + # via sqlalchemy grpcio==1.65.1 # via google-api-core # via grpcio-status @@ -99,6 +100,7 @@ httplib2==0.22.0 # via google-auth-httplib2 httpx==0.27.0 # via langchain-mistralai + # via ollama # via openai httpx-sse==0.4.0 # via langchain-mistralai @@ -125,18 +127,21 @@ langchain-aws==0.1.12 # via scrapegraphai langchain-community==0.2.10 # via scrapegraphai -langchain-core==0.2.33 +langchain-core==0.2.37 # via langchain # via langchain-aws # via langchain-community # via langchain-google-genai # via langchain-mistralai + # via langchain-ollama # via langchain-openai # via langchain-text-splitters langchain-google-genai==1.0.8 # via scrapegraphai langchain-mistralai==0.1.12 # via scrapegraphai +langchain-ollama==0.1.3 + # via scrapegraphai langchain-openai==0.1.22 # via scrapegraphai langchain-text-splitters==0.2.2 @@ -166,6 +171,8 @@ numpy==1.26.4 # via langchain-aws # via langchain-community # via pandas +ollama==0.3.2 + # via langchain-ollama openai==1.41.0 # via langchain-openai orjson==3.10.6 diff --git a/requirements.txt b/requirements.txt index 80cb0767..8a29f1c8 100644 --- a/requirements.txt +++ b/requirements.txt @@ -17,3 +17,4 @@ playwright>=1.43.0 undetected-playwright>=0.3.0 google>=3.0.0 semchunk>=1.0.1 +langchain-ollama>=0.1.3 diff --git a/tests/graphs/abstract_graph_test.py b/tests/graphs/abstract_graph_test.py index 60c8ab4c..62a9c583 100644 --- a/tests/graphs/abstract_graph_test.py +++ b/tests/graphs/abstract_graph_test.py @@ -10,7 +10,7 @@ from scrapegraphai.nodes import ( ) from scrapegraphai.models import OneApi, DeepSeek from langchain_openai import ChatOpenAI, AzureChatOpenAI -from langchain_community.chat_models import ChatOllama +from langchain_ollama import ChatOllama from langchain_google_genai import ChatGoogleGenerativeAI @@ -35,6 +35,7 @@ class TestGraph(AbstractGraph): input="doc", output=["parsed_doc"], node_config={ + "llm_model": self.llm_model, "chunk_size": self.model_token } )