agents icon indicating copy to clipboard operation
agents copied to clipboard

Add retrieval config support for google LLM

Open chenghao-mou opened this issue 3 weeks ago • 0 comments

This should close #4407

Tested with the following setup.

    }
    session = AgentSession(
        ...
        llm=google.LLM(
            model="gemini-2.5-flash",
            vertexai=True,
            location="global",
            # retrieval_config=types.RetrievalConfig(
            #     lat_lng=types.LatLng(latitude=53.350140, longitude=-6.266155)
            # ),
            # or
            retrieval_config={
                "lat_lng": {
                    "latitude": 53.350140,
                    "longitude": -6.266155,
                }
            },
        ),
        ...
        tools=[
            google.tools.GoogleMaps(),
        ],
    )

chenghao-mou avatar Dec 29 '25 16:12 chenghao-mou