diff --git a/src/haiku/rag/client.py b/src/haiku/rag/client.py index d25ddb2b..920f262a 100644 --- a/src/haiku/rag/client.py +++ b/src/haiku/rag/client.py @@ -88,7 +88,7 @@ class HaikuRAG: if not source_path.exists(): raise ValueError(f"File does not exist: {source_path}") - uri = str(source_path.resolve()) + uri = source_path.as_uri() md5_hash = hashlib.md5(source_path.read_bytes()).hexdigest() # Check if document already exists diff --git a/tests/test_client.py b/tests/test_client.py index 77430e24..086facc0 100644 --- a/tests/test_client.py +++ b/tests/test_client.py @@ -98,7 +98,7 @@ async def test_client_create_document_from_source(): assert doc.id is not None assert doc.content == test_content - assert doc.uri == str(temp_path.resolve()) + assert doc.uri == temp_path.as_uri() assert doc.metadata["source_type"] == "file" assert "contentType" in doc.metadata assert "md5" in doc.metadata @@ -109,7 +109,7 @@ async def test_client_create_document_from_source(): assert doc2.id is not None assert doc2.content == test_content - assert doc2.uri == str(temp_path.resolve()) + assert doc2.uri == temp_path.as_uri() assert "contentType" in doc2.metadata assert "md5" in doc2.metadata