#!/usr/bin/env python3import json, uuid, time, os, sys
try:
import chromadb; from openai import OpenAI
except ImportError: print("ERROR: pip install chromadb openai"); exit(1)
INPUT = sys.argv[1] if len(sys.argv) > 1 else "raw_ltm.jsonl"
client = OpenAI()
db = chromadb.PersistentClient(path="./chroma_db")
col = db.get_or_create_collection("memories")
def embed(text): return client.embeddings.create(input=text, model="text-embedding-3-large").data[0].embedding
with open(INPUT) as f: entries = [json.loads(line) for line in f if line.strip()]
print(f"Importing {len(entries)} entries..."); BATCH = 50
for i in range(0, len(entries), BATCH): batch = entries[i:i+BATCH]; ids = [str(uuid.uuid4()) for _ in batch]; docs = [e.get("content","") for e in batch]; metas = [{"time": e.get("time","")} for e in batch]; embs = [embed(d) for d in docs]; col.upsert(ids=ids, documents=docs, metadatas=metas, embeddings=embs); print(f"  {i+len(batch)}/{len(entries)}"); time.sleep(0.5)
print(f"Done. Collection now has {col.count()} entries.")