Skip to content

upstash/wikipedia-semantic-search

Repository files navigation

Indexing Millions of Wikipedia Articles With Upstash Vector

This repository contains the code and documentation for our project on indexing millions of Wikipedia articles using Upstash Vector, as described in our blog post.

Project Overview

We've created a semantic search engine and Upstash RAG Chat SDK using Wikipedia data to demonstrate the capabilities of Upstash Vector and RAG Chat SDK. The project involves:

  1. Preparing and embedding Wikipedia articles
  2. Indexing the vectors using Upstash Vector
  3. Building a Wikipedia semantic search engine
  4. Implementing a RAG chatbot

Key Features

  • Indexed over 144 million vectors from Wikipedia articles in 11 languages
  • Used BGE-M3 embedding model for multilingual support
  • Implemented semantic search with cross-lingual capabilities
  • Created a RAG chatbot using Upstash RAG Chat SDK

Technologies Used

Development

To run the project locally, follow these steps:

  1. Go to Upstash Console to manage your databases:
    • Create a new Vector database with embedding model support. You can choose the BGE-M3 model for multilingual support.
    • Create a new Redis database for storing chat sessions.
    • Copy the credentials for both Redis and Vector. Also copy the QStash credentials for using the upstash hosted LLM models.

Put the credentials in a .env file in the root of the project. Your .env file should look like this:

UPSTASH_VECTOR_REST_URL=
UPSTASH_VECTOR_REST_TOKEN=

UPSTASH_REDIS_REST_TOKEN=
UPSTASH_REDIS_REST_URL=

QSTASH_TOKEN=
  1. Populate your Vector index.

This project uses namespaces to store articles in different languages. So you have to upsert the vectors in the correct namespace. For english, upsert your vectors into the en namespace.

  1. Install the dependencies:
pnpm install
  1. Run the development server:
pnpm dev

Contributing

We welcome contributions to improve this project. Please feel free to submit issues or pull requests.

Acknowledgements

  • Wikipedia for providing the dataset
  • Upstash for their vector database and RAG Chat SDK
  • All contributors to the open-source libraries used in this project

Contact

For any questions or feedback about the project or Upstash Vector, please reach out to us at (add contact information).

Check out our live demo to see the project in action!