Anthropic API

from anthropic import Anthropic

# 初始化客户端
client = Anthropic(
    base_url="https://ai.megallm.io",
    api_key="your-api-key"
)

# 创建消息
message = client.messages.create(
    model="claude-3.5-sonnet",
    max_tokens=100,
    messages=[
        {
            "role": "user",
            "content": "Explain the theory of relativity in simple terms"
        }
    ]
)

print(message.content[0].text)

# Anthropic 格式
messages = [
    {
        "role": "user",
        "content": "Hello, Claude!"
    }
]

# 系统消息是独立的
system = "You are a helpful assistant"

message = client.messages.create(
    model="claude-3.5-sonnet",
    max_tokens=100,
    system=system,  # 系统提示是独立的
    messages=messages
)

# Anthropic 响应结构
response = {
    "id": "msg_123",
    "type": "message",
    "role": "assistant",
    "content": [
        {
            "type": "text",
            "text": "Hello! How can I help you today?"
        }
    ],
    "model": "claude-3.5-sonnet",
    "usage": {
        "input_tokens": 10,
        "output_tokens": 25
    }
}

tools = [
    {
        "name": "get_weather",
        "description": "获取某个位置的天气",
        "input_schema": {  # 注意: input_schema,而不是 parameters
            "type": "object",
            "properties": {
                "location": {
                    "type": "string",
                    "description": "城市名称"
                }
            },
            "required": ["location"]
        }
    }
]

# 之前 (Anthropic Cloud)
client = Anthropic(api_key="sk-ant-...")

# 之后 (MegaLLM)
client = Anthropic(
    base_url="https://ai.megallm.io",
    api_key="your-api-key"
)

curl https://ai.megallm.io/v1/messages \
  -H "x-api-key: $MEGALLM_API_KEY" \
  -H "anthropic-version: 2023-06-01" \
  -H "content-type: application/json" \
  -d '{
    "model": "claude-3.5-sonnet",
    "max_tokens": 100,
    "messages": [
      {"role": "user", "content": "Hello!"}
    ]
  }'

{
  "type": "error",
  "error": {
    "type": "invalid_request_error",
    "message": "max_tokens is required"
  }
}

conversation = []

def chat(user_input):
    conversation.append({"role": "user", "content": user_input})

    response = client.messages.create(
        model="claude-3.5-sonnet",
        max_tokens=150,
        messages=conversation
    )

    assistant_message = response.content[0].text
    conversation.append({"role": "assistant", "content": assistant_message})

    return assistant_message

# 使用示例
print(chat("What's the capital of France?"))
print(chat("What's its population?"))  # 知道 "its" 指的是巴黎

# 更确定性
response = client.messages.create(
    model="claude-3.5-sonnet",
    max_tokens=100,
    temperature=0.0,  # 非常一致
    messages=messages
)

# 更有创造性
response = client.messages.create(
    model="claude-3.5-sonnet",
    max_tokens=100,
    temperature=1.0,  # 更多样化
    top_p=0.95,       # 核采样
    messages=messages
)

def analyze_document(document_text):
    response = client.messages.create(
        model="claude-3-opus-20240229",
        max_tokens=500,
        system="您是一位文档分析专家。",
        messages=[
            {
                "role": "user",
                "content": f"""分析这份文档并提供:
                1. 主要主题
                2. 关键见解
                3. 摘要

                文档: {document_text}"""
            }
        ]
    )
    return response.content[0].text

def review_code(code):
    response = client.messages.create(
        model="claude-3.5-sonnet",
        max_tokens=800,
        system="您是一位专业的代码审查员。",
        messages=[
            {
                "role": "user",
                "content": f"""审查这段代码的:
                - 错误
                - 性能问题
                - 最佳实践
                - 安全问题

                代码:
                ```python
                {code}
                ```"""
            }
        ]
    )
    return response.content[0].text