Home Technology peripherals AI Retrieval Augmented Generation in SQLite

Retrieval Augmented Generation in SQLite

Feb 26, 2025 am 02:49 AM

This two-part series explores using SQLite for machine learning. The previous article discussed SQLite's growing role in production-ready web applications. This article focuses on implementing retrieval-augmented generation (RAG) using SQLite.

For custom web applications with generative AI, visit losangelesaiapps.com

The code is available here.

Traditional RAG implementation often involves:

  1. Searching for tutorials on RAG.
  2. Selecting a popular framework (LangChain, LlamaIndex).
  3. Choosing a cloud vector database (Pinecone, Weaviate).
  4. Integrating these components.

While effective, this approach can be overly complex, especially for beginners. This article demonstrates a simpler method using SQLite with the sqlite-vec extension and the OpenAI API. Part 1 of this series provides a detailed overview of SQLite's capabilities. For this article, it suffices to understand SQLite's simplicity as a single-file database.

This approach eliminates the need for cloud vector databases and bulky frameworks.

SQLite-Vec: Extending SQLite's Power

SQLite's strength lies in its extensibility. Extensions, similar to Python libraries, add functionality written in C. A prime example is the Full-Text Search (FTS) extension. sqlite-vec adds vector search capabilities, enabling semantic understanding beyond keyword matching. Searching for "horses" might return "equestrian" or "pony."

sqlite-vec uses virtual tables, offering:

  • Custom Data Sources: Data can reside outside the database file (e.g., CSV, API).
  • Flexible Functionality: Supports specialized indexing and complex data types.
  • Seamless Integration: Integrates with standard SQLite query syntax.
  • Modules: Backend logic is implemented in a separate module.

Virtual tables are created using:

CREATE VIRTUAL TABLE my_table USING my_extension_module();
Copy after login
Copy after login

my_extension_module() specifies the module (here, vec0 from sqlite-vec).

Code Walkthrough

The code (repo link) uses .txt files as sample data (mostly physics-related). my_docs.db is the SQLite database file.

  1. Installation: requirements.txt lists the necessary libraries (sqlite-vec, openai, python-dotenv). Create a virtual environment and run pip install -r requirements.txt.

  2. OpenAI API Key: Obtain an OpenAI API key.

  3. Loading the Extension: The Python code loads the sqlite-vec extension and creates a virtual table:

CREATE VIRTUAL TABLE my_table USING my_extension_module();
Copy after login
Copy after login

The documents table stores embeddings (embedding), filenames (file_name), and content (content). denotes auxiliary fields.

  1. Embedding and Insertion: The code iterates through .txt files, generates embeddings using the OpenAI API, and inserts them into the database:
db.enable_load_extension(True)
sqlite_vec.load(db)
db.enable_load_extension(False)

db.execute('''
    CREATE VIRTUAL TABLE documents USING vec0(
        embedding float[1536],
        +file_name TEXT,
        +content TEXT
    )
''')
Copy after login
  1. RAG Query: A KNN query retrieves similar documents based on embedding similarity:
# ... (OpenAI embedding function) ...

for file_name in os.listdir("data"):
    # ... (Open file, get content, get embedding) ...
    db.execute(
        'INSERT INTO documents (embedding, file_name, content) VALUES (?, ?, ?)',
        (serialize_float32(embedding), file_name, content)
    )
db.commit()
Copy after login

The results are then used as context for an OpenAI chat completion to answer the query.

Conclusion

sqlite-vec simplifies RAG significantly. It eliminates the need for complex frameworks and cloud services, making it cost-effective and easy to iterate upon. While scaling might eventually require a more robust database, sqlite-vec offers a compelling solution for smaller projects. The extension supports multiple programming languages.

Retrieval Augmented Generation in SQLite

The above is the detailed content of Retrieval Augmented Generation in SQLite. For more information, please follow other related articles on the PHP Chinese website!

Statement of this Website
The content of this article is voluntarily contributed by netizens, and the copyright belongs to the original author. This site does not assume corresponding legal responsibility. If you find any content suspected of plagiarism or infringement, please contact admin@php.cn

Hot AI Tools

Undresser.AI Undress

Undresser.AI Undress

AI-powered app for creating realistic nude photos

AI Clothes Remover

AI Clothes Remover

Online AI tool for removing clothes from photos.

Undress AI Tool

Undress AI Tool

Undress images for free

Clothoff.io

Clothoff.io

AI clothes remover

Video Face Swap

Video Face Swap

Swap faces in any video effortlessly with our completely free AI face swap tool!

Hot Tools

Notepad++7.3.1

Notepad++7.3.1

Easy-to-use and free code editor

SublimeText3 Chinese version

SublimeText3 Chinese version

Chinese version, very easy to use

Zend Studio 13.0.1

Zend Studio 13.0.1

Powerful PHP integrated development environment

Dreamweaver CS6

Dreamweaver CS6

Visual web development tools

SublimeText3 Mac version

SublimeText3 Mac version

God-level code editing software (SublimeText3)

Top 5 GenAI Launches of February 2025: GPT-4.5, Grok-3 & More! Top 5 GenAI Launches of February 2025: GPT-4.5, Grok-3 & More! Mar 22, 2025 am 10:58 AM

February 2025 has been yet another game-changing month for generative AI, bringing us some of the most anticipated model upgrades and groundbreaking new features. From xAI’s Grok 3 and Anthropic’s Claude 3.7 Sonnet, to OpenAI’s G

How to Use YOLO v12 for Object Detection? How to Use YOLO v12 for Object Detection? Mar 22, 2025 am 11:07 AM

YOLO (You Only Look Once) has been a leading real-time object detection framework, with each iteration improving upon the previous versions. The latest version YOLO v12 introduces advancements that significantly enhance accuracy

Best AI Art Generators (Free & Paid) for Creative Projects Best AI Art Generators (Free & Paid) for Creative Projects Apr 02, 2025 pm 06:10 PM

The article reviews top AI art generators, discussing their features, suitability for creative projects, and value. It highlights Midjourney as the best value for professionals and recommends DALL-E 2 for high-quality, customizable art.

Is ChatGPT 4 O available? Is ChatGPT 4 O available? Mar 28, 2025 pm 05:29 PM

ChatGPT 4 is currently available and widely used, demonstrating significant improvements in understanding context and generating coherent responses compared to its predecessors like ChatGPT 3.5. Future developments may include more personalized interactions and real-time data processing capabilities, further enhancing its potential for various applications.

Best AI Chatbots Compared (ChatGPT, Gemini, Claude & More) Best AI Chatbots Compared (ChatGPT, Gemini, Claude & More) Apr 02, 2025 pm 06:09 PM

The article compares top AI chatbots like ChatGPT, Gemini, and Claude, focusing on their unique features, customization options, and performance in natural language processing and reliability.

Getting Started With Meta Llama 3.2 - Analytics Vidhya Getting Started With Meta Llama 3.2 - Analytics Vidhya Apr 11, 2025 pm 12:04 PM

Meta's Llama 3.2: A Leap Forward in Multimodal and Mobile AI Meta recently unveiled Llama 3.2, a significant advancement in AI featuring powerful vision capabilities and lightweight text models optimized for mobile devices. Building on the success o

Top AI Writing Assistants to Boost Your Content Creation Top AI Writing Assistants to Boost Your Content Creation Apr 02, 2025 pm 06:11 PM

The article discusses top AI writing assistants like Grammarly, Jasper, Copy.ai, Writesonic, and Rytr, focusing on their unique features for content creation. It argues that Jasper excels in SEO optimization, while AI tools help maintain tone consist

Guide to Uber's H3 for Spatial Indexing Guide to Uber's H3 for Spatial Indexing Mar 22, 2025 am 10:54 AM

In today’s data-driven world, efficient geospatial indexing is crucial for applications ranging from ride-sharing and logistics to environmental monitoring and disaster response. Uber’s H3, a powerful open-source spat

See all articles