Merge branch 'getzep:main' into main
This commit is contained in:
commit
ae0316ad7b
1 changed files with 2 additions and 2 deletions
|
|
@ -317,7 +317,7 @@ graphiti = Graphiti(
|
||||||
"neo4j",
|
"neo4j",
|
||||||
"password",
|
"password",
|
||||||
llm_client=OpenAIClient(
|
llm_client=OpenAIClient(
|
||||||
llm_config=azure_llm_config,
|
config=azure_llm_config,
|
||||||
client=llm_client_azure
|
client=llm_client_azure
|
||||||
),
|
),
|
||||||
embedder=OpenAIEmbedder(
|
embedder=OpenAIEmbedder(
|
||||||
|
|
@ -327,7 +327,7 @@ graphiti = Graphiti(
|
||||||
client=embedding_client_azure
|
client=embedding_client_azure
|
||||||
),
|
),
|
||||||
cross_encoder=OpenAIRerankerClient(
|
cross_encoder=OpenAIRerankerClient(
|
||||||
llm_config=LLMConfig(
|
config=LLMConfig(
|
||||||
model=azure_llm_config.small_model # Use small model for reranking
|
model=azure_llm_config.small_model # Use small model for reranking
|
||||||
),
|
),
|
||||||
client=llm_client_azure
|
client=llm_client_azure
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue