langtrace-python-sdk 1.2.7__py3-none-any.whl → 1.2.9__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/instrumentation/common.py

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

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 (

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.7"
+__version__ = "1.2.9"

langtrace_python_sdk-1.2.9.dist-info/METADATA

@@ -0,0 +1,130 @@
+Metadata-Version: 2.3
+Name: langtrace-python-sdk
+Version: 1.2.9
+Summary: Python SDK for LangTrace
+Project-URL: Homepage, https://github.com/Scale3-Labs/langtrace-python-sdk
+Author-email: Scale3 Labs <engineering@scale3labs.com>
+License-Expression: Apache-2.0
+License-File: LICENSE
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Requires-Python: >=3.9
+Requires-Dist: opentelemetry-api
+Requires-Dist: opentelemetry-instrumentation
+Requires-Dist: opentelemetry-sdk
+Requires-Dist: pinecone-client
+Requires-Dist: tiktoken
+Requires-Dist: trace-attributes
+Provides-Extra: dev
+Requires-Dist: anthropic; extra == 'dev'
+Requires-Dist: chromadb; extra == 'dev'
+Requires-Dist: langchain; extra == 'dev'
+Requires-Dist: langchain-openai; extra == 'dev'
+Requires-Dist: llama-index; extra == 'dev'
+Requires-Dist: openai; extra == 'dev'
+Requires-Dist: python-dotenv; extra == 'dev'
+Description-Content-Type: text/markdown
+
+# What is Langtrace?
+
+Langtrace stands as a developer-centric, open-source solution, fully compatible with OpenTelemetry. It enables developers to effortlessly trace, monitor, and debug their LLM applications, offering robust support for automatic instrumentation.
+
+## Supported LLM Modules
+
+Langtrace supports a comprehensive range of LLMs, VectorDBs, and frameworks, ensuring wide coverage for your development needs:
+
+### LLMs
+
+1. OpenAI
+2. Anthropic
+3. Azure OpenAI
+
+### VectorDBs
+
+1. Pinecone
+2. Chromadb
+
+### Frameworks
+
+1. LangChain
+2. LlamaIndex
+3. Haystack
+
+We are actively working to extend our support to additional libraries!
+
+## Getting Started
+
+To begin utilizing Langtrace, follow these straightforward steps:
+
+1. Install the package using `pip install langtrace-python-sdk`.
+2. Incorporate Langtrace into your project with `from langtrace_python_sdk import langtrace`.
+   - This import should precede any other LLM module imports (such as OpenAI, LlamaIndex, etc.) to ensure proper functionality.
+3. Initialize Langtrace by adding `langtrace.init(write_to_langtrace_cloud=false)` to your code.
+4. Congratulations, you've completed the basic setup! You will now begin to see traces from your LLM modules logged directly to the console.
+
+
+## Exporting Traces to Langtrace
+
+To configure trace exporting, you have two options:
+
+You'll need a Langtrace `api_key`, which can be acquired by logging into your Langtrace account.
+
+1. Direct Initialization: Utilize `langtrace.init(api_key=<YOUR_API_KEY>)`.
+2. Environment Variables: Set `LANGTRACE_API_KEY`, then add `langtrace.init()` at the beginning of your file.
+
+### Additional Customization
+
+- `@with_langtrace_root_span` - this decorator is designed to organize and relate different spans, in a hierarchical manner. When you're performing multiple operations that you want to monitor together as a unit, this function helps by establishing a "parent" (`LangtraceRootSpan` or whatever is passed to `name`) span. Then, any calls to the LLM APIs made within the given function (fn) will be considered "children" of this parent span. This setup is especially useful for tracking the performance or behavior of a group of operations collectively, rather than individually.
+
+```python
+from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
+
+@with_langtrace_root_span()
+def example():
+    response = client.chat.completions.create(
+        model="gpt-4",
+        messages=[{"role": "user", "content": "Say this is a test three times"}],
+        stream=False,
+    )
+    return response
+```
+
+
+- `with_additional_attributes` - this function is designed to enhance the traces by adding custom attributes to the current context. These custom attributes provide extra details about the operations being performed, making it easier to analyze and understand their behavior.
+
+```python
+from langtrace_python_sdk.utils.with_root_span import (
+    with_langtrace_root_span,
+    with_additional_attributes,
+)
+
+@with_additional_attributes({"user.id": "1234", "user.feedback.rating": 1})
+def api_call1():
+    response = client.chat.completions.create(
+        model="gpt-4",
+        messages=[{"role": "user", "content": "Say this is a test three times"}],
+        stream=False,
+    )
+    return response
+
+
+@with_additional_attributes({"user.id": "5678", "user.feedback.rating": -1})
+def api_call2():
+    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():
+   api_call1()
+   api_call2()
+```
+
+## Langtrace Cloud
+
+Currently under development 🚧