Unverified Commit 6eee7cb4 authored by John Wang's avatar John Wang Committed by GitHub

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

parent 11baff67
...@@ -44,6 +44,7 @@ class AzureProvider(BaseProvider): ...@@ -44,6 +44,7 @@ class AzureProvider(BaseProvider):
config['openai_api_type'] = 'azure' config['openai_api_type'] = 'azure'
if model_id == 'text-embedding-ada-002': if model_id == 'text-embedding-ada-002':
config['deployment'] = model_id.replace('.', '') if model_id else None config['deployment'] = model_id.replace('.', '') if model_id else None
config['chunk_size'] = 1
else: else:
config['deployment_name'] = model_id.replace('.', '') if model_id else None config['deployment_name'] = model_id.replace('.', '') if model_id else None
return config return config
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment