Skip to content

Commit 971d7d6

Browse files
authored
Merge pull request #17 from WasmEdge/juntao-patch-1
Use UUID for error_examples
2 parents abbdd96 + 54f4bc5 commit 971d7d6

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

app/load_data.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -49,15 +49,18 @@ def load_error_examples():
4949

5050
# Get embedding for error
5151
embedding = llm_client.get_embeddings([example["error"]])[0]
52+
53+
# Store in vector DB with proper UUID
54+
point_id = str(uuid.uuid4())
5255

5356
# Store in vector DB
5457
vector_store.upsert("error_examples",
55-
[{"id": os.path.basename(file_path),
58+
[{"id": point_id,
5659
"vector": embedding,
5760
"payload": example}])
5861

5962
print(f"Loaded error example: {example['error'][:50]}...")
6063

6164
if __name__ == "__main__":
6265
load_project_examples()
63-
load_error_examples()
66+
load_error_examples()

0 commit comments

Comments
 (0)