From ea01cd733172fa3e29b335875d54721aa6c47df9 Mon Sep 17 00:00:00 2001 From: Daniel Chalef <131175+danielchalef@users.noreply.github.com> Date: Thu, 13 Nov 2025 22:03:33 -0800 Subject: [PATCH] Update LLMConfig to include both model and small_model MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Both parameters are needed for proper LLM configuration. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude --- README.md | 2 +- examples/azure-openai/README.md | 2 +- examples/azure-openai/azure_openai_neo4j.py | 3 ++- 3 files changed, 4 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index a42e951b..1efd37df 100644 --- a/README.md +++ b/README.md @@ -409,7 +409,7 @@ azure_client = AsyncOpenAI( # Create LLM and Embedder clients llm_client = AzureOpenAILLMClient( azure_client=azure_client, - config=LLMConfig(small_model="gpt-5-mini") # Your Azure deployment name + config=LLMConfig(model="gpt-5-mini", small_model="gpt-5-mini") # Your Azure deployment name ) embedder_client = AzureOpenAIEmbedderClient( azure_client=azure_client, diff --git a/examples/azure-openai/README.md b/examples/azure-openai/README.md index b83c5e1e..28576967 100644 --- a/examples/azure-openai/README.md +++ b/examples/azure-openai/README.md @@ -92,7 +92,7 @@ azure_client = AsyncOpenAI( # Create LLM and Embedder clients llm_client = AzureOpenAILLMClient( azure_client=azure_client, - config=LLMConfig(small_model=azure_deployment) + config=LLMConfig(model=azure_deployment, small_model=azure_deployment) ) embedder_client = AzureOpenAIEmbedderClient( azure_client=azure_client, diff --git a/examples/azure-openai/azure_openai_neo4j.py b/examples/azure-openai/azure_openai_neo4j.py index e3e35b58..f8965a53 100644 --- a/examples/azure-openai/azure_openai_neo4j.py +++ b/examples/azure-openai/azure_openai_neo4j.py @@ -85,7 +85,8 @@ async def main(): # Create LLM and Embedder clients llm_client = AzureOpenAILLMClient( - azure_client=azure_client, config=LLMConfig(small_model=azure_deployment) + azure_client=azure_client, + config=LLMConfig(model=azure_deployment, small_model=azure_deployment), ) embedder_client = AzureOpenAIEmbedderClient( azure_client=azure_client, model=azure_embedding_deployment