feat: fix azure embedding Too many inputs problem (#530)

6eee7cb4 · John Wang · GitHub · 11baff67 · 6eee7cb4
Unverified Commit 6eee7cb4 authored Jul 07, 2023 by John Wang Committed by GitHub Jul 07, 2023
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

azure_provider.py api/core/llm/provider/azure_provider.py +1 -0

No files found.
--- a/api/core/llm/provider/azure_provider.py
+++ b/api/core/llm/provider/azure_provider.py
@@ -44,6 +44,7 @@ class AzureProvider(BaseProvider):
        config['openai_api_type'] = 'azure'
        if model_id == 'text-embedding-ada-002':
            config['deployment'] = model_id.replace('.', '') if model_id else None
+            config['chunk_size'] = 1
        else:
            config['deployment_name'] = model_id.replace('.', '') if model_id else None
        return config