langtrace-python-sdk 2.1.12__py3-none-any.whl → 2.1.14__py3-none-any.whl

examples/__init__.py

@@ -0,0 +1 @@
+

examples/anthropic_example/__init__.py

@@ -0,0 +1,8 @@
+from examples.anthropic_example.completion import messages_create
+from langtrace_python_sdk import with_langtrace_root_span
+
+
+class AnthropicRunner:
+    @with_langtrace_root_span("Anthropic")
+    def run(self):
+        messages_create()

examples/anthropic_example/completion.py

@@ -7,12 +7,11 @@ from langtrace_python_sdk import langtrace, with_langtrace_root_span
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(write_spans_to_console=True)
+langtrace.init()
 
 
 @with_langtrace_root_span("messages_create")
 def messages_create():
-
     client = anthropic.Anthropic()
 
     message = client.messages.create(

examples/chroma_example/__init__.py

@@ -1 +1,8 @@
-from .basic import basic
+from examples.chroma_example.basic import basic
+from langtrace_python_sdk import with_langtrace_root_span
+
+
+class ChromaRunner:
+    @with_langtrace_root_span("Chroma")
+    def run(self):
+        basic()

examples/chroma_example/basic.py

@@ -7,7 +7,7 @@ from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(write_spans_to_console=True)
+langtrace.init(write_spans_to_console=False)
 
 
 @with_langtrace_root_span()

examples/cohere_example/__init__.py

@@ -0,0 +1,17 @@
+from examples.cohere_example.chat import chat_comp
+from examples.cohere_example.chat_stream import chat_stream
+from examples.cohere_example.tools import tool_calling
+from examples.cohere_example.embed import embed
+from examples.cohere_example.rerank import rerank
+from langtrace_python_sdk import with_langtrace_root_span
+
+
+class CohereRunner:
+
+    @with_langtrace_root_span("Cohere")
+    def run(self):
+        chat_comp()
+        chat_stream()
+        tool_calling()
+        embed()
+        rerank()

examples/cohere_example/chat.py

@@ -7,7 +7,7 @@ from langtrace_python_sdk import langtrace
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(write_spans_to_console=True)
+langtrace.init()
 
 
 co = cohere.Client()

examples/cohere_example/chat_stream.py

@@ -5,7 +5,7 @@ from langtrace_python_sdk import langtrace
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(write_spans_to_console=True)
+langtrace.init()
 
 
 co = cohere.Client()

examples/cohere_example/embed.py

@@ -7,7 +7,7 @@ from langtrace_python_sdk import langtrace
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(write_spans_to_console=True)
+langtrace.init()
 
 
 co = cohere.Client()

examples/cohere_example/rerank.py

@@ -7,7 +7,7 @@ from langtrace_python_sdk import langtrace
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(write_spans_to_console=True)
+langtrace.init()
 
 
 co = cohere.Client()

examples/cohere_example/tools.py

@@ -7,7 +7,7 @@ from langtrace_python_sdk import langtrace
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(write_spans_to_console=True)
+langtrace.init()
 
 co = cohere.Client()
 

examples/fastapi_example/__init__.py

@@ -0,0 +1,6 @@
+from .basic_route import root
+
+
+class FastAPIRunner:
+    def run(self):
+        root()

examples/fastapi_example/basic_route.py

@@ -7,7 +7,9 @@ from langchain_openai import ChatOpenAI, OpenAIEmbeddings
 from openai import OpenAI
 
 from langtrace_python_sdk import langtrace
+from dotenv import load_dotenv
 
+load_dotenv()
 langtrace.init(write_spans_to_console=True)
 app = FastAPI()
 client = OpenAI()

examples/langchain_example/__init__.py

@@ -0,0 +1,10 @@
+from .basic import basic_app, rag, load_and_split
+from langtrace_python_sdk import with_langtrace_root_span
+
+
+class LangChainRunner:
+    @with_langtrace_root_span("LangChain")
+    def run(self):
+        basic_app()
+        rag()
+        load_and_split()

examples/langchain_example/basic.py

@@ -18,7 +18,7 @@ _ = load_dotenv(find_dotenv())
 langtrace.init()
 
 
-@with_additional_attributes({"user.id": "1234", "user.feedback.rating": 1})
+# @with_additional_attributes({"user.id": "1234", "user.feedback.rating": 1})
 def api_call_1():
     llm = ChatOpenAI()
     prompt = ChatPromptTemplate.from_messages(
@@ -33,7 +33,7 @@ def api_call_1():
     print(res)
 
 
-@with_additional_attributes({"user.id": "37373", "user.feedback.rating": 1})
+# @with_additional_attributes({"user.id": "37373", "user.feedback.rating": 1})
 def api_call_2():
     llm = ChatOpenAI()
     prompt = ChatPromptTemplate.from_messages(
@@ -49,7 +49,7 @@ def api_call_2():
 
 
 @with_langtrace_root_span()
-def basic():
+def basic_app():
     api_call_1()
     api_call_2()
 

examples/llamaindex_example/__init__.py

@@ -0,0 +1,11 @@
+from langtrace_python_sdk import with_langtrace_root_span
+
+
+class LlamaIndexRunner:
+    @with_langtrace_root_span("LlamaIndex")
+    def run(self):
+        from .basic import basic_app
+        from .agent import main
+
+        basic_app()
+        main()

examples/llamaindex_example/agent.py

@@ -12,7 +12,7 @@ import nest_asyncio
 
 nest_asyncio.apply()
 
-langtrace.init(write_spans_to_console=True)
+langtrace.init(write_spans_to_console=False)
 
 
 def multiply(a: int, b: int) -> int:
@@ -82,5 +82,6 @@ class YourOpenAIAgent:
 
 
 # agent = YourOpenAIAgent(tools=[multiply_tool, add_tool])
-llm = OpenAI(model="gpt-3.5-turbo-0613")
-agent = OpenAIAgent.from_tools([multiply_tool, add_tool], llm=llm, verbose=True)
+def main():
+    llm = OpenAI(model="gpt-3.5-turbo-0613")
+    agent = OpenAIAgent.from_tools([multiply_tool, add_tool], llm=llm, verbose=True)

examples/llamaindex_example/basic.py

@@ -7,11 +7,11 @@ from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
 _ = load_dotenv(find_dotenv())
 
 
-langtrace.init(write_spans_to_console=True)
+langtrace.init(write_spans_to_console=False)
 
 
 @with_langtrace_root_span()
-def basic():
+def basic_app():
     documents = SimpleDirectoryReader(
         "src/examples/llamaindex_example/data"
     ).load_data()

examples/openai_example/__init__.py

@@ -0,0 +1,21 @@
+from langtrace_python_sdk import with_langtrace_root_span
+
+
+class OpenAIRunner:
+    @with_langtrace_root_span("OpenAI")
+    def run(self):
+        import asyncio
+        from .async_tool_calling_nonstreaming import run_conversation
+        from .async_tool_calling_streaming import (
+            run_conversation as run_conversation_streaming,
+        )
+        from .chat_completion import chat_completion as chat_completion_example
+        from .embeddings_create import embeddings_create as embeddings_create_example
+        from .function_calling import function_calling as function_example
+        from .images_edit import image_edit
+
+        asyncio.run(run_conversation())
+        asyncio.run(run_conversation_streaming())
+        chat_completion_example()
+        embeddings_create_example()
+        function_example()

examples/openai_example/async_tool_calling_nonstreaming.py

@@ -3,13 +3,12 @@ import json
 from dotenv import find_dotenv, load_dotenv
 from openai import AsyncOpenAI
 
-from langtrace_python_sdk import langtrace
+from langtrace_python_sdk import langtrace, with_langtrace_root_span
 
-# from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(write_spans_to_console=True)
+langtrace.init()
 
 client = AsyncOpenAI()
 
@@ -30,6 +29,7 @@ def get_current_weather(location, unit="fahrenheit"):
         return json.dumps({"location": location, "temperature": "unknown"})
 
 
+@with_langtrace_root_span("Run Conversation")
 async def run_conversation():
     # Step 1: send the conversation and available functions to the model
     messages = [

examples/openai_example/async_tool_calling_streaming.py

@@ -3,7 +3,7 @@ import json
 from dotenv import find_dotenv, load_dotenv
 from openai import AsyncOpenAI
 
-from langtrace_python_sdk import langtrace
+from langtrace_python_sdk import langtrace, with_langtrace_root_span
 
 # from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
 
@@ -42,6 +42,7 @@ def get_current_time(location):
         return json.dumps({"location": location, "time": "unknown"})
 
 
+@with_langtrace_root_span("Run Conversation Streaming")
 async def run_conversation():
     # Step 1: send the conversation and available functions to the model
     messages = [

examples/openai_example/chat_completion.py

@@ -9,11 +9,11 @@ from langtrace_python_sdk.utils.with_root_span import (
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(write_spans_to_console=True)
+langtrace.init(write_spans_to_console=False)
 client = OpenAI()
 
 
-# @with_additional_attributes({"user.id": "1234", "user.feedback.rating": 1})
+@with_additional_attributes({"user.id": "1234", "user.feedback.rating": 1})
 def api():
     response = client.chat.completions.create(
         model="gpt-4",
@@ -27,7 +27,7 @@ def api():
     return response
 
 
-# @with_langtrace_root_span()
+@with_langtrace_root_span("Chat Completion")
 def chat_completion():
     response = api()
     # print(response)

examples/openai_example/embeddings_create.py

@@ -6,12 +6,12 @@ from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(write_spans_to_console=True)
+langtrace.init(write_spans_to_console=False)
 
 client = OpenAI()
 
 
-@with_langtrace_root_span()
+@with_langtrace_root_span("Embeddings Create")
 def embeddings_create():
     result = client.embeddings.create(
         model="text-embedding-ada-002",

examples/openai_example/function_calling.py

@@ -3,11 +3,11 @@
 from dotenv import find_dotenv, load_dotenv
 from openai import OpenAI
 
-from langtrace_python_sdk import langtrace
+from langtrace_python_sdk import langtrace, with_langtrace_root_span
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(write_spans_to_console=True)
+langtrace.init(write_spans_to_console=False)
 
 
 client = OpenAI()
@@ -34,6 +34,7 @@ student_custom_functions = [
 ]
 
 
+@with_langtrace_root_span("Function Calling")
 def function_calling():
     response = client.chat.completions.create(
         model="gpt-3.5-turbo",

examples/openai_example/images_edit.py

@@ -0,0 +1,40 @@
+from dotenv import find_dotenv, load_dotenv
+from openai import OpenAI
+
+from langtrace_python_sdk import langtrace
+
+# from PIL import Image
+
+
+_ = load_dotenv(find_dotenv())
+
+langtrace.init(write_spans_to_console=False)
+
+
+client = OpenAI()
+
+
+# use this to convert the image to RGBA
+# def convert_to_rgba():
+#     Image.open("./resources/image.png").convert("RGBA").save("./resources/image_with_alpha.png")
+
+
+def image_edit():
+
+    response = client.images.edit(
+        model="dall-e-2",
+        image=open("./resources/lounge_flamingo.png", "rb"),
+        mask=open("./resources/mask.png", "rb"),
+        prompt="A sunlit indoor lounge area with a pool and duck standing in side with flamingo.",
+        n=1,
+        size="1024x1024",
+        response_format="url",
+    )
+
+    image_url = response.data[0].url
+    print(image_url)
+    print(response)
+
+
+# image_edit()
+# convert_to_rgba()

examples/pinecone_example/__init__.py

@@ -0,0 +1,9 @@
+from langtrace_python_sdk import with_langtrace_root_span
+
+
+class PineconeRunner:
+    @with_langtrace_root_span("Pinecone")
+    def run(self):
+        from .basic import basic as basic_app
+
+        basic_app()

examples/pinecone_example/basic.py

@@ -6,7 +6,11 @@ from dotenv import find_dotenv, load_dotenv
 from openai import OpenAI
 from pinecone import Pinecone, ServerlessSpec
 
-from langtrace_python_sdk import langtrace, with_langtrace_root_span
+from langtrace_python_sdk import (
+    langtrace,
+    with_langtrace_root_span,
+    with_additional_attributes,
+)
 from langtrace_python_sdk.utils.with_root_span import SendUserFeedback
 
 _ = load_dotenv(find_dotenv())
@@ -28,7 +32,8 @@ def create_index():
     )
 
 
-@with_langtrace_root_span()
+@with_additional_attributes({"db.embedding_model": "text-embedding-ada-002"})
+@with_langtrace_root_span("Pinecone Basic")
 def basic(span_id=None, trace_id=None):
 
     result = client.embeddings.create(

examples/qdrant_example/__init__.py

@@ -0,0 +1,8 @@
+from langtrace_python_sdk import with_langtrace_root_span
+from .basic import basic as basic_app
+
+
+class QdrantRunner:
+    @with_langtrace_root_span("Qdrant")
+    def run(self):
+        basic_app()

examples/weaviate_example/__init__.py

@@ -0,0 +1,26 @@
+from .query_text import (
+    create,
+    insert,
+    query_data_bm25,
+    query_fetch_object_by_id,
+    query_fetch_objects,
+    query_hybrid,
+    query_near_object,
+    query_data_near_text,
+    query_data_near_vector,
+)
+from langtrace_python_sdk import with_langtrace_root_span
+
+
+class WeaviateRunner:
+    @with_langtrace_root_span("Weaviate")
+    def run(self):
+        create()
+        insert()
+        query_data_bm25()
+        # query_fetch_object_by_id()
+        # query_fetch_objects()
+        # query_hybrid()
+        # query_near_object()
+        # query_data_near_text()
+        # query_data_near_vector()

examples/weaviate_example/query_text.py

@@ -21,7 +21,9 @@ from weaviate.collections.classes.grpc import Move
 
 import langtrace_python_sdk.langtrace as langtrace
 from langtrace_python_sdk import with_langtrace_root_span
+from dotenv import load_dotenv
 
+load_dotenv()
 # Set these environment variables
 WCS_DEMO_URL = os.environ["WCS_DEMO_URL"]
 WCS_DEMO_RO_KEY = os.environ["WCS_DEMO_RO_KEY"]
@@ -37,17 +39,19 @@ client = weaviate.connect_to_wcs(
 langtrace.init()
 
 
-@with_langtrace_root_span()
+@with_langtrace_root_span("create")
 def create():
-    questions = client.collections.create(
-        name="Question",
-        # Set the vectorizer to "text2vec-openai" to use the OpenAI API for vector-related operations
-        vectorizer_config=wvc.config.Configure.Vectorizer.text2vec_openai(),
-        # Ensure the `generative-openai` module is used for generative queries
-        generative_config=wvc.config.Configure.Generative.openai(),
-    )
+    if not client.collections.get("Question"):
+        questions = client.collections.create(
+            name="Question",
+            # Set the vectorizer to "text2vec-openai" to use the OpenAI API for vector-related operations
+            vectorizer_config=wvc.config.Configure.Vectorizer.text2vec_openai(),
+            # Ensure the `generative-openai` module is used for generative queries
+            generative_config=wvc.config.Configure.Generative.openai(),
+        )
 
 
+@with_langtrace_root_span("insert")
 def insert():
     resp = requests.get(
         "https://raw.githubusercontent.com/weaviate-tutorials/quickstart/main/data/jeopardy_tiny.json"

langtrace_python_sdk/__init__.py

@@ -15,13 +15,18 @@ limitations under the License.
 """
 
 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,
+)
+
 from langtrace_python_sdk.utils.prompt_registry import get_prompt_from_registry
 from langtrace_python_sdk.utils.with_root_span import SendUserFeedback
 
 __all__ = [
     "langtrace",
     "with_langtrace_root_span",
+    "with_additional_attributes",
     "get_prompt_from_registry",
     "SendUserFeedback",
 ]

langtrace_python_sdk/constants/instrumentation/openai.py

@@ -40,6 +40,10 @@ APIS = {
         "METHOD": OpenAIMethods.IMAGES_GENERATION.value,
         "ENDPOINT": "/images/generations",
     },
+    "IMAGES_EDIT": {
+        "METHOD": OpenAIMethods.IMAGES_EDIT.value,
+        "ENDPOINT": "/images/edits",
+    },
     "EMBEDDINGS_CREATE": {
         "METHOD": OpenAIMethods.EMBEDDINGS_CREATE.value,
         "ENDPOINT": "/embeddings",

langtrace_python_sdk/instrumentation/openai/instrumentation.py

@@ -28,6 +28,7 @@ from langtrace_python_sdk.instrumentation.openai.patch import (
     async_images_generate,
     chat_completions_create,
     embeddings_create,
+    images_edit,
     images_generate,
 )
 
@@ -37,7 +38,7 @@ logging.basicConfig(level=logging.FATAL)
 class OpenAIInstrumentation(BaseInstrumentor):
 
     def instrumentation_dependencies(self) -> Collection[str]:
-        return ["openai >= 0.27.0"]
+        return ["openai >= 0.27.0", "trace-attributes >= 4.0.5"]
 
     def _instrument(self, **kwargs):
         tracer_provider = kwargs.get("tracer_provider")
@@ -69,6 +70,13 @@ class OpenAIInstrumentation(BaseInstrumentor):
             "AsyncImages.generate",
             async_images_generate("openai.images.generate", version, tracer),
         )
+
+        wrap_function_wrapper(
+            "openai.resources.images",
+            "Images.edit",
+            images_edit("openai.images.edit", version, tracer),
+        )
+
         wrap_function_wrapper(
             "openai.resources.embeddings",
             "Embeddings.create",

langtrace_python_sdk/instrumentation/openai/patch.py

@@ -16,20 +16,19 @@ limitations under the License.
 
 import json
 
+from importlib_metadata import version as v
 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 import LANGTRACE_SDK_NAME
 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 importlib_metadata import version as v
-
-from langtrace_python_sdk.constants import LANGTRACE_SDK_NAME
 
 
 def images_generate(original_method, version, tracer):
@@ -187,6 +186,89 @@ def async_images_generate(original_method, version, tracer):
     return traced_method
 
 
+def images_edit(original_method, version, tracer):
+    """
+    Wrap the `edit` method of the `Images` class to trace it.
+    """
+
+    def traced_method(wrapped, instance, args, kwargs):
+        base_url = (
+            str(instance._client._base_url)
+            if hasattr(instance, "_client") and hasattr(instance._client, "_base_url")
+            else ""
+        )
+        service_provider = SERVICE_PROVIDERS["OPENAI"]
+        extra_attributes = baggage.get_baggage(LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY)
+
+        span_attributes = {
+            "langtrace.sdk.name": "langtrace-python-sdk",
+            "langtrace.service.name": service_provider,
+            "langtrace.service.type": "llm",
+            "langtrace.service.version": version,
+            "langtrace.version": v(LANGTRACE_SDK_NAME),
+            "url.full": base_url,
+            "llm.api": APIS["IMAGES_EDIT"]["ENDPOINT"],
+            "llm.model": kwargs.get("model"),
+            "llm.response_format": kwargs.get("response_format"),
+            "llm.image.size": kwargs.get("size"),
+            "llm.prompts": json.dumps(
+                [
+                    {
+                        "role": kwargs.get("user", "user"),
+                        "content": kwargs.get("prompt", []),
+                    }
+                ]
+            ),
+            "llm.top_k": kwargs.get("n"),
+            **(extra_attributes if extra_attributes is not None else {}),
+        }
+
+        attributes = LLMSpanAttributes(**span_attributes)
+
+        with tracer.start_as_current_span(
+            APIS["IMAGES_EDIT"]["METHOD"], kind=SpanKind.CLIENT
+        ) as span:
+            for field, value in attributes.model_dump(by_alias=True).items():
+                if value is not None:
+                    span.set_attribute(field, value)
+            try:
+                # Attempt to call the original method
+                result = wrapped(*args, **kwargs)
+
+                response = []
+                # Parse each image object
+                for each_data in result.data:
+                    response.append(
+                        {
+                            "role": "assistant",
+                            "content": {
+                                "url": each_data.url,
+                                "revised_prompt": each_data.revised_prompt,
+                                "base64": each_data.b64_json,
+                            },
+                        }
+                    )
+
+                span.add_event(
+                    name="response",
+                    attributes={"llm.responses": json.dumps(response)},
+                )
+
+                span.set_status(StatusCode.OK)
+                return result
+            except Exception as err:
+                # Record the exception in the span
+                span.record_exception(err)
+
+                # Set the span status to indicate an error
+                span.set_status(Status(StatusCode.ERROR, str(err)))
+
+                # Reraise the exception to ensure it's not swallowed
+                raise
+
+    return traced_method
+
+
 def chat_completions_create(original_method, version, tracer):
     """Wrap the `create` method of the `ChatCompletion` class to trace it."""
 

langtrace_python_sdk/version.py

@@ -1 +1 @@
-__version__ = "2.1.12"
+__version__ = "2.1.14"

{langtrace_python_sdk-2.1.12.dist-info → langtrace_python_sdk-2.1.14.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: langtrace-python-sdk
-Version: 2.1.12
+Version: 2.1.14
 Summary: Python SDK for LangTrace
 Project-URL: Homepage, https://github.com/Scale3-Labs/langtrace-python-sdk
 Author-email: Scale3 Labs <engineering@scale3labs.com>
@@ -19,7 +19,7 @@ Requires-Dist: opentelemetry-instrumentation>=0.46b0
 Requires-Dist: opentelemetry-sdk>=1.25.0
 Requires-Dist: sqlalchemy
 Requires-Dist: tiktoken>=0.1.1
-Requires-Dist: trace-attributes>=4.0.4
+Requires-Dist: trace-attributes<6.0.0,>=5.0.0
 Provides-Extra: dev
 Requires-Dist: anthropic; extra == 'dev'
 Requires-Dist: chromadb; extra == 'dev'
@@ -193,7 +193,7 @@ langtrace.init(custom_remote_exporter=<your_exporter>, batch=<True or False>)
 - `@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
+from langtrace_python_sdk import with_langtrace_root_span
 
 @with_langtrace_root_span()
 def example():
@@ -208,10 +208,8 @@ def example():
 - `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,
-)
+from langtrace_python_sdk import with_langtrace_root_span, with_additional_attributes
+
 
 @with_additional_attributes({"user.id": "1234", "user.feedback.rating": 1})
 def api_call1():
@@ -277,6 +275,9 @@ Langtrace automatically captures traces from the following vendors:
 
 We welcome contributions to this project. To get started, fork this repository and start developing. To get involved, join our [Discord](https://discord.langtrace.ai) workspace.
 
+If you want to run any of the examples go to `run_example.py` file, you will find `ENABLED_EXAMPLES`. choose the example you want to run and just toggle the flag to `True` and run the file using `python src/run_example.py`
+
+
 ---
 
 ## Security

{langtrace_python_sdk-2.1.12.dist-info → langtrace_python_sdk-2.1.14.dist-info}/RECORD

@@ -1,44 +1,49 @@
-examples/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-examples/anthropic_example/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-examples/anthropic_example/completion.py,sha256=PVxAth2vqbX_QN2DGVOt2mTGuUiGk4O5w7RFhHzsxA8,714
-examples/chroma_example/__init__.py,sha256=Tq6pae7fHA7dwuDslabB5MTNedL21gu2RaZicpxSyLU,25
-examples/chroma_example/basic.py,sha256=qngW9EQtIeuBFjyWdPPiJAE1xnVimKqkglJQz9cqueM,867
-examples/cohere_example/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-examples/cohere_example/chat.py,sha256=0iNVE3sxX6fZBSRozzx9a8M7HI4QtP1BIjNguBO32rw,915
-examples/cohere_example/chat_stream.py,sha256=9hT5whG_W002jxHSXgpQR6imaoQJ10ydte3cFuXc8fE,638
-examples/cohere_example/embed.py,sha256=9yNyl2_aJun-0n1v4B_H6PlCeO7LQDIuMagRzwC3OWs,523
-examples/cohere_example/rerank.py,sha256=Zy1H7TwXHOGulYAMkfOe-hB0YmJIfwfPurYxj__l_98,1145
-examples/cohere_example/tools.py,sha256=I_uCDHQiLRD4uCmeXRUTYNtTgO_efy809Pik1j-Usfw,1683
-examples/fastapi_example/basic_route.py,sha256=VXb9r8usZRqQOwS4bzonsP0jFYhvfZecbK8zYcOxAl4,1125
+examples/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+examples/anthropic_example/__init__.py,sha256=03us1YuvAJR6fqXX8NH2kROBfTmyz7KzFVY01UlL-tM,237
+examples/anthropic_example/completion.py,sha256=LXcuXxRx9hy4b9Woq0aCBx_PS0QYmYi2bT2N96CgS_Q,686
+examples/chroma_example/__init__.py,sha256=Mrf8KptW1hhzu6WDdRRTxbaB-0kM7x5u-Goc_zR7G5c,203
+examples/chroma_example/basic.py,sha256=oO7-__8HptnFXLVKbnPgoz02yM-CAPi721xsbUb_gYg,868
+examples/cohere_example/__init__.py,sha256=oYpsnS-dKOlWLkPEUWhXxi1vfxa77bt_DOdkJHg__7g,502
+examples/cohere_example/chat.py,sha256=1CPrqr45UmhByxAn9on0L06ixVqS3jc9tLT88pduTzw,888
+examples/cohere_example/chat_stream.py,sha256=daWGrj2yucN9sWKGiPzTw18clh23Grv-MIJT_OpUkpA,611
+examples/cohere_example/embed.py,sha256=p9BJvOg09JVb8BfTCb63v3uh_wOsi_OyrCAJdXXrE6E,496
+examples/cohere_example/rerank.py,sha256=e7OU0A2FzfiQDuOmCy3Kg5LLNYGRmRIK5LqeLnTWlP4,1118
+examples/cohere_example/tools.py,sha256=a5uvS058tcwU6PJbF9EDO6LPVmPj2LoW4Vn8Web3Iq8,1656
+examples/fastapi_example/__init__.py,sha256=INIfvJP7zC_KkJCtulS1qbh61-MJTPAHnzAgzeKi0yU,87
+examples/fastapi_example/basic_route.py,sha256=_IRXjkOtJQ-bTIGa1WbvUF_2LF4bjghjyXt4YrHaRvw,1170
 examples/hiveagent_example/basic.py,sha256=Sd7I5w8w5Xx7ODaydTY30yiq9HwJDMKHQywrZjgehP0,441
-examples/langchain_example/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-examples/langchain_example/basic.py,sha256=mSVcE-Kx7jNS4U_zrm5WZI98xyw6WTrhd2GMVGgn9Cg,2653
+examples/langchain_example/__init__.py,sha256=t28923ZDDejZabsvINnYwpLyb63WFwPde-Nmzdg-kH4,254
+examples/langchain_example/basic.py,sha256=988C40-bcMq-AM7xL_4v6FHgYliQL1SJM_YcLY-3WMM,2661
 examples/langchain_example/groq_example.py,sha256=9dBbOWZCVJ29PpB1NbJhc9MThmtwrlOpdNIxPl7d8f8,787
 examples/langchain_example/langgraph_example.py,sha256=7C2a4Sg0PKbbab03CVkStO3MzT7C-O1UtdmObvBXurM,2005
 examples/langchain_example/tool.py,sha256=8T8_IDbgA58XbsfyH5_xhA8ZKQfyfyFxF8wor-PsRjA,2556
-examples/llamaindex_example/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-examples/llamaindex_example/agent.py,sha256=41uxHpMKqIB4irD8YHu6Y0QI3pDfz79huUHroQwhfDA,2774
-examples/llamaindex_example/basic.py,sha256=kcl2RykIzq1Lv9KePzp_TCcdp_PFXxt3vsspb5p-5HI,651
+examples/llamaindex_example/__init__.py,sha256=4w8Hz5pfmMzhkHAbBim6jwxHxMicaN4xi1Of9pODO8g,252
+examples/llamaindex_example/agent.py,sha256=AukhBJwPC5_KFYQrDbEE00c-_6Yt2SCPZHii5B3mh-o,2795
+examples/llamaindex_example/basic.py,sha256=bDp2TlnfzviS9wi6ox-vfPo3e79QuStHD0sJJQXr_w0,656
 examples/llamaindex_example/data/abramov.txt,sha256=Ou-GyWZm5AjHLgxviBoRE9ikNv5MScsF0cd--0vVVhI,32667
-examples/openai_example/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-examples/openai_example/async_tool_calling_nonstreaming.py,sha256=7mP6Anonj7B8owPBfbhxyux4mdgp3XCuym1xAx6SLls,3874
-examples/openai_example/async_tool_calling_streaming.py,sha256=PQMHSWAN1DfgjK88b8LVSredmov0uH9Fgn1FHoCVv1M,7054
-examples/openai_example/chat_completion.py,sha256=f2rMHJU1Dph_fAOv9qCksvtldl1s9UitEeLsvyoSpGA,1253
-examples/openai_example/embeddings_create.py,sha256=d05rArTWxsmCxOdTRaGw3n9Yt1HXxOHHCtciCHIwBGc,502
-examples/openai_example/function_calling.py,sha256=rrnrFw2HyV_gv4TdwUr299baP4tIupz6EQWWmLDtJkU,2265
+examples/openai_example/__init__.py,sha256=6l2FePb3vYBJVwIT4qfdtOLobKLZ6iO16Fe31TUwNz8,827
+examples/openai_example/async_tool_calling_nonstreaming.py,sha256=FhbDX91RLhTrNd5IGPI0CxRdCjTRcNLlWkYdLqHGbVs,3838
+examples/openai_example/async_tool_calling_streaming.py,sha256=LaSKmn_Unv55eTHXYdEmKjo39eNuB3ASOBV-m8U1HfU,7136
+examples/openai_example/chat_completion.py,sha256=4JUv8VPcXTfYydCFxPOE3Nb5jYlbzJlvcErB8qmMano,1267
+examples/openai_example/embeddings_create.py,sha256=sqz-zh3pEaB17gVYeY0QE76QxRwsmo6wV8yjAfg1ljg,522
+examples/openai_example/function_calling.py,sha256=zz-JdCcpP7uCXG21EYXF1Y39IKj6gYt2fOP5N_ywpnc,2338
+examples/openai_example/images_edit.py,sha256=0hG5OYTSUBrcwSxbqN67brPYjf1pRJs8vqGzu-UjK6c,887
 examples/openai_example/images_generate.py,sha256=SZNY8Visk7JUpx5QhNxTNINHmPAGdCUayF-Q7_iCr50,470
 examples/openai_example/tool_calling.py,sha256=_IV7KoSI_37u1TTZWdVa58BYjkDfhSurvM86xwaNNhY,2316
 examples/openai_example/tool_calling_nonstreaming.py,sha256=Yc848IooZRXNynHL6z0kOgJ4qbmL_NOufcb2VmWRukI,3847
 examples/openai_example/tool_calling_streaming.py,sha256=mV1RbyAoVhumGRPpqPWQ6PMhnJyeifrlELd2-K1qJ_w,7015
+examples/openai_example/resources/lounge_flamingo.png,sha256=aspniTtmWqwLp3YUhYqAe2ze8nJaq-bTSW7uUJudtd0,2416234
+examples/openai_example/resources/mask.png,sha256=mUE9Dfp-x8jI0Nh4WGr0P9pueUqEZfpjwxR-6Rxzxz4,2483660
 examples/perplexity_example/basic.py,sha256=bp7n27gaugJkaFVyt8pjaEfi66lYcqP6eFFjPewUShY,668
-examples/pinecone_example/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-examples/pinecone_example/basic.py,sha256=gkb1uQwKFLnWmRuM2KGvAeTpT1AYlHmz2kRey5_-Dqs,1437
-examples/qdrant_example/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+examples/pinecone_example/__init__.py,sha256=98UBOq9wQdSw6JK7Hfh_AUDVZzL76IAs-kV4sktoDYg,210
+examples/pinecone_example/basic.py,sha256=RpcdCrpNEExad3nmagkF5M211PgNoqu5j_edX8jw2No,1576
+examples/qdrant_example/__init__.py,sha256=Ze9xEzW8FiHUO58YBa8JeHNOwcmo3dpYH77AkdyglKU,197
 examples/qdrant_example/basic.py,sha256=DCMjHSuBZKkhEjCkwy5d5La9WMyW0lCWqtcZWiFCEm4,1425
-examples/weaviate_example/query_text.py,sha256=iE9OiHsibjsprbCGzabE03eZsGN06e6ym2iS1A9P3ig,64650
-langtrace_python_sdk/__init__.py,sha256=FuvyRuStRe_N2wo2SB2_ZQ0w7LGNIjV0lLi6S1IgGwY,958
+examples/weaviate_example/__init__.py,sha256=8JMDBsRSEV10HfTd-YC7xb4txBjD3la56snk-Bbg2Kw,618
+examples/weaviate_example/query_text.py,sha256=qz9o-fTDzX5AW5m8BJF-TfmBdokxh492NfnmnPUMU3s,64814
+langtrace_python_sdk/__init__.py,sha256=rted5SGmbbJyqFzT4_Sjfn-Eie2rQwAULa_jMJldAmE,1034
 langtrace_python_sdk/langtrace.py,sha256=pN-xJRXrtvJIenMOH0-xlNXcnqL9qMjg28SrW-PMRU0,6978
-langtrace_python_sdk/version.py,sha256=2ZypnXz9pLMxwCwfHNJV9M24HPgZwt56lxlCb2BYeEA,23
+langtrace_python_sdk/version.py,sha256=otJNk418s7IUoP3b0ZQy4k9fA8qhpkQusp3zKCHD2bQ,23
 langtrace_python_sdk/constants/__init__.py,sha256=P8QvYwt5czUNDZsKS64vxm9Dc41ptGbuF1TFtAF6nv4,44
 langtrace_python_sdk/constants/exporter/langtrace_exporter.py,sha256=5MNjnAOg-4am78J3gVMH6FSwq5N8TOj72ugkhsw4vi0,46
 langtrace_python_sdk/constants/instrumentation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -47,7 +52,7 @@ langtrace_python_sdk/constants/instrumentation/chroma.py,sha256=hiPGYdHS0Yj4Kh3e
 langtrace_python_sdk/constants/instrumentation/cohere.py,sha256=tf9sDfb5K3qOAHChEE5o8eYWPZ1io58VsOjZDCZPxfw,577
 langtrace_python_sdk/constants/instrumentation/common.py,sha256=TCHX8uqIJD2p9sONGrlX6Q3b5Mj859PTP-c729ogKcY,782
 langtrace_python_sdk/constants/instrumentation/groq.py,sha256=VFXmIl4aqGY_fS0PAmjPj_Qm7Tibxbx7Ur_e7rQpqXc,134
-langtrace_python_sdk/constants/instrumentation/openai.py,sha256=9VF6ic9Ed3bpSvdp6iNmrpx2Ppo6DPav3hoUcqSQSv0,1048
+langtrace_python_sdk/constants/instrumentation/openai.py,sha256=uEOH5UXapU2DSf2AdgXTRhhJEHGWXUNFkUGD5QafflM,1164
 langtrace_python_sdk/constants/instrumentation/pinecone.py,sha256=Xaqqw-xBO0JJLGk75hiCUQGztNm0HiVaLQvjtYK7VJM,472
 langtrace_python_sdk/constants/instrumentation/qdrant.py,sha256=yL7BopNQTXW7L7Z-gVM2PdusKD7r9qqcATvczFd7NtQ,1999
 langtrace_python_sdk/constants/instrumentation/weaviate.py,sha256=Iytf2OpB_irZYEmvOQ7Pf483EdG5Bh59GxaBlXck0yY,1501
@@ -82,8 +87,8 @@ langtrace_python_sdk/instrumentation/llamaindex/__init__.py,sha256=rHvuqpuQKLj57
 langtrace_python_sdk/instrumentation/llamaindex/instrumentation.py,sha256=8iAg-Oxwf2W4S60qRfO5mvzORYxublgq7FdGWqUB4q8,2965
 langtrace_python_sdk/instrumentation/llamaindex/patch.py,sha256=LwnI9DCC5wcnixPUOmwIWSE7ryTk1J0xFBYdJja8t_Q,4324
 langtrace_python_sdk/instrumentation/openai/__init__.py,sha256=VPHRNCQEdkizIVP2d0Uw_a7t8XOTSTprEIB8oboJFbs,95
-langtrace_python_sdk/instrumentation/openai/instrumentation.py,sha256=G2HSZfr6DuP2n-8v0h91M60m0DJWFBcru4-1FQJl-5A,2726
-langtrace_python_sdk/instrumentation/openai/patch.py,sha256=Og6_W3MqnNrzOfQ_WBLnCQUy0xsNZol1bsoScajiI2E,37479
+langtrace_python_sdk/instrumentation/openai/instrumentation.py,sha256=A0BJHRLcZ74TNVg6I0I9M5YWvSpAtXwMmME6N5CEQ_M,2945
+langtrace_python_sdk/instrumentation/openai/patch.py,sha256=ugf3y-20YVqJmHsheQaL3EEy7FPbOgEMtcG5dOYjxiE,40553
 langtrace_python_sdk/instrumentation/pinecone/__init__.py,sha256=DzXyGh9_MGWveJvXULkFwdkf7PbG2s3bAWtT1Dmz7Ok,99
 langtrace_python_sdk/instrumentation/pinecone/instrumentation.py,sha256=mxQXe3oAOPLsMJGlEzAe6zpgK7RtWfqmcNmGW_gQXX4,1900
 langtrace_python_sdk/instrumentation/pinecone/patch.py,sha256=nZUHZ_1HfLjByJS0gkvrDRj69JbdnfWpezk6zl1EMvU,5000
@@ -138,7 +143,7 @@ tests/pinecone/cassettes/test_query.yaml,sha256=b5v9G3ssUy00oG63PlFUR3JErF2Js-5A
 tests/pinecone/cassettes/test_upsert.yaml,sha256=neWmQ1v3d03V8WoLl8FoFeeCYImb8pxlJBWnFd_lITU,38607
 tests/qdrant/conftest.py,sha256=9n0uHxxIjWk9fbYc4bx-uP8lSAgLBVx-cV9UjnsyCHM,381
 tests/qdrant/test_qdrant.py,sha256=pzjAjVY2kmsmGfrI2Gs2xrolfuaNHz7l1fqGQCjp5_o,3353
-langtrace_python_sdk-2.1.12.dist-info/METADATA,sha256=Pg_yY_2SiEBShRlaXAHaTrLPcQtvPRkSmiu7EqIitKA,11860
-langtrace_python_sdk-2.1.12.dist-info/WHEEL,sha256=zEMcRr9Kr03x1ozGwg5v9NQBKn3kndp6LSoSlVg-jhU,87
-langtrace_python_sdk-2.1.12.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-langtrace_python_sdk-2.1.12.dist-info/RECORD,,
+langtrace_python_sdk-2.1.14.dist-info/METADATA,sha256=cPyxda06xLH5QI-6uAMqaifqB3zQN02e1qFzeY9kXPk,12038
+langtrace_python_sdk-2.1.14.dist-info/WHEEL,sha256=zEMcRr9Kr03x1ozGwg5v9NQBKn3kndp6LSoSlVg-jhU,87
+langtrace_python_sdk-2.1.14.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+langtrace_python_sdk-2.1.14.dist-info/RECORD,,