Mistral

安装

要使用 MistralModel，您需要安装 pydantic-ai，或者安装带有 mistral 可选组的 pydantic-ai-slim。

pipuv

pip install "pydantic-ai-slim[mistral]"

uv add "pydantic-ai-slim[mistral]"

配置

要通过 API 使用 Mistral，请访问 console.mistral.ai/api-keys/ 并按照指引生成一个 API 密钥。

LatestMistralModelNames 包含了一系列最受欢迎的 Mistral 模型。

环境变量

获得 API 密钥后，可以将其设置为环境变量

export MISTRAL_API_KEY='your-api-key'

然后您就可以按名称使用 MistralModel。

from pydantic_ai import Agent

agent = Agent('mistral:mistral-large-latest')
...

或者直接用模型名称初始化模型

from pydantic_ai import Agent
from pydantic_ai.models.mistral import MistralModel

model = MistralModel('mistral-small-latest')
agent = Agent(model)
...

provider 参数

您可以通过 provider 参数提供一个自定义的 Provider。

from pydantic_ai import Agent
from pydantic_ai.models.mistral import MistralModel
from pydantic_ai.providers.mistral import MistralProvider

model = MistralModel(
    'mistral-large-latest', provider=MistralProvider(api_key='your-api-key', base_url='https://<mistral-provider-endpoint>')
)
agent = Agent(model)
...

您还可以使用自定义的 httpx.AsyncHTTPClient 来自定义提供商。

from httpx import AsyncClient

from pydantic_ai import Agent
from pydantic_ai.models.mistral import MistralModel
from pydantic_ai.providers.mistral import MistralProvider

custom_http_client = AsyncClient(timeout=30)
model = MistralModel(
    'mistral-large-latest',
    provider=MistralProvider(api_key='your-api-key', http_client=custom_http_client),
)
agent = Agent(model)
...