diff --git a/.env.example b/.env.example index 469903b..c6a19af 100644 --- a/.env.example +++ b/.env.example @@ -1,7 +1,17 @@ +# Google Gemini (native API with thinking support) +# Usage: agent.config.model=google-gla:gemini-3-pro-preview +GEMINI_API_KEY="CHANGEME" +# or use GOOGLE_API_KEY (either works) +# GOOGLE_API_KEY="CHANGEME" + +# Azure OpenAI +# Usage: agent.config.model=azure:gpt-4o AZURE_OPENAI_ENDPOINT="https://ENDPOINT.azure-api.net" AZURE_OPENAI_API_KEY="CHANGEME" OPENAI_API_VERSION="2025-04-01-preview" +# AWS Bedrock +# Usage: agent.config.model=bedrock:anthropic.claude-3-5-sonnet-20241022-v2:0 AWS_ACCESS_KEY_ID="CHANGEME" AWS_SECRET_ACCESS_KEY="CHANGEME" AWS_SESSION_TOKEN="CHANGEME" @@ -9,5 +19,9 @@ EXPIRY_TIME="CHANGEME" SESSION_DURATION_SECONDS=43199 AWS_DEFAULT_REGION='us-west-2' +# Llama API +# Usage: agent.config.model=llama:llama3.3-70b +LLAMA_API_KEY="CHANGEME" + OPENAI_API_KEY="CHANGEME" ANTHROPIC_API_KEY="CHANGEME" diff --git a/pyproject.toml b/pyproject.toml index dbefbb8..e02a471 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -21,7 +21,7 @@ dependencies = [ "hydra-submitit-launcher>=1.2.0", "logfire>=4.7.0", "pandas>=2.0.0", - "pydantic-ai-slim[anthropic,bedrock,openai]>=1.1.0", + "pydantic-ai-slim[anthropic,bedrock,google,openai]>=1.1.0", "pyyaml>=6.0.2", "opentelemetry-api>=1.20.0", "opentelemetry-sdk>=1.20.0", diff --git a/uv.lock b/uv.lock index b2b3632..cbfba0f 100644 --- a/uv.lock +++ b/uv.lock @@ -2700,7 +2700,7 @@ dependencies = [ { name = "opentelemetry-exporter-otlp" }, { name = "opentelemetry-sdk" }, { name = "pandas" }, - { name = "pydantic-ai-slim", extra = ["anthropic", "bedrock", "openai"] }, + { name = "pydantic-ai-slim", extra = ["anthropic", "bedrock", "google", "openai"] }, { name = "pyyaml" }, { name = "tabulate" }, { name = "typing-extensions" }, @@ -2767,7 +2767,7 @@ requires-dist = [ { name = "pandas", specifier = ">=2.0.0" }, { name = "playwright", marker = "extra == 'playwright'", specifier = ">=1.54.0" }, { name = "prompt-siren", extras = ["agentdojo", "swebench", "docker", "playwright"], marker = "extra == 'all'" }, - { name = "pydantic-ai-slim", extras = ["anthropic", "bedrock", "openai"], specifier = ">=1.1.0" }, + { name = "pydantic-ai-slim", extras = ["anthropic", "bedrock", "google", "openai"], specifier = ">=1.1.0" }, { name = "pyyaml", specifier = ">=6.0.2" }, { name = "ruff-api", marker = "extra == 'ufmt'", specifier = "==0.1.0" }, { name = "stdlibs", marker = "extra == 'ufmt'", specifier = "==2024.1.28" }, @@ -3027,6 +3027,9 @@ anthropic = [ bedrock = [ { name = "boto3" }, ] +google = [ + { name = "google-genai" }, +] openai = [ { name = "openai" }, ]