Update LLMConfig to include both model and small_model
Both parameters are needed for proper LLM configuration. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
parent
73d2a7c28f
commit
ea01cd7331
3 changed files with 4 additions and 3 deletions
|
|
@ -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,
|
||||
|
|
|
|||
|
|
@ -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,
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue