-
Notifications
You must be signed in to change notification settings - Fork 4
Expand file tree
/
Copy pathhook_embed.py
More file actions
35 lines (25 loc) · 845 Bytes
/
hook_embed.py
File metadata and controls
35 lines (25 loc) · 845 Bytes
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
#!/usr/bin/env python3
"""Hook helper: embed un-indexed messages after a Stop event.
Called as a background process from stop.sh when embeddingEnabled=true.
Mirrors the pattern of hook_stop.py — thin wrapper over the embed module.
"""
import argparse
import os
import sys
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
import db
import embed
def main():
parser = argparse.ArgumentParser()
parser.add_argument("--session", default="")
parser.add_argument("--dir", default="")
args = parser.parse_args()
cfg = db.load_config()
if not cfg.get("embeddingEnabled", False):
return
session_id = args.session or None
if session_id and db.get_session_stateless(session_id):
return
embed.embed_messages_batch(cfg, session_id=session_id)
if __name__ == "__main__":
main()