agentmemory-python / plugin /scripts /notification.py
Yash030's picture
feat: add Python plugin system, multi-agent connect CLI, and MCP tool expansions
26a284a
#!/usr/bin/env python
import sys
import json
import time
from datetime import datetime
from hook_utils import resolve_project, is_sdk_child, api_call_bg
def main():
try:
input_data = sys.stdin.read()
if not input_data:
return
data = json.loads(input_data)
except Exception:
return
if is_sdk_child(data):
return
notification_type = data.get("notification_type") or data.get("notificationType")
if notification_type != "permission_prompt":
return
session_id = data.get("session_id") or data.get("sessionId") or "unknown"
payload = {
"hookType": "notification",
"sessionId": session_id,
"project": resolve_project(data.get("cwd")),
"cwd": data.get("cwd") or "",
"timestamp": datetime.utcnow().isoformat() + "Z",
"data": {
"notification_type": notification_type,
"title": data.get("title"),
"message": data.get("message")
}
}
api_call_bg("observe", payload)
time.sleep(0.5)
if __name__ == "__main__":
main()