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

2023-07-07 11:17:36 +08:00
parent 11baff6740
commit 6eee7cb42c
1 changed files with 1 additions and 0 deletions
--- 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