o2o-site-AEO/solution/backend/services/agent/tools.py
민헌 aaf97f2797 [fix] solution/backend: 대화 수정의 값 형식 검증 · 여러 요청의 못 한 것·남은 것 안내 — 조용히 틀리던 경로 정리
말로 고친 값이 형식 검증 없이 저장됐다 — "반려동물 이제 돼요" 가 "가능" 으로 들어가면
화면엔 "가능" 이 뜨는데 factBool 은 'true' 만 참으로 읽어 구조화 데이터가 거짓이 된다.
여러 요청을 한 번에 받을 때도 할 수 없는 것·멈춘 뒤의 것이 말없이 사라져 사장님은
전부 된 줄 알았다(message 는 actions 가 있으면 버려졌다).

- tools: set_fact 값을 스키마 형식으로 맞춘다(bool true/false · time HH:MM · number 숫자) —
  "3시" 처럼 오전·오후를 모르면 저장하지 않고 되묻는다. 알림 문구는 가능·불가로 말한다
- tools: enabled 를 모르면 끄지 않고 되묻는다 — "" 를 끄기로 읽어 "다시 보여줘" 가 섹션을 껐다
- tools: 대표 지정·사진 목록·프롬프트를 '나가는 사진' 기준으로 — 내린 사진을 대표로 지정하고
  "바꿨습니다" 라고 하던 것. 정확히 맞는 이름을 부분 일치보다 먼저 고른다
- tools: Unchanged 표시 — "이미 켜져 있어요" 에 재발행 안내·카톡 발행 대기가 붙던 것
- tools/runtime: 인자가 문자열·dict 가 아니어도 죽지 않는다(모델의 스키마 위반)
- runtime: 같은 대상은 마지막 하나로 합친다(Tool.target, 상한 세기 전). 멈춘 뒤 남은 요청과
  지어낸 도구를 코드가 만든 이름으로 알린다. 발행 확인 문구는 맨 끝에 선다
- prompts: 항목마다 형식을 싣고 skipped(할 수 없는 요청의 이름) 칸 추가 — 문장은 런타임이 만든다.
  다른 가게 이야기면 되묻는 규칙
- channel: 발화에 다른 내 가게 이름이 나오면 모델을 부르기 전에 고르게 한다. 기억한 가게가
  목록에 없으면 비우고 목록으로. 확인 경로의 AgentError 를 사장님 말로 옮긴다
- docs/AGENT.md: 값 형식 · 못 한 것·남은 것·겹친 것 · 나가는 사진 기준

테스트 48건 추가, 에이전트·카카오 153 passed. 전체 932 passed / 47 failed —
실패 47건은 변경 전과 동일(gemini·openai 키 미설정, search_console KeyError, weather_notes)

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
2026-09-29 15:11:00 +09:00

644 lines
30 KiB
Python

"""도구 레지스트리 — 에이전트가 할 수 있는 일의 **전부**가 여기 있다.
★★ 도구는 반드시 `services/*` 를 통과한다. `crud`·`models` 를 직접 부르면 업종 스키마
검증 · 출처 필수 · 정정본 보호 · 소유자 범위가 통째로 사라지는데, **아무 증상이 없다** —
값은 들어가고 빌드는 성공하고 화면도 뜬다. `collect_service.store_facts` 가
"크롤러가 우회할 수 있는 뒷문을 만들지 않는다" 로 막아 둔 그 문이고, 에이전트에게만
열어 줄 이유가 없다.
★ 결과 문구는 도구가 만든다. LLM 이 쓰게 두면 **하지 않은 일을 했다고 말할 수 있고**,
사장님에게는 그 말이 사실로 보인다.
★ 등급은 여기서 못 박는다. LLM 이 정하게 두면 프롬프트에 끼어든 한 줄이 확인 절차를
건너뛴다 — 되돌릴 수 없는 행위일수록 그 값을 모델에 맡기면 안 된다.
"""
import re
import uuid
from dataclasses import dataclass, field
from enum import Enum
from typing import Awaitable, Callable
from common.category_schema.loader import get_schema
from common.enums import ErrorType, PlaceCategory, SourceType
from common.models.gmodel import UserInfo
from crud.fact_crud import FactCRUD
from crud.job_crud import JobQueue
from crud.media_crud import MediaCRUD
from crud.place_crud import PlaceCRUD
from crud.site_crud import SiteCRUD
from router.v1.fact.protocol import Req_UpsertFact
from router.v1.site.protocol import Req_SiteTheme, Req_StartBuild
from services import site_payload
from services.fact_service import FactService
from services.media_service import MediaService
from services.site_service import SiteService
class ToolGrade(str, Enum):
"""되돌릴 수 있느냐가 승인 강도를 정한다 — 분류가 아니라 동작을 가르는 값이다."""
READ = "READ" # 승인 없음
REVERSIBLE = "REVERSIBLE" # 실행하고 알린다. 사장님이 다시 고치면 된다
SEMI = "SEMI" # 실행 전에 한 번 묻는다(되돌릴 수는 있으나 그 사이 밖에서 읽힌다)
@dataclass
class ToolContext:
user: UserInfo
place_id: str
place: object
@dataclass
class Tool:
name: str
grade: ToolGrade
summary: str
args: dict = field(default_factory=dict)
run: Callable[[ToolContext, dict], Awaitable[str]] = None
# SEMI 도구가 실행 전에 사장님께 보일 문장.
confirm: str = ""
# ★ 이 도구가 바꾼 것은 **재발행해야 사이트에 반영된다.** 안내 문구는 도구가 아니라
# 런타임이 **한 번만** 붙인다 — 도구마다 문장에 박아 두면 한 발화로 셋을 고쳤을 때
# 같은 말이 세 번 나온다.
republish: bool = False
# 사장님께 "아직 안 했어요" 라고 알릴 때의 이름. describe 가 있으면 인자로 더 좁혀 말한다.
title: str = ""
describe: Callable[[ToolContext, dict], str] | None = None
# ★ 무엇을 겨누는가 — 이 인자들이 같으면 같은 요청이다. 런타임이 "3시… 아니 4시" 를
# 마지막 하나로 합칠 때 쓴다. None 이면 인자 전체가 같을 때만 같은 요청이다.
target: tuple[str, ...] | None = None
class Unchanged(str):
"""도구가 할 일이 없었다는 표시 — "이미 켜져 있어요".
★ 문자열 그대로 쓰이고, 런타임은 이 표시로 **바뀐 것이 없음**을 안다. 모르면 재발행 안내가
붙고 카톡에는 발행 대기까지 걸려, 사장님은 무언가 바뀐 줄 안다."""
def _services():
"""서비스는 매 호출 새로 만든다 — 라우터가 Depends 로 받는 것과 같은 수명이다.
★ Depends 기본값에 기대지 않고 의존을 손으로 넣는다. FastAPI 밖에서 부르면
기본값이 `Depends(...)` 객체 그대로라 서비스가 조용히 엉뚱한 것을 들고 돈다."""
place_crud = PlaceCRUD()
return FactService(FactCRUD(), place_crud), SiteService(SiteCRUD(), place_crud, JobQueue())
def _media_service() -> MediaService:
return MediaService(MediaCRUD(), PlaceCRUD())
def _arg(args: dict, name: str) -> str:
"""인자 하나를 문자열로. ★ 모델은 스키마를 어길 수 있다 — true 를 불리언으로, 요금을 숫자로
보낸다. `(args.get(x) or "").strip()` 은 거기서 AttributeError 로 죽고, 사장님에게는
"처리할 수 없어요" 로만 보인다."""
value = args.get(name)
return "" if value is None else str(value).strip()
# ── 값 형식 ─────────────────────────────────────────────────────────────
#
# ★ 저장 형식은 수집 어댑터와 같다 — bool "true"/"false" · time "HH:MM" · number 숫자만.
# 렌더러(shared/src/lib/facts.ts factBool)는 'true' 만 참으로 읽는다. "가능" 으로 저장하면
# 화면에는 원문 "가능" 이 뜨는데 구조화 데이터는 거짓이 된다 — 빌드도 성공하는 조용한 틀림이다.
# ★ 알아볼 수 없으면 저장하지 않고 되묻는다. 추측해서 넣은 값이 곧 1차 출처가 된다.
_TRUE_WORDS = {"true", "1", "y", "yes", "o", "가능", "있음", "있어요", "돼요", "됩니다", "허용", "네", "예"}
_FALSE_WORDS = {"false", "0", "n", "no", "x", "불가", "불가능", "없음", "없어요", "안돼요", "안됩니다", "금지", "아니요"}
# "15:00" · "오후 3시" · "15시 30분" · "3시 반"
_TIME = re.compile(r"^(오전|오후|아침|낮|저녁|밤|새벽)?(\d{1,2})(?::(\d{2})|시(?:(\d{1,2})분|(반))?)$")
_PM_WORDS = {"오후", "저녁", "밤"}
_AM_WORDS = {"오전", "아침", "새벽"}
# "2만 5천원" · "20,000원" · "1.5시간"
_NUMBER = re.compile(r"^(?:(\d+)만)?(?:(\d+)천)?(\d+(?:\.\d+)?)?$")
_NUMBER_UNITS = re.compile(r"(원|명|대|개|분|시간|실|석|인|층|평|㎡)$")
def _parse_bool(raw: str) -> str | None:
word = re.sub(r"\s+", "", raw.lower())
if word in _TRUE_WORDS:
return "true"
if word in _FALSE_WORDS:
return "false"
return None
def _parse_time(raw: str) -> tuple[str | None, bool]:
"""(HH:MM, 모호한가). 모호함은 틀림과 다르게 묻는다 — "오전인지 오후인지" 가 답을 끌어낸다.
★ 콜론 표기는 24시간으로 읽는다(프롬프트가 그렇게 시킨다). '3시' 처럼 한국어로 1~11시만
말하면 오전·오후를 모른다 — 체크인 03:00 을 넣으면 손님이 새벽에 온다."""
m = _TIME.fullmatch(re.sub(r"\s+", "", raw))
if not m:
return None, False
when, hour, minute = m.group(1), int(m.group(2)), int(m.group(3) or m.group(4) or (30 if m.group(5) else 0))
if when in _PM_WORDS and hour < 12:
hour += 12
elif when in _AM_WORDS and hour == 12:
hour = 0
elif when is None and m.group(3) is None and 1 <= hour <= 11:
return None, True
if hour > 23 or minute > 59:
return None, False
return f"{hour:02d}:{minute:02d}", False
def _parse_number(raw: str) -> str | None:
word = re.sub(r"[\s,]", "", raw)
if word == "무료":
return "0"
word = _NUMBER_UNITS.sub("", word)
m = _NUMBER.fullmatch(word)
if not word or not m:
return None
man, cheon, rest = m.group(1), m.group(2), m.group(3)
if (man or cheon) and rest and "." in rest:
return None
value = int(man or 0) * 10000 + int(cheon or 0) * 1000
if man or cheon:
return str(value + int(rest or 0))
number = float(rest)
return str(int(number)) if number.is_integer() else rest
def _normalize(spec, raw: str) -> str:
"""스키마 형식으로 맞춘 값. 못 맞추면 ToolRejected — 사장님께 보일 문장을 담는다."""
if spec.type == "bool":
value = _parse_bool(raw)
if value is None:
raise ToolRejected(f"{spec.label} 은(는) 가능·불가로만 정할 수 있어요. 어느 쪽인지 알려 주세요.")
return value
if spec.type == "time":
value, vague = _parse_time(raw)
if vague:
raise ToolRejected(f"{spec.label} 을(를) 오전인지 오후인지 알려 주세요. 예) 오후 3시")
if value is None:
raise ToolRejected(f"{spec.label} 은(는) 시각으로 알려 주세요. 예) 15:00")
return value
if spec.type == "number":
value = _parse_number(raw)
if value is None:
raise ToolRejected(f"{spec.label} 은(는) 숫자로 알려 주세요. 예) 20000")
return value
return raw
def _display(spec, value: str) -> str:
"""사장님께 말할 값. ★ 발행본과 같은 말로 한다(site/src/lib/derive.ts displayValue) —
"true 로 바꿨습니다" 는 사장님이 못 읽는다."""
if spec.type == "bool":
if spec.label.endswith("여부"):
return "있음" if value == "true" else "없음"
return "가능" if value == "true" else "불가"
if spec.type == "number" and spec.unit:
return f"{value}{spec.unit}"
return value
# ── 읽기 ────────────────────────────────────────────────────────────────
async def _get_site_status(ctx: ToolContext, args: dict) -> str:
_fact, site_service = _services()
res = await site_service.get_site(ctx.user, ctx.place_id)
site = res.site
if site is None or site.published_at is None:
return "아직 발행 전입니다. 준비가 되면 발행해 드릴게요."
# ★ 주소는 site_payload 의 함수로 만든다. 문자열로 조립하면 canonical 과 갈린다
# (CLAUDE.md '슬러그 규칙은 두 곳에 있고 같아야 한다').
url = f"{site_payload.publish_origin()}/s/{site_payload.publish_slug(ctx.place, site)}"
when = site.published_at.strftime("%Y-%m-%d %H:%M")
return f"발행되어 있습니다.\n주소: {url}\n마지막 발행: {when}"
async def _list_facts(ctx: ToolContext, args: dict) -> str:
fact_service, _site = _services()
res = await fact_service.list_facts(ctx.user, ctx.place_id, publishable_only=True)
rows = [f for f in (res.facts or []) if (f.value or "").strip()]
schema = get_schema(PlaceCategory(ctx.place.category))
keyword = _arg(args, "keyword")
if keyword:
rows = [f for f in rows if keyword in f.key or keyword in ((schema.get(f.key).label if schema.get(f.key) else ""))]
if not rows:
return "저장된 가게 정보가 아직 없습니다." if not keyword else f"'{keyword}' 로 찾은 정보가 없습니다."
lines = []
for f in rows[:20]:
spec = schema.get(f.key)
lines.append(f"· {spec.label if spec else f.key}: {f.value}")
more = f"\n(그 밖에 {len(rows) - 20}개 더 있습니다)" if len(rows) > 20 else ""
return "지금 저장된 정보입니다.\n" + "\n".join(lines) + more
# ── 되돌릴 수 있는 쓰기 ──────────────────────────────────────────────────
async def _set_fact(ctx: ToolContext, args: dict) -> str:
key, value = _arg(args, "key"), _arg(args, "value")
if not key or not value:
raise ToolRejected("무엇을 어떤 값으로 바꿀지 알려 주세요.")
schema = get_schema(PlaceCategory(ctx.place.category))
spec = schema.get(key)
# ★ LLM 이 없는 key 를 지어낼 수 있다. 스키마가 최종 판정이다.
if spec is None:
raise ToolRejected("그 항목은 이 가게에서 쓰지 않는 정보라 고칠 수 없어요.")
if spec.scope != "place":
raise ToolRejected(f"{spec.label} 은 객실·메뉴마다 다른 값이라 대화로는 아직 고칠 수 없어요.")
value = _normalize(spec, value)
fact_service, _site = _services()
# ★ FactService 를 그대로 통과시킨다. source_type=OWNER 라 노출값을 즉시 교체하고,
# 정정본 잠금·업종 스키마 검증이 전부 거기서 걸린다.
res = await fact_service.upsert_fact(
ctx.user, ctx.place_id, Req_UpsertFact(key=key, value=value, source_type=SourceType.OWNER)
)
if not res.result.success:
raise ToolRejected("그 값을 저장하지 못했습니다. 형식을 확인해 주세요.")
# ★ fact 는 바뀌었지만 사이트는 안 바뀐다. 이 한 줄이 빠지면 사장님은 반영된 줄 알고
# 확인하러 갔다가 옛 값을 보고 "고장났네" 가 된다.
return f"{spec.label} 을(를) {_display(spec, value)} 로 바꿨습니다."
# ── 페이지 구성 ──────────────────────────────────────────────────────────
async def _sections_of(ctx: ToolContext) -> tuple[list, dict]:
"""지금 발행본에 서는 섹션 목록(해석된 결과)과 저장된 theme.
★ `site_payload._sections` 를 그대로 쓴다 — 발행본이 쓰는 바로 그 함수다. 표를 따로
만들면 에디터·발행본·대화 셋이 갈라지고, 사장님은 "껐는데 나온다" 를 겪는다.
★ 저장값이 없어도 업종 기본이 선다. 그래서 아직 한 번도 디자인을 만지지 않은
사업장에서도 대화가 바로 통한다."""
_fact, site_service = _services()
res = await site_service.get_site(ctx.user, ctx.place_id)
theme = dict((res.site.theme if res.site and res.site.theme else {}) or {})
spec = site_payload._DEFAULT_THEME[PlaceCategory(ctx.place.category).value]["sections"]
return site_payload._sections(theme.get("sections"), spec), theme
def _find_section(rows: list, wanted: str):
"""이름이나 id 로 찾는다. 사장님은 '후기' 처럼 줄여 말한다 — 부분 일치도 받는다.
★ 둘 이상 걸리면 **고르지 않는다**(None). 추측으로 고르면 엉뚱한 섹션을 끄고,
사장님은 그 사실을 발행하고 나서야 안다."""
wanted = (wanted or "").strip()
if not wanted:
return None
exact = [r for r in rows if r["id"] == wanted or r["name"] == wanted]
if len(exact) == 1:
return exact[0]
partial = [r for r in rows if wanted in r["name"]]
return partial[0] if len(partial) == 1 else None
async def _save_sections(ctx: ToolContext, theme: dict, rows: list) -> None:
"""★ theme 의 나머지 칸(colors·fontStyle·look…)을 그대로 들고 간다. sections 만 갈아끼운다 —
통째로 새로 쓰면 사장님이 고른 색과 서체가 말없이 사라진다."""
_fact, site_service = _services()
theme["sections"] = rows
res = await site_service.set_theme(ctx.user, ctx.place_id, Req_SiteTheme(theme=theme))
if not res.result.success:
raise ToolRejected("화면 구성을 저장하지 못했습니다. 빌더 화면에서 확인해 주세요.")
_ON_WORDS = {"true", "1", "on", "yes", "켜", "켜기", "켜줘", "보이기", "보여줘", "넣기", "넣어줘"}
_OFF_WORDS = {"false", "0", "off", "no", "꺼", "끄기", "꺼줘", "빼기", "빼줘", "숨기기", "숨겨줘"}
def _switch(raw: str) -> bool | None:
"""켜기 True · 끄기 False · 모르면 None.
★ 모르는 말을 '끄기' 로 읽지 않는다. 스키마가 모든 인자를 필수로 받아 모델이 enabled 를
"" 로 채울 수 있는데, 그걸 끄기로 읽으면 "후기 다시 보여줘" 가 후기를 끈다."""
word = re.sub(r"\s+", "", raw.lower())
if word in _ON_WORDS:
return True
if word in _OFF_WORDS:
return False
return None
def _layout_line(row: dict) -> str:
mark = "켜짐" if row["enabled"] else "꺼짐"
lock = " (항상 켜짐)" if row["locked"] else ""
return f"· {row['name']} — {mark}{lock}"
async def _list_sections(ctx: ToolContext, args: dict) -> str:
rows, _theme = await _sections_of(ctx)
body = "\n".join(_layout_line(r) for r in rows)
return f"지금 홈페이지는 위에서부터 이 순서입니다.\n{body}"
async def _toggle_section(ctx: ToolContext, args: dict) -> str:
rows, theme = await _sections_of(ctx)
row = _find_section(rows, _arg(args, "name"))
if row is None:
raise ToolRejected("어느 부분을 말씀하시는지 못 찾았어요. '목록' 이라고 하시면 보여드릴게요.")
on = _switch(_arg(args, "enabled"))
if on is None:
raise ToolRejected(f"{row['name']} 을(를) 켤지 끌지 알려 주세요.")
# ★ 잠긴 섹션은 끌 수 없다. SEO·필수 마크업 때문에 잠긴 것이라, 끄면 발행 게이트에 걸린다
# (site_payload._sections 가 어차피 켜서 내보낸다 — 화면만 거짓말하게 된다).
if row["locked"] and not on:
raise ToolRejected(f"{row['name']} 은(는) 홈페이지에 꼭 있어야 하는 부분이라 끌 수 없어요.")
if row["enabled"] == on:
return Unchanged(f"{row['name']} 은(는) 이미 {'켜져' if on else '꺼져'} 있어요.")
row["enabled"] = on
await _save_sections(ctx, theme, rows)
return f"{row['name']} 을(를) {'켰습니다' if on else '껐습니다'}."
async def _move_section(ctx: ToolContext, args: dict) -> str:
"""★ 배열 순서가 곧 발행본의 섹션 순서다(site_payload._sections)."""
rows, theme = await _sections_of(ctx)
row = _find_section(rows, _arg(args, "name"))
if row is None:
raise ToolRejected("어느 부분을 말씀하시는지 못 찾았어요. '목록' 이라고 하시면 보여드릴게요.")
to = _arg(args, "to")
rest = [r for r in rows if r["id"] != row["id"]]
if to in ("맨 위", "처음", "위", "top", "first"):
moved, where = [row] + rest, "맨 위로"
elif to in ("맨 아래", "마지막", "아래", "bottom", "last"):
moved, where = rest + [row], "맨 아래로"
else:
anchor = _find_section(rest, to)
if anchor is None:
raise ToolRejected("어디로 옮길지 못 찾았어요. '소개 다음으로' 처럼 말씀해 주세요.")
at = rest.index(anchor) + 1
moved, where = rest[:at] + [row] + rest[at:], f"{anchor['name']} 다음으로"
await _save_sections(ctx, theme, moved)
return f"{row['name']} 을(를) {where} 옮겼습니다."
# ── 사진 ─────────────────────────────────────────────────────────────────
#
# ★ 업로드·교체는 없다. 이미지 재게시 권리가 미결이라 저장 경로를 일부러 안 만들어 뒀다
# (docs/DECISIONS.md 1-2 · 5-3). 아래는 **이미 있는 사진의 노출과 순서**만 바꾼다.
def _photo_name(row) -> str:
"""사장님이 부를 이름. Vision 이 만든 라벨·alt 가 유일한 단서다."""
return (row.label or "").strip() or (row.alt_text or "").strip() or "이름 없는 사진"
async def _photos(ctx: ToolContext) -> list:
res = await _media_service().list_media(ctx.user, ctx.place_id)
return list(res.media or [])
def _primary(rows: list):
"""발행본의 대표 사진 — 나가는 사진 중 객실·메뉴 전용이 아닌 첫 장(site_payload.primary_media).
★ 목록의 첫 장이 아니다. 첫 장이 내린 사진이면 발행본에는 없고, 대표는 그 다음 장이다."""
return next((r for r in rows if r.publishable and not r.unit_id), None)
def _find_photo(rows: list, wanted: str):
"""★ 둘 이상 걸리면 고르지 않는다 — 추측으로 내리면 엉뚱한 사진이 사라지고,
사장님은 발행하고 나서야 안다(섹션과 같은 규칙).
★ 이름이 정확히 맞는 한 장이 있으면 그걸 고른다 — '객실' 과 '객실 욕실' 이 있을 때
'객실' 은 모호하지 않다(섹션의 _find_section 과 같은 순서)."""
wanted = (wanted or "").strip()
if not wanted:
return None
exact = [r for r in rows if _photo_name(r) == wanted]
if len(exact) == 1:
return exact[0]
hits = [r for r in rows if wanted in _photo_name(r)]
return hits[0] if len(hits) == 1 else None
def _pick_photo(rows: list, wanted: str):
"""(사진, 나가는가). 나가는 사진에서 먼저 찾고, 없으면 안 나가는 사진에서 찾는다.
★ 한꺼번에 찾지 않는다. 내린 '객실' 과 나가는 '객실' 이 함께 있으면 모호해져서
사장님이 가리킨 나가는 사진을 못 내린다."""
shown = _find_photo([r for r in rows if r.publishable], wanted)
if shown is not None:
return shown, True
return _find_photo([r for r in rows if not r.publishable], wanted), False
async def _list_photos(ctx: ToolContext, args: dict) -> str:
rows = await _photos(ctx)
if not rows:
return "아직 등록된 사진이 없어요."
primary = _primary(rows)
lines = []
for r in rows[:15]:
where = " (객실·메뉴 전용)" if r.unit_id else ""
mark = "" if r.publishable else " — 지금은 안 나감"
head = "대표 " if r is primary else ""
lines.append(f"· {head}{_photo_name(r)}{where}{mark}")
more = f"\n(그 밖에 {len(rows) - 15}장 더)" if len(rows) > 15 else ""
return "홈페이지에 있는 사진입니다.\n" + "\n".join(lines) + more
async def _hide_photo(ctx: ToolContext, args: dict) -> str:
rows = await _photos(ctx)
row, shown = _pick_photo(rows, _arg(args, "name"))
if row is None:
raise ToolRejected("어느 사진을 말씀하시는지 못 찾았어요. '사진 목록' 이라고 하시면 보여드릴게요.")
if not shown:
return Unchanged(f"'{_photo_name(row)}' 사진은 이미 사이트에 안 나가고 있어요.")
res = await _media_service().hide_media(ctx.user, ctx.place_id, str(row.media_id))
if not res.result.success:
raise ToolRejected("그 사진을 내리지 못했습니다. 빌더 화면에서 확인해 주세요.")
return f"'{_photo_name(row)}' 사진을 내렸습니다."
async def _set_primary_photo(ctx: ToolContext, args: dict) -> str:
rows = await _photos(ctx)
row, shown = _pick_photo(rows, _arg(args, "name"))
if row is None:
raise ToolRejected("어느 사진을 말씀하시는지 못 찾았어요. '사진 목록' 이라고 하시면 보여드릴게요.")
# ★ 안 나가는 사진의 순서만 당기면 "바꿨습니다" 라고 말하는데 발행본의 대표는 그대로다.
if not shown:
raise ToolRejected(f"'{_photo_name(row)}' 은(는) 지금 사이트에 안 나가는 사진이라 대표로 쓸 수 없어요.")
if row.unit_id:
raise ToolRejected(f"'{_photo_name(row)}' 은(는) 객실·메뉴 전용 사진이라 대표로 쓸 수 없어요.")
res = await _media_service().set_primary(ctx.user, ctx.place_id, str(row.media_id))
if not res.result.success:
raise ToolRejected("대표 사진을 바꾸지 못했습니다. 빌더 화면에서 확인해 주세요.")
return f"대표 사진을 '{_photo_name(row)}' 으로 바꿨습니다."
# ── 반쯤 되돌릴 수 있는 것 ───────────────────────────────────────────────
async def _publish(ctx: ToolContext, args: dict) -> str:
_fact, site_service = _services()
res = await site_service.start_build(ctx.user, ctx.place_id, Req_StartBuild(publish=True))
if not res.result.success:
if res.result.code == ErrorType.PLACE_NOT_VERIFIED.value:
raise ToolRejected("가게 확인이 끝나지 않아 발행할 수 없어요. 빌더 화면에서 가게 정보를 먼저 확인해 주세요.")
raise ToolRejected("발행을 시작하지 못했습니다. 빌더 화면에서 확인해 주세요.")
return "발행을 시작했습니다. 1분쯤 걸리고, 끝나면 사이트에 반영됩니다."
# ── 남은 요청의 이름 ─────────────────────────────────────────────────────
#
# ★ 코드가 만든다. 멈춘 뒤 남은 요청을 알릴 때 모델 문장을 실으면 하지 않은 일을 한 것처럼
# 말할 수 있다. 인자에서 이름만 빌려 쓴다.
def _describe_fact(ctx: ToolContext, args: dict) -> str:
spec = get_schema(PlaceCategory(ctx.place.category)).get(_arg(args, "key"))
return f"{spec.label if spec else '가게 정보'} 변경"
def _describe_toggle(ctx: ToolContext, args: dict) -> str:
on = _switch(_arg(args, "enabled"))
verb = "켜기" if on else "끄기" if on is False else "켜기·끄기"
return f"{_arg(args, 'name') or '화면 구성'} {verb}"
def _describe_move(ctx: ToolContext, args: dict) -> str:
return f"{_arg(args, 'name') or '화면 구성'} 옮기기"
def _describe_hide(ctx: ToolContext, args: dict) -> str:
name = _arg(args, "name")
return f"'{name}' 사진 내리기" if name else "사진 내리기"
def describe_action(ctx: ToolContext, tool: Tool, args: dict) -> str:
return tool.describe(ctx, args) if tool.describe else tool.title
class ToolRejected(RuntimeError):
"""도구가 실행을 거절했다 — 사장님께 그대로 보여 줄 한국어 문장을 담는다."""
REGISTRY: dict[str, Tool] = {
t.name: t
for t in [
Tool(
name="get_site_status",
grade=ToolGrade.READ,
title="발행 상태 보기",
summary="홈페이지가 발행됐는지, 주소와 마지막 발행 시각을 알려준다.",
run=_get_site_status,
),
Tool(
name="list_facts",
grade=ToolGrade.READ,
title="가게 정보 보기",
summary="지금 저장된 가게 정보를 보여준다.",
args={"keyword": "찾고 싶은 항목이 있으면 그 말(선택)"},
run=_list_facts,
),
Tool(
name="set_fact",
grade=ToolGrade.REVERSIBLE,
republish=True,
describe=_describe_fact,
target=("key",),
summary="가게 정보 한 항목을 고친다. 사이트에 반영되려면 발행이 따로 필요하다.",
args={"key": "아래 항목 목록의 key", "value": "바꿀 값"},
run=_set_fact,
),
Tool(
name="list_sections",
grade=ToolGrade.READ,
title="화면 구성 보기",
summary="홈페이지가 어떤 부분들로 어떤 순서로 되어 있는지 보여준다.",
run=_list_sections,
),
Tool(
name="toggle_section",
grade=ToolGrade.REVERSIBLE,
republish=True,
describe=_describe_toggle,
target=("name",),
summary="홈페이지의 한 부분을 켜거나 끈다(예: 후기, 사진 갤러리, 예약 안내).",
args={"name": "그 부분의 이름", "enabled": "켜면 true, 끄면 false"},
run=_toggle_section,
),
Tool(
name="move_section",
grade=ToolGrade.REVERSIBLE,
republish=True,
describe=_describe_move,
target=("name",),
summary="홈페이지에서 한 부분의 위치(순서)를 바꾼다.",
args={"name": "옮길 부분의 이름", "to": "'맨 위' · '맨 아래' · 또는 그 뒤에 올 부분의 이름"},
run=_move_section,
),
Tool(
name="list_photos",
grade=ToolGrade.READ,
title="사진 목록 보기",
summary="홈페이지에 올라가 있는 사진 목록을 보여준다(맨 앞이 대표 사진).",
run=_list_photos,
),
Tool(
name="hide_photo",
grade=ToolGrade.REVERSIBLE,
republish=True,
describe=_describe_hide,
target=("name",),
summary="사진 한 장을 홈페이지에서 내린다. 새 사진을 올리는 것은 아직 못 한다.",
args={"name": "그 사진의 이름(라벨)"},
run=_hide_photo,
),
Tool(
name="set_primary_photo",
grade=ToolGrade.REVERSIBLE,
republish=True,
title="대표 사진 바꾸기",
# 대표는 한 장이다 — 두 번 말하면 마지막 것이 남는다.
target=(),
summary="대표 사진을 바꾼다(검색 결과와 목록 카드에 나오는 그림).",
args={"name": "대표로 쓸 사진의 이름(라벨)"},
run=_set_primary_photo,
),
Tool(
name="publish",
grade=ToolGrade.SEMI,
title="발행",
target=(),
summary="바뀐 내용을 홈페이지에 반영한다(재발행).",
run=_publish,
confirm="지금 홈페이지를 다시 발행할까요? 바뀐 내용이 손님에게 보이게 됩니다.",
),
]
}
async def sections_of(ctx: ToolContext):
"""런타임이 프롬프트에 실을 섹션 목록. 도구가 쓰는 것과 같은 함수여야 한다 —
다르면 모델이 본 이름과 도구가 찾는 이름이 갈린다."""
return await _sections_of(ctx)
async def photo_names(ctx: ToolContext) -> list[str]:
"""런타임이 프롬프트에 실을 사진 이름. 도구가 찾는 이름과 **같은 함수**로 만든다 —
다르면 모델이 본 이름과 도구가 찾는 이름이 갈린다."""
# ★ 나가는 사진만, 대표를 맨 앞에 싣는다 — 프롬프트는 '맨 앞이 대표' 라고 말한다.
# 내린 사진이 섞이면 모델이 그걸 대표로 고르고, 객실 전용 사진이 앞에 있으면 대표를 잘못 안다.
rows = [r for r in await _photos(ctx) if r.publishable]
primary = _primary(rows)
if primary is not None:
rows = [primary] + [r for r in rows if r is not primary]
return [_photo_name(r) for r in rows[:15]]
def describe() -> list[dict]:
"""프롬프트에 실을 도구 목록. ★ 등급은 싣지 않는다 — 모델이 알 필요도, 정할 이유도 없다."""
return [{"name": t.name, "설명": t.summary, "args": t.args} for t in REGISTRY.values()]
def fields_of(place) -> list[dict]:
schema = get_schema(PlaceCategory(place.category))
return [
{"key": k, "label": spec.label, "type": spec.type, "unit": spec.unit}
for k, spec in schema.fields.items()
if spec.scope == "place"
]