搜索结果

from anthropic import Anthropic
from anthropic.types.beta import (
    BetaMessageParam,
    BetaTextBlockParam,
    BetaSearchResultBlockParam,
    BetaToolResultBlockParam
)

client = Anthropic()

# 定义知识库搜索工具
knowledge_base_tool = {
    "name": "search_knowledge_base",
    "description": "搜索公司知识库以获取信息",
    "input_schema": {
        "type": "object",
        "properties": {
            "query": {
                "type": "string",
                "description": "搜索查询"
            }
        },
        "required": ["query"]
    }
}

# 处理工具调用的函数
def search_knowledge_base(query):
    # 您的搜索逻辑在这里
    # 以正确格式返回搜索结果
    return [
        BetaSearchResultBlockParam(
            type="search_result",
            source="https://docs.company.com/product-guide",
            title="产品配置指南",
            content=[
                BetaTextBlockParam(
                    type="text",
                    text="要配置产品，请导航到设置 > 配置。默认超时时间为30秒，但可以根据您的需要在10-120秒之间调整。"
                )
            ],
            citations={"enabled": True}
        ),
        BetaSearchResultBlockParam(
            type="search_result",
            source="https://docs.company.com/troubleshooting",
            title="故障排除指南",
            content=[
                BetaTextBlockParam(
                    type="text",
                    text="如果遇到超时错误，请首先检查配置设置。常见原因包括网络延迟和不正确的超时值。"
                )
            ],
            citations={"enabled": True}
        )
    ]

# 使用工具创建消息
response = client.beta.messages.create(
    model="claude-sonnet-4-20250514",  # 适用于所有支持的模型
    max_tokens=1024,
    betas=["search-results-2025-06-09"],
    tools=[knowledge_base_tool],
    messages=[
        BetaMessageParam(
            role="user",
            content="如何配置超时设置？"
        )
    ]
)

# 当Claude调用工具时，提供搜索结果
if response.content[0].type == "tool_use":
    tool_result = search_knowledge_base(response.content[0].input["query"])
    
    # 将工具结果发送回去
    final_response = client.beta.messages.create(
        model="claude-sonnet-4-20250514",  # 适用于所有支持的模型
        max_tokens=1024,
        betas=["search-results-2025-06-09"],
        messages=[
            BetaMessageParam(role="user", content="如何配置超时设置？"),
            BetaMessageParam(role="assistant", content=response.content),
            BetaMessageParam(
                role="user",
                content=[
                    BetaToolResultBlockParam(
                        type="tool_result",
                        tool_use_id=response.content[0].id,
                        content=tool_result  # 搜索结果放在这里
                    )
                ]
            )
        ]
    )

from anthropic import Anthropic
from anthropic.types.beta import (
    BetaMessageParam,
    BetaTextBlockParam,
    BetaSearchResultBlockParam
)

client = Anthropic()

# 直接在用户消息中提供搜索结果
response = client.beta.messages.create(
    model="claude-opus-4-20250514",
    max_tokens=1024,
    betas=["search-results-2025-06-09"],
    messages=[
        BetaMessageParam(
            role="user",
            content=[
                BetaSearchResultBlockParam(
                    type="search_result",
                    source="https://docs.company.com/api-reference",
                    title="API参考 - 身份验证",
                    content=[
                        BetaTextBlockParam(
                            type="text",
                            text="所有API请求都必须在Authorization标头中包含API密钥。密钥可以从仪表板生成。速率限制：标准层每小时1000个请求，高级层10000个。"
                        )
                    ],
                    citations={"enabled": True}
                ),
                BetaSearchResultBlockParam(
                    type="search_result",
                    source="https://docs.company.com/quickstart",
                    title="入门指南",
                    content=[
                        BetaTextBlockParam(
                            type="text",
                            text="开始使用：1）注册账户，2）从仪表板生成API密钥，3）使用pip install company-sdk安装我们的SDK，4）使用您的API密钥初始化客户端。"
                        )
                    ],
                    citations={"enabled": True}
                ),
                BetaTextBlockParam(
                    type="text",
                    text="基于这些搜索结果，我如何验证API请求以及速率限制是什么？"
                )
            ]
        )
    ]
)

print(response.model_dump_json(indent=2))