File tree Expand file tree Collapse file tree 1 file changed +6
-12
lines changed
Expand file tree Collapse file tree 1 file changed +6
-12
lines changed Original file line number Diff line number Diff line change @@ -857,18 +857,12 @@ def get_embedding(
857857 embedding_model = "text-embedding-ada-002"
858858
859859 try :
860- if azure_credential is not None :
861- embeddings_client = EmbeddingsClient (
862- endpoint = inference_endpoint ,
863- credential = azure_credential ,
864- credential_scopes = ["https://cognitiveservices.azure.com/.default" ]
865- )
866- else :
867- api_key = embedding_model_key or os .getenv ("AZURE_OPENAI_API_KEY" )
868- embeddings_client = EmbeddingsClient (
869- endpoint = inference_endpoint ,
870- credential = AzureKeyCredential (api_key )
871- )
860+ credential = azure_credential if azure_credential is not None else AzureCliCredential ()
861+ embeddings_client = EmbeddingsClient (
862+ endpoint = inference_endpoint ,
863+ credential = credential ,
864+ credential_scopes = ["https://cognitiveservices.azure.com/.default" ]
865+ )
872866
873867 response = embeddings_client .embed (model = embedding_model , input = [text ])
874868 return response .data [0 ].embedding
You can’t perform that action at this time.
0 commit comments