fix: azure openai embedding model name error (#612)

This commit is contained in:
John Wang 2023-07-20 13:52:54 +08:00 committed by GitHub
parent cae15013e0
commit e9810a6df2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -18,7 +18,7 @@ class AzureProvider(BaseProvider):
credentials = self.get_credentials('text-embedding-ada-002') if not credentials else credentials credentials = self.get_credentials('text-embedding-ada-002') if not credentials else credentials
try: try:
result = openai.Embedding.create(input=['test'], result = openai.Embedding.create(input=['test'],
engine='text-embedding-ada-0021', engine='text-embedding-ada-002',
timeout=60, timeout=60,
api_key=str(credentials.get('openai_api_key')), api_key=str(credentials.get('openai_api_key')),
api_base=str(credentials.get('openai_api_base')), api_base=str(credentials.get('openai_api_base')),