실습 1 — 도구 하나 붙이기

import ast, json, operator
from openai import OpenAI

client = OpenAI(base_url="http://localhost:11434/v1", api_key="ollama")
MODEL = "qwen2.5:3b"

# 본문 "에이전트의 안전장치"의 계산기: eval 을 쓰지 않고 사칙연산만 계산한다
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"))

# 모델에게 보여 줄 도구 설명(JSON 스키마)
TOOLS = [{"type": "function", "function": {
    "name": "calculate",
    "description": "사칙연산 식을 계산한다. 숫자 계산이 필요하면 반드시 이 도구를 쓴다.",
    "parameters": {"type": "object",
                   "properties": {"expression": {"type": "string", "description": "계산할 식. 예: 1234*5678"}},
                   "required": ["expression"]}}}]
print(safe_calc("1234*5678"))

resp = client.chat.completions.create(model=MODEL, messages=[{"role": "user", "content": "1234 곱하기 5678 은 얼마예요?"}],
                                      tools=TOOLS, temperature=0, seed=42)
msg = resp.choices[0].message
print("글(content):", repr(msg.content))
for tc in msg.tool_calls or []:
    # 도구 이름과 인자(JSON 글자), 그리고 결과를 돌려줄 때 쓸 id
    print("도구:", tc.function.name, "· 인자:", tc.function.arguments, "· id:", tc.id[:12])
print("멈춘 까닭:", resp.choices[0].finish_reason)

for q in ["안녕하세요, 오늘 기분 어때요?", "세탁기와 건조기의 차이를 한 문장으로 알려 주세요.", "387 더하기 1,250 에서 99 를 빼면?"]:
    msg = client.chat.completions.create(model=MODEL, messages=[{"role": "user", "content": q}],
                                         tools=TOOLS, temperature=0, seed=42).choices[0].message
    calls = [(tc.function.name, tc.function.arguments) for tc in msg.tool_calls or []]
    print(q, "→", calls if calls else "도구 없이 답: " + (msg.content or "")[:40].replace("\n", " "))