Chat atelier : evenement SSE sources + affichage des pages utilisees

Le Brain emet un evenement sources (source_id, page, score des passages
retenus) AVANT le premier token ; le Core le relaie tel quel (JSON brut) ;
l UI affiche une ligne discrete sous la reponse (ex: 12, 47, 103),
prefixee du nom de fichier si plusieurs sources. Transparence pour le MJ
et diagnostic immediat quand le RAG repond a cote.

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
2026-06-10 15:45:56 +02:00
parent 0e4820a2f8
commit 3d1cf6e495
10 changed files with 102 additions and 10 deletions

View File

@@ -85,9 +85,14 @@ async def chat_notebook_stream(
async def event_stream() -> AsyncIterator[str]:
try:
async for token in use_case.stream(body.source_ids, messages, context=body.context, top_k=top_k):
if token:
yield sse_event("token", {"token": token})
async for ev in use_case.stream(body.source_ids, messages, context=body.context, top_k=top_k):
if ev["type"] == "token":
if ev.get("token"):
yield sse_event("token", {"token": ev["token"]})
else:
# 'sources' (et tout futur évènement typé) : relayé tel quel.
ev_type = ev.pop("type")
yield sse_event(ev_type, ev)
yield sse_event("done", {})
except (LLMProviderError, EmbeddingError) as exc:
yield sse_event("error", {"message": str(exc)})

View File

@@ -68,13 +68,25 @@ class NotebookChatUseCase:
messages: list[ChatMessage],
context: str = "",
top_k: int = 6,
) -> AsyncIterator[str]:
) -> AsyncIterator[dict]:
"""Yield des évènements : {type:'sources', sources:[…]} (une fois, avant la
réponse — transparence sur les passages utilisés), puis {type:'token', token}."""
# Question AUTONOME pour la recherche : sur une relance (« et ses
# faiblesses ? »), l'embedding du dernier message seul ne contient pas
# le sujet → on le résout depuis l'historique (best-effort, 1 appel léger,
# uniquement à partir du 2e tour). La réponse, elle, voit tout l'historique.
search_query = await standalone_question(self._llm, messages)
passages = await self._rag.retrieve(source_ids, search_query, top_k=top_k)
# Évènement 'sources' AVANT le premier token : l'UI peut afficher les
# pages utilisées (« 📖 p. 12, 47 ») dès le début de la réponse.
yield {"type": "sources", "sources": [
{
"source_id": p.get("source_id"),
"page": p.get("page"),
"score": round(float(p.get("score") or 0.0), 3),
}
for p in passages
]}
sources_block = (
"\n\n".join(self._format_passage(p) for p in passages)
if passages else "(aucun passage pertinent trouvé dans les sources)"
@@ -86,7 +98,7 @@ class NotebookChatUseCase:
system_prompt = _SYSTEM_PROMPT.format(
context_block=context_block, sources_block=sources_block)
async for token in self._llm.stream_chat(messages, system_prompt=system_prompt):
yield token
yield {"type": "token", "token": token}
@staticmethod
def _format_passage(p: dict) -> str:

View File

@@ -17,9 +17,10 @@ public interface NotebookChatStreamer {
/**
* Streame la réponse ancrée sur les sources. Les callbacks sont invoqués au fil
* de l'eau : {@code onToken} par fragment, {@code onProgress} (mode approfondi
* uniquement) pendant la lecture du document, {@code onDone} à la fin,
* {@code onError} en cas d'échec.
* de l'eau : {@code onSourcesJson} UNE fois avant le premier token (JSON brut des
* passages utilisés — transparence UI), {@code onToken} par fragment,
* {@code onProgress} (mode approfondi uniquement) pendant la lecture du document,
* {@code onDone} à la fin, {@code onError} en cas d'échec.
*
* @param deep true = analyse approfondie (map-reduce sur tout le document) ;
* false = chat RAG (top-k).
@@ -29,6 +30,7 @@ public interface NotebookChatStreamer {
List<Msg> messages,
String context,
boolean deep,
Consumer<String> onSourcesJson,
Consumer<String> onToken,
Consumer<Progress> onProgress,
Runnable onDone,

View File

@@ -51,6 +51,7 @@ public class BrainNotebookChatClient implements NotebookChatStreamer {
List<Msg> messages,
String context,
boolean deep,
Consumer<String> onSourcesJson,
Consumer<String> onToken,
Consumer<Progress> onProgress,
Runnable onDone,
@@ -75,7 +76,8 @@ public class BrainNotebookChatClient implements NotebookChatStreamer {
try {
flux
.timeout(Duration.ofSeconds(timeoutSeconds))
.doOnNext(sse -> handleEvent(sse, terminated, onToken, onProgress, onDone, onError))
.doOnNext(sse -> handleEvent(
sse, terminated, onSourcesJson, onToken, onProgress, onDone, onError))
.blockLast();
if (!terminated[0]) {
onDone.run(); // flux terminé sans event done explicite
@@ -93,6 +95,7 @@ public class BrainNotebookChatClient implements NotebookChatStreamer {
private void handleEvent(
ServerSentEvent<String> sse,
boolean[] terminated,
Consumer<String> onSourcesJson,
Consumer<String> onToken,
Consumer<Progress> onProgress,
Runnable onDone,
@@ -103,6 +106,10 @@ public class BrainNotebookChatClient implements NotebookChatStreamer {
if ("token".equals(event)) {
String token = readField(data, "token");
if (token != null && !token.isEmpty()) onToken.accept(token);
} else if ("sources".equals(event)) {
// Passages utilisés par le RAG : relayés tels quels (JSON brut) — le
// Core n'a pas besoin de les comprendre, seulement de les transmettre.
onSourcesJson.accept(data);
} else if ("progress".equals(event)) {
onProgress.accept(new Progress(readInt(data, "current"), readInt(data, "total")));
} else if ("done".equals(event)) {

View File

@@ -132,6 +132,7 @@ public class NotebookController {
StringBuilder assistant = new StringBuilder();
chatStreamer.stream(
sourceIds, history, context, deep,
sourcesJson -> sendSources(emitter, sourcesJson),
token -> { assistant.append(token); sendToken(emitter, token); },
progress -> sendProgress(emitter, progress),
() -> {
@@ -161,6 +162,15 @@ public class NotebookController {
}
}
private void sendSources(SseEmitter emitter, String sourcesJson) {
try {
// JSON brut du Brain ({"sources":[{source_id,page,score},…]}), relayé tel quel.
emitter.send(SseEmitter.event().name("sources").data(sourcesJson));
} catch (IOException | IllegalStateException e) {
// flux fermé/expiré : on cesse d'écrire
}
}
private void sendProgress(SseEmitter emitter, NotebookChatStreamer.Progress p) {
try {
emitter.send(SseEmitter.event().name("progress")

View File

@@ -91,6 +91,11 @@
(created)="onActionCreated()">
</app-notebook-action-card>
}
@if (sourcesLabel(m); as label) {
<div class="nbd-msg-sources" title="Pages de la source utilisées pour ancrer cette réponse">
📖 {{ label }}
</div>
}
}
} @else {
<div class="nbd-msg-content">{{ m.content }}</div>

View File

@@ -104,3 +104,10 @@
display: inline-flex; align-items: center; gap: 0.35rem;
font-size: 0.82rem; color: #c4a8ff; font-style: italic;
}
// Pages de la source utilisées par le RAG (transparence de la réponse)
.nbd-msg-sources {
margin-top: 0.35rem;
font-size: 0.75rem;
opacity: 0.65;
}

View File

@@ -107,7 +107,29 @@ export class NotebookDetailComponent implements OnInit {
return cache._parsed;
}
/** trackBy stable pour les cartes d'action (évite toute recréation parasite). */
/**
* Libellé compact des pages utilisées par le RAG pour une réponse
* (« p. 12, 47, 103 » — préfixé du nom du fichier si plusieurs sources).
*/
sourcesLabel(m: NotebookMessage): string {
const src = m.sources ?? [];
if (!src.length) return '';
const bySource = new Map<string, number[]>();
for (const s of src) {
if (s.page == null) continue;
const pages = bySource.get(s.sourceId) ?? [];
if (!pages.includes(s.page)) pages.push(s.page);
bySource.set(s.sourceId, pages);
}
if (bySource.size === 0) return '';
const nameOf = (id: string) => this.sources.find(x => x.id === id)?.filename ?? '';
return [...bySource.entries()]
.map(([id, pages]) => {
const label = `p. ${pages.sort((a, b) => a - b).join(', ')}`;
return bySource.size > 1 && nameOf(id) ? `${nameOf(id)}${label}` : label;
})
.join(' · ');
}
load(): void {
this.service.get(this.notebookId).subscribe({
@@ -162,6 +184,7 @@ export class NotebookDetailComponent implements OnInit {
this.service.streamChat(this.notebookId, text, deep).subscribe({
next: (ev) => {
if (ev.type === 'token') { this.deepProgress = null; assistant.content += ev.value; }
else if (ev.type === 'sources') assistant.sources = ev.sources;
else if (ev.type === 'progress') this.deepProgress = { current: ev.current, total: ev.total };
else if (ev.type === 'error') assistant.content += (assistant.content ? '\n\n' : '') + `⚠️ ${ev.message}`;
},

View File

@@ -16,12 +16,22 @@ export interface NotebookSource {
pageCount: number;
}
/** Passage source utilisé par le RAG pour ancrer une réponse (transparence). */
export interface NotebookChatSource {
sourceId: string;
/** Numéro de page 1-based, null si inconnu. */
page: number | null;
score: number;
}
export interface NotebookMessage {
id?: string;
notebookId?: string;
/** "user" | "assistant" */
role: string;
content: string;
/** Passages utilisés (réponses streamées uniquement — non persisté). */
sources?: NotebookChatSource[];
}
export interface NotebookDetail {
@@ -35,6 +45,7 @@ export interface NotebookDetail {
/** Évènements du chat ancré streamé. */
export type NotebookChatEvent =
| { type: 'token'; value: string }
| { type: 'sources'; sources: NotebookChatSource[] }
| { type: 'progress'; current: number; total: number }
| { type: 'done' }
| { type: 'error'; message: string };

View File

@@ -78,6 +78,16 @@ export class NotebookService {
if (name === 'token') {
try { emit({ type: 'token', value: (JSON.parse(currentData) as { token?: string }).token ?? '' }); }
catch { /* ignore */ }
} else if (name === 'sources') {
try {
const o = JSON.parse(currentData) as { sources?: { source_id?: string; page?: number | null; score?: number }[] };
emit({
type: 'sources',
sources: (o.sources ?? []).map(s => ({
sourceId: s.source_id ?? '', page: s.page ?? null, score: s.score ?? 0
}))
});
} catch { /* ignore */ }
} else if (name === 'progress') {
try {
const o = JSON.parse(currentData) as { current?: number; total?: number };