NeoAnd/adk-python
1
0
Fork 0
mirror of https://github.com/EvolutionAPI/adk-python.git synced 2025-07-16 04:02:55 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: failed unit test for Gemini model headers.

Browse Source 
Root cause is that GenAI SDK updated their logic of adding headers in the latest version.

PiperOrigin-RevId: 761751413
This commit is contained in:
Liang Wu 2025-05-21 18:39:41 -07:00 committed by Copybara-Service
parent 4542af5650
commit 09f30afe10
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
tests/unittests/models/test_google_llm.py
View File

@ -15,12 +15,13 @@
import sys import sys
from unittest import mock from unittest import mock
from google.adk import version from google.adk import version as adk_version
from google.adk.models.gemini_llm_connection import GeminiLlmConnection from google.adk.models.gemini_llm_connection import GeminiLlmConnection
from google.adk.models.google_llm import Gemini from google.adk.models.google_llm import Gemini
from google.adk.models.llm_request import LlmRequest from google.adk.models.llm_request import LlmRequest
from google.adk.models.llm_response import LlmResponse from google.adk.models.llm_response import LlmResponse
from google.genai import types from google.genai import types
from google.genai import version as genai_version
from google.genai.types import Content from google.genai.types import Content
from google.genai.types import Part from google.genai.types import Part
import pytest import pytest
@ -73,10 +74,14 @@ def test_supported_models():
def test_client_version_header(): def test_client_version_header():
model = Gemini(model="gemini-1.5-flash") model = Gemini(model="gemini-1.5-flash")
client = model.api_client client = model.api_client
expected_header = ( adk_header = (
f"google-adk/{version.__version__}" f"google-adk/{adk_version.__version__} gl-python/{sys.version.split()[0]}"
f" gl-python/{sys.version.split()[0]} google-genai-sdk/"
) )
genai_header = (
f"google-genai-sdk/{genai_version.__version__} gl-python/{sys.version.split()[0]} "
)
expected_header = genai_header + adk_header
assert ( assert (
expected_header expected_header
in client._api_client._http_options.headers["x-goog-api-client"] in client._api_client._http_options.headers["x-goog-api-client"]