From 54f4bc5dbba429ba322dae46ec66ce34e2a80db3 Mon Sep 17 00:00:00 2001 From: Michael Yuan Date: Sat, 31 May 2025 12:52:03 -0500 Subject: [PATCH] Use UUID for error_examples Signed-off-by: Michael Yuan --- app/load_data.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/app/load_data.py b/app/load_data.py index c1f8207..63176b8 100644 --- a/app/load_data.py +++ b/app/load_data.py @@ -49,10 +49,13 @@ def load_error_examples(): # Get embedding for error embedding = llm_client.get_embeddings([example["error"]])[0] + + # Store in vector DB with proper UUID + point_id = str(uuid.uuid4()) # Store in vector DB vector_store.upsert("error_examples", - [{"id": os.path.basename(file_path), + [{"id": point_id, "vector": embedding, "payload": example}]) @@ -60,4 +63,4 @@ def load_error_examples(): if __name__ == "__main__": load_project_examples() - load_error_examples() \ No newline at end of file + load_error_examples()