Skip to content

aaronmoruf7/weaviate_vector_search

Folders and files

NameName
Last commit message
Last commit date

Latest commit

ย 

History

2 Commits
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 

Repository files navigation

This project uses CrewAI + OpenAI + Weaviate to dynamically search a sales database.

Instructions

git clone https://github.com/Anchor-Co-Pilot/agentic-rag-poc.git
pip install -r requirements.txt

Create a .env file

OPENAI_API_KEY=your-openai-api-key
WEAVIATE_URL=your-weaviate-instance-url
WEAVIATE_API_KEY=your-weaviate-api-key

Vector Database Schema (Weaviate)

Collection name: sales_rag_docs

Field Type
description text
invoice_no text
quantity text
unit_price text
country text
return_status text

๐Ÿš€ How to Run

  1. Run python src/preprocessing/pandas_preprocessing.py to upload data to the vector database and create embeddings

  2. Edit tests/test_crew.py

Change the line:

user_input = "Find all electronics in Germany"

to any query you want. Note that for this poc, it only allows one filter at a time

  1. Run the crew:
PYTHONPATH=src python tests/test_crew.py

โœ… Example Output

[
  {
    "description": "Headphones",
    "invoice_no": "231932",
    "return_status": "Returned",
    "unit_price": "29.11",
    "quantity": "49",
    "country": "Germany"
  },
  ...
]

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages