Unverified Commit e9810a6d authored by John Wang's avatar John Wang Committed by GitHub

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

parent cae15013
...@@ -18,7 +18,7 @@ class AzureProvider(BaseProvider): ...@@ -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')),
......
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