fix test errors

landing-ai · Jan 6, 2025 · 1c0fd60 · 1c0fd60
1 parent a4f8a9e
commit 1c0fd60
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 7 deletions.
diff --git a/vision_agent/tools/tool_utils.py b/vision_agent/tools/tool_utils.py
@@ -13,7 +13,6 @@
 from urllib3.util.retry import Retry
 
 from vision_agent.tools.tools_types import BoundingBoxes
-from vision_agent.utils import should_report_tool_traces
 from vision_agent.utils.exceptions import RemoteToolCallFailed
 from vision_agent.utils.execute import Error, MimeType
 from vision_agent.utils.image_utils import normalize_bbox
@@ -26,6 +25,10 @@
 _LND_API_URL_v2 = f"{_LND_BASE_URL}/v1/tools"
 
 
+def should_report_tool_traces() -> bool:
+    return bool(os.environ.get("REPORT_TOOL_TRACES", False))
+
+
 class ToolCallTrace(BaseModel):
     endpoint_url: str
     type: str

diff --git a/vision_agent/tools/tools.py b/vision_agent/tools/tools.py
@@ -32,10 +32,10 @@
     nms,
     send_inference_request,
     send_task_inference_request,
+    should_report_tool_traces,
     single_nms,
 )
 from vision_agent.tools.tools_types import JobStatus
-from vision_agent.utils import should_report_tool_traces
 from vision_agent.utils.exceptions import FineTuneModelIsNotReady
 from vision_agent.utils.execute import FileSerializer, MimeType
 from vision_agent.utils.image_utils import (

diff --git a/vision_agent/utils/__init__.py b/vision_agent/utils/__init__.py
@@ -1,4 +1,3 @@
-import os
 from .execute import (
     CodeInterpreter,
     CodeInterpreterFactory,
@@ -8,7 +7,3 @@
     Result,
 )
 from .sim import AzureSim, OllamaSim, Sim, load_sim, merge_sim
-
-
-def should_report_tool_traces() -> bool:
-    return bool(os.environ.get("REPORT_TOOL_TRACES", False))