gmakstutis/graphiti
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

#491 (#493)

Browse source 
* #491 Added small_model=

* #491 Renamed OpenAIEmbedderConfig parameter `model` to `embedding_model`
This commit is contained in:
Adam Katav 2025-05-20 18:03:22 +03:00 • committed by GitHub
parent 4babd5cc4f
commit 576faa5e7c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
mcp_server/graphiti_mcp_server.py
View file

@ -301,7 +301,7 @@ class GraphitiLLMConfig(BaseModel):
if not self.api_key:
return None
llm_client_config = LLMConfig(api_key=self.api_key, model=self.model)
llm_client_config = LLMConfig(api_key=self.api_key, model=self.model, small_model=self.model)
# Set temperature
llm_client_config.temperature = self.temperature
@ -408,7 +408,7 @@ class GraphitiEmbedderConfig(BaseModel):
if not self.api_key:
return None
embedder_config = OpenAIEmbedderConfig(api_key=self.api_key, model=self.model)
embedder_config = OpenAIEmbedderConfig(api_key=self.api_key, embedding_model=self.model)
return OpenAIEmbedder(config=embedder_config)