Skip to content

Commit f81442b

Browse files
committed
removed unused if
1 parent dde0c7e commit f81442b

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

scrapegraphai/graphs/abstract_graph.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,8 @@ def __init__(self, prompt: str, config: dict,
6969
self.config = config
7070
self.schema = schema
7171
self.llm_model = self._create_llm(config["llm"], chat=True)
72-
self.embedder_model = self._create_default_embedder(llm_config=config["llm"] ) if "embeddings" not in config else self._create_embedder(
72+
self.embedder_model = self._create_default_embedder(llm_config=config["llm"]
73+
) if "embeddings" not in config else self._create_embedder(
7374
config["embeddings"])
7475
self.verbose = False if config is None else config.get(
7576
"verbose", False)
@@ -101,7 +102,6 @@ def __init__(self, prompt: str, config: dict,
101102
"llm_model": self.llm_model,
102103
"embedder_model": self.embedder_model
103104
}
104-
105105
self.set_common_params(common_params, overwrite=False)
106106

107107
# set burr config
@@ -291,8 +291,6 @@ def _create_default_embedder(self, llm_config=None) -> object:
291291
)
292292
if isinstance(self.llm_model, OpenAI):
293293
return OpenAIEmbeddings(api_key=self.llm_model.openai_api_key)
294-
elif isinstance(self.llm_model, DeepSeek):
295-
return OpenAIEmbeddings(api_key=self.llm_model.openai_api_key)
296294
elif isinstance(self.llm_model, AzureOpenAIEmbeddings):
297295
return self.llm_model
298296
elif isinstance(self.llm_model, AzureOpenAI):

0 commit comments

Comments
 (0)