From 0228f392843bc3bbefeed156b8cc9e852e1579ff Mon Sep 17 00:00:00 2001
From: Rafid <rafid@cohere.com>
Date: Wed, 4 Dec 2024 20:29:36 +0000
Subject: [PATCH] Fix some issues in GitHub workflows

---
 .github/workflows/pre-commit-checks.yml | 12 ++++++++++--
 .github/workflows/test.yml              | 10 ++++------
 poetry.lock                             |  2 +-
 3 files changed, 15 insertions(+), 9 deletions(-)

diff --git a/.github/workflows/pre-commit-checks.yml b/.github/workflows/pre-commit-checks.yml
index 17046bc..150ffea 100644
--- a/.github/workflows/pre-commit-checks.yml
+++ b/.github/workflows/pre-commit-checks.yml
@@ -25,10 +25,18 @@ jobs:
         with:
           python-version: ${{ matrix.python-version }}
 
-      - name: Upgrade pip & install requirements
+      - name: Install poetry
+        run: |
+          pip install poetry
+
+      - name: Install dependencies
+        run: |
+          poetry install
+
+      - name: Install pre-commit
         run: |
           pip install pre-commit
 
-      - name: Formatting
+      - name: Run pre-commit
         run: |
           pre-commit run --all-files
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index b8986f2..e83f90b 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -51,17 +51,15 @@ jobs:
           cache-dependency-path: |
             poetry.lock
 
-      - name: Install dependencies (tests)
+      - name: Install poetry
         run: |
-          pip install pytest pytest-asyncio pytest-mock requests-mock
+          pip install poetry
 
       - name: Install dependencies
-        working-directory: .
         run: |
-          pip install -e .
+          poetry install
 
       - name: Run tests
         working-directory: .
         run: |
-          echo $COHERE_API_KEY
-          pytest -sv tests/test_compass_client.py
+          poetry run pytest -sv
diff --git a/poetry.lock b/poetry.lock
index 553c941..ebe7cd0 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -625,4 +625,4 @@ zstd = ["zstandard (>=0.18.0)"]
 [metadata]
 lock-version = "2.0"
 python-versions = ">=3.9,<4.0"
-content-hash = "43bc27b284619efe01aeb37dd0b6a2eac9d927a713c20c23ebfa19a9e3c297bd"
+content-hash = "1e01950e92aed31912006f9f27462f6f5f4810254789d4bef83b851466cb7b33"