From 765d28bede59417883264670b6039869c2339308 Mon Sep 17 00:00:00 2001
From: bryan davis <bryan@monkeytronics.org>
Date: Wed, 3 Dec 2025 12:11:28 -0600
Subject: [PATCH] add method signatures

---
 haiku_rag_slim/haiku/rag/client.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/haiku_rag_slim/haiku/rag/client.py b/haiku_rag_slim/haiku/rag/client.py
index 1bf21aba..a11d9cd1 100644
--- a/haiku_rag_slim/haiku/rag/client.py
+++ b/haiku_rag_slim/haiku/rag/client.py
@@ -105,6 +105,8 @@ class HaikuRAG:
         title: str | None = None,
         metadata: dict | None = None,
         chunks: list[Chunk] | None = None,
+        docling_document_json: str | None = None,
+        docling_version: str | None = None,
     ) -> Document:
         """Create a new document with optional URI and metadata.
 
@@ -142,6 +144,8 @@ class HaikuRAG:
                 uri=uri,
                 title=title,
                 metadata=metadata or {},
+                docling_document_json=docling_document_json,
+                docling_version=docling_version,
             )
 
             return await self.document_repository._create_and_chunk(
@@ -457,6 +461,8 @@ class HaikuRAG:
         metadata: dict | None = None,
         chunks: list[Chunk] | None = None,
         title: str | None = None,
+        docling_document_json: str | None = None,
+        docling_version: str | None = None,
     ) -> Document:
         """Update specific fields of a document by ID.
 
@@ -482,6 +488,10 @@ class HaikuRAG:
             existing_doc.title = title
         if metadata is not None:
             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
         if content is not None or chunks is not None: