haiku.rag 0.7.5__tar.gz → 0.7.7__tar.gz

{haiku_rag-0.7.5 → haiku_rag-0.7.7}/.pre-commit-config.yaml

@@ -21,12 +21,12 @@ repos:
     hooks:
       - id: pyright
 
-  # - repo: https://github.com/RodrigoGonzalez/check-mkdocs
-  #   rev: v1.2.0
-  #   hooks:
-  #     - id: check-mkdocs
-  #       name: check-mkdocs
-  #       args: ["--config", "mkdocs.yml"] # Optional, mkdocs.yml is the default
-  #       # If you have additional plugins or libraries that are not included in
-  #       # check-mkdocs, add them here
-  #       additional_dependencies: ["mkdocs-material"]
+  - repo: https://github.com/RodrigoGonzalez/check-mkdocs
+    rev: v1.2.0
+    hooks:
+      - id: check-mkdocs
+        name: check-mkdocs
+        args: ["--config", "mkdocs.yml"] # Optional, mkdocs.yml is the default
+        # If you have additional plugins or libraries that are not included in
+        # check-mkdocs, add them here
+        additional_dependencies: ["mkdocs-material"]

{haiku_rag-0.7.5 → haiku_rag-0.7.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: haiku.rag
-Version: 0.7.5
+Version: 0.7.7
 Summary: Retrieval Augmented Generation (RAG) with LanceDB
 Author-email: Yiorgis Gozadinos <ggozadinos@gmail.com>
 License: MIT

{haiku_rag-0.7.5 → haiku_rag-0.7.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "haiku.rag"
-version = "0.7.5"
+version = "0.7.7"
 description = "Retrieval Augmented Generation (RAG) with LanceDB"
 authors = [{ name = "Yiorgis Gozadinos", email = "ggozadinos@gmail.com" }]
 license = { text = "MIT" }

{haiku_rag-0.7.5 → haiku_rag-0.7.7}/src/haiku/rag/app.py

@@ -170,7 +170,7 @@ class HaikuRAGApp:
                 if transport == "stdio":
                     await server.run_stdio_async()
                 elif transport == "sse":
-                    await server.run_sse_async("sse")
+                    await server.run_sse_async()
                 else:
                     await server.run_http_async("streamable-http")
             except KeyboardInterrupt:

{haiku_rag-0.7.5 → haiku_rag-0.7.7}/src/haiku/rag/embeddings/__init__.py

@@ -27,4 +27,9 @@ def get_embedder() -> EmbedderBase:
 
         return OpenAIEmbedder(Config.EMBEDDINGS_MODEL, Config.EMBEDDINGS_VECTOR_DIM)
 
+    if Config.EMBEDDINGS_PROVIDER == "vllm":
+        from haiku.rag.embeddings.vllm import Embedder as VllmEmbedder
+
+        return VllmEmbedder(Config.EMBEDDINGS_MODEL, Config.EMBEDDINGS_VECTOR_DIM)
+
     raise ValueError(f"Unsupported embedding provider: {Config.EMBEDDINGS_PROVIDER}")

{haiku_rag-0.7.5 → haiku_rag-0.7.7}/tests/test_app.py

@@ -200,7 +200,7 @@ async def test_serve(app: HaikuRAGApp, monkeypatch, transport):
     if transport == "stdio":
         mock_server.run_stdio_async.assert_called_once()
     elif transport == "sse":
-        mock_server.run_sse_async.assert_called_once_with("sse")
+        mock_server.run_sse_async.assert_called_once()
     else:
         mock_server.run_http_async.assert_called_once_with("streamable-http")
 

{haiku_rag-0.7.5 → haiku_rag-0.7.7}/uv.lock

@@ -951,7 +951,7 @@ wheels = [
 
 [[package]]
 name = "haiku-rag"
-version = "0.7.5"
+version = "0.7.7"
 source = { editable = "." }
 dependencies = [
     { name = "docling" },