Revert "feat: add session ID support through HTTP headers"

src/langtrace_python_sdk/constants/exporter/langtrace_exporter.py

@@ -1,2 +1 @@
-LANGTRACE_REMOTE_URL = "https://app.langtrace.ai"
-LANGTRACE_SESSION_ID_HEADER = "x-langtrace-session-id"
+LANGTRACE_REMOTE_URL = "https://app.langtrace.ai"

src/langtrace_python_sdk/extensions/langtrace_exporter.py

@@ -9,7 +9,6 @@
 
 from langtrace_python_sdk.constants.exporter.langtrace_exporter import (
     LANGTRACE_REMOTE_URL,
-    LANGTRACE_SESSION_ID_HEADER,
 )
 from colorama import Fore
 from requests.exceptions import RequestException
@@ -52,14 +51,12 @@ class LangTraceExporter(SpanExporter):
     api_key: str
     api_host: str
     disable_logging: bool
-    session_id: str
 
     def __init__(
         self,
         api_host,
         api_key: str = None,
         disable_logging: bool = False,
-        session_id: str = None,
     ) -> None:
         self.api_key = api_key or os.environ.get("LANGTRACE_API_KEY")
         self.api_host = (
@@ -68,7 +65,6 @@ def __init__(
             else api_host
         )
         self.disable_logging = disable_logging
-        self.session_id = session_id or os.environ.get("LANGTRACE_SESSION_ID")
 
     def export(self, spans: typing.Sequence[ReadableSpan]) -> SpanExportResult:
         """
@@ -86,10 +82,6 @@ def export(self, spans: typing.Sequence[ReadableSpan]) -> SpanExportResult:
             "User-Agent": "LangtraceExporter",
         }
 
-        # Add session ID if available
-        if self.session_id:
-            headers[LANGTRACE_SESSION_ID_HEADER] = self.session_id
-
         # Check if the OTEL_EXPORTER_OTLP_HEADERS environment variable is set
         otel_headers = os.getenv("OTEL_EXPORTER_OTLP_HEADERS", None)
         if otel_headers:

src/langtrace_python_sdk/langtrace.py

@@ -39,7 +39,6 @@
 )
 from langtrace_python_sdk.constants.exporter.langtrace_exporter import (
     LANGTRACE_REMOTE_URL,
-    LANGTRACE_SESSION_ID_HEADER,
 )
 from langtrace_python_sdk.instrumentation import (
     AnthropicInstrumentation,
@@ -99,7 +98,6 @@ def __init__(self, **kwargs):
             or os.environ.get("LANGTRACE_HEADERS")
             or os.environ.get("OTEL_EXPORTER_OTLP_HEADERS")
         )
-        self.session_id = kwargs.get("session_id") or os.environ.get("LANGTRACE_SESSION_ID")
 
 
 def get_host(config: LangtraceConfig) -> str:
@@ -136,19 +134,15 @@ def setup_tracer_provider(config: LangtraceConfig, host: str) -> TracerProvider:
 
 
 def get_headers(config: LangtraceConfig):
-    headers = {
-        "x-api-key": config.api_key,
-    }
-
-    if config.session_id:
-        headers[LANGTRACE_SESSION_ID_HEADER] = config.session_id
+    if not config.headers:
+        return {
+            "x-api-key": config.api_key,
+        }
 
     if isinstance(config.headers, str):
-        headers.update(parse_env_headers(config.headers, liberal=True))
-    elif config.headers:
-        headers.update(config.headers)
+        return parse_env_headers(config.headers, liberal=True)
 
-    return headers
+    return config.headers
 
 
 def get_exporter(config: LangtraceConfig, host: str):
@@ -221,7 +215,6 @@ def init(
     service_name: Optional[str] = None,
     disable_logging: bool = False,
     headers: Dict[str, str] = {},
-    session_id: Optional[str] = None,
 ):
 
     check_if_sdk_is_outdated()
@@ -236,7 +229,6 @@ def init(
         service_name=service_name,
         disable_logging=disable_logging,
         headers=headers,
-        session_id=session_id,
     )
 
     if config.disable_logging:

src/langtrace_python_sdk/utils/with_root_span.py

@@ -61,11 +61,6 @@ def sync_wrapper(*args, **kwargs):
                 span_id = str(span.get_span_context().span_id)
                 trace_id = str(span.get_span_context().trace_id)
 
-                # Attach session ID if available
-                session_id = os.environ.get("LANGTRACE_SESSION_ID")
-                if session_id:
-                    span.set_attribute("session.id", session_id)
-
                 if (
                     "span_id" in func.__code__.co_varnames
                     and "trace_id" in func.__code__.co_varnames
@@ -87,12 +82,6 @@ async def async_wrapper(*args, **kwargs):
             ) as span:
                 span_id = span.get_span_context().span_id
                 trace_id = span.get_span_context().trace_id
-
-                # Attach session ID if available
-                session_id = os.environ.get("LANGTRACE_SESSION_ID")
-                if session_id:
-                    span.set_attribute("session.id", session_id)
-
                 if (
                     "span_id" in func.__code__.co_varnames
                     and "trace_id" in func.__code__.co_varnames