Skip to content

Commit 5e3eb6e

Browse files
committed
refactor(OpenAI): integrate new LangChain chat init
1 parent d177afb commit 5e3eb6e

File tree

2 files changed

+3
-20
lines changed

2 files changed

+3
-20
lines changed

scrapegraphai/graphs/abstract_graph.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88
from pydantic import BaseModel
99

1010
from langchain_community.chat_models import ChatOllama
11+
from langchain_openai import ChatOpenAI
1112

1213
from langchain_aws import BedrockEmbeddings
1314
from langchain_community.embeddings import HuggingFaceHubEmbeddings, OllamaEmbeddings
@@ -21,7 +22,6 @@
2122
from ..models import (
2223
Anthropic,
2324
AzureOpenAI,
24-
OpenAI,
2525
Bedrock,
2626
Gemini,
2727
Groq,
@@ -37,7 +37,7 @@
3737
from ..utils.logging import set_verbosity_debug, set_verbosity_warning, set_verbosity_info
3838

3939
from ..helpers import models_tokens
40-
from ..models import AzureOpenAI, OpenAI, Bedrock, Gemini, Groq, HuggingFace, Anthropic, DeepSeek
40+
from ..models import AzureOpenAI, Bedrock, Gemini, Groq, HuggingFace, Anthropic, DeepSeek
4141

4242

4343
class AbstractGraph(ABC):
@@ -311,7 +311,7 @@ def _create_default_embedder(self, llm_config=None) -> object:
311311
return GoogleGenerativeAIEmbeddings(
312312
google_api_key=llm_config["api_key"], model="models/embedding-001"
313313
)
314-
if isinstance(self.llm_model, OpenAI):
314+
if isinstance(self.llm_model, ChatOpenAI):
315315
return OpenAIEmbeddings(api_key=self.llm_model.openai_api_key,
316316
base_url=self.llm_model.openai_api_base)
317317
elif isinstance(self.llm_model, DeepSeek):

scrapegraphai/models/openai.py

Lines changed: 0 additions & 17 deletions
This file was deleted.

0 commit comments

Comments
 (0)