Chuyển đến nội dung chính

Tổng quan

AutoGen có thể sử dụng TokenLab thông qua OpenAIChatCompletionClient, đây là cùng một đường dẫn client tương thích với OpenAI mà nó sử dụng cho các endpoint Chat Completions tùy chỉnh khác.
Loại: Khung tác nhân (Agent Framework)Đường dẫn chính: Chat completions tương thích với OpenAIĐộ tin cậy hỗ trợ: Hỗ trợ đường dẫn endpoint tùy chỉnh

Cài đặt

pip install -U autogen-agentchat "autogen-ext[openai]" pyyaml

Môi trường

export TOKENLAB_API_KEY="sk-your-tokenlab-key"

Cấu hình mô hình YAML

Sử dụng endpoint /v1 của TokenLab làm base_url:
provider: autogen_ext.models.openai.OpenAIChatCompletionClient
config:
  model: claude-sonnet-5
  base_url: https://api.tokenlab.sh/v1
  api_key: ${TOKENLAB_API_KEY}
  model_info:
    function_calling: true
    json_output: true
    vision: false
    family: claude

Python Client

import os

from autogen_agentchat.agents import AssistantAgent
from autogen_ext.models.openai import OpenAIChatCompletionClient

model_client = OpenAIChatCompletionClient(
    model="claude-sonnet-5",
    base_url="https://api.tokenlab.sh/v1",
    api_key=os.environ["TOKENLAB_API_KEY"],
    model_info={
        "function_calling": True,
        "json_output": True,
        "vision": False,
        "family": "claude",
    },
)

agent = AssistantAgent("assistant", model_client=model_client)

Lưu ý về Endpoint

Phần mở rộng OpenAI của AutoGen yêu cầu định dạng yêu cầu tương thích với OpenAI. Đối với các tính năng gốc của nhà cung cấp như Anthropic Messages hoặc generateContent của Gemini, hãy sử dụng các endpoint gốc của TokenLab từ một client hỗ trợ trực tiếp các giao thức đó.