forked from SpeedyFoxAi/jarvis-memory
Initial commit: Jarvis Memory system
This commit is contained in:
70
skills/qdrant-memory/scripts/q_save.py
Executable file
70
skills/qdrant-memory/scripts/q_save.py
Executable file
@@ -0,0 +1,70 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
Q Save - Trigger conversation storage (Mem0-style)
|
||||
|
||||
Usage:
|
||||
q_save.py --user-id "rob" "User message" "AI response" [--turn N]
|
||||
|
||||
Called when user says "save q" or "q save" to immediately
|
||||
store the current conversation to Qdrant.
|
||||
|
||||
Mem0-style: user_id is REQUIRED and persistent across all chats.
|
||||
"""
|
||||
|
||||
import argparse
|
||||
import subprocess
|
||||
import sys
|
||||
from pathlib import Path
|
||||
|
||||
SCRIPT_DIR = Path(__file__).parent.resolve()
|
||||
BACKGROUND_STORE = SCRIPT_DIR / "background_store.py"
|
||||
|
||||
def q_save(
|
||||
user_id: str,
|
||||
user_message: str,
|
||||
ai_response: str,
|
||||
turn: int = None
|
||||
):
|
||||
"""Save conversation to Qdrant (background, zero delay)"""
|
||||
|
||||
cmd = [
|
||||
sys.executable,
|
||||
str(BACKGROUND_STORE),
|
||||
user_message,
|
||||
ai_response,
|
||||
"--user-id", user_id
|
||||
]
|
||||
|
||||
if turn:
|
||||
cmd.extend(["--turn", str(turn)])
|
||||
|
||||
# Fire and forget
|
||||
subprocess.Popen(
|
||||
cmd,
|
||||
stdout=subprocess.DEVNULL,
|
||||
stderr=subprocess.DEVNULL,
|
||||
start_new_session=True
|
||||
)
|
||||
|
||||
return True
|
||||
|
||||
def main():
|
||||
parser = argparse.ArgumentParser(
|
||||
description='Q Save - Mem0-style trigger (user-centric)'
|
||||
)
|
||||
parser.add_argument("--user-id", required=True,
|
||||
help="REQUIRED: User ID (e.g., 'rob')")
|
||||
parser.add_argument("user_message", help="User's message")
|
||||
parser.add_argument("ai_response", help="AI's response")
|
||||
parser.add_argument("--turn", type=int, help="Turn number")
|
||||
|
||||
args = parser.parse_args()
|
||||
|
||||
if q_save(args.user_id, args.user_message, args.ai_response, args.turn):
|
||||
print(f"✅ Saved for user '{args.user_id}'")
|
||||
else:
|
||||
print("❌ Failed to save", file=sys.stderr)
|
||||
sys.exit(1)
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user