File tree Expand file tree Collapse file tree 1 file changed +4
-3
lines changed
sdk/evaluation/azure-ai-evaluation/azure/ai/evaluation/simulator/_model_tools Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Original file line number Diff line number Diff line change @@ -208,7 +208,7 @@ async def request_api(
208
208
flag = True
209
209
while flag :
210
210
try :
211
- response = session .evaluations .operation_results (operation_id , headers = headers )
211
+ response = session .red_teams .operation_results (operation_id , headers = headers )
212
212
except Exception as e :
213
213
from types import SimpleNamespace # pylint: disable=forgotten-debug-statement
214
214
@@ -217,9 +217,10 @@ async def request_api(
217
217
response_data = response
218
218
flag = False
219
219
break
220
- if response . status_code == 200 :
221
- response_data = cast ( List [ Dict ], response . json ())
220
+ if not isinstance ( response , SimpleNamespace ) and response . get ( "object" ) == "chat.completion" :
221
+ response_data = response
222
222
flag = False
223
+ break
223
224
else :
224
225
request_count += 1
225
226
sleep_time = RAIService .SLEEP_TIME ** request_count
You can’t perform that action at this time.
0 commit comments