langtrace-python-sdk 1.2.6__py3-none-any.whl → 1.2.8__py3-none-any.whl

examples/langchain_example/basic.py

@@ -8,15 +8,18 @@ from langchain_core.runnables import RunnablePassthrough
 from langchain_openai import ChatOpenAI, OpenAIEmbeddings
 
 from langtrace_python_sdk import langtrace
-from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
+from langtrace_python_sdk.utils.with_root_span import (
+    with_langtrace_root_span,
+    with_additional_attributes,
+)
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(batch=False, debug_log_to_console=True, write_to_langtrace_cloud=False)
+langtrace.init()
 
 
-@with_langtrace_root_span()
-def basic():
+@with_additional_attributes({"user.id": "1234", "user.feedback.rating": 1})
+def api_call_1():
     llm = ChatOpenAI()
     prompt = ChatPromptTemplate.from_messages(
         [
@@ -30,6 +33,27 @@ def basic():
     print(res)
 
 
+@with_additional_attributes({"user.id": "37373", "user.feedback.rating": 1})
+def api_call_2():
+    llm = ChatOpenAI()
+    prompt = ChatPromptTemplate.from_messages(
+        [
+            ("system", "You are world class technical documentation writer."),
+            ("user", "{input}"),
+        ]
+    )
+    output_parser = StrOutputParser()
+    chain = prompt | llm | output_parser
+    res = chain.invoke({"input": "how can langsmith help with testing?"})
+    print(res)
+
+
+@with_langtrace_root_span()
+def basic():
+    api_call_1()
+    api_call_2()
+
+
 @with_langtrace_root_span()
 def rag():
     vectorstore = FAISS.from_texts(

examples/openai/chat_completion.py

@@ -2,38 +2,58 @@ from dotenv import find_dotenv, load_dotenv
 from openai import OpenAI
 
 from langtrace_python_sdk import langtrace
-from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
+from langtrace_python_sdk.utils.with_root_span import (
+    with_additional_attributes, with_langtrace_root_span)
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(batch=False, debug_log_to_console=True, write_to_langtrace_cloud=False)
-
+langtrace.init()
 client = OpenAI()
 
 
-@with_langtrace_root_span()
-def chat_completion():
+@with_additional_attributes({"user.id": "1234", "user.feedback.rating": 1})
+def api1():
     response = client.chat.completions.create(
         model="gpt-4",
         messages=[{"role": "user", "content": "Say this is a test three times"}],
-        stream=True,
+        stream=False,
     )
-    print(response)
-    # stream = client.chat.completions.create(
-    #     model="gpt-4",
-    #     messages=[{"role": "user", "content": "Say this is a test three times"}, {"role": "assistant", "content": "This is a test. This is a test. This is a test"},
-    #               {"role": "user", "content": "Say this is a mock 4 times"}],
-    #     stream=False,
-    # )
-
-    # result = []
-    # for chunk in response:
-    #     if chunk.choices[0].delta.content is not None:
-    #         content = [
-    #             choice.delta.content if choice.delta and
-    #             choice.delta.content else ""
-    #             for choice in chunk.choices]
-    #         result.append(
-    #             content[0] if len(content) > 0 else "")
-
-    # print("".join(result))
+    return response
+
+
+@with_additional_attributes({"user.id": "5678", "user.feedback.rating": -1})
+def api2():
+    response = client.chat.completions.create(
+        model="gpt-4",
+        messages=[{"role": "user", "content": "Say this is a test three times"}],
+        stream=False,
+    )
+    return response
+
+
+@with_langtrace_root_span()
+def chat_completion():
+    response = api1()
+    response = api2()
+    return response
+
+
+# print(response)
+# stream = client.chat.completions.create(
+#     model="gpt-4",
+#     messages=[{"role": "user", "content": "Say this is a test three times"}, {"role": "assistant", "content": "This is a test. This is a test. This is a test"},
+#               {"role": "user", "content": "Say this is a mock 4 times"}],
+#     stream=False,
+# )
+
+# result = []
+# for chunk in response:
+#     if chunk.choices[0].delta.content is not None:
+#         content = [
+#             choice.delta.content if choice.delta and
+#             choice.delta.content else ""
+#             for choice in chunk.choices]
+#         result.append(
+#             content[0] if len(content) > 0 else "")
+
+# print("".join(result))

langtrace_python_sdk/constants/exporter/langtrace_exporter.py

@@ -0,0 +1 @@
+LANGTRACE_REMOTE_URL = "https://app.langtrace.ai/api/trace"

langtrace_python_sdk/constants/instrumentation/common.py

@@ -17,3 +17,5 @@ SERVICE_PROVIDERS = {
     "OPENAI": "OpenAI",
     "PINECONE": "Pinecone",
 }
+
+LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY = "langtrace_additional_attributes"

langtrace_python_sdk/extensions/langtrace_exporter.py

@@ -2,9 +2,11 @@ import json
 import os
 import typing
 
+from langtrace_python_sdk.constants.exporter.langtrace_exporter import (
+    LANGTRACE_REMOTE_URL,
+)
 import requests
-from opentelemetry.sdk.trace.export import (ReadableSpan, SpanExporter,
-                                            SpanExportResult)
+from opentelemetry.sdk.trace.export import ReadableSpan, SpanExporter, SpanExportResult
 from opentelemetry.trace.span import format_trace_id
 
 
@@ -22,7 +24,6 @@ class LangTraceExporter(SpanExporter):
     **Attributes:**
 
     * `api_key` (str): An API key to authenticate with the LangTrace collector (required).
-    * `url` (str): The URL of the LangTrace collector endpoint (required if `write_to_remote_url` is True).
     * `write_to_remote_url` (bool): A flag indicating whether to send spans to the remote URL (defaults to False).
 
     **Methods:**
@@ -46,22 +47,15 @@ class LangTraceExporter(SpanExporter):
     """
 
     api_key: str
-    url: str
     write_to_remote_url: bool
 
-    def __init__(
-        self, api_key: str = None, url: str = None, write_to_remote_url: bool = False
-    ) -> None:
-        self.api_key = api_key if api_key else os.environ.get("API_KEY")
-        self.url = url if url else os.environ.get("URL")
+    def __init__(self, api_key: str = None, write_to_remote_url: bool = False) -> None:
+        self.api_key = api_key if api_key else os.environ.get("LANGTRACE_API_KEY")
         self.write_to_remote_url = write_to_remote_url
 
         if self.write_to_remote_url and not self.api_key:
             raise ValueError("No API key provided")
 
-        if self.write_to_remote_url and not self.url:
-            raise ValueError("No URL provided")
-
     def export(self, spans: typing.Sequence[ReadableSpan]) -> SpanExportResult:
         """
         Exports a batch of telemetry data.
@@ -92,11 +86,11 @@ class LangTraceExporter(SpanExporter):
         # Send data to remote URL
         try:
             requests.post(
-                url=self.url,
+                url=LANGTRACE_REMOTE_URL,
                 data=json.dumps(data),
                 headers={"Content-Type": "application/json", "x-api-key": self.api_key},
             )
-            print(f"Traces sent To {self.url}")
+            print(f"Traces sent To {LANGTRACE_REMOTE_URL}")
             return SpanExportResult.SUCCESS
         except Exception as e:
             print("Error sending data to remote URL", e)

langtrace_python_sdk/instrumentation/anthropic/patch.py

@@ -4,12 +4,13 @@ This module contains the patching logic for the Anthropic library."""
 import json
 
 from langtrace.trace_attributes import Event, LLMSpanAttributes
+from opentelemetry import baggage
 from opentelemetry.trace import SpanKind
 from opentelemetry.trace.status import Status, StatusCode
 
 from langtrace_python_sdk.constants.instrumentation.anthropic import APIS
-from langtrace_python_sdk.constants.instrumentation.common import \
-    SERVICE_PROVIDERS
+from langtrace_python_sdk.constants.instrumentation.common import (
+    LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY, SERVICE_PROVIDERS)
 
 
 def messages_create(original_method, version, tracer):
@@ -31,6 +32,7 @@ def messages_create(original_method, version, tracer):
             prompts = json.dumps(
                 [{"role": "system", "content": system}] + kwargs.get("messages", [])
             )
+        extra_attributes = baggage.get_baggage(LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY)
 
         span_attributes = {
             "langtrace.service.name": service_provider,
@@ -42,6 +44,7 @@ def messages_create(original_method, version, tracer):
             "llm.model": kwargs.get("model"),
             "llm.prompts": prompts,
             "llm.stream": kwargs.get("stream"),
+            **(extra_attributes if extra_attributes is not None else {})
         }
 
         attributes = LLMSpanAttributes(**span_attributes)

langtrace_python_sdk/instrumentation/chroma/patch.py

@@ -3,11 +3,13 @@ This module contains the patching logic for the Chroma client.
 """
 
 from langtrace.trace_attributes import DatabaseSpanAttributes
+from opentelemetry import baggage
 from opentelemetry.trace import SpanKind
 from opentelemetry.trace.status import Status, StatusCode
 
 from langtrace_python_sdk.constants.instrumentation.chroma import APIS
-from langtrace_python_sdk.constants.instrumentation.common import SERVICE_PROVIDERS
+from langtrace_python_sdk.constants.instrumentation.common import (
+    LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY, SERVICE_PROVIDERS)
 
 
 def collection_patch(method, version, tracer):
@@ -18,6 +20,8 @@ def collection_patch(method, version, tracer):
     def traced_method(wrapped, instance, args, kwargs):
         api = APIS[method]
         service_provider = SERVICE_PROVIDERS["CHROMA"]
+        extra_attributes = baggage.get_baggage(LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY)
+
         span_attributes = {
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "vectordb",
@@ -25,6 +29,7 @@ def collection_patch(method, version, tracer):
             "langtrace.version": "1.0.0",
             "db.system": "chromadb",
             "db.operation": api["OPERATION"],
+            **(extra_attributes if extra_attributes is not None else {})
         }
 
         if hasattr(instance, "name") and instance.name is not None:

langtrace_python_sdk/instrumentation/langchain/patch.py

@@ -5,10 +5,12 @@ This module contains the patching logic for the langchain package.
 import json
 
 from langtrace.trace_attributes import FrameworkSpanAttributes
+from opentelemetry import baggage
 from opentelemetry.trace import SpanKind, StatusCode
 from opentelemetry.trace.status import Status
 
-from langtrace_python_sdk.constants.instrumentation.common import SERVICE_PROVIDERS
+from langtrace_python_sdk.constants.instrumentation.common import (
+    LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY, SERVICE_PROVIDERS)
 
 
 def generic_patch(
@@ -20,12 +22,15 @@ def generic_patch(
 
     def traced_method(wrapped, instance, args, kwargs):
         service_provider = SERVICE_PROVIDERS["LANGCHAIN"]
+        extra_attributes = baggage.get_baggage(LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY)
+
         span_attributes = {
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "framework",
             "langtrace.service.version": version,
             "langtrace.version": "1.0.0",
             "langchain.task.name": task,
+            **(extra_attributes if extra_attributes is not None else {})
         }
 
         if len(args) > 0 and trace_input:

langtrace_python_sdk/instrumentation/langchain_community/patch.py

@@ -1,10 +1,12 @@
 import json
 
 from langtrace.trace_attributes import FrameworkSpanAttributes
+from opentelemetry import baggage
 from opentelemetry.trace import SpanKind
 from opentelemetry.trace.status import Status, StatusCode
 
-from langtrace_python_sdk.constants.instrumentation.common import SERVICE_PROVIDERS
+from langtrace_python_sdk.constants.instrumentation.common import (
+    LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY, SERVICE_PROVIDERS)
 
 
 def generic_patch(
@@ -12,12 +14,15 @@ def generic_patch(
 ):
     def traced_method(wrapped, instance, args, kwargs):
         service_provider = SERVICE_PROVIDERS["LANGCHAIN_COMMUNITY"]
+        extra_attributes = baggage.get_baggage(LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY)
+
         span_attributes = {
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "framework",
             "langtrace.service.version": version,
             "langtrace.version": "1.0.0",
             "langchain.task.name": task,
+            **(extra_attributes if extra_attributes is not None else {})
         }
 
         if trace_input and len(args) > 0:

langtrace_python_sdk/instrumentation/langchain_core/patch.py

@@ -5,10 +5,12 @@ This module contains the patching functions for the langchain_core package.
 import json
 
 from langtrace.trace_attributes import FrameworkSpanAttributes
+from opentelemetry import baggage
 from opentelemetry.trace import SpanKind, StatusCode
 from opentelemetry.trace.status import Status
 
-from langtrace_python_sdk.constants.instrumentation.common import SERVICE_PROVIDERS
+from langtrace_python_sdk.constants.instrumentation.common import (
+    LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY, SERVICE_PROVIDERS)
 
 
 def generic_patch(
@@ -26,12 +28,15 @@ def generic_patch(
 
     def traced_method(wrapped, instance, args, kwargs):
         service_provider = SERVICE_PROVIDERS["LANGCHAIN_CORE"]
+        extra_attributes = baggage.get_baggage(LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY)
+
         span_attributes = {
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "framework",
             "langtrace.service.version": version,
             "langtrace.version": "1.0.0",
             "langchain.task.name": task,
+            **(extra_attributes if extra_attributes is not None else {})
         }
 
         if len(args) > 0 and trace_input:

langtrace_python_sdk/instrumentation/llamaindex/patch.py

@@ -3,10 +3,12 @@ This module contains a generic patch method that wraps a function with a span.
 """
 
 from langtrace.trace_attributes import FrameworkSpanAttributes
+from opentelemetry import baggage
 from opentelemetry.trace import SpanKind
 from opentelemetry.trace.status import Status, StatusCode
 
-from langtrace_python_sdk.constants.instrumentation.common import SERVICE_PROVIDERS
+from langtrace_python_sdk.constants.instrumentation.common import (
+    LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY, SERVICE_PROVIDERS)
 
 
 def generic_patch(method, task, tracer, version):
@@ -15,12 +17,15 @@ def generic_patch(method, task, tracer, version):
 
     def traced_method(wrapped, instance, args, kwargs):
         service_provider = SERVICE_PROVIDERS["LLAMAINDEX"]
+        extra_attributes = baggage.get_baggage(LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY)
+
         span_attributes = {
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "framework",
             "langtrace.service.version": version,
             "langtrace.version": "1.0.0",
             "llamaindex.task.name": task,
+            **(extra_attributes if extra_attributes is not None else {})
         }
 
         attributes = FrameworkSpanAttributes(**span_attributes)

langtrace_python_sdk/instrumentation/openai/patch.py

@@ -4,12 +4,15 @@ This module contains the patching logic for the OpenAI library."""
 import json
 
 from langtrace.trace_attributes import Event, LLMSpanAttributes
+from opentelemetry import baggage, trace
 from opentelemetry.trace import SpanKind
 from opentelemetry.trace.status import Status, StatusCode
 
-from langtrace_python_sdk.constants.instrumentation.common import SERVICE_PROVIDERS
+from langtrace_python_sdk.constants.instrumentation.common import (
+    LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY, SERVICE_PROVIDERS)
 from langtrace_python_sdk.constants.instrumentation.openai import APIS
-from langtrace_python_sdk.utils.llm import calculate_prompt_tokens, estimate_tokens
+from langtrace_python_sdk.utils.llm import (calculate_prompt_tokens,
+                                            estimate_tokens)
 
 
 def images_generate(original_method, version, tracer):
@@ -24,6 +27,8 @@ def images_generate(original_method, version, tracer):
             else ""
         )
         service_provider = SERVICE_PROVIDERS["OPENAI"]
+        extra_attributes = baggage.get_baggage(LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY)
+
         span_attributes = {
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "llm",
@@ -34,6 +39,7 @@ def images_generate(original_method, version, tracer):
             "llm.model": kwargs.get("model"),
             "llm.stream": kwargs.get("stream"),
             "llm.prompts": json.dumps([kwargs.get("prompt", [])]),
+            **(extra_attributes if extra_attributes is not None else {})
         }
 
         attributes = LLMSpanAttributes(**span_attributes)
@@ -90,6 +96,8 @@ def chat_completions_create(original_method, version, tracer):
             else ""
         )
         service_provider = SERVICE_PROVIDERS["OPENAI"]
+        extra_attributes = baggage.get_baggage(LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY)
+
         span_attributes = {
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "llm",
@@ -99,6 +107,7 @@ def chat_completions_create(original_method, version, tracer):
             "llm.api": APIS["CHAT_COMPLETION"]["ENDPOINT"],
             "llm.prompts": json.dumps(kwargs.get("messages", [])),
             "llm.stream": kwargs.get("stream"),
+            **(extra_attributes if extra_attributes is not None else {})
         }
 
         attributes = LLMSpanAttributes(**span_attributes)
@@ -130,10 +139,12 @@ def chat_completions_create(original_method, version, tracer):
                     responses = [
                         {
                             "message": {
-                                    "role": choice.message.role
+                                "role": (
+                                    choice.message.role
                                     if choice.message and choice.message.role
-                                    else "assistant",
-                                    "content": (
+                                    else "assistant"
+                                ),
+                                "content": (
                                     choice.message.content
                                     if choice.message and choice.message.content
                                     else (
@@ -299,6 +310,8 @@ def embeddings_create(original_method, version, tracer):
         )
 
         service_provider = SERVICE_PROVIDERS["OPENAI"]
+        extra_attributes = baggage.get_baggage(LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY)
+
         span_attributes = {
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "llm",
@@ -308,6 +321,7 @@ def embeddings_create(original_method, version, tracer):
             "llm.api": APIS["EMBEDDINGS_CREATE"]["ENDPOINT"],
             "llm.model": kwargs.get("model"),
             "llm.prompts": "",
+            **(extra_attributes if extra_attributes is not None else {})
         }
 
         attributes = LLMSpanAttributes(**span_attributes)
@@ -323,6 +337,9 @@ def embeddings_create(original_method, version, tracer):
         with tracer.start_as_current_span(
             APIS["EMBEDDINGS_CREATE"]["METHOD"], kind=SpanKind.CLIENT
         ) as span:
+
+            print("Inside embeddings_create", trace.get_current_span())
+
             for field, value in attributes.model_dump(by_alias=True).items():
                 if value is not None:
                     span.set_attribute(field, value)

langtrace_python_sdk/instrumentation/pinecone/patch.py

@@ -2,10 +2,12 @@
 This module contains the patching logic for the Pinecone client."""
 
 from langtrace.trace_attributes import DatabaseSpanAttributes
+from opentelemetry import baggage
 from opentelemetry.trace import SpanKind
 from opentelemetry.trace.status import Status, StatusCode
 
-from langtrace_python_sdk.constants.instrumentation.common import SERVICE_PROVIDERS
+from langtrace_python_sdk.constants.instrumentation.common import (
+    LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY, SERVICE_PROVIDERS)
 from langtrace_python_sdk.constants.instrumentation.pinecone import APIS
 
 
@@ -16,6 +18,8 @@ def generic_patch(original_method, method, version, tracer):
     def traced_method(wrapped, instance, args, kwargs):
         api = APIS[method]
         service_provider = SERVICE_PROVIDERS["PINECONE"]
+        extra_attributes = baggage.get_baggage(LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY)
+
         span_attributes = {
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "vectordb",
@@ -23,6 +27,7 @@ def generic_patch(original_method, method, version, tracer):
             "langtrace.version": "1.0.0",
             "db.system": "pinecone",
             "db.operation": api["OPERATION"],
+            **(extra_attributes if extra_attributes is not None else {})
         }
 
         attributes = DatabaseSpanAttributes(**span_attributes)

langtrace_python_sdk/langtrace.py

@@ -1,3 +1,4 @@
+
 from opentelemetry import trace
 from opentelemetry.sdk.trace import TracerProvider
 from opentelemetry.sdk.trace.export import (
@@ -37,17 +38,19 @@ def init(
     api_key: str = None,
     batch: bool = True,
     write_to_langtrace_cloud: bool = True,
-    remote_url: str = None,
     debug_log_to_console: bool = False,
-    custom_remote_exporter = None
+    custom_remote_exporter=None,
 ):
 
     provider = TracerProvider()
 
-    remote_write_exporter = LangTraceExporter(api_key, remote_url, write_to_langtrace_cloud) if custom_remote_exporter is None else custom_remote_exporter
+    remote_write_exporter = (
+        LangTraceExporter(api_key, write_to_langtrace_cloud)
+        if custom_remote_exporter is None
+        else custom_remote_exporter
+    )
     console_exporter = ConsoleSpanExporter()
     batch_processor_remote = BatchSpanProcessor(remote_write_exporter)
-    simple_processor_remote = SimpleSpanProcessor(remote_write_exporter)
     batch_processor_console = BatchSpanProcessor(console_exporter)
     simple_processor_console = SimpleSpanProcessor(console_exporter)
 

langtrace_python_sdk/utils/with_root_span.py

@@ -1,21 +1,27 @@
 import asyncio
 from functools import wraps
 
-from opentelemetry import trace
+
+from langtrace_python_sdk.constants.instrumentation.common import (
+    LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY,
+)
 from opentelemetry.trace import SpanKind
+from opentelemetry import trace, context, baggage
 
 
 def with_langtrace_root_span(
-    name="LangtraceRootSpan", kind=SpanKind.INTERNAL, attributes={}
+    name="LangtraceRootSpan",
+    kind=SpanKind.INTERNAL,
 ):
+
     def decorator(func):
         @wraps(func)
         def sync_wrapper(*args, **kwargs):
             tracer = trace.get_tracer(__name__)
             operation_name = name if name else func.__name__
-            with tracer.start_as_current_span(
-                operation_name, kind=kind, attributes=attributes
-            ):
+
+            with tracer.start_as_current_span(operation_name, kind=kind):
+
                 return func(*args, **kwargs)
 
         @wraps(func)
@@ -31,3 +37,29 @@ def with_langtrace_root_span(
             return sync_wrapper
 
     return decorator
+
+
+def with_additional_attributes(attributes={}):
+    def decorator(func):
+        @wraps(func)
+        def sync_wrapper(*args, **kwargs):
+            new_ctx = baggage.set_baggage(
+                LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY, attributes
+            )
+            context.attach(new_ctx)
+            return func(*args, **kwargs)
+
+        @wraps(func)
+        async def async_wrapper(*args, **kwargs):
+            new_ctx = baggage.set_baggage(
+                LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY, attributes
+            )
+            context.attach(new_ctx)
+            return await func(*args, **kwargs)
+
+        if asyncio.iscoroutinefunction(func):
+            return async_wrapper
+        else:
+            return sync_wrapper
+
+    return decorator

langtrace_python_sdk/version.py

@@ -1 +1 @@
-__version__ = "1.2.6"
+__version__ = "1.2.8"