pk-ai-tools 0.2.2

Simple RAG pipeline + document ingestion utilities (LangChain + Chroma + Ollama).

pk-ai-tools

pk-ai-tools is a small Python library that provides a reusable RAG (Retrieval-Augmented Generation) pipeline and a flexible document ingestion system.

It is built on top of:

• LangChain
• Chroma
• Ollama

The main goal of this project is to make it easy to:

• Ingest documents from many formats (PDF, Word, Excel, CSV, Markdown, HTML, etc.)
• Build or update a Chroma vector database
• Ask questions using a simple RAG pipeline backed by local Ollama models

This library was originally created for personal use and has since been generalized to be reusable across projects.

---

Installation

pip install pk-ai-tools
Requirements
Python 3.10 – 3.13

Python 3.14+ is currently not supported due to upstream dependencies (Chroma / onnxruntime)

Ollama installed and running

https://ollama.com

At least one local LLM downloaded (for example: llama3)

Quick example
python
Kopiera kod
from pk_ai_tools import RAGPipeline

rag = RAGPipeline(
    doc_folder="./data",
    language="en",
    uuid="demo-user",
    model_name="llama3"
)

answer = rag.ask("What is this documentation about?")
print(answer)
How it works (high level)
Documents in doc_folder are ingested and split into chunks

Chunks are embedded and stored in a Chroma vector database

Queries are expanded and retrieved using LangChain

Answers are generated by a local Ollama model

Supported document types
The document ingestion system supports:

PDF

Word (.docx)

Excel (.xlsx)

CSV

Markdown

HTML

Plain text

Notes
This library is designed for local, private RAG setups

No cloud APIs are required

Vector databases are stored locally

Ollama must be running before querying

Roadmap (informal)
Improve Python 3.14+ compatibility when dependencies allow

Optional dependency groups (lighter installs)

Better configuration presets

More examples and docs