import os
from pydantic_ai import Agent
from pydantic_ai.models.openai import OpenAIChatModel
from pydantic_ai.providers.openai import OpenAIProvider
from pydantic_ai.usage import UsageLimits
model = OpenAIChatModel(
"z-ai/glm-5.2",
provider=OpenAIProvider(
base_url="https://api.redpill.ai/v1",
api_key=os.environ["REDPILL_AI_API_KEY"],
),
)
agent = Agent(model, system_prompt="Be concise and state uncertainty.")
@agent.tool_plain
def deployment_status(service: str) -> str:
"""Return a sample deployment status."""
return f"{service}: healthy"
result = agent.run_sync(
"Check the gateway deployment status.",
usage_limits=UsageLimits(request_limit=3, tool_calls_limit=1),
)
print(result.output)