Spaces:
Running
Running
File size: 1,615 Bytes
3619c2d | 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 | #!/usr/bin/env python
import sys
import json
import time
import os
from datetime import datetime, timezone
from hook_utils import resolve_project, is_sdk_child, api_call_bg
def main():
# Log to file for debugging - write early before any potential errors
log_path = os.path.expanduser("~/.agentmemory/hook_log.txt")
try:
with open(log_path, "a", encoding="utf-8") as f:
f.write(f"[{datetime.now(timezone.utc).isoformat()}] prompt_submit hook called\n")
f.write(f"sys.argv: {sys.argv}\n")
except Exception as e:
pass
try:
input_data = sys.stdin.read()
if not input_data:
return
data = json.loads(input_data)
except Exception as e:
try:
with open(log_path, "a", encoding="utf-8") as f:
f.write(f"Error parsing stdin: {e}\n")
except:
pass
return
if is_sdk_child(data):
return
session_id = data.get("session_id") or data.get("sessionId") or "unknown"
cwd = data.get("cwd") or ""
project = resolve_project(cwd)
prompt = data.get("prompt") or data.get("userPrompt") or ""
payload = {
"hookType": "prompt_submit",
"sessionId": session_id,
"project": project,
"cwd": cwd,
"timestamp": datetime.utcnow().isoformat() + "Z",
"data": {
"prompt": prompt
}
}
api_call_bg("observe", payload)
# Allow background request to be flushed
time.sleep(0.5)
if __name__ == "__main__":
main()
|