Hello-Agents: Datawhale의 오픈소스 AI 에이전트 튜토리얼이 제로에서 프로덕션급 에이전트 구축을 돕는 방법

 1# 온라인 읽기
 2# https://datawhalechina.github.io/hello-agents/
 3
 4# 로컬 설정
 5git clone https://github.com/datawhalechina/hello-agents.git
 6cd hello-agents
 7# Extra-Chapter/07 참고하여 환경 구성
 8
 9# 4장 ReAct Agent 실행
10python code/chapter4/react_agent.py

 1import openai
 2import json
 3
 4tools = [
 5    {
 6        "type": "function",
 7        "function": {
 8            "name": "search_web",
 9            "description": "웹에서 정보 검색",
10            "parameters": {
11                "type": "object",
12                "properties": {"query": {"type": "string"}},
13                "required": ["query"]
14            }
15        }
16    },
17    {
18        "type": "function",
19        "function": {
20            "name": "calculate",
21            "description": "수학 계산 수행",
22            "parameters": {
23                "type": "object",
24                "properties": {"expression": {"type": "string"}},
25                "required": ["expression"]
26            }
27        }
28    }
29]
30
31def search_web(query): return f"검색 결과: {query}"
32def calculate(expr): return str(eval(expr))
33
34messages = [
35    {"role": "system", "content": "당신은 유용한 어시스턴트입니다. 필요할 때 도구를 사용하세요."},
36    {"role": "user", "content": "도쿄 인구를 1000으로 나눈 값은?"}
37]
38
39for step in range(5):
40    response = openai.chat.completions.create(
41        model="gpt-4", messages=messages, tools=tools, tool_choice="auto"
42    )
43    message = response.choices[0].message
44    messages.append(message)
45    
46    if message.tool_calls:
47        for tc in message.tool_calls:
48            name = tc.function.name
49            args = json.loads(tc.function.arguments)
50            result = search_web(**args) if name == "search_web" else calculate(**args)
51            messages.append({"role": "tool", "tool_call_id": tc.id, "content": result})
52    else:
53        print("최종 답변:", message.content)
54        break