refactor: import fixes

shah-siddd · gustavocidornelas · commit bd78a73b6d69 · 2025-09-10T10:28:18.000-03:00
diff --git a/src/openlayer/lib/tracing/__init__.py b/src/openlayer/lib/tracing/__init__.py
@@ -1,4 +1,4 @@
-"""OpenLayer tracing module."""
+"""Openlayer tracing module."""
 
 from .tracer import (
     trace,
@@ -13,13 +13,6 @@
     create_step,
 )
 
-from .context import (
-    set_user_session_context,
-    update_trace_user_session,
-    get_current_user_id,
-    get_current_session_id,
-    clear_user_session_context,
-)
 
 __all__ = [
     # Core tracing functions
@@ -33,12 +26,5 @@
     "get_current_trace",
     "get_current_step", 
     "create_step",
-    
-    # User and session context functions
-    "set_user_session_context",
-    "update_trace_user_session",
-    "get_current_user_id",
-    "get_current_session_id",
-    "clear_user_session_context",
 ]
 
diff --git a/src/openlayer/lib/tracing/context.py b/src/openlayer/lib/tracing/context.py
@@ -1,5 +1,5 @@
 """
-Streamlined user and session context management for OpenLayer tracing.
+Streamlined user and session context management for Openlayer tracing.
 
 This module provides simple functions to set user_id and session_id in middleware
 and override them anywhere in your traced code.
diff --git a/src/openlayer/types/inference_pipelines/data_stream_params.py b/src/openlayer/types/inference_pipelines/data_stream_params.py
@@ -89,12 +89,6 @@ class ConfigLlmData(TypedDict, total=False):
     used.
     """
 
-    user_id_column_name: Annotated[str, PropertyInfo(alias="userIdColumnName")]
-    """Name of the column with the user IDs."""
-
-    session_id_column_name: Annotated[str, PropertyInfo(alias="sessionIdColumnName")]
-    """Name of the column with the session IDs."""
-
 
 class ConfigTabularClassificationData(TypedDict, total=False):
     class_names: Required[Annotated[SequenceNotStr[str], PropertyInfo(alias="classNames")]]
@@ -235,4 +229,4 @@ class ConfigTextClassificationData(TypedDict, total=False):
 
 Config: TypeAlias = Union[
     ConfigLlmData, ConfigTabularClassificationData, ConfigTabularRegressionData, ConfigTextClassificationData
-]
+]
