Skip to content

Commit 9e795f4

Browse files
VinciGit00f-aguzzi
andcommitted
removed unused init files
Co-Authored-By: Federico Aguzzi <62149513+f-aguzzi@users.noreply.github.com>
1 parent 5e3eb6e commit 9e795f4

File tree

4 files changed

+3
-6
lines changed

4 files changed

+3
-6
lines changed

requirements-dev.lock

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -180,7 +180,6 @@ graphviz==0.20.3
180180
# via scrapegraphai
181181
greenlet==3.0.3
182182
# via playwright
183-
# via sqlalchemy
184183
groq==0.9.0
185184
# via langchain-groq
186185
grpc-google-iam-v1==0.13.1

requirements.lock

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,6 @@ graphviz==0.20.3
128128
# via scrapegraphai
129129
greenlet==3.0.3
130130
# via playwright
131-
# via sqlalchemy
132131
groq==0.9.0
133132
# via langchain-groq
134133
grpc-google-iam-v1==0.13.1

scrapegraphai/graphs/abstract_graph.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -163,9 +163,10 @@ def _create_llm(self, llm_config: dict, chat=False) -> object:
163163
if "gpt-" in llm_params["model"]:
164164
try:
165165
self.model_token = models_tokens["openai"][llm_params["model"]]
166+
llm_params["model_provider"] = "openai"
166167
except KeyError as exc:
167168
raise KeyError("Model not supported") from exc
168-
return OpenAI(llm_params)
169+
return init_chat_model(**llm_params)
169170
elif "oneapi" in llm_params["model"]:
170171
# take the model after the last dash
171172
llm_params["model"] = llm_params["model"].split("/")[-1]
@@ -455,4 +456,4 @@ def run(self) -> str:
455456
"""
456457
Abstract method to execute the graph and return the result.
457458
"""
458-
pass
459+
pass

scrapegraphai/models/__init__.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,12 +2,10 @@
22
__init__.py file for models folder
33
"""
44

5-
from .openai import OpenAI
65
from .azure_openai import AzureOpenAI
76
from .openai_itt import OpenAIImageToText
87
from .openai_tts import OpenAITextToSpeech
98
from .gemini import Gemini
10-
from .ollama import Ollama
119
from .hugging_face import HuggingFace
1210
from .groq import Groq
1311
from .bedrock import Bedrock

0 commit comments

Comments
 (0)