Add CI with ruff and pytest, run format

.github/workflows/ci.yaml

@@ -0,0 +1,71 @@
+name: CI
+
+on:
+  pull_request:
+    branches:
+      - main
+
+jobs:
+  lint:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v2
+
+      - name: Set up Python
+        uses: actions/setup-python@v2
+        with:
+          python-version: "3.10"
+
+      - name: Install project dependencies
+        run: pip install -r requirements-ci.txt
+
+      - name: Run Ruff lint
+        run: ruff check .
+
+  format:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v2
+
+      - name: Set up Python
+        uses: actions/setup-python@v2
+        with:
+          python-version: "3.10"
+
+      - name: Install project dependencies
+        run: pip install -r requirements-ci.txt
+
+      - name: Run Ruff format
+        run: ruff format .
+
+      - name: Check for uncommitted changes (formatting)
+        run: git diff --exit-code
+
+  test:
+    runs-on: ubuntu-latest
+    strategy:
+      fail-fast: false
+      matrix:
+        python_version:
+          - "3.9"
+          - "3.10"
+          - "3.11"
+          - "3.12"
+
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v2
+
+      - name: Set up Python
+        uses: actions/setup-python@v2
+        with:
+          python-version: ${{ matrix.python_version }}
+
+      - name: Install project dependencies
+        run: pip install -r requirements-ci.txt
+
+      - name: Run tests
+        run: pytest
+

README.md

@@ -1,7 +1,7 @@
 
 # rerankers
 
-![Python Versions](https://img.shields.io/badge/Python-3.8_3.9_3.10_3.11-blue)
+![Python Versions](https://img.shields.io/badge/Python-3.9_3.10_3.11_3.12-blue)
 [![Downloads](https://static.pepy.tech/badge/rerankers/month)](https://pepy.tech/project/rerankers)
 [![Twitter Follow](https://img.shields.io/twitter/follow/bclavie?style=social)](https://twitter.com/bclavie)
 

examples/langchain_integration.ipynb

@@ -262,7 +262,6 @@
    ],
    "source": [
     "from langchain.retrievers import ContextualCompressionRetriever\n",
-    "from langchain_openai import OpenAI\n",
     "\n",
     "text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)\n",
     "texts = text_splitter.split_documents(documents)\n",
@@ -274,7 +273,7 @@
     "\n",
     "\n",
     "compressed_docs = compression_retriever.get_relevant_documents(\n",
-    "     \"What did the president say about the minimum wage?\"\n",
+    "    \"What did the president say about the minimum wage?\"\n",
     ")\n",
     "\n",
     "pretty_print_docs(compressed_docs)"