Skip to content

Commit cdab8dc

Browse files
Remove Keybase authentication from get_embedding
1 parent fb47027 commit cdab8dc

File tree

1 file changed

+6
-12
lines changed

1 file changed

+6
-12
lines changed

scripts/data_utils.py

Lines changed: 6 additions & 12 deletions
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)