CodeCow/Scrapegraph-ai
1
0
Fork 0
mirror of https://github.com/VinciGit00/Scrapegraph-ai.git synced 2026-06-28 21:01:55 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix(Ollama): instance model from correct package

Browse Source
This commit is contained in:
Federico Aguzzi 2024-09-02 12:28:36 +02:00
parent 74dfc693f6
commit 398b2c556f
5 changed files with 20 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pyproject.toml
View File

@ -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"

9
requirements-dev.lock
View File

@ -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

9
requirements.lock
View File

@ -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

1
requirements.txt
View File

@ -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

3
tests/graphs/abstract_graph_test.py
View File

@ -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
}
)