From d047a9b2c901b3dec5f322f777c425c54c1bbd59 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Fri, 22 Nov 2024 10:47:55 -0800 Subject: [PATCH] e2e testing for /gemini pass through --- .../test_google_ai_passthrough.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) create mode 100644 tests/pass_through_tests/test_google_ai_passthrough.py diff --git a/tests/pass_through_tests/test_google_ai_passthrough.py b/tests/pass_through_tests/test_google_ai_passthrough.py new file mode 100644 index 000000000..b8d115f21 --- /dev/null +++ b/tests/pass_through_tests/test_google_ai_passthrough.py @@ -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