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()