실습 4 — 자율 루프 만들기

import ast, json, operator, re, time
import numpy as np
from sentence_transformers import SentenceTransformer
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"))
rules = [l for l in open("company_rules.txt", encoding="utf-8").read().split("\n") if re.match(r"^제\d+조\(", l)]
emb = SentenceTransformer("intfloat/multilingual-e5-small", device="cpu")
R = emb.encode(["passage: " + r for r in rules], normalize_embeddings=True)
def search_rules(query):
    q = emb.encode("query: " + query, normalize_embeddings=True)
    return [rules[i] for i in np.argsort(-(R @ q))[:2]]
IMPL = {"calculate": lambda expression: safe_calc(expression), "search_rules": search_rules, "get_today": lambda: "2026-10-27"}
def tool(name, desc, props=None, req=None):
    return {"type": "function", "function": {"name": name, "description": desc,
            "parameters": {"type": "object", "properties": props or {}, "required": req or []}}}
TOOLS = [tool("calculate", "사칙연산 식을 계산한다. 숫자 계산이 필요하면 반드시 이 도구를 쓴다.",
              {"expression": {"type": "string", "description": "계산할 식. 예: 5*12"}}, ["expression"]),
         tool("search_rules", "솔내기술(가상 회사)의 사내 규정(근무, 휴가, 출장, 경비, 복지, 교육, 보안)을 찾는다.",
              {"query": {"type": "string", "description": "찾을 내용. 예: 회식비 한도"}}, ["query"]),
         tool("get_today", "오늘 날짜를 YYYY-MM-DD 로 돌려준다.")]
SYSTEM = ("당신은 솔내기술(가상 회사)의 사내 도우미입니다. 규정은 search_rules 로 찾아 확인하고, 계산은 calculate 로 하세요. "
          "도구로 확인한 내용만으로 짧게 답하세요.")
print("도구:", [t["function"]["name"] for t in TOOLS])

def run_agent(question, max_steps=6, show=True):
    """도구 호출이 없을 때까지 '묻기 → 실행 → 결과 덧붙이기'를 되풀이한다"""
    messages = [{"role": "system", "content": SYSTEM}, {"role": "user", "content": question}]
    for step in range(1, max_steps + 1):
        msg = client.chat.completions.create(model=MODEL, messages=messages, tools=TOOLS, temperature=0, seed=42).choices[0].message
        if not msg.tool_calls:                                   # 도구를 더 부르지 않으면 끝
            return msg.content, step
        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:
            try:
                args = json.loads(tc.function.arguments)
                result = IMPL[tc.function.name](**args)
            except Exception as e:                                # 이름이 없거나 인자가 틀리면 오류를 돌려준다
                result = {"오류": str(e)[:80]}
            if show:
                print(f"  걸음 {step}: {tc.function.name}({tc.function.arguments}) → {str(result)[:70]}")
            messages.append({"role": "tool", "tool_call_id": tc.id, "content": json.dumps(result, ensure_ascii=False)})
    return "[걸음 한도에 닿았습니다]", max_steps

answer, steps = run_agent("37 곱하기 48 은?")
print("답:", answer, "· 걸음:", steps)

for q in ["팀원 6명이 회식하면 회식비는 최대 얼마까지 쓸 수 있어요?", "해외 출장을 3일 가면 일비는 모두 몇 달러예요?"]:
    print("질문:", q)
    start = time.time()
    answer, steps = run_agent(q)
    print("답:", answer.replace("\n", " ")[:120], f"· 걸음 {steps} · {time.time() - start:.1f}초")

# 걸음 한도를 1로 줄이면, 검색만 하고 계산까지 가지 못한다
answer, steps = run_agent("팀원 6명이 회식하면 회식비는 최대 얼마까지 쓸 수 있어요?", max_steps=1)
print("답:", answer, "· 걸음:", steps)