Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Feature/revive integration tests #1343

Merged
merged 30 commits into from
Oct 5, 2024
Merged
Changes from 2 commits
Commits
Show all changes
30 commits
Select commit Hold shift + click to select a range
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
157 changes: 26 additions & 131 deletions .github/workflows/integration-test-workflow-debian.yml
Original file line number Diff line number Diff line change
@@ -1,10 +1,14 @@
name: R2R CLI Integration Test (Debian GNU/Linux 12 (bookworm) amd64)
name: R2R CLI Integration (ubuntu latest)

on:
pull_request:
branches:
- 'dev*'
- 'dev-minor*'
push:
branches:
- '**'
workflow_dispatch:
- 'dev*'
- 'dev-minor*'

jobs:
build-and-test:
Expand All @@ -16,133 +20,24 @@ jobs:
env:
OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
TELEMETRY_ENABLED: false
POSTGRES_USER: ${{ secrets.POSTGRES_USER }}
POSTGRES_PASSWORD: ${{ secrets.POSTGRES_PASSWORD }}
POSTGRES_DBNAME: ${{ secrets.POSTGRES_DBNAME }}
POSTGRES_HOST: ${{ secrets.POSTGRES_HOST }}
POSTGRES_PORT: ${{ secrets.POSTGRES_PORT }}
R2R_PROJECT_NAME: ${{ secrets.R2R_PROJECT_NAME }}
POSTGRES_HOST: localhost
POSTGRES_DBNAME: postgres
POSTGRES_PORT: 5432
POSTGRES_PASSWORD: postgres
POSTGRES_USER: postgres

steps:
- uses: actions/checkout@v4

- name: Set up Python
uses: actions/setup-python@v4
with:
python-version: '3.x'

- name: Install Poetry
run: |
curl -sSL https://install.python-poetry.org | python3 -

- name: Install dependencies
working-directory: ./py
run: |
poetry install -E core -E ingestion-bundle

- name: Start R2R server
working-directory: ./py
run: |
poetry run r2r serve &
echo "Waiting for services to start..."
sleep 30

- name: Run integration tests
working-directory: ./py
run: |
echo "R2R Version"
poetry run r2r version

- name: Walkthrough
working-directory: ./py
run: |
echo "Ingest Data"
poetry run r2r ingest-sample-files

echo "Get Documents Overview"
poetry run r2r documents-overview

echo "Get Document Chunks"
poetry run r2r document-chunks --document-id=9fbe403b-c11c-5aae-8ade-ef22980c3ad1

echo "Delete Documents"
poetry run r2r delete --filter=document_id:eq:9fbe403b-c11c-5aae-8ade-ef22980c3ad1

echo "Update Document"
poetry run r2r update-files core/examples/data/aristotle_v2.txt --document-ids=9fbe403b-c11c-5aae-8ade-ef22980c3ad1

echo "Vector Search"
poetry run r2r search --query="What was Uber's profit in 2020?"

echo "Hybrid Search"
r2r search --query="What was Uber's profit in 2020?" --use-hybrid-search

echo "Basic RAG"
poetry run r2r rag --query="What was Uber's profit in 2020?"

echo "RAG with Hybrid Search"
poetry run r2r rag --query="Who is Jon Snow?" --use-hybrid-search

echo "Streaming RAG"
poetry run r2r rag --query="who was aristotle" --use-hybrid-search --stream

echo "User Registration"
curl -X POST http://localhost:7272/v2/register \
-H "Content-Type: application/json" \
-d '{
"email": "[email protected]",
"password": "password123"
}'

echo "User Login"
curl -X POST http://localhost:7272/v2/login \
-H "Content-Type: application/x-www-form-urlencoded" \
-d "[email protected]&password=password123"

echo "Users Overview"
poetry run r2r users-overview

echo "Logging"
poetry run r2r logs

echo "Analytics"
poetry run r2r analytics --filters '{"search_latencies": "search_latency"}' --analysis-types '{"search_latencies": ["basic_statistics", "search_latency"]}'

- name: GraphRAG
working-directory: ./py
run: |
echo "Create Knowledge Graph"
poetry run r2r create-graph --document-ids=9fbe403b-c11c-5aae-8ade-ef22980c3ad1

echo "Inspect Knowledge Graph"
poetry run r2r inspect-knowledge-graph

echo "Graph Enrichment"
poetry run r2r enrich-graph

echo "Local Search"
r2r search --query="Who is Aristotle?" --use-kg-search --kg-search-type=local

echo "Global Search"
r2r search --query="What were Aristotles key contributions to philosophy?" --use-kg-search --kg-search-type=global --max-llm-queries-for-global-search=100

echo "RAG"
r2r rag --query="What are the key contributions of Aristotle to modern society?" --use-kg-search --kg-search-type=global --max-llm-queries-for-global-search=100






- name: Advanced RAG
working-directory: ./py
run: |
echo "HyDE"
poetry run r2r rag --query="who was aristotle" --use-hybrid-search --stream --search-strategy=hyde

echo "Rag-Fusion"
r2r rag --query="Explain the theory of relativity" --use-hybrid-search --stream --search-strategy=rag_fusion

- name: Stop R2R server
run: |
pkill -f "r2r serve"
- name: Install and configure PostgreSQL
run: |
sudo apt-get update
sudo apt-get install -y postgresql
sudo systemctl start postgresql.service
sudo -u postgres psql -c "ALTER USER postgres PASSWORD 'postgres';"

# Add your other steps here
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The workflow is missing steps to set up Python, install dependencies, and run the integration tests. These steps were present in the original workflow and are necessary for the integration tests to run.


- name: Cleanup PostgreSQL
if: always() # Run this step even if the job fails
run: |
sudo systemctl stop postgresql.service
sudo rm -rf /var/lib/postgresql/