Skip to content

Commit 5ebc313

Browse files
authored
Allow cognitiveservices endpoint for index building (#33900)
1 parent 503f32e commit 5ebc313

File tree

2 files changed

+7
-9
lines changed

2 files changed

+7
-9
lines changed

sdk/ai/azure-ai-generative/azure/ai/generative/evaluate/_constants.py

Lines changed: 6 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -2,21 +2,19 @@
22
# Copyright (c) Microsoft Corporation. All rights reserved.
33
# ---------------------------------------------------------
44

5-
from azureml.metrics import constants
6-
75
QA = "qa"
86
CHAT = "chat"
97

108
SUPPORTED_TASK_TYPE = [QA, CHAT]
119

1210
SUPPORTED_TO_METRICS_TASK_TYPE_MAPPING = {
13-
QA: constants.QUESTION_ANSWERING,
14-
CHAT: constants.RAG_EVALUATION,
11+
QA: "qa",
12+
CHAT: "rag-evaluation",
1513
}
1614

1715
TYPE_TO_KWARGS_MAPPING = {
18-
constants.QUESTION_ANSWERING: ["questions", "contexts", "y_pred", "y_test"],
19-
constants.RAG_EVALUATION: ["y_pred"]
16+
"qa": ["questions", "contexts", "y_pred", "y_test"],
17+
"rag-evaluation": ["y_pred"]
2018
}
2119

2220

@@ -74,6 +72,6 @@ class ChatMetrics:
7472

7573

7674
TASK_TYPE_TO_METRICS_MAPPING = {
77-
constants.QUESTION_ANSWERING: QaMetrics,
78-
constants.RAG_EVALUATION: ChatMetrics
75+
"qa": QaMetrics,
76+
"rag-evaluation": ChatMetrics
7977
}

sdk/ai/azure-ai-generative/azure/ai/generative/index/_models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ def split_details(details):
4545
config = {**split_details(details), **config}
4646
config["kind"] = "open_ai"
4747
if "endpoint" in config:
48-
if ".openai." in config["endpoint"] or ".api.cognitive." in config["endpoint"]:
48+
if ".openai." in config["endpoint"] or ".api.cognitive." in config["endpoint"] or ".cognitiveservices." in config["endpoint"]:
4949
config["api_base"] = config["endpoint"].rstrip("/")
5050
else:
5151
config["api_base"] = f"https://{config['endpoint']}.openai.azure.com"

0 commit comments

Comments
 (0)