mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-08-28 05:25:04 +00:00
630f26b119
* feat(search): add tool_context-scoped chunk dedup with TTL
Introduce _ToolContextDedupMixin shared by search/vector_search/bm25_search
to skip already-seen chunks within one agent tool_context. Per-context state
lives in app_context.metadata with configurable TTL (default 24h).
* feat(search): unify chunk answer rendering with merge and explicit empty messages
- Refactor SearchStep/VectorSearchStep/Bm25SearchStep to share format_chunks_answer for consistent source rendering and adjacent session-chunk merging.
- Distinguish empty results: ALL_RETURNED_MESSAGE when dedup removes everything vs NO_RESULTS_MESSAGE when nothing matched.
- Bump JsonlFileChunker default max_chars to 4000.
- Add unit tests for source-format merge and empty-result messages.
* refactor(config): reorganize file_chunker components and move jsonl max_chars into config
- Register explicit markdown/json/jsonl chunkers in beam.yaml and lme.yaml with markdown options (embed_toc, max_ast_sections, frontmatter handling) and jsonl max_chars=4000.
- Restrict default chunker to txt/log extensions.
- Revert JsonlFileChunker code default max_chars back to 2000; the 4000 value now lives in config.
* chore(benchmark): increase longmemeval num_items from 64 to 500
* refactor(search): split SearchStep into simplified and v2 variants, extract counter utility
- Extract global_counter_next from ApplicationContext into reme/utils/counter.py
as a standalone function operating on metadata dict with lazy initialization.
- Split SearchStep into two variants:
- SearchStep (simplified): inline chunk.id dedup, single-branch vector/keyword
optimization based on vector_weight, inline answer formatting.
- SearchV2Step (full): preserves _ToolContextDedupMixin with interval-subset-aware
dedup and format_chunks_answer with session-aware chunk merging.
- Update beam.yaml and lme.yaml to use search_v2_step for benchmark jobs.
- Rename existing search tests to test_search_v2_step_* and add new
test_search_step_* tests covering the simplified variant.
* fix: normalise missing trailing newline in _build_union_chunk to prevent line collision
* refactor: lazy-init counter tree in ApplicationContext metadata
- Remove hardcoded _counter_tree and _counter_tree_lock initialization
from ApplicationContext.metadata; rely on lazy initialization in
reme.utils.counter.global_counter_next on first call
- Set longmemeval num_items back to 500
- Remove obsolete trailing-newline collision tests
---------
Co-authored-by: sa-buc <jiangniurou.xyf@dail-algo011164204033.ET135>
42 lines
1.1 KiB
Python
42 lines
1.1 KiB
Python
"""Utility modules."""
|
|
|
|
from .common_utils import (
|
|
hash_text,
|
|
execute_stream_task,
|
|
mock_reme_server,
|
|
call_action,
|
|
call_and_check,
|
|
)
|
|
from .env_utils import load_env, parse_env_file
|
|
from .link_expansion import expand_links, render_expansion_lines
|
|
from .logger_utils import get_logger
|
|
from .logo_utils import print_logo
|
|
from .service_utils import find_reme, locate_reme, precheck_start, cli_find_reme, running_service_config
|
|
from .similarity_utils import cosine_similarity, batch_cosine_similarity
|
|
from .token_utils import estimate_token_count
|
|
from .agent_state_io import AsStateHandler
|
|
from .counter import global_counter_next
|
|
|
|
__all__ = [
|
|
"hash_text",
|
|
"execute_stream_task",
|
|
"mock_reme_server",
|
|
"call_action",
|
|
"call_and_check",
|
|
"load_env",
|
|
"parse_env_file",
|
|
"expand_links",
|
|
"render_expansion_lines",
|
|
"get_logger",
|
|
"print_logo",
|
|
"find_reme",
|
|
"locate_reme",
|
|
"precheck_start",
|
|
"cli_find_reme",
|
|
"running_service_config",
|
|
"cosine_similarity",
|
|
"batch_cosine_similarity",
|
|
"estimate_token_count",
|
|
"AsStateHandler",
|
|
"global_counter_next",
|
|
]
|