Skip to content

Commit 1ef6321

Browse files
committed
add deepseek embeddings
1 parent 659fad7 commit 1ef6321

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

scrapegraphai/graphs/abstract_graph.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -231,6 +231,8 @@ def _create_default_embedder(self, llm_config=None) -> object:
231231
model="models/embedding-001")
232232
if isinstance(self.llm_model, OpenAI):
233233
return OpenAIEmbeddings(api_key=self.llm_model.openai_api_key)
234+
elif isinstance(self.llm_model, DeepSeek):
235+
return OpenAIEmbeddings(api_key=self.llm_model.openai_api_key)
234236
elif isinstance(self.llm_model, AzureOpenAIEmbeddings):
235237
return self.llm_model
236238
elif isinstance(self.llm_model, AzureOpenAI):

0 commit comments

Comments
 (0)