ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(azure.py): fix raw response dump

Browse Source
This commit is contained in:
Krrish Dholakia 2024-08-27 13:44:38 -07:00
parent 18731cf42b
commit 63adb3f940
2 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
litellm/llms/azure.py
View File

@ -1130,7 +1130,8 @@ class AzureChatCompletion(BaseLLM):
else:
azure_client = client
## COMPLETION CALL
response = azure_client.embeddings.with_raw_response.create(**data, timeout=timeout) # type: ignore
raw_response = azure_client.embeddings.with_raw_response.create(**data, timeout=timeout) # type: ignore
response = raw_response.parse()
## LOGGING
logging_obj.post_call(
input=input,

7
litellm/llms/openai.py
View File

@ -2126,7 +2126,8 @@ class OpenAITextCompletion(BaseLLM):
openai_client = client
try:
response = openai_client.completions.with_raw_response.create(**data)
raw_response = openai_client.completions.with_raw_response.create(**data)
response = raw_response.parse()
except Exception as e:
status_code = getattr(e, "status_code", 500)
error_headers = getattr(e, "headers", None)
@ -2170,8 +2171,8 @@ class OpenAITextCompletion(BaseLLM):
else:
openai_client = client
response = await openai_client.completions.with_raw_response.create(**data)
raw_response = await openai_client.completions.with_raw_response.create(**data)
response = raw_response.parse()
streamwrapper = CustomStreamWrapper(
completion_stream=response,
model=model,