248 lines
9.5 KiB
Python
248 lines
9.5 KiB
Python
import curses
|
|
import json
|
|
import threading
|
|
from datetime import datetime
|
|
import config
|
|
from .theme import init_colors
|
|
from .render import draw
|
|
from .input import handle_key
|
|
from . import keycodes, menubar
|
|
from .agent import log, WELCOME_ART
|
|
from services.session_manager_neo import NeoSessionManager, NeoSession
|
|
from lib import ragroleplay, moop
|
|
from lib import personality as personality_mod
|
|
|
|
|
|
class HendrikTUI:
|
|
def __init__(self, llm_client, tools_definition, TOOLS, TOOL_HANDLERS,
|
|
build_system_prompt, agent_max_iterations):
|
|
self.llm = llm_client
|
|
self.tools_def = tools_definition
|
|
self.TOOLS = TOOLS
|
|
self.TOOL_HANDLERS = TOOL_HANDLERS
|
|
self.build_system_prompt = build_system_prompt
|
|
self.agent_max_iterations = agent_max_iterations
|
|
self.character_name = config.AGENT_CHARACTER or "AI Agent"
|
|
|
|
self.messages = None
|
|
self.log = []
|
|
self.input_buffer = [""]
|
|
self.input_line = 0
|
|
self.input_col = 0
|
|
self.scroll = 0
|
|
self.processing = False
|
|
self.running = True
|
|
self.h, self.w = 0, 0
|
|
|
|
# State menu bar
|
|
self.menu_active = False # menu bar sedang fokus (F10)
|
|
self.menu_open = False # dropdown menu sedang terbuka
|
|
self.active_menu = 0 # index menu aktif (0-3)
|
|
self.menu_sel = 0 # index item yang diseleksi di dropdown
|
|
self.cursor_yx = None # posisi kursor input (untuk restore)
|
|
|
|
self.agent_thread: threading.Thread | None = None
|
|
self.agent_done = threading.Event()
|
|
|
|
self.session_mgr = NeoSessionManager()
|
|
self.current_session: NeoSession | None = None
|
|
|
|
def switch_model_set(self, set_id):
|
|
default_chain = moop.default_chain(None, "llm")
|
|
target = moop.get_set(None, set_id)
|
|
if not target:
|
|
return
|
|
chain = moop.resolve_chain(None, set_id)
|
|
if not chain:
|
|
log(self, "error", "Model set tidak memiliki model yang aktif.")
|
|
return
|
|
if not default_chain:
|
|
moop.set_default(None, "llm", set_id)
|
|
self.llm.set_chains({"llm": chain})
|
|
if self.current_session:
|
|
self.session_mgr.update_model_info(
|
|
self.current_session.doc_id, self._model_info()
|
|
)
|
|
candidate = chain[0]
|
|
log(self, "system",
|
|
f"Model \u2192 {candidate['provider']} / {candidate['model']}")
|
|
|
|
def _model_info(self) -> dict:
|
|
return {
|
|
"provider": moop.resolve_provider(None, self.llm.base_url, self.llm.model),
|
|
"base_url": self.llm.base_url,
|
|
"model": self.llm.model,
|
|
}
|
|
|
|
def _inject_user_context(self) -> str | None:
|
|
user_id = config.TUI_USER_ID
|
|
if not user_id:
|
|
return None
|
|
|
|
try:
|
|
results = ragroleplay.user_load(
|
|
config.ragroleplay_db_path,
|
|
user_id=user_id,
|
|
character=personality_mod.PERSONALITY.name
|
|
)
|
|
if results:
|
|
u = results[0]
|
|
return (
|
|
f'[User Context]\n'
|
|
f'ID: {u["id"]}\n'
|
|
f'Nama: {u["fullname"]} ({u["nickname"]})\n'
|
|
f'Family: {u.get("familyname", "-") or "-"}\n'
|
|
f'Character: {u.get("character", "-") or "-"}\n'
|
|
f'Alias: {u.get("alias", "-") or "-"}\n'
|
|
f'Salutation: {u.get("salutation", "-") or "-"}\n'
|
|
f'Persona: {u.get("persona", "-") or "-"}\n'
|
|
f'Telegram: {u.get("telegram_id", "-") or "-"} / @{u.get("telegram_username", "-") or "-"}\n'
|
|
f'XMPP: {u.get("xmpp_username", "-") or "-"}\n'
|
|
f'[/User Context]\n'
|
|
f'[PENTING: Kamu SUDAH mengenal user ini. WAJIB panggil memories_latest(character="{personality_mod.PERSONALITY.name}", user_id="{u["id"]}", limit=10) untuk mengambil riwayat percakapan terbaru. GUNAKAN data kondisi emotional/physical dari memori terbaru untuk melanjutkan state character secara natural sebelum merespon.]'
|
|
)
|
|
else:
|
|
return (
|
|
f'[User Context: Pengguna baru — belum ada di database]\n'
|
|
f'[Platform: TUI]\n'
|
|
f'[PENTING: Kamu BELUM mengenal user ini. WAJIB tanya nama sebagai pembuka. '
|
|
f'Setelah nama diketahui, simpan via users_store. '
|
|
f'Lanjutkan percakapan natural dan proaktif melengkapi data user lainnya di pesan berikutnya.]'
|
|
)
|
|
except Exception:
|
|
return None
|
|
|
|
def new_session(self):
|
|
self.current_session = None
|
|
|
|
system_prompt = self.build_system_prompt(
|
|
tools_definition=self.tools_def,
|
|
character=config.AGENT_CHARACTER or None,
|
|
skills=config.AGENT_SKILLS.split(",") if config.AGENT_SKILLS else None,
|
|
)
|
|
messages = [{"role": "system", "content": system_prompt}]
|
|
|
|
context = self._inject_user_context()
|
|
if context:
|
|
messages.append({"role": "system", "content": context})
|
|
|
|
self.messages = messages
|
|
# ---------------------------------
|
|
|
|
self.log.clear()
|
|
self.scroll = 0
|
|
self.input_buffer = [""]
|
|
self.input_line = 0
|
|
self.input_col = 0
|
|
log(self, "welcome", WELCOME_ART)
|
|
|
|
def switch_session(self, doc_id: int):
|
|
session = self.session_mgr.get(doc_id)
|
|
if not session:
|
|
return
|
|
self.current_session = session
|
|
self.messages = list(session.messages)
|
|
self.log.clear()
|
|
self.scroll = 0
|
|
log(self, "welcome", WELCOME_ART)
|
|
for i, msg in enumerate(session.messages):
|
|
if msg["role"] == "user":
|
|
if i > 0:
|
|
log(self, "sep", "")
|
|
log(self, "user", msg["content"])
|
|
mi = msg.get("model_info")
|
|
if mi:
|
|
self.log[-1]["model_info"] = (mi.get("provider"), mi.get("model"))
|
|
elif msg["role"] == "assistant":
|
|
# Log tool calls jika ada
|
|
if msg.get("tool_calls"):
|
|
for tc in msg["tool_calls"]:
|
|
log(self, "tool_call", json.dumps({
|
|
"name": tc["function"]["name"],
|
|
"arguments": tc["function"]["arguments"],
|
|
}))
|
|
# Log AI content jika ada dan ini adalah response final
|
|
next_role = session.messages[i + 1]["role"] if i + 1 < len(session.messages) else None
|
|
if next_role in ("user", None):
|
|
if msg.get("content"):
|
|
log(self, "ai", msg["content"])
|
|
elif msg["role"] == "tool":
|
|
# Tool result - tidak perlu di-log secara eksplisit
|
|
pass
|
|
if session.messages and session.messages[-1]["role"] == "assistant":
|
|
log(self, "sep", "")
|
|
|
|
def run(self):
|
|
try:
|
|
curses.wrapper(self._main)
|
|
except KeyboardInterrupt:
|
|
pass
|
|
|
|
def _main(self, stdscr):
|
|
curses.use_default_colors()
|
|
init_colors()
|
|
stdscr.keypad(True)
|
|
curses.raw() # Ctrl+C sebagai key code 3, bukan SIGINT → KeyboardInterrupt
|
|
curses.set_escdelay(keycodes.ESC_PEEK_MS) # ESC/Alt/F-key langsung diproses, tidak menunggu 1s
|
|
stdscr.refresh()
|
|
|
|
self.messages = [{"role": "system", "content": self.build_system_prompt(
|
|
tools_definition=self.tools_def,
|
|
character=config.AGENT_CHARACTER or None,
|
|
skills=config.AGENT_SKILLS.split(",") if config.AGENT_SKILLS else None,
|
|
)}]
|
|
context = self._inject_user_context()
|
|
if context:
|
|
self.messages.append({"role": "system", "content": context})
|
|
log(self, "welcome", WELCOME_ART)
|
|
|
|
keycodes.enable_extended_keys()
|
|
try:
|
|
self._run_loop(stdscr)
|
|
finally:
|
|
keycodes.disable_extended_keys()
|
|
|
|
def _run_loop(self, stdscr):
|
|
while self.running:
|
|
self.h, self.w = stdscr.getmaxyx()
|
|
if self.h < 14 or self.w < 40:
|
|
stdscr.erase()
|
|
stdscr.addstr(0, 0, "Terminal too small (min 40x14)")
|
|
stdscr.refresh()
|
|
stdscr.getch()
|
|
continue
|
|
|
|
draw(self, stdscr)
|
|
curses.curs_set(2)
|
|
|
|
timeout_ms = 100 if self.processing else -1
|
|
stdscr.timeout(timeout_ms)
|
|
|
|
try:
|
|
key = keycodes.read_key(stdscr, timeout_ms)
|
|
except KeyboardInterrupt:
|
|
if self.processing:
|
|
self.llm.cancel_requested = True
|
|
self.agent_done.set()
|
|
else:
|
|
break
|
|
key = -1
|
|
|
|
# Routing key:
|
|
# 1. menu bar fokus → navigasi menu bar
|
|
# 2. shortcut global → aksi fitur (Ctrl+N, F2, F4, F6, ...)
|
|
# 3. mnemonic Alt+ → fokus menu
|
|
# 4. sisanya → editing input
|
|
if self.menu_active:
|
|
menubar.handle_menubar_key(self, stdscr, key)
|
|
else:
|
|
if not menubar.dispatch_global_key(self, stdscr, key):
|
|
if not menubar.handle_alt_menu_key(self, stdscr, key):
|
|
handle_key(self, stdscr, key)
|
|
|
|
if self.agent_done.is_set():
|
|
self.agent_thread.join()
|
|
self.agent_done.clear()
|
|
self.processing = False
|
|
self.agent_thread = None
|