openrouter
This commit is contained in:
@@ -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,
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user