remove unused vars

Author: bboynton97

tests/unit/client/api/versions/test_v4.py

@@ -1,10 +1,9 @@
 import pytest
-from unittest.mock import Mock, patch, MagicMock
+from unittest.mock import Mock, patch
 from requests.models import Response
 
 from agentops.client.api.versions.v4 import V4Client
 from agentops.exceptions import ApiServerException
-from agentops.client.api.types import UploadedObjectResponse
 
 
 class TestV4Client:

tests/unit/helpers/test_version.py

@@ -1,5 +1,4 @@
-import pytest
-from unittest.mock import Mock, patch, MagicMock
+from unittest.mock import Mock, patch
 from importlib.metadata import PackageNotFoundError
 
 from agentops.helpers.version import get_agentops_version, check_agentops_update

tests/unit/instrumentation/common/test_streaming.py

@@ -1,8 +1,5 @@
 import pytest
-from unittest.mock import Mock, patch, MagicMock
-import time
-import asyncio
-from typing import AsyncGenerator
+from unittest.mock import Mock, patch
 from types import SimpleNamespace
 
 from agentops.instrumentation.common.streaming import (

tests/unit/instrumentation/openai_core/test_instrumentor.py

@@ -107,9 +107,9 @@ def test_instrument_method_wraps_response_api(self, instrumentor):
             instrumentor_obj._custom_wrap()
 
             # Verify wrap_function_wrapper was called for Response API methods
-            assert mock_wfw.call_count >= 2, (
-                f"Expected at least 2 calls to wrap_function_wrapper, got {mock_wfw.call_count}"
-            )
+            assert (
+                mock_wfw.call_count >= 2
+            ), f"Expected at least 2 calls to wrap_function_wrapper, got {mock_wfw.call_count}"
 
             # Find Response API calls
             response_api_calls = []

tests/unit/sdk/instrumentation_tester.py

@@ -45,7 +45,8 @@ def reset_trace_globals():
 
 class HasAttributesViaProperty(Protocol):
     @property
-    def attributes(self) -> Attributes: ...
+    def attributes(self) -> Attributes:
+        ...
 
 
 class HasAttributesViaAttr(Protocol):

tests/unit/sdk/test_attributes.py

@@ -4,13 +4,10 @@
 This module tests all attribute management functions for telemetry contexts.
 """
 
-import os
 import platform
-from unittest.mock import Mock, patch, MagicMock
-from typing import Any
+from unittest.mock import Mock, patch
 
 import pytest
-import psutil
 
 from agentops.sdk.attributes import (
     get_system_resource_attributes,

tests/unit/sdk/test_exporters.py

@@ -3,8 +3,7 @@
 """
 
 import unittest
-from unittest.mock import Mock, patch, MagicMock
-from typing import List
+from unittest.mock import Mock, patch
 
 import requests
 from opentelemetry.sdk.trace import ReadableSpan

tests/unit/sdk/test_factory.py

@@ -1,4 +1,3 @@
-from typing import AsyncGenerator
 import asyncio
 import inspect
 import pytest
@@ -642,7 +641,7 @@ def test_decorator_with_builtin_function(self, instrumentation: InstrumentationT
         assert result == 5
 
         # Built-in functions may not be instrumented the same way
-        spans = instrumentation.get_finished_spans()
+        _ = instrumentation.get_finished_spans()
         # The behavior may vary depending on the implementation
 
     def test_decorator_with_coroutine_function(self, instrumentation: InstrumentationTester):

tests/unit/test_init_py.py

@@ -344,7 +344,7 @@ def test_start_trace_auto_init_failure():
 def test_start_trace_auto_init_still_not_initialized():
     with (
         patch("agentops.tracer") as mock_tracer,
-        patch("agentops.init") as mock_init,
+        patch("agentops.init") as _,
         patch("agentops.logger") as mock_logger,
     ):
         mock_tracer.initialized = False
@@ -372,35 +372,6 @@ def test_all_exports_importable():
     from agentops import (
         init,
         configure,
-        get_client,
-        record,
-        start_trace,
-        end_trace,
-        update_trace_metadata,
-        start_session,
-        end_session,
-        track_agent,
-        track_tool,
-        end_all_sessions,
-        ToolEvent,
-        ErrorEvent,
-        ActionEvent,
-        LLMEvent,
-        Session,
-        trace,
-        session,
-        agent,
-        task,
-        workflow,
-        operation,
-        guardrail,
-        tracer,
-        tool,
-        TraceState,
-        SUCCESS,
-        ERROR,
-        UNSET,
-        StatusCode,
     )
 
     assert callable(init)