danvics/haiku.rag
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add method signatures

Browse source
This commit is contained in:
bryan davis 2025-12-03 12:11:28 -06:00
parent 4436dd80cd
commit 765d28bede
1 changed files with 10 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
haiku_rag_slim/haiku/rag/client.py
View file

@ -105,6 +105,8 @@ class HaikuRAG:
title: str | None = None, title: str | None = None,
metadata: dict | None = None, metadata: dict | None = None,
chunks: list[Chunk] | None = None, chunks: list[Chunk] | None = None,
docling_document_json: str | None = None,
docling_version: str | None = None,
) -> Document: ) -> Document:
"""Create a new document with optional URI and metadata. """Create a new document with optional URI and metadata.
@ -142,6 +144,8 @@ class HaikuRAG:
uri=uri, uri=uri,
title=title, title=title,
metadata=metadata or {}, metadata=metadata or {},
docling_document_json=docling_document_json,
docling_version=docling_version,
) )
return await self.document_repository._create_and_chunk( return await self.document_repository._create_and_chunk(
@ -457,6 +461,8 @@ class HaikuRAG:
metadata: dict | None = None, metadata: dict | None = None,
chunks: list[Chunk] | None = None, chunks: list[Chunk] | None = None,
title: str | None = None, title: str | None = None,
docling_document_json: str | None = None,
docling_version: str | None = None,
) -> Document: ) -> Document:
"""Update specific fields of a document by ID. """Update specific fields of a document by ID.
@ -482,6 +488,10 @@ class HaikuRAG:
existing_doc.title = title existing_doc.title = title
if metadata is not None: if metadata is not None:
existing_doc.metadata = metadata existing_doc.metadata = metadata
if docling_document_json is not None:
existing_doc.docling_document_json = docling_document_json
if docling_version is not None:
existing_doc.docling_version = docling_version
# Determine if we need to rechunk # Determine if we need to rechunk
if content is not None or chunks is not None: if content is not None or chunks is not None: