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

e2e testing for /gemini pass through

Browse source
This commit is contained in:
Ishaan Jaff 2024-11-22 10:47:55 -08:00
parent d8e5134935
commit d047a9b2c9
1 changed files with 15 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

15
tests/pass_through_tests/test_google_ai_passthrough.py Normal file
View file

@ -0,0 +1,15 @@
import google.generativeai as genai
import pytest
genai.configure(
api_key="sk-1234",
client_options={"api_endpoint": "http://0.0.0.0:4000/gemini"},
transport="rest",
)
def test_basic_non_streaming():
model = genai.GenerativeModel("gemini-1.5-flash")
response = model.generate_content("Explain how AI works")
print("response", response)
assert response.text is not None