75 lines
2.8 KiB
Python
Executable File
75 lines
2.8 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""
|
|
Parse OpenClaw session transcripts and extract conversation turns
|
|
"""
|
|
import json
|
|
import sys
|
|
from typing import List, Dict, Any
|
|
|
|
def extract_conversation_turns(session_file: str) -> List[Dict[str, str]]:
|
|
"""Extract conversation turns from a session file"""
|
|
turns = []
|
|
current_user_message = None
|
|
|
|
try:
|
|
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:
|
|
message = data['message']
|
|
role = message.get('role')
|
|
content = message.get('content', [])
|
|
|
|
if role == 'user':
|
|
# Extract text content from user message
|
|
text_content = []
|
|
for item in content:
|
|
if item.get('type') == 'text':
|
|
text_content.append(item.get('text', ''))
|
|
current_user_message = '\n'.join(text_content)
|
|
|
|
elif role == 'assistant' and current_user_message:
|
|
# Extract text content from assistant message (skip tool calls)
|
|
text_content = []
|
|
for item in content:
|
|
if item.get('type') == 'text':
|
|
text_content.append(item.get('text', ''))
|
|
|
|
if text_content:
|
|
assistant_text = '\n'.join(text_content)
|
|
turns.append({
|
|
'user': current_user_message,
|
|
'assistant': assistant_text,
|
|
'agent_id': 'case',
|
|
'session': 'main'
|
|
})
|
|
current_user_message = None
|
|
|
|
except json.JSONDecodeError:
|
|
continue
|
|
|
|
except Exception as e:
|
|
print(f"Error reading session file: {e}", file=sys.stderr)
|
|
return []
|
|
|
|
return turns
|
|
|
|
def main():
|
|
if len(sys.argv) != 2:
|
|
print("Usage: parse-session.py <session_file>")
|
|
sys.exit(1)
|
|
|
|
session_file = sys.argv[1]
|
|
turns = extract_conversation_turns(session_file)
|
|
|
|
# Get last 10 assistant turns
|
|
last_turns = turns[-10:] if len(turns) >= 10 else turns
|
|
|
|
# Output as JSON for the auto-memory hook
|
|
for turn in last_turns:
|
|
print(json.dumps(turn))
|
|
|
|
if __name__ == "__main__":
|
|
main() |