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

add mock test for ai21

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-27 14:42:13 -07:00
parent bae54ca642
commit 2436dfbbc0
2 changed files with 112 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/llms/databricks.py
View file

@ -363,6 +363,12 @@ class DatabricksChatCompletion(BaseLLM):
except Exception as e:
raise DatabricksError(status_code=500, message=str(e))
logging_obj.post_call(
input=messages,
api_key="",
original_response=response_json,
additional_args={"complete_input_dict": data},
)
response = ModelResponse(**response_json)
if base_model is not None: