검색 결과

from anthropic import Anthropic
from anthropic.types import (
    MessageParam,
    TextBlockParam,
    SearchResultBlockParam,
    ToolResultBlockParam
)

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 [
        SearchResultBlockParam(
            type="search_result",
            source="https://docs.company.com/product-guide",
            title="제품 구성 가이드",
            content=[
                TextBlockParam(
                    type="text",
                    text="제품을 구성하려면 설정 > 구성으로 이동하세요. 기본 타임아웃은 30초이지만, 필요에 따라 10-120초 사이에서 조정할 수 있습니다."
                )
            ],
            citations={"enabled": True}
        ),
        SearchResultBlockParam(
            type="search_result",
            source="https://docs.company.com/troubleshooting",
            title="문제 해결 가이드",
            content=[
                TextBlockParam(
                    type="text",
                    text="타임아웃 오류가 발생하면 먼저 구성 설정을 확인하세요. 일반적인 원인으로는 네트워크 지연과 잘못된 타임아웃 값이 있습니다."
                )
            ],
            citations={"enabled": True}
        )
    ]

# 도구와 함께 메시지 생성
response = client.messages.create(
    model="claude-sonnet-4-20250514",  # 지원되는 모든 모델에서 작동
    max_tokens=1024,
    tools=[knowledge_base_tool],
    messages=[
        MessageParam(
            role="user",
            content="타임아웃 설정을 어떻게 구성하나요?"
        )
    ]
)

# Claude가 도구를 호출할 때 검색 결과 제공
if response.content[0].type == "tool_use":
    tool_result = search_knowledge_base(response.content[0].input["query"])
    
    # 도구 결과를 다시 보내기
    final_response = client.messages.create(
        model="claude-sonnet-4-20250514",  # 지원되는 모든 모델에서 작동
        max_tokens=1024,
        messages=[
            MessageParam(role="user", content="타임아웃 설정을 어떻게 구성하나요?"),
            MessageParam(role="assistant", content=response.content),
            MessageParam(
                role="user",
                content=[
                    ToolResultBlockParam(
                        type="tool_result",
                        tool_use_id=response.content[0].id,
                        content=tool_result  # 검색 결과가 여기에 들어감
                    )
                ]
            )
        ]
    )

from anthropic import Anthropic
from anthropic.types import (
    MessageParam,
    TextBlockParam,
    SearchResultBlockParam
)

client = Anthropic()

# 사용자 메시지에서 직접 검색 결과 제공
response = client.messages.create(
    model="claude-opus-4-1-20250805",
    max_tokens=1024,
    messages=[
        MessageParam(
            role="user",
            content=[
                SearchResultBlockParam(
                    type="search_result",
                    source="https://docs.company.com/api-reference",
                    title="API 참조 - 인증",
                    content=[
                        TextBlockParam(
                            type="text",
                            text="모든 API 요청은 Authorization 헤더에 API 키를 포함해야 합니다. 키는 대시보드에서 생성할 수 있습니다. 속도 제한: 표준 등급은 시간당 1000개 요청, 프리미엄은 10000개입니다."
                        )
                    ],
                    citations={"enabled": True}
                ),
                SearchResultBlockParam(
                    type="search_result",
                    source="https://docs.company.com/quickstart",
                    title="시작하기 가이드",
                    content=[
                        TextBlockParam(
                            type="text",
                            text="시작하려면: 1) 계정에 가입, 2) 대시보드에서 API 키 생성, 3) pip install company-sdk를 사용하여 SDK 설치, 4) API 키로 클라이언트 초기화."
                        )
                    ],
                    citations={"enabled": True}
                ),
                TextBlockParam(
                    type="text",
                    text="이 검색 결과를 바탕으로 API 요청을 어떻게 인증하고 속도 제한은 무엇인가요?"
                )
            ]
        )
    ]
)

print(response.model_dump_json(indent=2))