From 8ac702957314271b94ae2e777d39221f44a33a20 Mon Sep 17 00:00:00 2001
From: Leo Ribeiro <leordev@gmail.com>
Date: Mon, 25 Nov 2024 13:07:42 -0500
Subject: [PATCH] ci: Lock CI uv version to 0.5.4 and fix file test_session.py
 format (#317)

---
 .github/workflows/ci.yaml | 18 ++++++++++++------
 tests/cli/test_session.py |  5 +++--
 2 files changed, 15 insertions(+), 8 deletions(-)

diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
index 5d0e0aba..28a2a3a7 100644
--- a/.github/workflows/ci.yaml
+++ b/.github/workflows/ci.yaml
@@ -16,8 +16,10 @@ jobs:
     steps:
       - uses: actions/checkout@v4
 
-      - name: Install UV
-        run: curl -LsSf https://astral.sh/uv/install.sh | sh
+      - name: Install uv
+        uses: astral-sh/setup-uv@v3
+        with:
+          version: "0.5.4"
 
       - name: Source Cargo Environment
         run: source $HOME/.cargo/env
@@ -38,8 +40,10 @@ jobs:
     steps:
       - uses: actions/checkout@v4
 
-      - name: Install UV
-        run: curl -LsSf https://astral.sh/uv/install.sh | sh
+      - name: Install uv
+        uses: astral-sh/setup-uv@v3
+        with:
+          version: "0.5.4"
 
       - name: Source Cargo Environment
         run: source $HOME/.cargo/env
@@ -71,8 +75,10 @@ jobs:
     steps:
       - uses: actions/checkout@v4
 
-      - name: Install UV
-        run: curl -LsSf https://astral.sh/uv/install.sh | sh
+      - name: Install uv
+        uses: astral-sh/setup-uv@v3
+        with:
+          version: "0.5.4"
 
       - name: Source Cargo Environment
         run: source $HOME/.cargo/env
diff --git a/tests/cli/test_session.py b/tests/cli/test_session.py
index d6cfb758..573f72a3 100644
--- a/tests/cli/test_session.py
+++ b/tests/cli/test_session.py
@@ -271,8 +271,9 @@ def test_observer_plugin_called(create_session_with_mock_configs):
     observer_manager_mock = MagicMock(spec=ObserverManager)
     observer_manager_mock._observers = [observer_mock]
 
-    with patch("exchange.observers.ObserverManager.get_instance", return_value=observer_manager_mock), patch(
-        "exchange.Exchange.generate", return_value=Message.assistant("test response")
+    with (
+        patch("exchange.observers.ObserverManager.get_instance", return_value=observer_manager_mock),
+        patch("exchange.Exchange.generate", return_value=Message.assistant("test response")),
     ):
         session = create_session_with_mock_configs({"name": SESSION_NAME})