finalize

Scale3-Labs · Jun 13, 2024 · ed8aa99 · ed8aa99
1 parent 57a6fee
commit ed8aa99
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 5 deletions.
diff --git a/src/examples/llamaindex_example/agent.py b/src/examples/llamaindex_example/agent.py
@@ -12,7 +12,7 @@
 
 nest_asyncio.apply()
 
-langtrace.init(write_spans_to_console=False)
+langtrace.init()
 
 
 def multiply(a: int, b: int) -> int:

diff --git a/src/examples/llamaindex_example/basic.py b/src/examples/llamaindex_example/basic.py
@@ -8,8 +8,9 @@
 
 
 langtrace.init(
-    write_spans_to_console=False,
-    disable_tracing_for_methods={"open_ai": ["openai.embeddings.create"]},
+    disable_tracing_for_methods={
+        "open_ai": ["openai.chat.completions.create"],
+    }
 )
 
 

diff --git a/src/langtrace_python_sdk/utils/langtrace_sampler.py b/src/langtrace_python_sdk/utils/langtrace_sampler.py
@@ -12,7 +12,6 @@
 
 class LangtraceSampler(Sampler):
     _disabled_methods_names: set
-    _seen: set = set()
 
     def __init__(
         self,
@@ -42,7 +41,10 @@ def should_sample(
             return SamplingResult(decision=Decision.RECORD_AND_SAMPLE)
 
         if parent_context:
-            if parent_span_context.trace_flags != TraceFlags.SAMPLED:
+            if (
+                parent_span_context.span_id != 0
+                and parent_span_context.trace_flags != TraceFlags.SAMPLED
+            ):
                 return SamplingResult(decision=Decision.DROP)
 
         if name in self._disabled_methods_names: