Преглед на файлове

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

John Wang преди 1 година
родител
ревизия
6eee7cb42c
променени са 1 файла, в които са добавени 1 реда и са изтрити 0 реда
  1. 1 0
      api/core/llm/provider/azure_provider.py

+ 1 - 0
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