Command-R

import cohere

# Cohere 클라이언트 초기화
co = cohere.Client("your-api-key")

# RAG용 참조 문서 준비
documents = [
    {"title": "회사 연차 정책", "text": "입사 1년 이상 직원은 연 15일의 유급 휴가가 부여됩니다. 미사용 연차는 이월되지 않습니다."},
    {"title": "재택근무 가이드", "text": "주 2회까지 재택근무가 가능하며, 팀장 사전 승인이 필요합니다."},
    {"title": "경비 처리 규정", "text": "출장비는 월 50만원 한도 내에서 사후 정산됩니다. 영수증 첨부 필수입니다."}
]

# Grounded Generation으로 RAG 수행
response = co.chat(
    model="command-r-plus",
    message="연차 휴가는 며칠인가요?",
    documents=documents,  # 참조 문서 전달
    temperature=0.3,
    citation_quality="accurate"  # 정확한 인용 생성
)

print("답변:", response.text)
print("\n인용 정보:")
for citation in response.citations:
    print(f"  - [{citation.start}:{citation.end}] 문서: {citation.document_ids}")

# Tool Use (함수 호출) 기능
tools = [{
    "name": "search_database",
    "description": "사내 데이터베이스에서 정보를 검색합니다",
    "parameters": {
        "type": "object",
        "properties": {
            "query": {"type": "string", "description": "검색어"}
        },
        "required": ["query"]
    }
}]

response = co.chat(
    model="command-r-plus",
    message="최근 프로젝트 현황을 알려줘",
    tools=tools
)

if response.tool_calls:
    print(f"호출할 함수: {response.tool_calls[0].name}")