Anthropic
安装
要使用 AnthropicModel
模型,您需要安装 pydantic-ai
,或者安装带有 anthropic
可选组的 pydantic-ai-slim
。
pip install "pydantic-ai-slim[anthropic]"
uv add "pydantic-ai-slim[anthropic]"
配置
要通过其 API 使用 Anthropic,请前往 console.anthropic.com/settings/keys 生成一个 API 密钥。
AnthropicModelName
包含可用的 Anthropic 模型列表。
环境变量
获得 API 密钥后,可以将其设置为环境变量
export ANTHROPIC_API_KEY='your-api-key'
然后您可以通过名称使用 AnthropicModel
。
from pydantic_ai import Agent
agent = Agent('anthropic:claude-3-5-sonnet-latest')
...
或者直接用模型名称初始化模型
from pydantic_ai import Agent
from pydantic_ai.models.anthropic import AnthropicModel
model = AnthropicModel('claude-3-5-sonnet-latest')
agent = Agent(model)
...
provider
参数
您可以通过 provider
参数提供一个自定义的 Provider
。
from pydantic_ai import Agent
from pydantic_ai.models.anthropic import AnthropicModel
from pydantic_ai.providers.anthropic import AnthropicProvider
model = AnthropicModel(
'claude-3-5-sonnet-latest', provider=AnthropicProvider(api_key='your-api-key')
)
agent = Agent(model)
...
自定义 HTTP 客户端
您可以使用自定义的 httpx.AsyncClient
来定制 AnthropicProvider
。
from httpx import AsyncClient
from pydantic_ai import Agent
from pydantic_ai.models.anthropic import AnthropicModel
from pydantic_ai.providers.anthropic import AnthropicProvider
custom_http_client = AsyncClient(timeout=30)
model = AnthropicModel(
'claude-3-5-sonnet-latest',
provider=AnthropicProvider(api_key='your-api-key', http_client=custom_http_client),
)
agent = Agent(model)
...