fixes

Author: Dwij1704

agentops/__init__.py

@@ -1,3 +1,12 @@
+# For backwards compatibility
+from agentops.legacy import (
+    start_session,
+    end_session,
+    track_agent,
+    track_tool,
+    end_all_sessions,
+)  # type: ignore
+
 from typing import List, Optional, Union
 from agentops.client import Client
 
@@ -158,10 +167,6 @@ def configure(**kwargs):
     _client.configure(**kwargs)
 
 
-# For backwards compatibility
-
-from agentops.legacy import *  # type: ignore
-
 __all__ = [
     "init",
     "configure",
@@ -171,4 +176,5 @@ def configure(**kwargs):
     "end_session",
     "track_agent",
     "track_tool",
+    "end_all_sessions",
 ]

agentops/client/api/__init__.py

@@ -4,7 +4,7 @@
 This module provides the client for the AgentOps API.
 """
 
-from typing import Dict, Optional, Type, TypeVar, cast
+from typing import Dict, Type, TypeVar, cast
 
 from agentops.client.api.base import BaseApiClient
 from agentops.client.api.types import AuthTokenResponse

agentops/client/api/base.py

@@ -8,14 +8,14 @@
 
 import requests
 
-from agentops.client.api.types import AuthTokenResponse
 from agentops.client.http.http_client import HttpClient
 
 
 class TokenFetcher(Protocol):
     """Protocol for token fetching functions"""
 
-    def __call__(self, api_key: str) -> str: ...
+    def __call__(self, api_key: str) -> str:
+        ...
 
 
 class BaseApiClient:

agentops/client/api/versions/v3.py

@@ -4,10 +4,6 @@
 This module provides the client for the V3 version of the AgentOps API.
 """
 
-from typing import Any, Dict, List, Optional
-
-import requests
-
 from agentops.client.api.base import BaseApiClient
 from agentops.client.api.types import AuthTokenResponse
 from agentops.exceptions import ApiServerException

agentops/client/client.py

@@ -1,9 +1,8 @@
-from typing import List, Optional, Union
 import atexit
 
 from agentops.client.api import ApiClient
 from agentops.config import Config
-from agentops.exceptions import AgentOpsClientNotInitializedException, NoApiKeyException, NoSessionException
+from agentops.exceptions import NoApiKeyException
 from agentops.instrumentation import instrument_all
 from agentops.logging import logger
 from agentops.logging.config import configure_logging, intercept_opentelemetry_logging

agentops/client/http/http_adapter.py

@@ -1,12 +1,9 @@
-from typing import Callable, Dict, Optional, Union
+from typing import Optional
 
 from requests.adapters import HTTPAdapter
 from urllib3.util import Retry
 
 # from agentops.client.auth_manager import AuthManager
-from agentops.exceptions import AgentOpsApiJwtExpiredException, ApiServerException
-from agentops.logging import logger
-from agentops.client.api.types import AuthTokenResponse
 
 
 class BaseHTTPAdapter(HTTPAdapter):

agentops/client/http/http_client.py

@@ -1,11 +1,9 @@
-from typing import Callable, Dict, Optional
+from typing import Dict, Optional
 
 import requests
 
 from agentops.client.http.http_adapter import BaseHTTPAdapter
-from agentops.exceptions import AgentOpsApiJwtExpiredException, ApiServerException
 from agentops.logging import logger
-from agentops.semconv import ResourceAttributes
 
 
 class HttpClient:

agentops/config.py

@@ -2,8 +2,8 @@
 import logging
 import os
 import sys
-from dataclasses import asdict, dataclass, field
-from typing import Any, List, Optional, Set, TypedDict, Union
+from dataclasses import dataclass, field
+from typing import List, Optional, Set, TypedDict, Union
 from uuid import UUID
 
 from opentelemetry.sdk.trace import SpanProcessor
@@ -13,8 +13,6 @@
 from agentops.helpers.env import get_env_bool, get_env_int, get_env_list
 from agentops.helpers.serialization import AgentOpsJSONEncoder
 
-from .logging.config import logger
-
 
 class ConfigDict(TypedDict):
     api_key: Optional[str]

agentops/exceptions.py

@@ -1,6 +1,3 @@
-from agentops.logging import logger
-
-
 class MultiSessionException(Exception):
     def __init__(self, message):
         super().__init__(message)

agentops/instrumentation/__init__.py

@@ -1,4 +1,4 @@
-from typing import Any, Optional
+from typing import Optional
 from types import ModuleType
 from dataclasses import dataclass
 import importlib