langwatch 0.2.17__tar.gz → 0.2.19__tar.gz

{langwatch-0.2.17 → langwatch-0.2.19}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langwatch
-Version: 0.2.17
+Version: 0.2.19
 Summary: LangWatch Python SDK, for monitoring your LLMs
 Author-email: Langwatch Engineers <engineering@langwatch.ai>
 License: MIT

{langwatch-0.2.17 → langwatch-0.2.19}/examples/strands_bot.py

@@ -1,5 +1,5 @@
 import os
-from strands import Agent
+from strands import Agent, tool
 from strands.models.litellm import LiteLLMModel
 import langwatch
 
@@ -31,6 +31,16 @@ langwatch.setup()  # The api key is set from the environment variable automatica
 # langwatch.setup(skip_open_telemetry_setup=True)
 
 
+
+@tool
+@langwatch.span(type="tool")
+def get_user_location() -> str:
+    """Get the user's location."""
+
+    # Implement user location lookup logic here
+    return "London, UK"
+
+
 class KiteAgent:
     def __init__(self):
         self.model = LiteLLMModel(
@@ -42,7 +52,8 @@ class KiteAgent:
         self.agent = Agent(
             name="kite-agent",
             model=self.model,
-            tools=[],
+            system_prompt="Always use the get_user_location tool before answering any questions.",
+            tools=[get_user_location],
             trace_attributes={
                 "custom.model_id": "openai/gpt-5-mini",
                 "custom.example.attribute": "swift",
@@ -62,7 +73,7 @@ async def main(message: cl.Message):
 
     langwatch.get_current_trace().update(
         metadata={
-            "custom.example.attribute": "swift",
+            "custom.example.attribute2": "langwatch",
         }
     )
 

{langwatch-0.2.17 → langwatch-0.2.19}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "langwatch"
-version = "0.2.17" # remember to also update it in src/langwatch/__version__.py
+version = "0.2.19" # remember to also update it in src/langwatch/__version__.py
 description = "LangWatch Python SDK, for monitoring your LLMs"
 authors = [{ name = "Langwatch Engineers", email = "engineering@langwatch.ai" }]
 requires-python = ">=3.10,<3.14"

{langwatch-0.2.17 → langwatch-0.2.19}/src/langwatch/__version__.py

@@ -1,3 +1,3 @@
 """Version information for LangWatch."""
 
-__version__ = "0.2.17"
+__version__ = "0.2.19"

{langwatch-0.2.17 → langwatch-0.2.19}/src/langwatch/telemetry/context.py

@@ -1,8 +1,9 @@
 import threading
-from typing import TYPE_CHECKING, List
+from typing import TYPE_CHECKING, Dict, List
 import contextvars
 import warnings
 from opentelemetry import trace as trace_api
+from opentelemetry import context
 
 from langwatch.utils.initialization import ensure_setup
 
@@ -44,7 +45,19 @@ def get_current_trace(
 
     trace = LangWatchTrace()
     if start_if_none:
-        return trace.__enter__()
+        otel_span = trace_api.get_current_span()
+        otel_span_id = otel_span.get_span_context().span_id
+        trace.__enter__()
+
+        # Keep the previous span in the context if we are starting a new trace not at root level
+        if otel_span_id != 0:
+            ctx = trace_api.set_span_in_context(otel_span)
+            context.attach(ctx)
+        otel_span = trace_api.get_current_span()
+
+        trace.__exit__(None, None, None)
+
+        return trace
     return trace
 
 
@@ -57,20 +70,20 @@ def get_current_span() -> "LangWatchSpan":
     """
     ensure_setup()
 
-    # First try getting from LangWatch context
-    span = stored_langwatch_span.get(None)
-    if span is not None:
-        return span
-
-    if _is_on_child_thread() and len(main_thread_langwatch_span) > 0:
-        return main_thread_langwatch_span[-1]
-
-    # Fall back to OpenTelemetry context
     otel_span = trace_api.get_current_span()
-    trace = get_current_trace()
+    otel_span_id = otel_span.get_span_context().span_id
+
+    # If on a child thread and there is no parent, try to find a parent from the main thread
+    if (
+        _is_on_child_thread()
+        and len(main_thread_langwatch_span) > 0
+        and otel_span_id == 0
+    ):
+        return main_thread_langwatch_span[-1]
 
     from langwatch.telemetry.span import LangWatchSpan
 
+    trace = get_current_trace()
     return LangWatchSpan.wrap_otel_span(otel_span, trace)
 
 
@@ -91,11 +104,8 @@ def _set_current_span(span: "LangWatchSpan"):
     if not _is_on_child_thread():
         main_thread_langwatch_span.append(span)
 
-    try:
-        return stored_langwatch_span.set(span)
-    except Exception as e:
-        warnings.warn(f"Failed to set LangWatch span context: {e}")
-        return None
+    # Dummy token, just for the main thread span list
+    return "token"
 
 
 def _reset_current_trace(token: contextvars.Token):
@@ -112,19 +122,12 @@ def _reset_current_trace(token: contextvars.Token):
             warnings.warn(f"Failed to reset LangWatch trace context: {e}")
 
 
-def _reset_current_span(token: contextvars.Token):
+def _reset_current_span(_token: str):
     global main_thread_langwatch_span
     if not _is_on_child_thread():
         if len(main_thread_langwatch_span) > 0:
             main_thread_langwatch_span.pop()
 
-    try:
-        stored_langwatch_span.reset(token)
-    except Exception as e:
-        # Only warn if it's not a context error
-        if "different Context" not in str(e):
-            warnings.warn(f"Failed to reset LangWatch span context: {e}")
-
 
 def _is_on_child_thread() -> bool:
     return threading.current_thread() != threading.main_thread()

{langwatch-0.2.17 → langwatch-0.2.19}/src/langwatch/utils/initialization.py

@@ -1,6 +1,7 @@
 """Utility module for handling LangWatch initialization."""
 
 import logging
+import os
 import sys
 from typing import List, Optional, Sequence
 from opentelemetry.sdk.trace import TracerProvider
@@ -102,13 +103,18 @@ def ensure_setup(api_key: Optional[str] = None) -> None:
     If no client is setup, this will create a default client using environment variables.
     Validates that we have a working tracer provider to prevent silent failures.
     """
+
+    # We want to skip auto-setup if langwatch api key is not available to avoid throwing errors
+    if not os.getenv("LANGWATCH_API_KEY", api_key):
+        return
+
     client = get_instance()
     if client is None:
         logger.debug("No LangWatch client found, creating default client")
         client = setup(
-            debug=True,
+            debug=True,  # Enable debug logging for auto-created clients
             api_key=api_key,
-        )  # Enable debug logging for auto-created clients
+        )
 
     # Verify we have a valid tracer provider
     tracer_provider = trace.get_tracer_provider()