Merge pull request #438 from Scale3-Labs/ali/s3en-2927-customer-requestfix-signoz-issue

Ali/s3en 2927 customer requestfix signoz issue

Author: alizenhom

src/langtrace_python_sdk/instrumentation/dspy/patch.py

@@ -120,8 +120,6 @@ def traced_method(wrapped, instance, args, kwargs):
             **get_extra_attributes(),
         }
 
-        trace_checkpoint = os.environ.get("TRACE_DSPY_CHECKPOINT", "true").lower()
-
         if instance.__class__.__name__:
             span_attributes["dspy.signature.name"] = instance.__class__.__name__
             span_attributes["dspy.signature"] = str(instance.signature)
@@ -143,9 +141,6 @@ def traced_method(wrapped, instance, args, kwargs):
                         "dspy.signature.result",
                         json.dumps(result.toDict()),
                     )
-                    if trace_checkpoint == "true":
-                        print(Fore.RED + "Note: DSPy checkpoint tracing is enabled in Langtrace. To disable it, set the env var, TRACE_DSPY_CHECKPOINT to false" + Fore.RESET)
-                        set_span_attribute(span, "dspy.checkpoint", ujson.dumps(instance.dump_state(False), indent=2))
                     span.set_status(Status(StatusCode.OK))
 
                 span.end()

src/langtrace_python_sdk/langtrace.py

@@ -99,7 +99,9 @@ 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")
+        self.session_id = kwargs.get("session_id") or os.environ.get(
+            "LANGTRACE_SESSION_ID"
+        )
 
 
 def get_host(config: LangtraceConfig) -> str:
@@ -156,9 +158,16 @@ def get_exporter(config: LangtraceConfig, host: str):
         return config.custom_remote_exporter
 
     headers = get_headers(config)
-    host = f"{host}/api/trace" if host == LANGTRACE_REMOTE_URL else host
-    if "http" in host.lower() or "https" in host.lower():
-        return HTTPExporter(endpoint=host, headers=headers)
+    exporter_protocol = os.environ.get("OTEL_EXPORTER_OTLP_PROTOCOL", "http")
+    if "http" in exporter_protocol.lower():
+        return HTTPExporter(
+            endpoint=(
+                f"{host}/api/trace"
+                if host == LANGTRACE_REMOTE_URL
+                else f"{host}/v1/traces"
+            ),
+            headers=headers,
+        )
     else:
         return GRPCExporter(endpoint=host, headers=headers)
 

src/langtrace_python_sdk/version.py

@@ -1 +1 @@
-__version__ = "3.3.15"
+__version__ = "3.3.16"