{{ error }}
+{% endif %} + +{% endblock %} diff --git a/chat/templates/bot_list.html b/chat/templates/bot_list.html new file mode 100644 index 0000000..f3b4562 --- /dev/null +++ b/chat/templates/bot_list.html @@ -0,0 +1,17 @@ +{% extends "base.html" %} +{% block title %}Bots - chat{% endblock %} +{% block content %} +No bots yet. Create your first bot.
+{% endif %} +{% endblock %} diff --git a/chat/web/__init__.py b/chat/web/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/chat/web/bots.py b/chat/web/bots.py new file mode 100644 index 0000000..bbf077d --- /dev/null +++ b/chat/web/bots.py @@ -0,0 +1,116 @@ +from __future__ import annotations +import sqlite3 +from pathlib import Path +from fastapi import APIRouter, Depends, Form, HTTPException, Request +from fastapi.responses import RedirectResponse, HTMLResponse +from fastapi.templating import Jinja2Templates + +from chat.eventlog.log import append_event +from chat.eventlog.projector import project +from chat.state.entities import list_bots + +TEMPLATES = Jinja2Templates(directory=str(Path(__file__).resolve().parent.parent / "templates")) + +router = APIRouter() + +REQUIRED_FIELDS = ("id", "name", "persona", "initial_relationship_to_you", "kickoff_prose") + + +def get_conn(request: Request): + settings = request.app.state.settings + db_path: Path = settings.db_path + db_path.parent.mkdir(parents=True, exist_ok=True) + conn = sqlite3.connect(db_path, check_same_thread=False) + conn.execute("PRAGMA journal_mode=WAL") + conn.execute("PRAGMA foreign_keys=ON") + try: + yield conn + conn.commit() + finally: + conn.close() + + +def _split_voice_samples(text: str) -> list[str]: + if not text or not text.strip(): + return [] + # Split on a line containing only "---" (with optional surrounding whitespace). + parts: list[str] = [] + buf: list[str] = [] + for line in text.splitlines(): + if line.strip() == "---": + if buf: + parts.append("\n".join(buf).strip()) + buf = [] + continue + buf.append(line) + if buf: + parts.append("\n".join(buf).strip()) + return [p for p in parts if p] + + +def _split_traits(text: str) -> list[str]: + if not text or not text.strip(): + return [] + items: list[str] = [] + for line in text.splitlines(): + line = line.strip() + if not line: + continue + if "," in line: + items.extend(p.strip() for p in line.split(",")) + else: + items.append(line) + return [t for t in items if t] + + +@router.get("/bots", response_class=HTMLResponse) +async def bots_list(request: Request, conn=Depends(get_conn)): + bots = list_bots(conn) + return TEMPLATES.TemplateResponse(request, "bot_list.html", {"bots": bots}) + + +@router.get("/bots/new", response_class=HTMLResponse) +async def bot_form(request: Request): + return TEMPLATES.TemplateResponse(request, "bot_form.html", {"values": {}, "error": None}) + + +@router.post("/bots/new") +async def bot_create( + request: Request, + id: str = Form(""), + name: str = Form(""), + persona: str = Form(""), + voice_samples: str = Form(""), + traits: str = Form(""), + backstory: str = Form(""), + initial_relationship_to_you: str = Form(""), + kickoff_prose: str = Form(""), + conn=Depends(get_conn), +): + values = { + "id": id, + "name": name, + "persona": persona, + "voice_samples": voice_samples, + "traits": traits, + "backstory": backstory, + "initial_relationship_to_you": initial_relationship_to_you, + "kickoff_prose": kickoff_prose, + } + missing = [f for f in REQUIRED_FIELDS if not values[f].strip()] + if missing: + raise HTTPException(status_code=400, detail=f"missing required: {', '.join(missing)}") + + payload = { + "id": id.strip(), + "name": name.strip(), + "persona": persona.strip(), + "voice_samples": _split_voice_samples(voice_samples), + "traits": _split_traits(traits), + "backstory": backstory.strip(), + "initial_relationship_to_you": initial_relationship_to_you.strip(), + "kickoff_prose": kickoff_prose.strip(), + } + append_event(conn, kind="bot_authored", payload=payload) + project(conn) + return RedirectResponse(url=f"/bots/{payload['id']}/kickoff", status_code=303) diff --git a/pyproject.toml b/pyproject.toml index c0593a9..b72bc90 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -13,6 +13,7 @@ dependencies = [ "tiktoken>=0.7", "jinja2>=3.1", "aiosqlite>=0.20", + "python-multipart>=0.0.9", ] [project.optional-dependencies] diff --git a/tests/test_bot_authoring.py b/tests/test_bot_authoring.py new file mode 100644 index 0000000..5b8a82d --- /dev/null +++ b/tests/test_bot_authoring.py @@ -0,0 +1,132 @@ +from __future__ import annotations + +import pytest +from fastapi.testclient import TestClient + +from chat.app import app + + +@pytest.fixture +def client(tmp_path, monkeypatch): + config_path = tmp_path / "config.toml" + config_path.write_text('featherless_api_key = "test"\n') + monkeypatch.setenv("CHAT_CONFIG_PATH", str(config_path)) + monkeypatch.setenv("CHAT_DB_PATH", str(tmp_path / "test.db")) + with TestClient(app) as c: + yield c + + +def test_get_new_bot_form_renders(client): + response = client.get("/bots/new") + assert response.status_code == 200 + body = response.text.lower() + assert "