Fixed test

Author: are-ces

tests/unit/rag/test_rag_query.py

@@ -124,7 +124,6 @@ async def test_query_adds_vector_db_id_to_chunk_metadata(self):
         )
 
         result = await rag_tool.query(content="test", vector_db_ids=vector_db_ids)
-
         returned_chunks = result.metadata["chunks"]
         returned_scores = result.metadata["scores"]
         returned_doc_ids = result.metadata["document_ids"]
@@ -133,12 +132,17 @@ async def test_query_adds_vector_db_id_to_chunk_metadata(self):
         assert returned_scores == (0.9, 0.8)  
         assert returned_doc_ids == ["doc1", "doc2"]
 
-        metadata_texts = [
-            item.text
+        # Parse metadata from query result
+        def parse_metadata(s):
+            import ast, re
+            match = re.search(r"Metadata:\s*(\{.*\})", s)
+            if not match:
+                raise ValueError(f"No metadata found in string: {s}")
+            return ast.literal_eval(match.group(1))
+
+        returned_metadata = [
+            parse_metadata(item.text)["vector_db_id"]
             for item in result.content
             if "Metadata:" in item.text
         ]
-
-        assert any("'vector_db_id': 'db1'" in t for t in metadata_texts)
-        assert any("'vector_db_id': 'db2'" in t for t in metadata_texts)
-
+        assert returned_metadata == ["db1", "db2"]