프로젝트 1 — 회의실 예약 에이전트

import json, re, time, urllib.request
import pandas as pd
from openai import OpenAI, APIError

client = OpenAI(base_url="http://localhost:11434/v1", api_key="ollama")
MODEL = "qwen2.5:3b"
rooms = pd.read_csv("rooms.csv")
tasks = [json.loads(l) for l in open("agent_tasks.jsonl", encoding="utf-8")]
NAMES = ", ".join(rooms["방"])

def overlap(s1, e1, s2, e2):
    return s1 < e2 and s2 < e1                       # 두 시간이 겹치나

class RoomDB:
    """요청마다 새로 만드는 가짜 예약 저장소. new 에 새로 만든 예약이 쌓인다"""
    def __init__(self, helpful=False):
        self.bookings = pd.read_csv("room_bookings.csv").to_dict("records")
        self.new, self.helpful = [], helpful           # helpful: 오류 문구에 고치는 법을 넣을지(판 2)
    def is_free(self, room, date, start, end):
        return not any(b["방"] == room and b["날짜"] == date and overlap(start, end, b["시작"], b["끝"]) for b in self.bookings)
    def find_free_rooms(self, date, start, end, people=1, equipment="", **_):
        ok = rooms[(rooms["정원"] >= int(people or 1)) & rooms["장비"].str.contains(equipment or "")]
        return {"빈 방": [r for r in ok["방"] if self.is_free(r, date, start, end)]}
    def book_room(self, room, date, start, end, title="회의", **_):
        tip = lambda s: s if self.helpful else ""
        if room not in set(rooms["방"]):
            return {"오류": f"'{room}' 라는 방은 없습니다." + tip(f" 방 이름은 {NAMES} 가운데 하나입니다. find_free_rooms 로 먼저 찾으세요.")}
        if not start < end:
            return {"오류": f"시작 {start} 가 끝 {end} 보다 늦거나 같습니다." + tip(" 24시간 꼴(오후 3시 = 15:00)로 다시 적으세요.")}
        if not self.is_free(room, date, start, end):
            return {"오류": f"{room} 은 그 시간에 이미 예약이 있습니다." + tip(" find_free_rooms 로 다른 방을 찾으세요.")}
        rec = {"날짜": date, "방": room, "시작": start, "끝": end, "제목": title}
        self.bookings.append(rec)
        self.new.append(rec)
        return {"예약됨": rec}

db = RoomDB()
print(db.find_free_rooms("2026-10-28", "14:00", "15:00", 6, "화상회의"))
print(db.book_room("숲", "2026-10-28", "14:00", "15:00", "시험"))
print(db.book_room("별빛", "2026-10-28", "14:00", "15:00", "시험"))

def fn(name, desc, props, req):
    return {"type": "function", "function": {"name": name, "description": desc,
            "parameters": {"type": "object", "properties": props, "required": req}}}
P = {"room": {"type": "string", "description": "방 이름. 예: 숲"}, "date": {"type": "string", "description": "날짜 YYYY-MM-DD"},
     "start": {"type": "string", "description": "시작 시각 HH:MM (24시간)"}, "end": {"type": "string", "description": "끝 시각 HH:MM (24시간)"},
     "people": {"type": "integer", "description": "참석 인원"}, "title": {"type": "string", "description": "회의 제목"},
     "equipment": {"type": "string", "description": "화상회의, TV, 화이트보드 가운데 하나. 없으면 빈 글자"}}
TOOLS = [fn("find_free_rooms", "날짜와 시간에 비어 있고 인원과 장비 조건에 맞는 회의실 이름 목록을 돌려준다",
            {k: P[k] for k in ["date", "start", "end", "people", "equipment"]}, ["date", "start", "end", "people"]),
         fn("book_room", "회의실을 예약한다", {k: P[k] for k in ["room", "date", "start", "end", "title"]},
            ["room", "date", "start", "end", "title"])]
SYSTEM_1 = ("당신은 솔내기술의 회의실 예약 도우미입니다. 오늘은 2026-10-27(화)입니다. 도구로 확인한 사실만 말하세요. "
            "날짜, 시작과 끝 시각, 인원 가운데 빠진 것이 있으면 예약하지 말고 되물으세요. "
            "조건에 맞는 빈 방이 없으면 예약하지 말고 알리세요. 사용자가 예약하지 말라고 하면 예약하지 마세요.")

def unload(model):
    body = json.dumps({"model": model, "keep_alive": 0}).encode()
    urllib.request.urlopen(urllib.request.Request("http://localhost:11434/api/generate", data=body,
                                                  headers={"Content-Type": "application/json"})).read()

def chat(**kw):
    """7차시의 ask 처럼, 오류가 나면 모델을 내렸다가 다시 보낸다"""
    for _ in range(3):
        try:
            return client.chat.completions.create(model=MODEL, temperature=0, seed=42, **kw).choices[0].message
        except APIError:
            unload(MODEL)
            time.sleep(2)
    raise RuntimeError("Ollama 가 계속 오류를 냅니다")

def run_agent(request, system, db, max_steps=6):
    messages = [{"role": "system", "content": system}, {"role": "user", "content": request}]
    calls = []
    for _ in range(max_steps):
        msg = chat(messages=messages, tools=TOOLS)
        if not msg.tool_calls:
            return msg.content, calls
        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 = getattr(db, tc.function.name)(**args)
            except Exception as e:
                args, result = {}, {"오류": str(e)[:80]}
            calls.append((tc.function.name, args, result))
            messages.append({"role": "tool", "tool_call_id": tc.id, "content": json.dumps(result, ensure_ascii=False)})
    return "[걸음 한도]", calls

def grade(task, db):
    """끝난 뒤의 예약 목록으로 채점한다(본문 '에이전트 평가')"""
    if task["기대"] == "예약 안 함":
        return len(db.new) == 0
    if len(db.new) != 1:
        return False
    b, c = db.new[0], task["조건"]
    room = rooms[rooms["방"] == b["방"]].iloc[0]
    return (b["날짜"], b["시작"], b["끝"]) == (c["날짜"], c["시작"], c["끝"]) and room["정원"] >= c["인원"] \
        and c["장비"] in room["장비"] and c.get("방", b["방"]) == b["방"]

def evaluate_agent(system, helpful, label):
    start, ok, n_calls = time.time(), 0, 0
    for t in tasks:
        db = RoomDB(helpful=helpful)
        answer, calls = run_agent(t["요청"], system, db)
        good = grade(t, db)
        ok, n_calls = ok + good, n_calls + len(calls)
        print(" ", "O" if good else "X", t["요청"][:28], "|", [(c[0], c[1].get("room", "")) for c in calls], "| 새 예약:", [b["방"] for b in db.new])
    row = {"판": label, "성공(/10)": ok, "호출 수": n_calls, "초": round(time.time() - start)}
    print(row)
    return row

row1 = evaluate_agent(SYSTEM_1, helpful=False, label="1. 짧은 지시문")

SYSTEM_2 = SYSTEM_1 + (" 예약은 늘 이 순서로 합니다. 1) find_free_rooms 로 조건에 맞는 빈 방을 찾는다. "
                       "2) 사용자가 방을 정했으면 그 방이 목록에 있는지 보고, 정하지 않았으면 목록의 첫 방을 고른다. "
                       "3) book_room 으로 예약한다. 시각은 24시간 꼴(오후 2시 = 14:00)로 적는다. "
                       "도구가 오류를 돌려주면 오류 문구대로 고쳐 다시 시도한다.")
row2 = evaluate_agent(SYSTEM_2, helpful=True, label="2. 순서와 친절한 오류")

EXTRACT = ("오늘은 2026-10-27(화)입니다. 사용자의 말에서 회의실 예약 조건을 뽑아 JSON 하나만 쓰세요. 키: "
           "\"의도\"(예약/조회/모름), \"날짜\"(YYYY-MM-DD), \"시작\"(HH:MM, 24시간), \"끝\"(HH:MM), \"인원\"(숫자), "
           "\"장비\"(화상회의/TV/화이트보드/빈 글자), \"방\"(사용자가 정한 방 이름 또는 빈 글자), \"제목\". 말에 없는 값은 null 로 둡니다.")

def workflow(request, db):
    """LLM 은 조건 뽑기 한 번만. 검증, 빈 방 찾기, 예약은 코드가 한다"""
    s = json.loads(chat(messages=[{"role": "system", "content": EXTRACT}, {"role": "user", "content": request}],
                        response_format={"type": "json_object"}).content)
    # 코드로 검증: 말에 없는 방과 장비는 지우고, 말에 나온 방 이름은 규칙으로도 찾는다
    named = [r for r in rooms["방"] if r in request]
    s["방"] = named[0] if named else ""
    if s.get("장비") not in ("화상회의", "TV", "화이트보드") or s.get("장비") not in request:
        s["장비"] = ""
    try:
        s["인원"] = int(s.get("인원") or 1)
    except (TypeError, ValueError):
        s["인원"] = 1
    if s.get("의도") != "예약":
        return s, f"예약하지 않음(의도: {s.get('의도')})"
    if not re.search(r"\d", request) or not all(s.get(k) for k in ["날짜", "시작", "끝"]):
        return s, "되묻기: 날짜와 시각을 알려 주세요"              # 말에 숫자가 없으면 시각을 지어낸 것이다
    free = db.find_free_rooms(s["날짜"], s["시작"], s["끝"], s["인원"], s["장비"])["빈 방"]
    if s["방"]:
        free = [r for r in free if r == s["방"]]
    if not free:
        return s, "빈 방 없음"
    return s, db.book_room(free[0], s["날짜"], s["시작"], s["끝"], s.get("제목") or "회의")

start, ok = time.time(), 0
for t in tasks:
    db = RoomDB()
    slots, result = workflow(t["요청"], db)
    good = grade(t, db)
    ok += good
    print(" ", "O" if good else "X", t["요청"][:28], "|", {k: slots.get(k) for k in ["의도", "날짜", "시작", "인원", "방"]}, "|", str(result)[:40])
row3 = {"판": "3. 워크플로", "성공(/10)": ok, "호출 수": len(tasks), "초": round(time.time() - start)}
print(row3)

print(pd.DataFrame([row1, row2, row3]).to_string(index=False))