phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-24 18:24:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(tests): fixing response objects for testing

Browse source
This commit is contained in:
Krrish Dholakia 2023-11-13 14:38:41 -08:00
parent 9776126c8d
commit 8a3b771e50
6 changed files with 188 additions and 104 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/llms/anthropic.py
View file

@ -125,11 +125,18 @@ def completion(
data=json.dumps(data),
stream=optional_params["stream"],
)
if response.status_code != 200:
raise AnthropicError(status_code=response.status_code, message=response.text)
return response.iter_lines()
else:
response = requests.post(
api_base, headers=headers, data=json.dumps(data)
)
if response.status_code != 200:
raise AnthropicError(status_code=response.status_code, message=response.text)
## LOGGING
logging_obj.post_call(
input=prompt,