aiqa-client 0.1.4__py3-none-any.whl → 0.1.6__py3-none-any.whl

aiqa/__init__.py

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

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.dist-info → aiqa_client-0.1.6.dist-info}/METADATA

@@ -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.6.dist-info/RECORD

@@ -0,0 +1,9 @@
+aiqa/__init__.py,sha256=KXcAhHf_aZCgs6bn5Gw3IHccunqZzjbj8md0RuWSRAg,470
+aiqa/aiqa_exporter.py,sha256=vXyX6Q_iOjrDz3tCPOMXuBTQg7ocACdOOqzpkUqhy9g,19131
+aiqa/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+aiqa/tracing.py,sha256=mWYIZ2Pla91oUY5a6OXOJWUR2eYtjoQkYPaJdBTpho8,13522
+aiqa_client-0.1.6.dist-info/licenses/LICENSE,sha256=kIzkzLuzG0HHaWYm4F4W5FeJ1Yxut3Ec6bhLWyw798A,1062
+aiqa_client-0.1.6.dist-info/METADATA,sha256=Yu-H7t101WTN5rdwJl8-7AAVemL9hHZVxSGjZifurKk,3772
+aiqa_client-0.1.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aiqa_client-0.1.6.dist-info/top_level.txt,sha256=nwcsuVVSuWu27iLxZd4n1evVzv1W6FVTrSnCXCc-NQs,5
+aiqa_client-0.1.6.dist-info/RECORD,,

aiqa_client-0.1.4.dist-info/RECORD

@@ -1,9 +0,0 @@
-aiqa/__init__.py,sha256=MwDG0A3kszFsknTCxQ3l8lX1QB2soTMSMZ2YhI7FcYU,470
-aiqa/aiqa_exporter.py,sha256=vXyX6Q_iOjrDz3tCPOMXuBTQg7ocACdOOqzpkUqhy9g,19131
-aiqa/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-aiqa/tracing.py,sha256=Aq4VbX6czSO7LtIcA6tTXghtw9-upZ802g3DNylzoq8,12680
-aiqa_client-0.1.4.dist-info/licenses/LICENSE,sha256=kIzkzLuzG0HHaWYm4F4W5FeJ1Yxut3Ec6bhLWyw798A,1062
-aiqa_client-0.1.4.dist-info/METADATA,sha256=WU-tzSni5NhyJfUHdq4f0lX6JHOPW0VRGrtkW6XL7mo,3772
-aiqa_client-0.1.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aiqa_client-0.1.4.dist-info/top_level.txt,sha256=nwcsuVVSuWu27iLxZd4n1evVzv1W6FVTrSnCXCc-NQs,5
-aiqa_client-0.1.4.dist-info/RECORD,,