Spaces:
Running
Running
File size: 2,701 Bytes
aa686c9 25d86d2 aa686c9 da7768b aa686c9 25d86d2 aa686c9 25d86d2 aa686c9 da7768b 25d86d2 da7768b aa686c9 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 | #!/usr/bin/env python3
"""Patch hermes-agent to support auto-detecting document file paths for native delivery.
By default, extract_local_files() only detects image/video extensions (.png, .jpg, etc.)
which means .md, .pdf, .docx, .xlsx and other document files created by the agent
are NOT sent as attachments.
This patch adds document extensions to _LOCAL_MEDIA_EXTS so that when Hermes creates
a file like /tmp/weather_report.md, it gets automatically sent as a native attachment
on Feishu and WeChat instead of appearing as a plain text path.
"""
import re
import sys
import os
import glob
def patch_file(filepath: str):
with open(filepath, 'r') as f:
content = f.read()
# Find and replace _LOCAL_MEDIA_EXTS
old = """\
_LOCAL_MEDIA_EXTS = (
'.png', '.jpg', '.jpeg', '.gif', '.webp',
'.mp4', '.mov', '.avi', '.mkv', '.webm',
)"""
new = """\
_LOCAL_MEDIA_EXTS = (
# Images
'.png', '.jpg', '.jpeg', '.gif', '.webp', '.bmp', '.svg',
# Videos
'.mp4', '.mov', '.avi', '.mkv', '.webm',
# Audio
'.ogg', '.opus', '.mp3', '.wav', '.m4a', '.silk', '.flac',
# Documents
'.md', '.txt', '.csv', '.json', '.xml', '.yaml', '.yml', '.toml', '.log',
'.pdf',
'.doc', '.docx',
'.xls', '.xlsx',
'.ppt', '.pptx',
'.html', '.htm',
# Archives
'.zip', '.tar', '.gz', '.7z', '.rar',
)"""
if old not in content:
print(f"WARNING: Could not find _LOCAL_MEDIA_EXTS in {filepath}", file=sys.stderr)
print("The upstream code may have changed. Skipping this patch.", file=sys.stderr)
sys.exit(0)
content = content.replace(old, new, 1)
with open(filepath, 'w') as f:
f.write(content)
print(f"Patched {filepath}: added document/audio/archive extensions to _LOCAL_MEDIA_EXTS")
if __name__ == "__main__":
# hermes-agent is installed in editable mode (-e), so source is in /app/hermes-agent/
candidates = [
"/app/hermes-agent/gateway/platforms/base.py",
]
# Also search venv site-packages as fallback
candidates.extend(glob.glob("/app/venv/lib/**/gateway/platforms/base.py", recursive=True))
filepath = None
for c in candidates:
if os.path.isfile(c):
filepath = c
break
if not filepath:
print("WARNING: base.py not found in any candidate location", file=sys.stderr)
print(f"Checked: {candidates}", file=sys.stderr)
print("Skipping patch_file_delivery.", file=sys.stderr)
sys.exit(0)
patch_file(filepath)
|