검색 결과

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))