Added pytest to test dependencies in indexer and retriever

OpenNyAI · Aug 5, 2024 · dbfe0e3 · dbfe0e3
1 parent 79676c8
commit dbfe0e3
Show file tree

Hide file tree

Showing 6 changed files with 10 additions and 4 deletions.
diff --git a/.github/workflows/trigger-tests.yml b/.github/workflows/trigger-tests.yml
@@ -46,9 +46,15 @@ jobs:
       - name: Test flow with pytest
         working-directory: ./flow
         run: poetry run pytest
+      - name: Install dependencies indexer
+        working-directory: ./indexer
+        run: poetry install --with test
       - name: Test indexer with pytest
         working-directory: ./indexer
         run: poetry run pytest
+      - name: Install dependencies retriever
+        working-directory: ./retriever
+        run: poetry install --with test
       - name: Test retriever with pytest
         working-directory: ./retriever
         run: poetry run pytest
diff --git a/indexer/poetry.lock b/indexer/poetry.lock
diff --git a/indexer/pyproject.toml b/indexer/pyproject.toml
@@ -32,6 +32,7 @@ lib = {path = "../jb-lib", develop = true}
 [tool.poetry.group.test.dependencies]
 pytest-mock = "*"
 pytest-asyncio = "^0.23.7"
+pytest = "^8.2.2"
 
 [build-system]
 requires = ["poetry-core"]

diff --git a/retriever/main.py b/retriever/main.py
@@ -10,7 +10,6 @@
 from langchain_openai import AzureOpenAIEmbeddings, OpenAIEmbeddings
 from lib.data_models import RAG, Flow
 from lib.kafka_utils import KafkaConsumer, KafkaProducer
-from openai import OpenAI
 from r2r import R2R, VectorSearchSettings
 
 load_dotenv()
@@ -32,7 +31,6 @@
 
 logger = logging.getLogger("retriever")
 
-client = OpenAI()
 db_name = os.getenv("POSTGRES_DATABASE_NAME")
 db_user = os.getenv("POSTGRES_DATABASE_USERNAME")
 db_password = os.getenv("POSTGRES_DATABASE_PASSWORD")

diff --git a/retriever/poetry.lock b/retriever/poetry.lock
diff --git a/retriever/pyproject.toml b/retriever/pyproject.toml
@@ -26,6 +26,7 @@ lib = {path = "../jb-lib", develop = true}
 [tool.poetry.group.test.dependencies]
 pytest-mock = "*"
 pytest-asyncio = "^0.23.7"
+pytest = "^8.2.2"
 
 [build-system]
 requires = ["poetry-core"]