forked from phoenix/litellm-mirror
15 lines
401 B
Python
15 lines
401 B
Python
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
|