fix(ai): mode agent utilise les outils natifs au lieu du bloc obsigate-action (BUG-053)
This commit is contained in:
+32
-2
@@ -486,12 +486,17 @@ def build_system_prompt(context: dict[str, Any], scope: str = "directory", vault
|
||||
return prompt
|
||||
|
||||
|
||||
GENERAL_SYSTEM_PROMPT = """Tu es l'assistant intégré d'ObsiGate, une application web auto-hébergée pour consulter, rechercher et éditer des vaults Obsidian (Markdown).
|
||||
GENERAL_SYSTEM_HEADER = """Tu es l'assistant intégré d'ObsiGate, une application web auto-hébergée pour consulter, rechercher et éditer des vaults Obsidian (Markdown).
|
||||
|
||||
Tes deux rôles :
|
||||
1. **Aider sur l'application** : expliquer la navigation, la recherche (full-text, filtres `tag:`, `created:`, `path:`), l'éditeur (CodeMirror, autosave, raccourcis), les onglets et le split view, les sauvegardes et la restauration, le partage public, l'export (HTML/Markdown/ePub/PDF), Mermaid, Excalidraw, les plugins, les thèmes, le mode hors-ligne, le MFA, etc.
|
||||
2. **Proposer des actions concrètes** : créer un fichier ou un dossier dans un vault.
|
||||
"""
|
||||
|
||||
# Text action protocol — used by the classic (non-agent) chat endpoint, where
|
||||
# the model has no native tool calling; the frontend turns each block into a
|
||||
# clickable “Apply” card.
|
||||
GENERAL_ACTION_TEXT_PROTOCOL = """
|
||||
Quand l'utilisateur demande explicitement de créer un fichier, inclus un bloc de ce type dans ta réponse (un bloc par fichier, et rien d'autre à l'intérieur du bloc) :
|
||||
|
||||
```obsigate-action
|
||||
@@ -513,6 +518,25 @@ Règles :
|
||||
- Réponds dans la langue de l'utilisateur, de façon concise et structurée (Markdown).
|
||||
"""
|
||||
|
||||
# Agent mode: the model has native tools, so it must call them (function
|
||||
# calling) instead of emitting the text `obsigate-action` blocks — otherwise
|
||||
# the requested file is never created (BUG-053).
|
||||
GENERAL_ACTION_TOOL_PROTOCOL = """
|
||||
Tu disposes d'outils natifs (function calling) pour lire, chercher et modifier les vaults : `create_file`, `create_directory`, `append_to_file`, `edit_file`, `read_file`, `search_fulltext`, etc.
|
||||
|
||||
Quand l'utilisateur demande explicitement de créer un fichier, **appelle directement l'outil `create_file`** avec `{"vault": "<nom du vault>", "path": "<chemin/relatif.md>", "content": "<contenu markdown>"}`. Pour créer un dossier, appelle `create_directory`.
|
||||
|
||||
Règles :
|
||||
- N'écris **jamais** de bloc ```obsigate-action``` : en mode agent, toutes les actions passent par les outils natifs.
|
||||
- Écris le contenu **complet** demandé dans l'argument `content` (ne le tronque pas, pas de « … » ni de ligne omise).
|
||||
- Pour créer un fichier dans un nouveau dossier, un seul appel `create_file` avec le chemin complet suffit (les dossiers parents sont créés automatiquement).
|
||||
- N'invente jamais un nom de vault : utilise l'un des vaults disponibles listés ci-dessous.
|
||||
- Réponds dans la langue de l'utilisateur, de façon concise et structurée (Markdown).
|
||||
"""
|
||||
|
||||
# Backwards-compatible alias (classic chat prompt).
|
||||
GENERAL_SYSTEM_PROMPT = GENERAL_SYSTEM_HEADER + GENERAL_ACTION_TEXT_PROTOCOL
|
||||
|
||||
|
||||
def _format_app_context(app_context: dict[str, Any] | None, recent_files: list[dict[str, Any]] | None) -> str:
|
||||
"""Render the live application state for the General assistant prompt.
|
||||
@@ -588,14 +612,20 @@ def build_general_system_prompt(
|
||||
vaults: list[str] | None = None,
|
||||
app_context: dict[str, Any] | None = None,
|
||||
recent_files: list[dict[str, Any]] | None = None,
|
||||
agent: bool = False,
|
||||
) -> str:
|
||||
"""System prompt for the General assistant (app help + actions).
|
||||
|
||||
``app_context`` carries the live UI state (open documents, current
|
||||
directory, active search) and ``recent_files`` the last modified files, so
|
||||
the assistant knows what the user is doing rather than answering blind.
|
||||
|
||||
``agent`` selects the action protocol: the classic chat endpoint (no native
|
||||
tools) uses the text ``obsigate-action`` blocks, while the tool-calling
|
||||
agent endpoint must invoke the native tools instead (BUG-053).
|
||||
"""
|
||||
prompt = GENERAL_SYSTEM_PROMPT
|
||||
protocol = GENERAL_ACTION_TOOL_PROTOCOL if agent else GENERAL_ACTION_TEXT_PROTOCOL
|
||||
prompt = GENERAL_SYSTEM_HEADER + protocol
|
||||
if vaults:
|
||||
prompt += "\nVaults disponibles : " + ", ".join(sorted(vaults)) + "\n"
|
||||
else:
|
||||
|
||||
@@ -193,10 +193,12 @@ def _recent_files_for_prompt(current_user, limit: int = 10) -> list[dict[str, An
|
||||
return []
|
||||
|
||||
|
||||
def _resolve_system_prompt(req, current_user) -> str:
|
||||
def _resolve_system_prompt(req, current_user, agent: bool = False) -> str:
|
||||
"""Resolve the vault access and build the assistant system prompt.
|
||||
|
||||
Shared by the classic chat endpoint and the tool-calling agent endpoint.
|
||||
``agent=True`` selects the native-tool action protocol (no text
|
||||
``obsigate-action`` blocks) for the General/empty-directory prompts.
|
||||
"""
|
||||
mode = _normalize_mode(req.mode)
|
||||
vault_path: Path | None = None
|
||||
@@ -222,6 +224,7 @@ def _resolve_system_prompt(req, current_user) -> str:
|
||||
list(index.keys()),
|
||||
app_context=_submitted_app_context(req),
|
||||
recent_files=_recent_files_for_prompt(current_user),
|
||||
agent=agent,
|
||||
)
|
||||
elif effective_mode == "documents":
|
||||
prompt = build_system_prompt(context, scope="documents", vault_name=req.vault)
|
||||
@@ -233,6 +236,7 @@ def _resolve_system_prompt(req, current_user) -> str:
|
||||
list(index.keys()),
|
||||
app_context=_submitted_app_context(req),
|
||||
recent_files=_recent_files_for_prompt(current_user),
|
||||
agent=agent,
|
||||
)
|
||||
prompt += (
|
||||
f"\n## Dossier vide\nLe dossier « {req.directory or '/'} » "
|
||||
@@ -243,6 +247,14 @@ def _resolve_system_prompt(req, current_user) -> str:
|
||||
else:
|
||||
prompt = build_system_prompt(context, scope="directory", vault_name=req.vault)
|
||||
|
||||
if agent and effective_mode != "general" and context["file_count"] > 0:
|
||||
prompt += (
|
||||
"\n## Mode agent\n"
|
||||
"Utilise les outils natifs (function calling) pour agir sur les fichiers "
|
||||
"(`create_file`, `create_directory`, `append_to_file`, `edit_file`, …). "
|
||||
"N'écris jamais de bloc ```obsigate-action```."
|
||||
)
|
||||
|
||||
skill_id = getattr(req, "skill", None)
|
||||
if skill_id:
|
||||
skill_prompt = get_skill_prompt(skill_id, current_user)
|
||||
@@ -499,7 +511,7 @@ async def api_bookslm_agent(
|
||||
``confirm`` / ``confirm_messages``.
|
||||
"""
|
||||
_validate_vision_support(req)
|
||||
system_prompt = _resolve_system_prompt(req, current_user)
|
||||
system_prompt = _resolve_system_prompt(req, current_user, agent=True)
|
||||
vault_path = _resolve_optional_vault_path(req, current_user)
|
||||
|
||||
messages: list[dict] = [{"role": "system", "content": system_prompt}]
|
||||
@@ -519,7 +531,9 @@ async def api_bookslm_agent(
|
||||
provider=req.provider,
|
||||
model=req.model,
|
||||
temperature=0.3,
|
||||
max_tokens=4096,
|
||||
# Tool-call arguments can carry a whole file body (e.g. a generated
|
||||
# table): leave more room than the plain-chat default.
|
||||
max_tokens=8192,
|
||||
)
|
||||
|
||||
async def generate_sse():
|
||||
|
||||
Reference in New Issue
Block a user