chng name

Scale3-Labs · Jun 13, 2024 · 9f16e04 · 9f16e04
1 parent ed8aa99
commit 9f16e04
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/src/examples/langchain_example/basic.py b/src/examples/langchain_example/basic.py
@@ -15,7 +15,7 @@
 
 langtrace.init(
     write_spans_to_console=False,
-    disable_tracing_for_methods={"langchain": ["RunnableSequence.invoke"]},
+    disable_tracing_for_functions={"langchain": ["RunnableSequence.invoke"]},
 )
 
 

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

diff --git a/src/langtrace_python_sdk/langtrace.py b/src/langtrace_python_sdk/langtrace.py
@@ -64,15 +64,15 @@ def init(
     custom_remote_exporter=None,
     api_host: Optional[str] = LANGTRACE_REMOTE_URL,
     disable_instrumentations: Optional[DisableInstrumentations] = None,
-    disable_tracing_for_methods: Optional[InstrumentationMethods] = None,
+    disable_tracing_for_functions: Optional[InstrumentationMethods] = None,
 ):
 
     host = (
         os.environ.get("LANGTRACE_API_HOST", None) or api_host or LANGTRACE_REMOTE_URL
     )
     check_if_sdk_is_outdated()
     print(Fore.GREEN + "Initializing Langtrace SDK.." + Fore.RESET)
-    sampler = LangtraceSampler(disabled_methods=disable_tracing_for_methods)
+    sampler = LangtraceSampler(disabled_methods=disable_tracing_for_functions)
     provider = TracerProvider(
         resource=Resource.create({"service.name": sys.argv[0]}),
         sampler=sampler,