search

BerriAI / litellm

Python SDK, Proxy Server to call 100+ LLM APIs using the OpenAI format - [Bedrock, Azure, OpenAI, VertexAI, Cohere, Anthropic, Sagemaker, HuggingFace, Replicate, Groq]
https://docs.litellm.ai/docs/
Other
12.2k stars 1.42k forks source link

[Bug]: Have mock responses return a pydantic object #938

Open krrishdholakia opened 9 months ago

krrishdholakia commented 9 months ago

What happened?

Mock responses returns a dict (old openai format).

Update it to return a pydantic object.

Relevant log output

N/A

Twitter / LinkedIn details

No response

krrishdholakia commented 9 months ago

Relevant code - https://github.com/BerriAI/litellm/blob/main/litellm/main.py#L208