Pydantic AI

O Pydantic AI é um framework de agentes em Python, do time do Pydantic, construído em torno de saídas tipadas. A Bayle se conecta pelo provider OpenAI e pelo modelo de chat.

Copia as instruções de configuração desta página, para colar no seu agente de código.

Ver o prompt

Me ajude a configurar isto para usar a Bayle AI, uma API compatível com a OpenAI. Guia completo: https://docs.bayle.ai/pt-br/guides/sdks/pydantic-ai/

Objetivo: construir os meus agentes do Pydantic AI (https://pydantic.dev/docs/ai/overview/) em um modelo da Bayle.

1. Instale o pacote se `python -c "import pydantic_ai"` falhar:
   pip install "pydantic-ai-slim[openai]"
2. Escolha um modelo. Saídas tipadas e ferramentas dependem de function calling, então escolha um que suporte:
   curl -s "https://api.bayle.ai/models?type=text" \
     | jq -r '.data[] | select(.model_spec.capabilities.supportsFunctionCalling) | .id'
   Me mostre a lista, deixe eu escolher e use esse id como MODEL_ID.
3. No meu código, monte o modelo assim. Use OpenAIChatModel: a string de modelo `openai:` resolve para o modelo da Responses API, que a Bayle não serve:
   import os
   from pydantic_ai import Agent
   from pydantic_ai.models.openai import OpenAIChatModel
   from pydantic_ai.providers.openai import OpenAIProvider

   model = OpenAIChatModel(
       "MODEL_ID",
       provider=OpenAIProvider(
           base_url="https://api.bayle.ai",
           api_key=os.environ["BAYLE_API_KEY"],
       ),
   )

   agent = Agent(model)
   Se eu já tiver um Agent, mude só o modelo que ele recebe.
4. Verifique rodando:
   print(agent.run_sync("In one sentence, what is an embedding?").output)
   e me mostre a saída.

Regras:
1. A base URL é https://api.bayle.ai exatamente como está escrita. Não acrescente nenhum caminho a ela.
2. Leia a key da variável de ambiente BAYLE_API_KEY. Nunca escreva a key em um arquivo que você criar, em um comando, em um commit nem no chat. Onde a ferramenta guarda a key na própria tela de configurações, me diga onde colar. Se BAYLE_API_KEY não estiver definida, pare e peça que eu crie uma key em https://chat.bayle.ai/developers e a exporte.
3. Pegue os ids de modelo só de GET https://api.bayle.ai/models (sem key). Nunca chute um.
4. Termine com uma requisição de teste de verdade, me mostre o resultado e liste os arquivos que você alterou.

Instalação

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

Veja o guia oficial de instalação para os outros extras.

Uso

Ainda não tem uma chave da Bayle?Obter chave de API

import os
from pydantic_ai import Agent
from pydantic_ai.models.openai import OpenAIChatModel
from pydantic_ai.providers.openai import OpenAIProvider

model = OpenAIChatModel(
    "MODEL_ID",
    provider=OpenAIProvider(
        base_url="https://api.bayle.ai",
        api_key=os.environ["BAYLE_API_KEY"],
    ),
)

agent = Agent(model)
print(agent.run_sync("In one sentence, what is an embedding?").output)

Use OpenAIChatModel. A string de modelo openai: resolve para o modelo da Responses API, que a Bayle não serve.
Saídas tipadas e ferramentas dependem de function calling, então escolha um modelo que suporte.

Escolha um modelo

curl -s "https://api.bayle.ai/models?type=text" \
  | jq -r '.data[] | select(.model_spec.capabilities.supportsFunctionCalling) | .id'