fixing uv lock

Author: Roxanne Farhad

src/agentex/lib/cli/commands/agents.py

@@ -26,9 +26,8 @@
 from agentex.lib.cli.handlers.agent_handlers import (
     run_agent,
     build_agent,
-    prepare_cloud_build_context,
     parse_build_args,
-    CloudBuildContext,
+    prepare_cloud_build_context,
 )
 from agentex.lib.cli.handlers.deploy_handlers import (
     HelmError,

src/agentex/lib/cli/handlers/agent_handlers.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
-from pathlib import Path
 from typing import NamedTuple
+from pathlib import Path
 
 from rich.console import Console
 from python_on_whales import DockerException, docker

src/agentex/lib/sdk/config/deployment_config.py

@@ -2,7 +2,7 @@
 
 from typing import Any, Dict
 
-from pydantic import ConfigDict, Field
+from pydantic import Field
 
 from agentex.lib.utils.model_utils import BaseModel
 

tests/lib/cli/test_agent_handlers.py

@@ -6,17 +6,17 @@
 import tarfile
 import tempfile
 from pathlib import Path
+from collections.abc import Iterator
 
 import pytest
 from typer.testing import CliRunner
 
+from agentex.lib.cli.commands.agents import agents
 from agentex.lib.cli.handlers.agent_handlers import (
     CloudBuildContext,
     parse_build_args,
     prepare_cloud_build_context,
 )
-from agentex.lib.cli.commands.agents import agents
-
 
 runner = CliRunner()
 
@@ -54,7 +54,7 @@ class TestPrepareCloudBuildContext:
     """Tests for prepare_cloud_build_context function."""
 
     @pytest.fixture
-    def temp_agent_dir(self) -> Path:
+    def temp_agent_dir(self) -> Iterator[Path]:
         """Create a temporary agent directory with minimal required files."""
         with tempfile.TemporaryDirectory() as tmpdir:
             agent_dir = Path(tmpdir)
@@ -96,7 +96,7 @@ def temp_agent_dir(self) -> Path:
             yield agent_dir
 
     @pytest.fixture
-    def temp_agent_dir_no_deployment(self) -> Path:
+    def temp_agent_dir_no_deployment(self) -> Iterator[Path]:
         """Create a temporary agent directory without deployment config."""
         with tempfile.TemporaryDirectory() as tmpdir:
             agent_dir = Path(tmpdir)
@@ -260,7 +260,7 @@ class TestPackageCommand:
     """Tests for the 'agentex agents package' CLI command."""
 
     @pytest.fixture
-    def temp_agent_dir(self) -> Path:
+    def temp_agent_dir(self) -> Iterator[Path]:
         """Create a temporary agent directory with minimal required files."""
         with tempfile.TemporaryDirectory() as tmpdir:
             agent_dir = Path(tmpdir)