phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix(utils.py): fix fix pydantic obj to schema creation for vertex en… (#6071)

Browse source 
* fix(utils.py): fix  fix pydantic obj to schema creation for vertex endpoints

Fixes https://github.com/BerriAI/litellm/issues/6027

* test(test_completion.pyu): skip test - avoid hitting gemini rate limits

* fix(common_utils.py): fix ruff linting error
This commit is contained in:
Krish Dholakia 2024-10-06 00:25:55 -04:00 • committed by GitHub
parent 29da2d49d6
commit 49d8b2be46
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 143 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

25
tests/local_testing/test_completion.py
View file

@ -1711,6 +1711,31 @@ def test_completion_perplexity_api():
# test_completion_perplexity_api()
@pytest.mark.skip(
reason="too many requests. Hitting gemini rate limits. Convert to mock test."
)
def test_completion_pydantic_obj_2():
from pydantic import BaseModel
litellm.set_verbose = True
class CalendarEvent(BaseModel):
name: str
date: str
participants: list[str]
class EventsList(BaseModel):
events: list[CalendarEvent]
messages = [
{"role": "user", "content": "List important events from the 20th century."}
]
response = litellm.completion(
model="gemini/gemini-1.5-pro", messages=messages, response_format=EventsList
)
@pytest.mark.skip(reason="this test is flaky")
def test_completion_perplexity_api_2():
try: