Fix CI usage of uv (huggingface#970)

Author: albertvillanova

.github/workflows/quality.yml

@@ -2,6 +2,9 @@ name: Quality Check
 
 on: [pull_request]
 
+env:
+  UV_SYSTEM_PYTHON: 1
+
 jobs:
   check_code_quality:
     runs-on: ubuntu-latest
@@ -16,15 +19,14 @@ jobs:
           python-version: "3.12"
 
       # Setup venv
-      - name: Setup venv + uv
+      - name: Setup uv
         run: |
           pip install --upgrade uv
-          uv venv
 
       - name: Install dependencies
         run: uv pip install "smolagents[quality] @ ."
 
       # Equivalent of "make quality" but step by step
-      - run: uv run ruff check examples src tests utils # linter
-      - run: uv run ruff format --check examples src tests utils # formatter
-      - run: uv run python utils/check_tests_in_ci.py
+      - run: ruff check examples src tests utils # linter
+      - run: ruff format --check examples src tests utils # formatter
+      - run: python utils/check_tests_in_ci.py

.github/workflows/tests.yml

@@ -2,6 +2,9 @@ name: Python tests
 
 on: [pull_request]
 
+env:
+  UV_SYSTEM_PYTHON: 1
+
 jobs:
   build-ubuntu:
     runs-on: ubuntu-latest
@@ -21,10 +24,9 @@ jobs:
           python-version: ${{ matrix.python-version }}
 
       # Setup venv
-      - name: Setup venv + uv
+      - name: Setup uv
         run: |
           pip install --upgrade uv
-          uv venv
 
       # Install dependencies
       - name: Install dependencies
@@ -36,89 +38,89 @@ jobs:
       # See https://stackoverflow.com/a/62112985
       - name: Import tests
         run: |
-          uv run pytest ./tests/test_import.py
+          pytest ./tests/test_import.py
         if: ${{ success() || failure() }}
 
       - name: Agent tests
         run: |
-          uv run pytest ./tests/test_agents.py
+          pytest ./tests/test_agents.py
         if: ${{ success() || failure() }}
 
       - name: Default tools tests
         run: |
-          uv run pytest ./tests/test_default_tools.py
+          pytest ./tests/test_default_tools.py
         if: ${{ success() || failure() }}
 
       # - name: Docs tests # Disabled for now (slow test + requires API keys)
       #   run: |
-      #     uv run pytest ./tests/test_all_docs.py
+      #     pytest ./tests/test_all_docs.py
 
       - name: CLI tests
         run: |
-          uv run pytest ./tests/test_cli.py
+          pytest ./tests/test_cli.py
         if: ${{ success() || failure() }}
 
       - name: Final answer tests
         run: |
-          uv run pytest ./tests/test_final_answer.py
+          pytest ./tests/test_final_answer.py
         if: ${{ success() || failure() }}
 
       - name: Models tests
         run: |
-          uv run pytest ./tests/test_models.py
+          pytest ./tests/test_models.py
         if: ${{ success() || failure() }}
 
       - name: Memory tests
         run: |
-          uv run pytest ./tests/test_memory.py
+          pytest ./tests/test_memory.py
         if: ${{ success() || failure() }}
 
       - name: Monitoring tests
         run: |
-          uv run pytest ./tests/test_monitoring.py
+          pytest ./tests/test_monitoring.py
         if: ${{ success() || failure() }}
 
       - name: Local Python executor tests
         run: |
-          uv run pytest ./tests/test_local_python_executor.py
+          pytest ./tests/test_local_python_executor.py
         if: ${{ success() || failure() }}
 
       - name: Remote executor tests
         run: |
-          uv run pytest ./tests/test_remote_executors.py
+          pytest ./tests/test_remote_executors.py
         if: ${{ success() || failure() }}
 
       - name: Search tests
         run: |
-          uv run pytest ./tests/test_search.py
+          pytest ./tests/test_search.py
         if: ${{ success() || failure() }}
 
       - name: Tools tests
         run: |
-          uv run pytest ./tests/test_tools.py
+          pytest ./tests/test_tools.py
         if: ${{ success() || failure() }}
 
       - name: Tool validation tests
         run: |
-          uv run pytest ./tests/test_tool_validation.py
+          pytest ./tests/test_tool_validation.py
         if: ${{ success() || failure() }}
 
       - name: Types tests
         run: |
-          uv run pytest ./tests/test_types.py
+          pytest ./tests/test_types.py
         if: ${{ success() || failure() }}
 
       - name: Utils tests
         run: |
-          uv run pytest ./tests/test_utils.py
+          pytest ./tests/test_utils.py
         if: ${{ success() || failure() }}
 
       - name: Gradio UI tests
         run: |
-          uv run pytest ./tests/test_gradio_ui.py
+          pytest ./tests/test_gradio_ui.py
         if: ${{ success() || failure() }}
 
       - name: Function type hints utils tests
         run: |
-          uv run pytest ./tests/test_function_type_hints_utils.py
+          pytest ./tests/test_function_type_hints_utils.py
         if: ${{ success() || failure() }}

tests/test_import.py

@@ -1,11 +1,27 @@
+import os
 import subprocess
+import tempfile
 
 
-def test_import_smolagents_without_extras():
-    # Run the import statement in an isolated virtual environment
-    result = subprocess.run(
-        ["uv", "run", "--isolated", "--no-editable", "-"], input="import smolagents", text=True, capture_output=True
-    )
+def test_import_smolagents_without_extras(monkeypatch):
+    monkeypatch.delenv("VIRTUAL_ENV", raising=False)
+    with tempfile.TemporaryDirectory() as temp_dir:
+        # Create a virtual environment
+        venv_dir = os.path.join(temp_dir, "venv")
+        subprocess.run(["uv", "venv", venv_dir], check=True)
+
+        # Install smolagents in the virtual environment
+        subprocess.run(
+            ["uv", "pip", "install", "--python", os.path.join(venv_dir, "bin", "python"), "smolagents @ ."], check=True
+        )
+
+        # Run the import test in the virtual environment
+        result = subprocess.run(
+            [os.path.join(venv_dir, "bin", "python"), "-c", "import smolagents"],
+            capture_output=True,
+            text=True,
+        )
+
     # Check if the import was successful
     assert result.returncode == 0, (
         "Import failed with error: "

tests/test_tools.py

@@ -30,6 +30,8 @@
 from smolagents.agent_types import _AGENT_TYPE_MAPPING, AgentAudio, AgentImage, AgentText
 from smolagents.tools import AUTHORIZED_TYPES, Tool, ToolCollection, tool
 
+from .utils.markers import require_run_all
+
 
 if is_torch_available():
     import torch
@@ -506,6 +508,7 @@ def test_from_mcp(self, mock_server_parameters, mock_mcp_adapt, mock_smolagents_
             assert "tool1" in tool_collection.tools
             assert "tool2" in tool_collection.tools
 
+    @require_run_all
     def test_integration_from_mcp(self):
         # define the most simple mcp server with one tool that echoes the input text
         mcp_server_script = dedent("""\