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

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')),