cortexdb-py 0.2.2

Python SDK for CortexDB — a RAG-powered memory database

CortexDB Python SDK

A Python client for the CortexDB RAG backend, providing easy access to ingestion, querying, and LLM orchestration.

Features

• Setup API: Configure LLM providers (Gemini, OpenAI, Azure).
• Ingest API: Ingest documents with automatic embedding generation.
• Query API: Perform semantic search, entity retrieval, and graph traversals.
• Native LLM Integration: Uses google-genai for native Gemini support.

Installation

pip install -e .

Usage

See main.py for a complete example of setting up a medical chatbot.

from cortexdb import CortexDB

db = CortexDB("http://localhost:8080")

# Setup
db.setup.configure(provider="GEMINI", api_key="...", chat_model="gemini-2.0-flash", embed_model="gemini-embedding-001")

# Ingest
db.ingest.document(uid="user-1", document_title="Test Doc", document_text="Hello world")
db.ingest.prompt(uid="user-1", converser="USER", text="What is this?")

# Query
results = db.query.search_contexts("Hello", limit=5)

Testing

pip install pytest respx
pytest tests/