실습 3 — 도구 여러 개에서 고르기

import ast, json, operator, re
from datetime import date
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"))

# 사내 규정 검색 도구: 7장의 규정집을 조항 단위로 나눠 의미 검색한다
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]]       # 가까운 조항 2개

def get_today():
    return "2026-10-27"          # 교안의 '오늘'. 실제 서비스라면 date.today().isoformat()

IMPL = {"calculate": lambda expression: safe_calc(expression), "search_rules": search_rules, "get_today": get_today}
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 로 돌려준다.")]
print(len(rules), "조항 ·", search_rules("재택 근무")[0][:40])

SYSTEM = "당신은 솔내기술(가상 회사)의 사내 도우미입니다. 필요하면 도구를 쓰고, 도구 없이 알 수 있으면 바로 답하세요."
questions = ["37 곱하기 48 은?", "재택 근무는 일주일에 며칠까지 돼요?", "오늘이 며칠이에요?",
             "안녕하세요!", "팀원 6명이 회식하면 회식비는 최대 얼마까지 쓸 수 있어요?", "해외 출장을 3일 가면 일비는 모두 몇 달러예요?"]
for q in questions:
    msg = client.chat.completions.create(model=MODEL, messages=[{"role": "system", "content": SYSTEM}, {"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(f"{q:32s} →", calls if calls else "도구 없음")