mirror of
https://github.com/R0m1k3/Loki.git
synced 2026-10-23 06:58:14 +02:00
Phase 4 : boucle agentique, outils fichiers et aperçu live
Backend : - tools.py : read_file / write_file / list_dir confinés au workspace (anti-traversal) - agent.py : boucle de tool-calling itérative au-dessus d'Ollama, génère des événements (token, tool_call, tool_result, final) - routes/chat.py : pilote run_agent, relaie les événements en SSE, persiste la réponse finale + le récap des outils (meta) - routes/files.py : arborescence et contenu du workspace (confiné) - db.py : colonne meta sur messages (+ migration douce), list_messages_for_model Frontend : - ToolCard : carte d'appel d'outil dans le fil (statut en cours/terminé/échec) - streamChat : événements tool_call / tool_result - store : streamTools, fileTree, preview (ouverture auto d'un HTML écrit) - LeftPanel : arborescence réelle du workspace, clic -> aperçu - PreviewPanel : rendu HTML en iframe + onglet Code - ChatPanel : rendu des outils dans les bulles agent (streaming + persisté) Tests manuels : outils + confinement, boucle agent (mock Ollama), chat HTTP de bout en bout (SSE, persistance meta, fichiers écrits, confinement route). Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01SVay7z3y7q2gEe54ByAE6N
This commit is contained in:
1 parent
48cbcaafaf
commit
b6bf4edf97
13 files changed
+677
-80
No files matched your search
+40
-32
@@ -1,29 +1,30 @@
|
||||
"""Route de conversation : streaming token par token depuis Ollama (SSE).
|
||||
"""Route de conversation agentique : boucle d'outils + streaming SSE.
|
||||
|
||||
Flux :
|
||||
1. on enregistre le message utilisateur ;
|
||||
2. on rejoue l'historique de la session vers Ollama en streaming ;
|
||||
3. on relaie chaque token au client (SSE) ;
|
||||
4. on enregistre la réponse complète de l'assistant.
|
||||
2. on lance la boucle agentique (run_agent) sur l'historique de la session ;
|
||||
3. on relaie tokens et événements d'outils au client (SSE) ;
|
||||
4. on enregistre la réponse finale de l'assistant + le récap des outils.
|
||||
"""
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
|
||||
import httpx
|
||||
from fastapi import APIRouter, HTTPException
|
||||
from fastapi.responses import StreamingResponse
|
||||
from pydantic import BaseModel
|
||||
|
||||
from .. import db
|
||||
from ..agent import run_agent
|
||||
from ..config import settings
|
||||
from ..ollama_client import ollama
|
||||
|
||||
router = APIRouter(prefix="/api", tags=["chat"])
|
||||
|
||||
SYSTEM_PROMPT = (
|
||||
"Tu es Loki, un assistant de développement local. Tu écris du code clair, "
|
||||
"commenté en français, et tu réponds de façon concise et utile."
|
||||
"Tu es Loki, un assistant de développement local agentique. Tu disposes "
|
||||
"d'outils pour lire, écrire et lister des fichiers dans le workspace. "
|
||||
"Utilise-les pour accomplir les tâches concrètement, puis réponds de façon "
|
||||
"concise en français. Après avoir écrit un fichier, propose un aperçu."
|
||||
)
|
||||
|
||||
|
||||
@@ -32,6 +33,7 @@ class ChatRequest(BaseModel):
|
||||
content: str
|
||||
model: str | None = None
|
||||
options: dict | None = None
|
||||
tools_enabled: bool = True
|
||||
|
||||
|
||||
def _sse(event: str, data: dict) -> str:
|
||||
@@ -46,39 +48,45 @@ async def chat(req: ChatRequest) -> StreamingResponse:
|
||||
|
||||
model = req.model or session.get("model") or settings.default_model
|
||||
|
||||
# Premier message : on titre la session avec un extrait.
|
||||
history = db.list_messages(req.session_id)
|
||||
if not history:
|
||||
# Premier message : titre la session avec un extrait.
|
||||
if not db.list_messages(req.session_id):
|
||||
title = req.content.strip().split("\n")[0][:60] or "Nouvelle session"
|
||||
db.rename_session(req.session_id, title)
|
||||
|
||||
db.add_message(req.session_id, "user", req.content, None)
|
||||
|
||||
# Contexte envoyé à Ollama : invite système + historique complet.
|
||||
convo = [{"role": "system", "content": SYSTEM_PROMPT}]
|
||||
convo += [
|
||||
{"role": m["role"], "content": m["content"]}
|
||||
for m in db.list_messages(req.session_id)
|
||||
]
|
||||
convo += db.list_messages_for_model(req.session_id)
|
||||
|
||||
async def event_stream():
|
||||
yield _sse("start", {"model": model})
|
||||
full = ""
|
||||
try:
|
||||
async for chunk in ollama.chat(
|
||||
model, convo, options=req.options, stream=True
|
||||
):
|
||||
token = chunk.get("message", {}).get("content", "")
|
||||
if token:
|
||||
full += token
|
||||
yield _sse("token", {"content": token})
|
||||
if chunk.get("done"):
|
||||
break
|
||||
except (httpx.HTTPError, OSError) as exc:
|
||||
yield _sse("error", {"message": str(exc)})
|
||||
final_content = ""
|
||||
tools_meta: list[dict] = []
|
||||
|
||||
if full:
|
||||
db.add_message(req.session_id, "assistant", full, model)
|
||||
yield _sse("done", {"content": full, "model": model})
|
||||
async for ev in run_agent(
|
||||
model, convo, options=req.options, tools_enabled=req.tools_enabled
|
||||
):
|
||||
etype = ev.pop("type")
|
||||
if etype == "token":
|
||||
yield _sse("token", ev)
|
||||
elif etype == "tool_call":
|
||||
yield _sse("tool_call", ev)
|
||||
elif etype == "tool_result":
|
||||
yield _sse("tool_result", ev)
|
||||
elif etype == "error":
|
||||
yield _sse("error", ev)
|
||||
elif etype == "final":
|
||||
final_content = ev["content"]
|
||||
tools_meta = ev["tools"]
|
||||
|
||||
if final_content or tools_meta:
|
||||
db.add_message(
|
||||
req.session_id,
|
||||
"assistant",
|
||||
final_content,
|
||||
model,
|
||||
meta={"tools": tools_meta} if tools_meta else None,
|
||||
)
|
||||
yield _sse("done", {"content": final_content, "tools": tools_meta})
|
||||
|
||||
return StreamingResponse(event_stream(), media_type="text/event-stream")
|
||||
@@ -0,0 +1,49 @@
|
||||
"""Routes de lecture du workspace (arborescence + contenu d'un fichier)."""
|
||||
from __future__ import annotations
|
||||
|
||||
import os
|
||||
|
||||
from fastapi import APIRouter, HTTPException
|
||||
|
||||
from ..config import settings
|
||||
from ..tools import ToolError, _safe_path
|
||||
|
||||
router = APIRouter(prefix="/api/files", tags=["files"])
|
||||
|
||||
|
||||
def _tree(path: str) -> list[dict]:
|
||||
"""Arborescence triée (dossiers d'abord) du workspace."""
|
||||
items = []
|
||||
for name in sorted(os.listdir(path)):
|
||||
if name.startswith("."):
|
||||
continue
|
||||
full = os.path.join(path, name)
|
||||
rel = os.path.relpath(full, os.path.abspath(settings.workspace_dir))
|
||||
if os.path.isdir(full):
|
||||
items.append({"name": name, "path": rel, "type": "dir",
|
||||
"children": _tree(full)})
|
||||
else:
|
||||
items.append({"name": name, "path": rel, "type": "file",
|
||||
"size": os.path.getsize(full)})
|
||||
# Dossiers avant fichiers
|
||||
items.sort(key=lambda x: (x["type"] != "dir", x["name"]))
|
||||
return items
|
||||
|
||||
|
||||
@router.get("")
|
||||
async def list_files() -> dict:
|
||||
root = os.path.abspath(settings.workspace_dir)
|
||||
os.makedirs(root, exist_ok=True)
|
||||
return {"tree": _tree(root)}
|
||||
|
||||
|
||||
@router.get("/content")
|
||||
async def file_content(path: str) -> dict:
|
||||
try:
|
||||
target = _safe_path(path)
|
||||
except ToolError as exc:
|
||||
raise HTTPException(400, str(exc))
|
||||
if not os.path.isfile(target):
|
||||
raise HTTPException(404, "fichier introuvable")
|
||||
with open(target, "r", encoding="utf-8", errors="replace") as f:
|
||||
return {"path": path, "content": f.read()}
|
||||
Reference in new issue
Block a user