실습 2 — 도구 결과를 돌려주고 답 받기

import ast, json, operator
from openai import OpenAI

client = OpenAI(base_url="http://localhost:11434/v1", api_key="ollama")
MODEL = "qwen2.5:3b"
OPS = {ast.Add: operator.add, ast.Sub: operator.sub, ast.Mult: operator.mul, ast.Div: operator.truediv}
def safe_calc(expression):
    def walk(node):
        if isinstance(node, ast.Expression):
            return walk(node.body)
        if isinstance(node, ast.Constant) and isinstance(node.value, (int, float)):
            return node.value
        if isinstance(node, ast.BinOp) and type(node.op) in OPS:
            return OPS[type(node.op)](walk(node.left), walk(node.right))
        if isinstance(node, ast.UnaryOp) and isinstance(node.op, ast.USub):
            return -walk(node.operand)
        raise ValueError("숫자와 + - * / 괄호만 쓸 수 있습니다")
    return walk(ast.parse(expression, mode="eval"))
TOOLS = [{"type": "function", "function": {
    "name": "calculate", "description": "사칙연산 식을 계산한다. 숫자 계산이 필요하면 반드시 이 도구를 쓴다.",
    "parameters": {"type": "object", "properties": {"expression": {"type": "string", "description": "계산할 식. 예: 1234*5678"}},
                   "required": ["expression"]}}}]
question = "한 상자에 48개씩 든 사과 1,375상자가 있어요. 사과는 모두 몇 개예요?"
print(question, "· 정답:", 48 * 1375)

messages = [{"role": "user", "content": question}]
msg = client.chat.completions.create(model=MODEL, messages=messages, tools=TOOLS, temperature=0, seed=42).choices[0].message
# 모델의 응답(도구 호출 요청)을 그대로 대화에 덧붙인다
messages.append({"role": "assistant", "content": msg.content or "", "tool_calls": [tc.model_dump() for tc in msg.tool_calls]})
for tc in msg.tool_calls:
    args = json.loads(tc.function.arguments)              # 인자는 JSON 글자로 온다
    result = safe_calc(args["expression"])                  # 우리 코드가 실행한다
    print("실행:", tc.function.name, args, "→", result)
    messages.append({"role": "tool", "tool_call_id": tc.id, "content": json.dumps({"결과": result})})
print("쌓인 메시지:", [m["role"] for m in messages])

final = client.chat.completions.create(model=MODEL, messages=messages, tools=TOOLS, temperature=0, seed=42).choices[0].message
print("도구를 쓴 답:", final.content)

# 같은 질문을 도구 없이: 모델이 머릿속으로 계산한다
plain = client.chat.completions.create(model=MODEL, messages=[{"role": "user", "content": question}], temperature=0, seed=42)
print("도구 없는 답:", plain.choices[0].message.content.replace("\n", " ")[:120])