mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-07-21 16:04:41 +00:00
refactor(memory): make query rewrite provider-agnostic
Move query_rewrite from the honcho plugin to plugins/memory/ and rename the auxiliary task key honcho_query_rewrite -> memory_query_rewrite so any memory provider can use the same rewrite path and model/timeout config block. No behavior change.
This commit is contained in:
parent
8ab4cb9d0d
commit
e7fb51d5ac
5 changed files with 10 additions and 6 deletions
|
|
@ -1652,7 +1652,7 @@ DEFAULT_CONFIG = {
|
|||
"reasoning_effort": "", # per-task thinking level: none|minimal|low|medium|high|xhigh|max|ultra (empty = provider default)
|
||||
"language": "",
|
||||
},
|
||||
"honcho_query_rewrite": {
|
||||
"memory_query_rewrite": {
|
||||
"provider": "auto", # fast/cheap model recommended
|
||||
"model": "",
|
||||
"base_url": "",
|
||||
|
|
|
|||
|
|
@ -3257,7 +3257,7 @@ _AUX_TASKS: list[tuple[str, str, str]] = [
|
|||
("approval", "Approval", "smart command approval"),
|
||||
("mcp", "MCP", "MCP tool reasoning"),
|
||||
("title_generation", "Title generation", "session titles"),
|
||||
("honcho_query_rewrite", "Honcho query rewrite", "memory retrieval queries"),
|
||||
("memory_query_rewrite", "Memory query rewrite", "memory retrieval queries"),
|
||||
("tts_audio_tags", "TTS audio tags", "Gemini TTS tag insertion"),
|
||||
("skills_hub", "Skills hub", "skills search/install"),
|
||||
("triage_specifier", "Triage specifier", "kanban spec fleshing"),
|
||||
|
|
|
|||
|
|
@ -1636,7 +1636,7 @@ class HonchoMemoryProvider(MemoryProvider):
|
|||
|
||||
def register(ctx) -> None:
|
||||
"""Register Honcho as a memory provider plugin."""
|
||||
from plugins.memory.honcho.query_rewrite import rewrite_dialectic_query
|
||||
from plugins.memory.query_rewrite import rewrite_dialectic_query
|
||||
|
||||
ctx.register_memory_provider(
|
||||
HonchoMemoryProvider(query_rewriter=rewrite_dialectic_query)
|
||||
|
|
|
|||
|
|
@ -1,4 +1,8 @@
|
|||
"""Rewrite the latest user message into a clean Honcho retrieval query."""
|
||||
"""Rewrite the latest user message into a clean memory-retrieval query.
|
||||
|
||||
Provider-agnostic: any memory provider can pass ``rewrite_dialectic_query``
|
||||
as its query rewriter. Model/timeout are configured under
|
||||
``auxiliary.memory_query_rewrite`` in config.yaml."""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
|
|
@ -9,7 +13,7 @@ from typing import Any
|
|||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
TASK_KEY = "honcho_query_rewrite"
|
||||
TASK_KEY = "memory_query_rewrite"
|
||||
|
||||
_MAX_INPUT_CHARS = 4_000
|
||||
_MAX_QUERY_CHARS = 320
|
||||
|
|
@ -6,7 +6,7 @@ from unittest.mock import MagicMock, patch
|
|||
import pytest
|
||||
|
||||
from plugins.memory.honcho import HonchoMemoryProvider, register
|
||||
from plugins.memory.honcho.query_rewrite import (
|
||||
from plugins.memory.query_rewrite import (
|
||||
TASK_KEY,
|
||||
_bounded_user_message,
|
||||
_normalize_rewrite,
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue