mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-05-18 04:41:56 +00:00
fix(yuanbao): resolve quoted file/image via transcript lookup when quote desc lacks ybres
When a user quotes a file message (type=3) and @bot, the quote's desc field only contains the filename without a ybres:// resource reference. The existing QuoteContextMiddleware only extracted media refs from desc using the ybres regex, which always returned empty for file quotes. Fix: add a transcript lookup fallback in QuoteContextMiddleware.handle() — when quote_media_refs is empty but reply_to_message_id is set, search the session transcript for the quoted message_id and extract ybres anchors from its content. Also fix message_type classification: when quote media resolves non-image files, override message_type to DOCUMENT so gateway/run.py's document injection logic properly prepends the file path and content for the agent.
This commit is contained in:
parent
3df26b925c
commit
fc2754dbdf
1 changed files with 33 additions and 1 deletions
|
|
@ -2216,6 +2216,34 @@ class QuoteContextMiddleware(InboundMiddleware):
|
|||
|
||||
async def handle(self, ctx: InboundContext, next_fn) -> None:
|
||||
ctx.reply_to_message_id, ctx.reply_to_text, ctx.quote_media_refs = self._extract_quote_context(ctx.cloud_custom_data)
|
||||
|
||||
# Fallback: if quote has a message_id but no media_refs extracted from desc,
|
||||
# look up the quoted message in transcript history by message_id to find ybres anchors.
|
||||
if ctx.reply_to_message_id and not ctx.quote_media_refs:
|
||||
store = getattr(ctx.adapter, "_session_store", None)
|
||||
if store:
|
||||
try:
|
||||
session_entry = store.get_or_create_session(ctx.source)
|
||||
history = store.load_transcript(session_entry.session_id)
|
||||
for msg in (history or []):
|
||||
mid = msg.get("message_id", "")
|
||||
if mid and mid == ctx.reply_to_message_id:
|
||||
content = msg.get("content", "")
|
||||
if isinstance(content, str) and "|ybres:" in content:
|
||||
for m in _YB_RES_REF_RE.finditer(content):
|
||||
head = m.group(1)
|
||||
rid = m.group(2)
|
||||
kind, _, filename = head.partition(":")
|
||||
kind = kind.strip()
|
||||
if kind in ("image", "file"):
|
||||
ctx.quote_media_refs.append((rid, kind, filename.strip()))
|
||||
break
|
||||
except Exception as exc:
|
||||
logger.warning(
|
||||
"[%s] QuoteContext transcript lookup failed: %s",
|
||||
ctx.adapter.name, exc,
|
||||
)
|
||||
|
||||
await next_fn()
|
||||
|
||||
|
||||
|
|
@ -2589,7 +2617,11 @@ class DispatchMiddleware(InboundMiddleware):
|
|||
|
||||
event = MessageEvent(
|
||||
text=_patched_event_text,
|
||||
message_type=ctx.msg_type,
|
||||
message_type=(
|
||||
MessageType.DOCUMENT
|
||||
if any(not mt.startswith("image/") for mt in media_types)
|
||||
else ctx.msg_type
|
||||
),
|
||||
source=ctx.source,
|
||||
message_id=ctx.msg_id or None,
|
||||
raw_message=ctx.push,
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue