haiku.rag-slim 0.19.5__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- haiku_rag_slim-0.19.5/.gitignore +31 -0
- haiku_rag_slim-0.19.5/LICENSE +7 -0
- haiku_rag_slim-0.19.5/PKG-INFO +130 -0
- haiku_rag_slim-0.19.5/README.md +73 -0
- haiku_rag_slim-0.19.5/haiku/rag/__init__.py +0 -0
- haiku_rag_slim-0.19.5/haiku/rag/app.py +631 -0
- haiku_rag_slim-0.19.5/haiku/rag/chunkers/__init__.py +31 -0
- haiku_rag_slim-0.19.5/haiku/rag/chunkers/base.py +28 -0
- haiku_rag_slim-0.19.5/haiku/rag/chunkers/docling_local.py +110 -0
- haiku_rag_slim-0.19.5/haiku/rag/chunkers/docling_serve.py +109 -0
- haiku_rag_slim-0.19.5/haiku/rag/cli.py +520 -0
- haiku_rag_slim-0.19.5/haiku/rag/client.py +862 -0
- haiku_rag_slim-0.19.5/haiku/rag/config/__init__.py +78 -0
- haiku_rag_slim-0.19.5/haiku/rag/config/loader.py +55 -0
- haiku_rag_slim-0.19.5/haiku/rag/config/models.py +170 -0
- haiku_rag_slim-0.19.5/haiku/rag/converters/__init__.py +31 -0
- haiku_rag_slim-0.19.5/haiku/rag/converters/base.py +59 -0
- haiku_rag_slim-0.19.5/haiku/rag/converters/docling_local.py +153 -0
- haiku_rag_slim-0.19.5/haiku/rag/converters/docling_serve.py +198 -0
- haiku_rag_slim-0.19.5/haiku/rag/converters/text_utils.py +127 -0
- haiku_rag_slim-0.19.5/haiku/rag/embeddings/__init__.py +56 -0
- haiku_rag_slim-0.19.5/haiku/rag/embeddings/base.py +25 -0
- haiku_rag_slim-0.19.5/haiku/rag/embeddings/lm_studio.py +28 -0
- haiku_rag_slim-0.19.5/haiku/rag/embeddings/ollama.py +28 -0
- haiku_rag_slim-0.19.5/haiku/rag/embeddings/openai.py +26 -0
- haiku_rag_slim-0.19.5/haiku/rag/embeddings/vllm.py +29 -0
- haiku_rag_slim-0.19.5/haiku/rag/embeddings/voyageai.py +27 -0
- haiku_rag_slim-0.19.5/haiku/rag/graph/__init__.py +26 -0
- haiku_rag_slim-0.19.5/haiku/rag/graph/agui/__init__.py +53 -0
- haiku_rag_slim-0.19.5/haiku/rag/graph/agui/cli_renderer.py +135 -0
- haiku_rag_slim-0.19.5/haiku/rag/graph/agui/emitter.py +201 -0
- haiku_rag_slim-0.19.5/haiku/rag/graph/agui/events.py +254 -0
- haiku_rag_slim-0.19.5/haiku/rag/graph/agui/server.py +310 -0
- haiku_rag_slim-0.19.5/haiku/rag/graph/agui/state.py +34 -0
- haiku_rag_slim-0.19.5/haiku/rag/graph/agui/stream.py +86 -0
- haiku_rag_slim-0.19.5/haiku/rag/graph/common/__init__.py +5 -0
- haiku_rag_slim-0.19.5/haiku/rag/graph/common/models.py +42 -0
- haiku_rag_slim-0.19.5/haiku/rag/graph/common/nodes.py +297 -0
- haiku_rag_slim-0.19.5/haiku/rag/graph/common/prompts.py +46 -0
- haiku_rag_slim-0.19.5/haiku/rag/graph/deep_qa/__init__.py +1 -0
- haiku_rag_slim-0.19.5/haiku/rag/graph/deep_qa/dependencies.py +27 -0
- haiku_rag_slim-0.19.5/haiku/rag/graph/deep_qa/graph.py +249 -0
- haiku_rag_slim-0.19.5/haiku/rag/graph/deep_qa/models.py +20 -0
- haiku_rag_slim-0.19.5/haiku/rag/graph/deep_qa/prompts.py +59 -0
- haiku_rag_slim-0.19.5/haiku/rag/graph/deep_qa/state.py +56 -0
- haiku_rag_slim-0.19.5/haiku/rag/graph/research/__init__.py +3 -0
- haiku_rag_slim-0.19.5/haiku/rag/graph/research/common.py +87 -0
- haiku_rag_slim-0.19.5/haiku/rag/graph/research/dependencies.py +151 -0
- haiku_rag_slim-0.19.5/haiku/rag/graph/research/graph.py +312 -0
- haiku_rag_slim-0.19.5/haiku/rag/graph/research/models.py +166 -0
- haiku_rag_slim-0.19.5/haiku/rag/graph/research/prompts.py +107 -0
- haiku_rag_slim-0.19.5/haiku/rag/graph/research/state.py +85 -0
- haiku_rag_slim-0.19.5/haiku/rag/inspector/__init__.py +8 -0
- haiku_rag_slim-0.19.5/haiku/rag/inspector/app.py +194 -0
- haiku_rag_slim-0.19.5/haiku/rag/inspector/widgets/__init__.py +5 -0
- haiku_rag_slim-0.19.5/haiku/rag/inspector/widgets/chunk_list.py +67 -0
- haiku_rag_slim-0.19.5/haiku/rag/inspector/widgets/detail_view.py +92 -0
- haiku_rag_slim-0.19.5/haiku/rag/inspector/widgets/document_list.py +64 -0
- haiku_rag_slim-0.19.5/haiku/rag/inspector/widgets/search_modal.py +151 -0
- haiku_rag_slim-0.19.5/haiku/rag/logging.py +56 -0
- haiku_rag_slim-0.19.5/haiku/rag/mcp.py +245 -0
- haiku_rag_slim-0.19.5/haiku/rag/monitor.py +206 -0
- haiku_rag_slim-0.19.5/haiku/rag/qa/__init__.py +29 -0
- haiku_rag_slim-0.19.5/haiku/rag/qa/agent.py +71 -0
- haiku_rag_slim-0.19.5/haiku/rag/qa/prompts.py +60 -0
- haiku_rag_slim-0.19.5/haiku/rag/reranking/__init__.py +63 -0
- haiku_rag_slim-0.19.5/haiku/rag/reranking/base.py +13 -0
- haiku_rag_slim-0.19.5/haiku/rag/reranking/cohere.py +35 -0
- haiku_rag_slim-0.19.5/haiku/rag/reranking/mxbai.py +31 -0
- haiku_rag_slim-0.19.5/haiku/rag/reranking/vllm.py +44 -0
- haiku_rag_slim-0.19.5/haiku/rag/reranking/zeroentropy.py +60 -0
- haiku_rag_slim-0.19.5/haiku/rag/store/__init__.py +4 -0
- haiku_rag_slim-0.19.5/haiku/rag/store/engine.py +393 -0
- haiku_rag_slim-0.19.5/haiku/rag/store/models/__init__.py +4 -0
- haiku_rag_slim-0.19.5/haiku/rag/store/models/chunk.py +17 -0
- haiku_rag_slim-0.19.5/haiku/rag/store/models/document.py +17 -0
- haiku_rag_slim-0.19.5/haiku/rag/store/repositories/__init__.py +9 -0
- haiku_rag_slim-0.19.5/haiku/rag/store/repositories/chunk.py +501 -0
- haiku_rag_slim-0.19.5/haiku/rag/store/repositories/document.py +263 -0
- haiku_rag_slim-0.19.5/haiku/rag/store/repositories/settings.py +165 -0
- haiku_rag_slim-0.19.5/haiku/rag/store/upgrades/__init__.py +62 -0
- haiku_rag_slim-0.19.5/haiku/rag/store/upgrades/v0_10_1.py +64 -0
- haiku_rag_slim-0.19.5/haiku/rag/store/upgrades/v0_9_3.py +112 -0
- haiku_rag_slim-0.19.5/haiku/rag/utils.py +415 -0
- haiku_rag_slim-0.19.5/pyproject.toml +65 -0
- haiku_rag_slim-0.19.5/test_agui_server.py +101 -0
|
@@ -0,0 +1,31 @@
|
|
|
1
|
+
# Python-generated files
|
|
2
|
+
__pycache__/
|
|
3
|
+
*.py[oc]
|
|
4
|
+
build/
|
|
5
|
+
dist/
|
|
6
|
+
wheels/
|
|
7
|
+
*.egg-info
|
|
8
|
+
|
|
9
|
+
# Virtual environments
|
|
10
|
+
.venv
|
|
11
|
+
|
|
12
|
+
# tests
|
|
13
|
+
.coverage*
|
|
14
|
+
evaluations/evaluations/data/
|
|
15
|
+
tests/data/
|
|
16
|
+
.pytest_cache/
|
|
17
|
+
.ruff_cache/
|
|
18
|
+
|
|
19
|
+
# environment variables and config files
|
|
20
|
+
.env
|
|
21
|
+
haiku.rag.yaml
|
|
22
|
+
TODO.md
|
|
23
|
+
PLAN.md
|
|
24
|
+
DEVNOTES.md
|
|
25
|
+
|
|
26
|
+
# mcp registry
|
|
27
|
+
.mcpregistry_github_token
|
|
28
|
+
.mcpregistry_registry_token
|
|
29
|
+
|
|
30
|
+
# MkDocs site directory when doing local docs builds
|
|
31
|
+
site/
|
|
@@ -0,0 +1,7 @@
|
|
|
1
|
+
Copyright 2025 Yiorgis Gozadinos
|
|
2
|
+
|
|
3
|
+
Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the “Software”), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
|
|
4
|
+
|
|
5
|
+
The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
|
|
6
|
+
|
|
7
|
+
THE SOFTWARE IS PROVIDED “AS IS”, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
@@ -0,0 +1,130 @@
|
|
|
1
|
+
Metadata-Version: 2.4
|
|
2
|
+
Name: haiku.rag-slim
|
|
3
|
+
Version: 0.19.5
|
|
4
|
+
Summary: Opinionated agentic RAG powered by LanceDB, Pydantic AI, and Docling - Minimal dependencies
|
|
5
|
+
Author-email: Yiorgis Gozadinos <ggozadinos@gmail.com>
|
|
6
|
+
License: MIT
|
|
7
|
+
License-File: LICENSE
|
|
8
|
+
Keywords: RAG,lancedb,mcp,ml,vector-database
|
|
9
|
+
Classifier: Development Status :: 4 - Beta
|
|
10
|
+
Classifier: Environment :: Console
|
|
11
|
+
Classifier: Intended Audience :: Developers
|
|
12
|
+
Classifier: Operating System :: MacOS
|
|
13
|
+
Classifier: Operating System :: Microsoft :: Windows :: Windows 10
|
|
14
|
+
Classifier: Operating System :: Microsoft :: Windows :: Windows 11
|
|
15
|
+
Classifier: Operating System :: POSIX :: Linux
|
|
16
|
+
Classifier: Programming Language :: Python :: 3.12
|
|
17
|
+
Classifier: Programming Language :: Python :: 3.13
|
|
18
|
+
Classifier: Typing :: Typed
|
|
19
|
+
Requires-Python: >=3.12
|
|
20
|
+
Requires-Dist: docling-core==2.51.1
|
|
21
|
+
Requires-Dist: httpx>=0.28.1
|
|
22
|
+
Requires-Dist: lancedb==0.25.3
|
|
23
|
+
Requires-Dist: pathspec>=0.12.1
|
|
24
|
+
Requires-Dist: pydantic-ai-slim[ag-ui,fastmcp,logfire,openai]==1.19.0
|
|
25
|
+
Requires-Dist: pydantic>=2.12.4
|
|
26
|
+
Requires-Dist: python-dotenv>=1.2.1
|
|
27
|
+
Requires-Dist: pyyaml>=6.0.3
|
|
28
|
+
Requires-Dist: rich>=14.2.0
|
|
29
|
+
Requires-Dist: typer<0.20.0,>=0.19.2
|
|
30
|
+
Requires-Dist: watchfiles>=1.1.1
|
|
31
|
+
Provides-Extra: anthropic
|
|
32
|
+
Requires-Dist: pydantic-ai-slim[anthropic]; extra == 'anthropic'
|
|
33
|
+
Provides-Extra: bedrock
|
|
34
|
+
Requires-Dist: pydantic-ai-slim[bedrock]; extra == 'bedrock'
|
|
35
|
+
Provides-Extra: cohere
|
|
36
|
+
Requires-Dist: cohere>=5.20.0; extra == 'cohere'
|
|
37
|
+
Provides-Extra: docling
|
|
38
|
+
Requires-Dist: docling==2.62.0; extra == 'docling'
|
|
39
|
+
Requires-Dist: opencv-python-headless>=4.11.0.86; extra == 'docling'
|
|
40
|
+
Provides-Extra: google
|
|
41
|
+
Requires-Dist: pydantic-ai-slim[google]; extra == 'google'
|
|
42
|
+
Provides-Extra: groq
|
|
43
|
+
Requires-Dist: pydantic-ai-slim[groq]; extra == 'groq'
|
|
44
|
+
Provides-Extra: inspector
|
|
45
|
+
Requires-Dist: textual>=1.0.0; extra == 'inspector'
|
|
46
|
+
Provides-Extra: mistral
|
|
47
|
+
Requires-Dist: pydantic-ai-slim[mistral]; extra == 'mistral'
|
|
48
|
+
Provides-Extra: mxbai
|
|
49
|
+
Requires-Dist: mxbai-rerank>=0.1.6; extra == 'mxbai'
|
|
50
|
+
Provides-Extra: vertexai
|
|
51
|
+
Requires-Dist: pydantic-ai-slim[vertexai]; extra == 'vertexai'
|
|
52
|
+
Provides-Extra: voyageai
|
|
53
|
+
Requires-Dist: voyageai>=0.3.5; extra == 'voyageai'
|
|
54
|
+
Provides-Extra: zeroentropy
|
|
55
|
+
Requires-Dist: zeroentropy>=0.1.0a6; extra == 'zeroentropy'
|
|
56
|
+
Description-Content-Type: text/markdown
|
|
57
|
+
|
|
58
|
+
# haiku.rag-slim
|
|
59
|
+
|
|
60
|
+
Opinionated agentic RAG powered by LanceDB, Pydantic AI, and Docling - Core package with minimal dependencies.
|
|
61
|
+
|
|
62
|
+
`haiku.rag-slim` is the core package for users who want to install only the dependencies they need. Document processing (docling), and reranker support are all optional extras.
|
|
63
|
+
|
|
64
|
+
**For most users, we recommend installing [`haiku.rag`](https://pypi.org/project/haiku.rag/) instead**, which includes all features out of the box.
|
|
65
|
+
|
|
66
|
+
## Installation
|
|
67
|
+
|
|
68
|
+
**Python 3.12 or newer required**
|
|
69
|
+
|
|
70
|
+
### Minimal Installation
|
|
71
|
+
|
|
72
|
+
```bash
|
|
73
|
+
uv pip install haiku.rag-slim
|
|
74
|
+
```
|
|
75
|
+
|
|
76
|
+
Core functionality with OpenAI/Ollama support, MCP server, and Logfire observability. Document processing (docling) is optional.
|
|
77
|
+
|
|
78
|
+
### With Document Processing
|
|
79
|
+
|
|
80
|
+
```bash
|
|
81
|
+
uv pip install haiku.rag-slim[docling]
|
|
82
|
+
```
|
|
83
|
+
|
|
84
|
+
Adds support for 40+ file formats including PDF, DOCX, HTML, and more.
|
|
85
|
+
|
|
86
|
+
### Available Extras
|
|
87
|
+
|
|
88
|
+
**Document Processing:**
|
|
89
|
+
- `docling` - PDF, DOCX, HTML, and 40+ file formats
|
|
90
|
+
|
|
91
|
+
**Embedding Providers:**
|
|
92
|
+
- `voyageai` - VoyageAI embeddings
|
|
93
|
+
|
|
94
|
+
**Rerankers:**
|
|
95
|
+
- `mxbai` - MixedBread AI
|
|
96
|
+
- `cohere` - Cohere
|
|
97
|
+
- `zeroentropy` - Zero Entropy
|
|
98
|
+
|
|
99
|
+
**Model Providers:**
|
|
100
|
+
- OpenAI/Ollama - included in core (OpenAI-compatible APIs)
|
|
101
|
+
- `anthropic` - Anthropic Claude
|
|
102
|
+
- `groq` - Groq
|
|
103
|
+
- `google` - Google Gemini
|
|
104
|
+
- `mistral` - Mistral AI
|
|
105
|
+
- `bedrock` - AWS Bedrock
|
|
106
|
+
- `vertexai` - Google Vertex AI
|
|
107
|
+
|
|
108
|
+
|
|
109
|
+
```bash
|
|
110
|
+
# Common combinations
|
|
111
|
+
uv pip install haiku.rag-slim[docling,anthropic,mxbai]
|
|
112
|
+
uv pip install haiku.rag-slim[docling,groq,logfire]
|
|
113
|
+
```
|
|
114
|
+
|
|
115
|
+
## Usage
|
|
116
|
+
|
|
117
|
+
See the main [`haiku.rag`](https://github.com/ggozad/haiku.rag) repository for:
|
|
118
|
+
- Quick start guide
|
|
119
|
+
- CLI examples
|
|
120
|
+
- Python API usage
|
|
121
|
+
- MCP server setup
|
|
122
|
+
|
|
123
|
+
## Documentation
|
|
124
|
+
|
|
125
|
+
Full documentation: https://ggozad.github.io/haiku.rag/
|
|
126
|
+
|
|
127
|
+
- [Installation](https://ggozad.github.io/haiku.rag/installation/) - Provider setup
|
|
128
|
+
- [Configuration](https://ggozad.github.io/haiku.rag/configuration/) - YAML configuration
|
|
129
|
+
- [CLI](https://ggozad.github.io/haiku.rag/cli/) - Command reference
|
|
130
|
+
- [Python API](https://ggozad.github.io/haiku.rag/python/) - Complete API docs
|
|
@@ -0,0 +1,73 @@
|
|
|
1
|
+
# haiku.rag-slim
|
|
2
|
+
|
|
3
|
+
Opinionated agentic RAG powered by LanceDB, Pydantic AI, and Docling - Core package with minimal dependencies.
|
|
4
|
+
|
|
5
|
+
`haiku.rag-slim` is the core package for users who want to install only the dependencies they need. Document processing (docling), and reranker support are all optional extras.
|
|
6
|
+
|
|
7
|
+
**For most users, we recommend installing [`haiku.rag`](https://pypi.org/project/haiku.rag/) instead**, which includes all features out of the box.
|
|
8
|
+
|
|
9
|
+
## Installation
|
|
10
|
+
|
|
11
|
+
**Python 3.12 or newer required**
|
|
12
|
+
|
|
13
|
+
### Minimal Installation
|
|
14
|
+
|
|
15
|
+
```bash
|
|
16
|
+
uv pip install haiku.rag-slim
|
|
17
|
+
```
|
|
18
|
+
|
|
19
|
+
Core functionality with OpenAI/Ollama support, MCP server, and Logfire observability. Document processing (docling) is optional.
|
|
20
|
+
|
|
21
|
+
### With Document Processing
|
|
22
|
+
|
|
23
|
+
```bash
|
|
24
|
+
uv pip install haiku.rag-slim[docling]
|
|
25
|
+
```
|
|
26
|
+
|
|
27
|
+
Adds support for 40+ file formats including PDF, DOCX, HTML, and more.
|
|
28
|
+
|
|
29
|
+
### Available Extras
|
|
30
|
+
|
|
31
|
+
**Document Processing:**
|
|
32
|
+
- `docling` - PDF, DOCX, HTML, and 40+ file formats
|
|
33
|
+
|
|
34
|
+
**Embedding Providers:**
|
|
35
|
+
- `voyageai` - VoyageAI embeddings
|
|
36
|
+
|
|
37
|
+
**Rerankers:**
|
|
38
|
+
- `mxbai` - MixedBread AI
|
|
39
|
+
- `cohere` - Cohere
|
|
40
|
+
- `zeroentropy` - Zero Entropy
|
|
41
|
+
|
|
42
|
+
**Model Providers:**
|
|
43
|
+
- OpenAI/Ollama - included in core (OpenAI-compatible APIs)
|
|
44
|
+
- `anthropic` - Anthropic Claude
|
|
45
|
+
- `groq` - Groq
|
|
46
|
+
- `google` - Google Gemini
|
|
47
|
+
- `mistral` - Mistral AI
|
|
48
|
+
- `bedrock` - AWS Bedrock
|
|
49
|
+
- `vertexai` - Google Vertex AI
|
|
50
|
+
|
|
51
|
+
|
|
52
|
+
```bash
|
|
53
|
+
# Common combinations
|
|
54
|
+
uv pip install haiku.rag-slim[docling,anthropic,mxbai]
|
|
55
|
+
uv pip install haiku.rag-slim[docling,groq,logfire]
|
|
56
|
+
```
|
|
57
|
+
|
|
58
|
+
## Usage
|
|
59
|
+
|
|
60
|
+
See the main [`haiku.rag`](https://github.com/ggozad/haiku.rag) repository for:
|
|
61
|
+
- Quick start guide
|
|
62
|
+
- CLI examples
|
|
63
|
+
- Python API usage
|
|
64
|
+
- MCP server setup
|
|
65
|
+
|
|
66
|
+
## Documentation
|
|
67
|
+
|
|
68
|
+
Full documentation: https://ggozad.github.io/haiku.rag/
|
|
69
|
+
|
|
70
|
+
- [Installation](https://ggozad.github.io/haiku.rag/installation/) - Provider setup
|
|
71
|
+
- [Configuration](https://ggozad.github.io/haiku.rag/configuration/) - YAML configuration
|
|
72
|
+
- [CLI](https://ggozad.github.io/haiku.rag/cli/) - Command reference
|
|
73
|
+
- [Python API](https://ggozad.github.io/haiku.rag/python/) - Complete API docs
|
|
File without changes
|