Skip to content

Commit

Permalink
fix lint
Browse files Browse the repository at this point in the history
  • Loading branch information
ArnoChenFx committed Oct 11, 2024
1 parent f2c502f commit b610438
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions server/graph_service/zep_graphiti.py
Original file line number Diff line number Diff line change
Expand Up @@ -83,8 +83,9 @@ async def get_graphiti(settings: ZepEnvDep):
client.llm_client.config.api_key = settings.openai_api_key
if settings.model_name is not None:
client.llm_client.model = settings.model_name
if settings.embedding_model_name is not None:
if hasattr(client.embedder, 'config') and hasattr(client.embedder.config, 'embedding_model'):
if settings.embedding_model_name is not None and \
hasattr(client.embedder, 'config') and \
hasattr(client.embedder.config, 'embedding_model'):
client.embedder.config.embedding_model = settings.embedding_model_name

try:
Expand Down

0 comments on commit b610438

Please sign in to comment.