Skip to content

GeminiClient

autogen.beta.config.gemini.gemini_client.GeminiClient #

GeminiClient(model, api_key=None, vertexai=None, credentials=None, project=None, location=None, streaming=False, create_config=None, cached_content=None)

Bases: LLMClient

Source code in autogen/beta/config/gemini/gemini_client.py 
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
def __init__(
    self,
    model: str,
    api_key: str | None = None,
    vertexai: bool | None = None,
    credentials: google.auth.credentials.Credentials | str | None = None,
    project: str | None = None,
    location: str | None = None,
    streaming: bool = False,
    create_config: CreateConfig | None = None,
    cached_content: str | None = None,
) -> None:
    if isinstance(credentials, str):
        # String indicates a json credentials file, load into credentials
        credentials = service_account.Credentials.from_service_account_file(
            credentials,
            scopes=["https://www.googleapis.com/auth/cloud-platform"],
        )
    self._client = genai.Client(
        vertexai=vertexai, api_key=api_key, credentials=credentials, project=project, location=location
    )
    self._model_name = model
    self._streaming = streaming
    self._create_config = create_config or {}
    self._cached_content = cached_content