89 lines
2.8 KiB
Python
Executable File
89 lines
2.8 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import json
|
|
import sys
|
|
import subprocess
|
|
from pathlib import Path
|
|
|
|
def extract_turns(session_file):
|
|
"""Extract user-assistant turn pairs from session transcript"""
|
|
turns = []
|
|
messages = []
|
|
|
|
# Read and parse JSONL
|
|
with open(session_file, 'r') as f:
|
|
for line in f:
|
|
try:
|
|
data = json.loads(line.strip())
|
|
if data.get('type') == 'message' and 'message' in data:
|
|
msg = data['message']
|
|
if msg.get('role') in ['user', 'assistant']:
|
|
messages.append(msg)
|
|
except:
|
|
continue
|
|
|
|
# Pair user messages with assistant responses
|
|
for i in range(len(messages) - 1):
|
|
if (messages[i].get('role') == 'user' and
|
|
messages[i+1].get('role') == 'assistant'):
|
|
|
|
user_content = extract_text_content(messages[i])
|
|
assistant_content = extract_text_content(messages[i+1])
|
|
|
|
if user_content and assistant_content:
|
|
turns.append({
|
|
'user': user_content,
|
|
'assistant': assistant_content,
|
|
'agent_id': 'case',
|
|
'session': 'main'
|
|
})
|
|
|
|
return turns[-10:] # Last 10 turns
|
|
|
|
def extract_text_content(message):
|
|
"""Extract text content from message"""
|
|
content = message.get('content', [])
|
|
if isinstance(content, str):
|
|
return content
|
|
|
|
text_parts = []
|
|
for part in content:
|
|
if isinstance(part, dict) and part.get('type') == 'text':
|
|
text_parts.append(part.get('text', ''))
|
|
|
|
return ' '.join(text_parts).strip()
|
|
|
|
def main():
|
|
if len(sys.argv) != 2:
|
|
print("Usage: extract_memory_turns.py <session_file>")
|
|
sys.exit(1)
|
|
|
|
session_file = sys.argv[1]
|
|
turns = extract_turns(session_file)
|
|
|
|
print(f"Extracted {len(turns)} turns from {session_file}")
|
|
|
|
# Send each turn to auto-memory-hook.py
|
|
for i, turn in enumerate(turns):
|
|
try:
|
|
proc = subprocess.Popen(
|
|
['python3', '/home/wdjones/.openclaw/workspace/tools/auto-memory-hook.py'],
|
|
stdin=subprocess.PIPE,
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE,
|
|
text=True
|
|
)
|
|
|
|
json_input = json.dumps(turn)
|
|
stdout, stderr = proc.communicate(input=json_input)
|
|
|
|
if proc.returncode == 0:
|
|
print(f"Turn {i+1}: Processed successfully")
|
|
else:
|
|
print(f"Turn {i+1}: Error - {stderr}")
|
|
|
|
except Exception as e:
|
|
print(f"Turn {i+1}: Failed to process - {e}")
|
|
|
|
if __name__ == '__main__':
|
|
main() |