aiqa-client 0.1.4__tar.gz → 0.1.6__tar.gz

{aiqa_client-0.1.4/aiqa_client.egg-info → aiqa_client-0.1.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aiqa-client
-Version: 0.1.4
+Version: 0.1.6
 Summary: OpenTelemetry-based Python client for tracing functions and sending traces to the AIQA server
 Author-email: AIQA <info@aiqa.dev>
 License: MIT

{aiqa_client-0.1.4 → aiqa_client-0.1.6}/aiqa/__init__.py

@@ -13,7 +13,7 @@ from .tracing import (
     exporter,
 )
 
-__version__ = "0.1.4"
+__version__ = "0.1.6"
 
 __all__ = [
     "WithTracing",

{aiqa_client-0.1.4 → aiqa_client-0.1.6}/aiqa/tracing.py

@@ -12,6 +12,7 @@ from functools import wraps
 from opentelemetry import trace
 from opentelemetry.sdk.trace import TracerProvider
 from opentelemetry.sdk.trace.export import BatchSpanProcessor
+from opentelemetry.sdk.trace.sampling import ALWAYS_ON
 from opentelemetry.sdk.resources import Resource
 from opentelemetry.semconv.resource import ResourceAttributes
 from opentelemetry.trace import Status, StatusCode
@@ -24,12 +25,14 @@ logger = logging.getLogger(__name__)
 exporter = AIQASpanExporter()
 
 # Initialize OpenTelemetry
+# Use ALWAYS_ON sampler to ensure all spans are recorded (important for FastAPI async contexts)
 provider = TracerProvider(
     resource=Resource.create(
         {
             ResourceAttributes.SERVICE_NAME: os.getenv("OTEL_SERVICE_NAME", "aiqa-service"),
         }
-    )
+    ),
+    sampler=ALWAYS_ON
 )
 
 provider.add_span_processor(BatchSpanProcessor(exporter))
@@ -185,11 +188,13 @@ def WithTracing(
                             del input_data[key]
                 
                 # Use start_as_current_span to ensure span is recorded by BatchSpanProcessor
-                # This automatically manages context and ends the span when exiting
                 with tracer.start_as_current_span(fn_name) as span:
-                    # Ensure span is recording
+                    # Check if span is recording - if not, spans won't be exported
                     if not span.is_recording():
-                        logger.warning(f"Span {fn_name} is not recording - tracing may not work")
+                        logger.warning(f"Span {fn_name} is not recording - will not be exported")
+                        return await fn(*args, **kwargs)
+                    
+                    logger.debug(f"Span {fn_name} is recording, trace_id={format(span.get_span_context().trace_id, '032x')}")
                     
                     if input_data is not None:
                         # Serialize for span attributes (OpenTelemetry only accepts primitives or JSON strings)
@@ -217,12 +222,16 @@ def WithTracing(
                         span.set_attribute("output", _serialize_for_span(output_data))
                         span.set_status(Status(StatusCode.OK))
                         
+                        logger.debug(f"Span {fn_name} completed successfully, is_recording={span.is_recording()}")
                         return result
                     except Exception as exception:
                         error = exception if isinstance(exception, Exception) else Exception(str(exception))
                         span.record_exception(error)
                         span.set_status(Status(StatusCode.ERROR, str(error)))
                         raise
+                    finally:
+                        # Log when span context exits - span should be ended automatically by context manager
+                        logger.debug(f"Span {fn_name} context exiting, is_recording={span.is_recording()}")
             
             async_traced_fn._is_traced = True
             logger.debug(f"Function {fn_name} is now traced (async)")
@@ -240,11 +249,13 @@ def WithTracing(
                             del input_data[key]
                 
                 # Use start_as_current_span to ensure span is recorded by BatchSpanProcessor
-                # This automatically manages context and ends the span when exiting
                 with tracer.start_as_current_span(fn_name) as span:
-                    # Ensure span is recording
+                    # Check if span is recording - if not, spans won't be exported
                     if not span.is_recording():
-                        logger.warning(f"Span {fn_name} is not recording - tracing may not work")
+                        logger.warning(f"Span {fn_name} is not recording - will not be exported")
+                        return fn(*args, **kwargs)
+                    
+                    logger.debug(f"Span {fn_name} is recording, trace_id={format(span.get_span_context().trace_id, '032x')}")
                     
                     if input_data is not None:
                         # Serialize for span attributes (OpenTelemetry only accepts primitives or JSON strings)

{aiqa_client-0.1.4 → aiqa_client-0.1.6/aiqa_client.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aiqa-client
-Version: 0.1.4
+Version: 0.1.6
 Summary: OpenTelemetry-based Python client for tracing functions and sending traces to the AIQA server
 Author-email: AIQA <info@aiqa.dev>
 License: MIT

{aiqa_client-0.1.4 → aiqa_client-0.1.6}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "aiqa-client"
-version = "0.1.4"
+version = "0.1.6"
 description = "OpenTelemetry-based Python client for tracing functions and sending traces to the AIQA server"
 readme = "README.md"
 requires-python = ">=3.8"