0.4.4

agentops/__init__.py

@@ -1,12 +1,29 @@
 from typing import Any, Dict, List, Optional, Union
 
-from agentops.legacy import ErrorEvent, ToolEvent, end_session, start_session
+from agentops.legacy import ActionEvent, ErrorEvent, ToolEvent, start_session, end_session
 
 from .client import Client
 
 # Client global instance; one per process runtime
 _client = Client()
 
+def record(event):
+    """
+    Legacy function to record an event. This is kept for backward compatibility.
+
+    In the current version, this simply sets the end_timestamp on the event.
+
+    Args:
+        event: The event to record
+    """
+    from agentops.helpers.time import get_ISO_time
+
+    # TODO: Manual timestamp assignment is a temporary fix; should use proper event lifecycle
+    if event and hasattr(event, 'end_timestamp'):
+        event.end_timestamp = get_ISO_time()
+
+    return event
+
 
 def init(
     api_key: Optional[str] = None,
@@ -139,6 +156,9 @@ def get_client() -> Client:
     "init",
     "configure",
     "get_client",
+    "record",
     "start_session",
     "end_session",
+    "track_agent",
+    "track_tool",
 ]

agentops/client/client.py

@@ -29,6 +29,8 @@
         self.config = Config()
 
     def init(self, **kwargs):
+        # Recreate the Config object to parse environment variables at the time of initialization
+        self.config = Config()
         self.configure(**kwargs)
 
         if not self.config.api_key:
@@ -56,10 +58,18 @@
 
         self.initialized = True
 
+        # Start a session if auto_start_session is True
+        session = None
         if self.config.auto_start_session:
             from agentops.legacy import start_session
 
-            start_session()
+            # Pass default_tags if they exist
+            if self.config.default_tags:
+                session = start_session(tags=list(self.config.default_tags))
+            else:
+                session = start_session()
+
+        return session
 
     def configure(self, **kwargs):
         """Update client configuration"""

agentops/config.py

@@ -20,6 +20,7 @@
     api_key: Optional[str]
     endpoint: Optional[str]
     max_wait_time: Optional[int]
+    export_flush_interval: Optional[int]
     max_queue_size: Optional[int]
     default_tags: Optional[List[str]]
     instrument_llm_calls: Optional[bool]
@@ -32,7 +33,7 @@
     prefetch_jwt_token: Optional[bool]
 
 
-@dataclass(slots=True)
+@dataclass
 class Config:
     api_key: Optional[str] = field(
         default_factory=lambda: os.getenv("AGENTOPS_API_KEY"),
@@ -48,6 +49,11 @@
         default_factory=lambda: get_env_int("AGENTOPS_MAX_WAIT_TIME", 5000),
         metadata={"description": "Maximum time in milliseconds to wait for API responses"},
     )
+
+    export_flush_interval: int = field(
+        default_factory=lambda: get_env_int("AGENTOPS_EXPORT_FLUSH_INTERVAL", 1000),
+        metadata={"description": "Time interval in milliseconds between automatic exports of telemetry data"},
+    )
 
     max_queue_size: int = field(
         default_factory=lambda: get_env_int("AGENTOPS_MAX_QUEUE_SIZE", 512),
@@ -65,7 +71,7 @@
     )
 
     auto_start_session: bool = field(
-        default_factory=lambda: get_env_bool("AGENTOPS_AUTO_START_SESSION", False),
+        default_factory=lambda: get_env_bool("AGENTOPS_AUTO_START_SESSION", True),
         metadata={"description": "Whether to automatically start a session when initializing"},
     )
 
@@ -85,7 +91,7 @@
     )
 
     log_level: Union[str, int] = field(
-        default_factory=lambda: os.getenv("AGENTOPS_LOG_LEVEL", "WARNING"),
+        default_factory=lambda: os.getenv("AGENTOPS_LOG_LEVEL", "INFO"),
         metadata={"description": "Logging level for AgentOps logs"},
     )
 
@@ -119,6 +125,7 @@
         api_key: Optional[str] = None,
         endpoint: Optional[str] = None,
         max_wait_time: Optional[int] = None,
+        export_flush_interval: Optional[int] = None,
         max_queue_size: Optional[int] = None,
         default_tags: Optional[List[str]] = None,
         instrument_llm_calls: Optional[bool] = None,
@@ -147,6 +154,9 @@
 
         if max_wait_time is not None:
             self.max_wait_time = max_wait_time
+
+        if export_flush_interval is not None:
+            self.export_flush_interval = export_flush_interval
 
         if max_queue_size is not None:
             self.max_queue_size = max_queue_size
@@ -170,7 +180,14 @@
             self.env_data_opt_out = env_data_opt_out
 
         if log_level is not None:
-            self.log_level = log_level
+            if isinstance(log_level, str):
+                log_level_str = log_level.upper()
+                if hasattr(logging, log_level_str):
+                    self.log_level = getattr(logging, log_level_str)
+                else:
+                    self.log_level = logging.INFO
+            else:
+                self.log_level = log_level
 
         if fail_safe is not None:
             self.fail_safe = fail_safe
@@ -195,6 +212,7 @@
             "api_key": self.api_key,
             "endpoint": self.endpoint,
             "max_wait_time": self.max_wait_time,
+            "export_flush_interval": self.export_flush_interval,
             "max_queue_size": self.max_queue_size,
             "default_tags": self.default_tags,
             "instrument_llm_calls": self.instrument_llm_calls,

agentops/instrumentation/__init__.py

@@ -3,7 +3,7 @@
 from dataclasses import dataclass
 import importlib
 
-from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
+from opentelemetry.instrumentation.instrumentor import BaseInstrumentor  # type: ignore
 
 from agentops.logging import logger
 from agentops.sdk.core import TracingCore