openrouter

This commit is contained in:
2025-12-16 06:56:46 +01:00
parent 7578c9aab0
commit 6e7d071dea
4 changed files with 7 additions and 330 deletions

View File

@@ -7,8 +7,6 @@ import logging
import httpx
from pydantic_ai import Agent
from pydantic_ai.models.google import GoogleModel
from pydantic_ai.providers.google import GoogleProvider
# Set up logging based on environment
log_level = logging.DEBUG if os.getenv("DEBUG") else logging.INFO
@@ -76,17 +74,8 @@ def create_agent(user_roles: list[str] | None = None):
else:
logger.info("MCP server unavailable - running without MCP tools")
# Use Vertex AI for higher rate limits (requires GOOGLE_APPLICATION_CREDENTIALS)
# Note: gemini-3-pro-preview requires location="global"
provider = GoogleProvider(
vertexai=True,
project=os.getenv("GOOGLE_PROJECT_ID"),
location=os.getenv("GOOGLE_LOCATION", "global"),
)
model = GoogleModel("gemini-3-pro-preview", provider=provider)
return Agent(
model=model,
model="openrouter:google/gemini-3-pro-preview",
toolsets=toolsets if toolsets else None,
instructions=AGENT_INSTRUCTIONS,
)