238 lines
8.2 KiB
Python
238 lines
8.2 KiB
Python
# router.py
|
|
|
|
import os
|
|
import logging
|
|
from fastapi import APIRouter
|
|
from pydantic import BaseModel
|
|
|
|
from reasoning.reasoning import reason_check
|
|
from reasoning.reflection import reflect_notes
|
|
from reasoning.refine import refine_answer
|
|
from persona.speak import speak
|
|
from persona.identity import load_identity
|
|
from context import collect_context, update_last_assistant_message
|
|
from intake.intake import add_exchange_internal
|
|
|
|
from autonomy.monologue.monologue import InnerMonologue
|
|
from autonomy.self.state import load_self_state
|
|
|
|
|
|
# -------------------------------------------------------------------
|
|
# Setup
|
|
# -------------------------------------------------------------------
|
|
VERBOSE_DEBUG = os.getenv("VERBOSE_DEBUG", "false").lower() == "true"
|
|
logger = logging.getLogger(__name__)
|
|
|
|
if VERBOSE_DEBUG:
|
|
logger.setLevel(logging.DEBUG)
|
|
|
|
console_handler = logging.StreamHandler()
|
|
console_handler.setFormatter(logging.Formatter(
|
|
'%(asctime)s [ROUTER] %(levelname)s: %(message)s',
|
|
datefmt='%H:%M:%S'
|
|
))
|
|
logger.addHandler(console_handler)
|
|
|
|
try:
|
|
os.makedirs('/app/logs', exist_ok=True)
|
|
file_handler = logging.FileHandler('/app/logs/cortex_verbose_debug.log', mode='a')
|
|
file_handler.setFormatter(logging.Formatter(
|
|
'%(asctime)s [ROUTER] %(levelname)s: %(message)s',
|
|
datefmt='%Y-%m-%d %H:%M:%S'
|
|
))
|
|
logger.addHandler(file_handler)
|
|
logger.debug("VERBOSE_DEBUG enabled for router.py")
|
|
except Exception as e:
|
|
logger.debug(f"File logging failed: {e}")
|
|
|
|
|
|
cortex_router = APIRouter()
|
|
inner_monologue = InnerMonologue()
|
|
|
|
|
|
# -------------------------------------------------------------------
|
|
# Models
|
|
# -------------------------------------------------------------------
|
|
class ReasonRequest(BaseModel):
|
|
session_id: str
|
|
user_prompt: str
|
|
temperature: float | None = None
|
|
|
|
|
|
# -------------------------------------------------------------------
|
|
# /reason endpoint
|
|
# -------------------------------------------------------------------
|
|
@cortex_router.post("/reason")
|
|
async def run_reason(req: ReasonRequest):
|
|
|
|
if VERBOSE_DEBUG:
|
|
logger.debug(f"\n{'='*80}")
|
|
logger.debug(f"[PIPELINE START] Session: {req.session_id}")
|
|
logger.debug(f"[PIPELINE START] User prompt: {req.user_prompt[:200]}...")
|
|
logger.debug(f"{'='*80}\n")
|
|
|
|
# ----------------------------------------------------------------
|
|
# STAGE 0 — Context
|
|
# ----------------------------------------------------------------
|
|
if VERBOSE_DEBUG:
|
|
logger.debug("[STAGE 0] Collecting unified context...")
|
|
|
|
context_state = await collect_context(req.session_id, req.user_prompt)
|
|
|
|
if VERBOSE_DEBUG:
|
|
logger.debug(f"[STAGE 0] Context collected - {len(context_state.get('rag', []))} RAG results")
|
|
|
|
# ----------------------------------------------------------------
|
|
# STAGE 0.5 — Identity
|
|
# ----------------------------------------------------------------
|
|
if VERBOSE_DEBUG:
|
|
logger.debug("[STAGE 0.5] Loading identity block...")
|
|
|
|
identity_block = load_identity(req.session_id)
|
|
|
|
if VERBOSE_DEBUG:
|
|
logger.debug(f"[STAGE 0.5] Identity loaded: {identity_block.get('name', 'Unknown')}")
|
|
|
|
# ----------------------------------------------------------------
|
|
# STAGE 0.6 — Inner Monologue (observer-only)
|
|
# ----------------------------------------------------------------
|
|
if VERBOSE_DEBUG:
|
|
logger.debug("[STAGE 0.6] Running inner monologue...")
|
|
|
|
try:
|
|
self_state = load_self_state()
|
|
|
|
mono_context = {
|
|
"user_message": req.user_prompt,
|
|
"session_id": req.session_id,
|
|
"self_state": self_state,
|
|
"context_summary": context_state,
|
|
}
|
|
|
|
inner_result = await inner_monologue.process(mono_context)
|
|
logger.info(f"[INNER_MONOLOGUE] {inner_result}")
|
|
|
|
except Exception as e:
|
|
logger.warning(f"[INNER_MONOLOGUE] failed: {e}")
|
|
|
|
# ----------------------------------------------------------------
|
|
# STAGE 1 — Intake summary
|
|
# ----------------------------------------------------------------
|
|
intake_summary = "(no context available)"
|
|
if context_state.get("intake"):
|
|
l20 = context_state["intake"].get("L20")
|
|
if isinstance(l20, dict):
|
|
intake_summary = l20.get("summary", intake_summary)
|
|
elif isinstance(l20, str):
|
|
intake_summary = l20
|
|
|
|
if VERBOSE_DEBUG:
|
|
logger.debug(f"[STAGE 1] Intake summary extracted (L20): {intake_summary[:150]}...")
|
|
|
|
# ----------------------------------------------------------------
|
|
# STAGE 2 — Reflection
|
|
# ----------------------------------------------------------------
|
|
if VERBOSE_DEBUG:
|
|
logger.debug("[STAGE 2] Running reflection...")
|
|
|
|
try:
|
|
reflection = await reflect_notes(intake_summary, identity_block=identity_block)
|
|
reflection_notes = reflection.get("notes", [])
|
|
except Exception as e:
|
|
reflection_notes = []
|
|
if VERBOSE_DEBUG:
|
|
logger.debug(f"[STAGE 2] Reflection failed: {e}")
|
|
|
|
# ----------------------------------------------------------------
|
|
# STAGE 3 — Reasoning (draft)
|
|
# ----------------------------------------------------------------
|
|
if VERBOSE_DEBUG:
|
|
logger.debug("[STAGE 3] Running reasoning (draft)...")
|
|
|
|
draft = await reason_check(
|
|
req.user_prompt,
|
|
identity_block=identity_block,
|
|
rag_block=context_state.get("rag", []),
|
|
reflection_notes=reflection_notes,
|
|
context=context_state
|
|
)
|
|
|
|
# ----------------------------------------------------------------
|
|
# STAGE 4 — Refinement
|
|
# ----------------------------------------------------------------
|
|
if VERBOSE_DEBUG:
|
|
logger.debug("[STAGE 4] Running refinement...")
|
|
|
|
result = await refine_answer(
|
|
draft_output=draft,
|
|
reflection_notes=reflection_notes,
|
|
identity_block=identity_block,
|
|
rag_block=context_state.get("rag", []),
|
|
)
|
|
|
|
final_neutral = result["final_output"]
|
|
|
|
# ----------------------------------------------------------------
|
|
# STAGE 5 — Persona
|
|
# ----------------------------------------------------------------
|
|
if VERBOSE_DEBUG:
|
|
logger.debug("[STAGE 5] Applying persona layer...")
|
|
|
|
persona_answer = await speak(final_neutral)
|
|
|
|
# ----------------------------------------------------------------
|
|
# STAGE 6 — Session update
|
|
# ----------------------------------------------------------------
|
|
update_last_assistant_message(req.session_id, persona_answer)
|
|
|
|
if VERBOSE_DEBUG:
|
|
logger.debug(f"\n{'='*80}")
|
|
logger.debug(f"[PIPELINE COMPLETE] Session: {req.session_id}")
|
|
logger.debug(f"[PIPELINE COMPLETE] Final answer length: {len(persona_answer)} chars")
|
|
logger.debug(f"{'='*80}\n")
|
|
|
|
# ----------------------------------------------------------------
|
|
# RETURN
|
|
# ----------------------------------------------------------------
|
|
return {
|
|
"draft": draft,
|
|
"neutral": final_neutral,
|
|
"persona": persona_answer,
|
|
"reflection": reflection_notes,
|
|
"session_id": req.session_id,
|
|
"context_summary": {
|
|
"rag_results": len(context_state.get("rag", [])),
|
|
"minutes_since_last": context_state.get("minutes_since_last_msg"),
|
|
"message_count": context_state.get("message_count"),
|
|
"mode": context_state.get("mode"),
|
|
}
|
|
}
|
|
|
|
|
|
# -------------------------------------------------------------------
|
|
# /ingest endpoint (internal)
|
|
# -------------------------------------------------------------------
|
|
class IngestPayload(BaseModel):
|
|
session_id: str
|
|
user_msg: str
|
|
assistant_msg: str
|
|
|
|
|
|
@cortex_router.post("/ingest")
|
|
async def ingest(payload: IngestPayload):
|
|
try:
|
|
update_last_assistant_message(payload.session_id, payload.assistant_msg)
|
|
except Exception as e:
|
|
logger.warning(f"[INGEST] Session update failed: {e}")
|
|
|
|
try:
|
|
add_exchange_internal({
|
|
"session_id": payload.session_id,
|
|
"user_msg": payload.user_msg,
|
|
"assistant_msg": payload.assistant_msg,
|
|
})
|
|
except Exception as e:
|
|
logger.warning(f"[INGEST] Intake update failed: {e}")
|
|
|
|
return {"status": "ok", "session_id": payload.session_id}
|