langtrace-python-sdk 1.3.1__tar.gz → 1.3.3__tar.gz

{langtrace_python_sdk-1.3.1 → langtrace_python_sdk-1.3.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: langtrace-python-sdk
-Version: 1.3.1
+Version: 1.3.3
 Summary: Python SDK for LangTrace
 Project-URL: Homepage, https://github.com/Scale3-Labs/langtrace-python-sdk
 Author-email: Scale3 Labs <engineering@scale3labs.com>
@@ -19,6 +19,7 @@ Requires-Dist: trace-attributes
 Provides-Extra: dev
 Requires-Dist: anthropic; extra == 'dev'
 Requires-Dist: chromadb; extra == 'dev'
+Requires-Dist: cohere; extra == 'dev'
 Requires-Dist: langchain; extra == 'dev'
 Requires-Dist: langchain-openai; extra == 'dev'
 Requires-Dist: llama-index; extra == 'dev'
@@ -144,7 +145,7 @@ pip install langtrace-python-sdk
 
 ``` python
 from langtrace_python_sdk import langtrace # Must precede any llm module imports
-langtrace.init(write_to_langtrace_cloud=False, batch=False)
+langtrace.init(write_to_langtrace_cloud=False)
 ```
 
 ## Langtrace self hosted custom exporter

{langtrace_python_sdk-1.3.1 → langtrace_python_sdk-1.3.3}/README.md

@@ -116,7 +116,7 @@ pip install langtrace-python-sdk
 
 ``` python
 from langtrace_python_sdk import langtrace # Must precede any llm module imports
-langtrace.init(write_to_langtrace_cloud=False, batch=False)
+langtrace.init(write_to_langtrace_cloud=False)
 ```
 
 ## Langtrace self hosted custom exporter

{langtrace_python_sdk-1.3.1 → langtrace_python_sdk-1.3.3}/pyproject.toml

@@ -37,7 +37,8 @@ dev = [
     "langchain-openai",
     "llama-index",
     "langchain-openai",
-    "chromadb"
+    "chromadb",
+    "cohere"
 ]
 
 

{langtrace_python_sdk-1.3.1 → langtrace_python_sdk-1.3.3}/src/examples/anthropic_example/completion.py

@@ -8,7 +8,7 @@ from langtrace_python_sdk import with_langtrace_root_span
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(batch=False, debug_log_to_console=True, write_to_langtrace_cloud=False)
+langtrace.init(write_to_langtrace_cloud=False)
 
 @with_langtrace_root_span("messages_create")
 def messages_create():

{langtrace_python_sdk-1.3.1 → langtrace_python_sdk-1.3.3}/src/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(batch=False, debug_log_to_console=True, write_to_langtrace_cloud=False)
+langtrace.init(write_to_langtrace_cloud=False)
 
 
 @with_langtrace_root_span()

{langtrace_python_sdk-1.3.1 → langtrace_python_sdk-1.3.3}/src/examples/cohere_example/chat.py

@@ -6,7 +6,7 @@ from langtrace_python_sdk import langtrace
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(batch=False, debug_log_to_console=True, write_to_langtrace_cloud=False)
+langtrace.init(write_to_langtrace_cloud=False)
 
 co = cohere.Client()
 

{langtrace_python_sdk-1.3.1 → langtrace_python_sdk-1.3.3}/src/examples/cohere_example/chat_stream.py

@@ -2,11 +2,12 @@ from dotenv import find_dotenv, load_dotenv
 import cohere
 
 from langtrace_python_sdk import langtrace
+
 # from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(batch=False, debug_log_to_console=True, write_to_langtrace_cloud=False)
+langtrace.init(write_to_langtrace_cloud=False)
 
 co = cohere.Client()
 

{langtrace_python_sdk-1.3.1 → langtrace_python_sdk-1.3.3}/src/examples/cohere_example/embed_create.py

@@ -2,11 +2,12 @@ from dotenv import find_dotenv, load_dotenv
 import cohere
 
 from langtrace_python_sdk import langtrace
+
 # from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(batch=False, debug_log_to_console=True, write_to_langtrace_cloud=False)
+langtrace.init(write_to_langtrace_cloud=False)
 
 co = cohere.Client()
 
@@ -14,8 +15,8 @@ co = cohere.Client()
 # @with_langtrace_root_span("embed_create")
 def embed_create():
     response = co.embed(
-        texts=['hello', 'goodbye'],
-        model='embed-english-v3.0',
-        input_type='classification'
+        texts=["hello", "goodbye"],
+        model="embed-english-v3.0",
+        input_type="classification",
     )
     # print(response)

{langtrace_python_sdk-1.3.1 → langtrace_python_sdk-1.3.3}/src/examples/fastapi_example/basic_route.py

@@ -8,14 +8,16 @@ from openai import OpenAI
 
 from langtrace_python_sdk import langtrace
 
-langtrace.init(write_to_langtrace_cloud=False, debug_log_to_console=True)
+langtrace.init(write_to_langtrace_cloud=False)
 app = FastAPI()
 client = OpenAI()
 
+
 @app.get("/")
 def root():
     vectorstore = FAISS.from_texts(
-        ["Langtrace helps you ship high quality AI Apps to production."], embedding=OpenAIEmbeddings()
+        ["Langtrace helps you ship high quality AI Apps to production."],
+        embedding=OpenAIEmbeddings(),
     )
     retriever = vectorstore.as_retriever()
 

langtrace_python_sdk-1.3.3/src/examples/hiveagent_example/basic.py

@@ -0,0 +1,23 @@
+from langtrace_python_sdk import langtrace
+from hive_agent import HiveAgent
+from dotenv import load_dotenv
+from openai import OpenAI
+
+load_dotenv()
+
+langtrace.init(
+    write_to_langtrace_cloud=False,
+    api_host="http://localhost:3000",
+)
+
+client = OpenAI()
+
+
+def basic():
+    my_agent = HiveAgent(
+        name="my_agent",
+        functions=[],
+        instruction="your instructions for this agent's goal",
+    )
+
+    my_agent.run_server()

langtrace_python_sdk-1.3.3/src/examples/llamaindex_example/agent.py

@@ -0,0 +1,86 @@
+import json
+from typing import Sequence, List
+
+from llama_index.agent.openai import OpenAIAgent
+from llama_index.llms.openai import OpenAI
+from llama_index.core.llms import ChatMessage
+from llama_index.core.tools import BaseTool, FunctionTool
+from openai.types.chat import ChatCompletionMessageToolCall
+from langtrace_python_sdk import langtrace
+
+import nest_asyncio
+
+nest_asyncio.apply()
+
+langtrace.init(write_to_langtrace_cloud=False)
+
+
+def multiply(a: int, b: int) -> int:
+    """Multiple two integers and returns the result integer"""
+    return a * b
+
+
+multiply_tool = FunctionTool.from_defaults(fn=multiply)
+
+
+def add(a: int, b: int) -> int:
+    """Add two integers and returns the result integer"""
+    return a + b
+
+
+add_tool = FunctionTool.from_defaults(fn=add)
+
+
+class YourOpenAIAgent:
+    def __init__(
+        self,
+        tools: Sequence[BaseTool] = [],
+        llm: OpenAI = OpenAI(temperature=0, model="gpt-3.5-turbo-0613"),
+        chat_history: List[ChatMessage] = [],
+    ) -> None:
+        self._llm = llm
+        self._tools = {tool.metadata.name: tool for tool in tools}
+        self._chat_history = chat_history
+
+    def reset(self) -> None:
+        self._chat_history = []
+
+    def chat(self, message: str) -> str:
+        chat_history = self._chat_history
+        chat_history.append(ChatMessage(role="user", content=message))
+        tools = [tool.metadata.to_openai_tool() for _, tool in self._tools.items()]
+
+        ai_message = self._llm.chat(chat_history, tools=tools).message
+        additional_kwargs = ai_message.additional_kwargs
+        chat_history.append(ai_message)
+
+        tool_calls = additional_kwargs.get("tool_calls", None)
+        # parallel function calling is now supported
+        if tool_calls is not None:
+            for tool_call in tool_calls:
+                function_message = self._call_function(tool_call)
+                chat_history.append(function_message)
+                ai_message = self._llm.chat(chat_history).message
+                chat_history.append(ai_message)
+
+        return ai_message.content
+
+    def _call_function(self, tool_call: ChatCompletionMessageToolCall) -> ChatMessage:
+        id_ = tool_call.id
+        function_call = tool_call.function
+        tool = self._tools[function_call.name]
+        output = tool(**json.loads(function_call.arguments))
+        return ChatMessage(
+            name=function_call.name,
+            content=str(output),
+            role="tool",
+            additional_kwargs={
+                "tool_call_id": id_,
+                "name": function_call.name,
+            },
+        )
+
+
+# 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)

{langtrace_python_sdk-1.3.1 → langtrace_python_sdk-1.3.3}/src/examples/llamaindex_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(batch=False, debug_log_to_console=True, write_to_langtrace_cloud=False)
+langtrace.init(write_to_langtrace_cloud=False)
 
 
 @with_langtrace_root_span()

{langtrace_python_sdk-1.3.1 → langtrace_python_sdk-1.3.3}/src/examples/openai/chat_completion.py

@@ -7,7 +7,7 @@ from langtrace_python_sdk.utils.with_root_span import (
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init()
+langtrace.init(write_to_langtrace_cloud=True)
 client = OpenAI()
 
 

{langtrace_python_sdk-1.3.1 → langtrace_python_sdk-1.3.3}/src/examples/openai/function_calling.py

@@ -8,7 +8,7 @@ from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(batch=False, debug_log_to_console=True, write_to_langtrace_cloud=False)
+langtrace.init(write_to_langtrace_cloud=False)
 
 client = OpenAI()
 

{langtrace_python_sdk-1.3.1 → langtrace_python_sdk-1.3.3}/src/examples/perplexity_example/basic.py

@@ -2,11 +2,13 @@ from openai import OpenAI
 
 from langtrace_python_sdk import langtrace
 from langtrace_python_sdk.utils.with_root_span import (
-    with_additional_attributes, with_langtrace_root_span)
+    with_additional_attributes,
+    with_langtrace_root_span,
+)
 
 # _ = load_dotenv(find_dotenv())
 
-langtrace.init(batch=False, write_to_langtrace_cloud=False, debug_log_to_console=True)
+langtrace.init(write_to_langtrace_cloud=False)
 client = OpenAI(base_url="https://api.perplexity.ai", api_key="PPLX_API_KEY")
 
 
@@ -18,4 +20,4 @@ def basic():
         stream=False,
     )
     print(response)
-    return response
+    return response

{langtrace_python_sdk-1.3.1 → langtrace_python_sdk-1.3.3}/src/examples/pinecone_example/basic.py

@@ -11,7 +11,7 @@ from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(batch=False, debug_log_to_console=True, write_to_langtrace_cloud=False)
+langtrace.init(write_to_langtrace_cloud=False)
 
 client = OpenAI()
 pinecone = Pinecone()

{langtrace_python_sdk-1.3.1 → langtrace_python_sdk-1.3.3}/src/langtrace_python_sdk/extensions/langtrace_exporter.py

@@ -6,7 +6,9 @@ import requests
 from opentelemetry.sdk.trace.export import ReadableSpan, SpanExporter, SpanExportResult
 from opentelemetry.trace.span import format_trace_id
 
-from langtrace_python_sdk.constants.exporter.langtrace_exporter import LANGTRACE_REMOTE_URL
+from langtrace_python_sdk.constants.exporter.langtrace_exporter import (
+    LANGTRACE_REMOTE_URL,
+)
 
 
 class LangTraceExporter(SpanExporter):
@@ -48,7 +50,12 @@ class LangTraceExporter(SpanExporter):
     api_key: str
     write_to_remote_url: bool
 
-    def __init__(self, api_key: str = None, write_to_remote_url: bool = False, api_host: typing.Optional[str] = None) -> None:
+    def __init__(
+        self,
+        api_key: str = None,
+        write_to_remote_url: bool = False,
+        api_host: typing.Optional[str] = None,
+    ) -> None:
         self.api_key = api_key or os.environ.get("LANGTRACE_API_KEY")
         self.write_to_remote_url = write_to_remote_url
         self.api_host: str = api_host or LANGTRACE_REMOTE_URL
@@ -89,6 +96,7 @@ class LangTraceExporter(SpanExporter):
                 data=json.dumps(data),
                 headers={"Content-Type": "application/json", "x-api-key": self.api_key},
             )
+            print(f"sent to {self.api_host}/api/trace with {len(data)} spans")
             return SpanExportResult.SUCCESS
         except Exception as e:
             return SpanExportResult.FAILURE

{langtrace_python_sdk-1.3.1 → langtrace_python_sdk-1.3.3}/src/langtrace_python_sdk/instrumentation/cohere/patch.py

@@ -9,7 +9,10 @@ from opentelemetry.trace import SpanKind
 from opentelemetry.trace.status import Status, StatusCode
 
 from langtrace_python_sdk.constants.instrumentation.cohere import APIS
-from langtrace_python_sdk.constants.instrumentation.common import (LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY, SERVICE_PROVIDERS)
+from langtrace_python_sdk.constants.instrumentation.common import (
+    LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY,
+    SERVICE_PROVIDERS,
+)
 from langtrace_python_sdk.utils.llm import estimate_tokens
 
 
@@ -33,7 +36,7 @@ def embed_create(original_method, version, tracer):
             "llm.embedding_dataset_id": kwargs.get("dataset_id"),
             "llm.embedding_input_type": kwargs.get("input_type"),
             "llm.embedding_job_name": kwargs.get("name"),
-            **(extra_attributes if extra_attributes is not None else {})
+            **(extra_attributes if extra_attributes is not None else {}),
         }
 
         attributes = LLMSpanAttributes(**span_attributes)
@@ -41,9 +44,7 @@ def embed_create(original_method, version, tracer):
         if kwargs.get("user") is not None:
             attributes.llm_user = kwargs.get("user")
 
-        span = tracer.start_span(
-            APIS["EMBED_CREATE"]["METHOD"], kind=SpanKind.CLIENT
-        )
+        span = tracer.start_span(APIS["EMBED_CREATE"]["METHOD"], kind=SpanKind.CLIENT)
         for field, value in attributes.model_dump(by_alias=True).items():
             if value is not None:
                 span.set_attribute(field, value)
@@ -59,6 +60,7 @@ def embed_create(original_method, version, tracer):
             span.set_status(Status(StatusCode.ERROR, str(error)))
             span.end()
             raise
+
     return traced_method
 
 
@@ -69,16 +71,12 @@ def chat_create(original_method, version, tracer):
         service_provider = SERVICE_PROVIDERS["COHERE"]
 
         message = kwargs.get("message", "")
-        prompts = json.dumps([
-            {
-                "role": "USER",
-                "content": message
-            }
-        ])
+        prompts = json.dumps([{"role": "USER", "content": message}])
         preamble = kwargs.get("preamble")
         if preamble:
             prompts = json.dumps(
-                [{"role": "system", "content": preamble}] + [{"role": "USER", "content": message}]
+                [{"role": "system", "content": preamble}]
+                + [{"role": "USER", "content": message}]
             )
 
         chat_history = kwargs.get("chat_history")
@@ -90,8 +88,10 @@ def chat_create(original_method, version, tracer):
                             item.get("role") if item.get("role") is not None else "USER"
                         ),
                         "content": (
-                            item.get("message") if item.get("message") is not None else ""
-                        )
+                            item.get("message")
+                            if item.get("message") is not None
+                            else ""
+                        ),
                     }
                 }
                 for item in chat_history
@@ -108,10 +108,12 @@ def chat_create(original_method, version, tracer):
             "langtrace.version": "1.0.0",
             "url.full": APIS["CHAT_CREATE"]["URL"],
             "llm.api": APIS["CHAT_CREATE"]["ENDPOINT"],
-            "llm.model": kwargs.get("model") if kwargs.get("model") is not None else "command-r",
+            "llm.model": (
+                kwargs.get("model") if kwargs.get("model") is not None else "command-r"
+            ),
             "llm.stream": False,
             "llm.prompts": prompts,
-            **(extra_attributes if extra_attributes is not None else {})
+            **(extra_attributes if extra_attributes is not None else {}),
         }
 
         attributes = LLMSpanAttributes(**span_attributes)
@@ -146,9 +148,7 @@ def chat_create(original_method, version, tracer):
             # stringify the list of objects
             attributes.llm_tool_results = json.dumps(kwargs.get("tool_results"))
 
-        span = tracer.start_span(
-            APIS["CHAT_CREATE"]["METHOD"], kind=SpanKind.CLIENT
-        )
+        span = tracer.start_span(APIS["CHAT_CREATE"]["METHOD"], kind=SpanKind.CLIENT)
 
         # Set the attributes on the span
         for field, value in attributes.model_dump(by_alias=True).items():
@@ -159,37 +159,47 @@ def chat_create(original_method, version, tracer):
             result = wrapped(*args, **kwargs)
 
             # Set the response attributes
-            if (hasattr(result, "generation_id")) and (result.generation_id is not None):
+            if (hasattr(result, "generation_id")) and (
+                result.generation_id is not None
+            ):
                 span.set_attribute("llm.generation_id", result.generation_id)
             if (hasattr(result, "response_id")) and (result.response_id is not None):
                 span.set_attribute("llm.response_id", result.response_id)
-            if (hasattr(result, "is_search_required")) and (result.is_search_required is not None):
+            if (hasattr(result, "is_search_required")) and (
+                result.is_search_required is not None
+            ):
                 span.set_attribute("llm.is_search_required", result.is_search_required)
 
             if kwargs.get("stream") is False or kwargs.get("stream") is None:
                 if hasattr(result, "text") and result.text is not None:
-                    if hasattr(result, "chat_history") and result.chat_history is not None:
+                    if (
+                        hasattr(result, "chat_history")
+                        and result.chat_history is not None
+                    ):
                         responses = [
                             {
                                 "message": {
                                     "role": (
-                                        item.role if hasattr(item, "role") and item.role is not None else "USER"
+                                        item.role
+                                        if hasattr(item, "role")
+                                        and item.role is not None
+                                        else "USER"
                                     ),
                                     "content": (
-                                        item.message if hasattr(item, "message") and item.message is not None else ""
-                                    )
+                                        item.message
+                                        if hasattr(item, "message")
+                                        and item.message is not None
+                                        else ""
+                                    ),
                                 }
                             }
                             for item in result.chat_history
                         ]
                         span.set_attribute("llm.responses", json.dumps(responses))
                     else:
-                        responses = [{
-                            "message": {
-                                "role": "CHATBOT",
-                                "content": result.text
-                            }
-                        }]
+                        responses = [
+                            {"message": {"role": "CHATBOT", "content": result.text}}
+                        ]
                         span.set_attribute("llm.responses", json.dumps(responses))
                 else:
                     responses = []
@@ -197,15 +207,33 @@ def chat_create(original_method, version, tracer):
 
                 # Get the usage
                 if hasattr(result, "meta") and result.meta is not None:
-                    if hasattr(result.meta, "billed_units") and result.meta.billed_units is not None:
+                    if (
+                        hasattr(result.meta, "billed_units")
+                        and result.meta.billed_units is not None
+                    ):
                         usage = result.meta.billed_units
                         if usage is not None:
                             usage_dict = {
-                                "input_tokens": usage.input_tokens if usage.input_tokens is not None else 0,
-                                "output_tokens": usage.output_tokens if usage.output_tokens is not None else 0,
-                                "total_tokens": usage.input_tokens + usage.output_tokens if usage.input_tokens is not None and usage.output_tokens is not None else 0,
+                                "input_tokens": (
+                                    usage.input_tokens
+                                    if usage.input_tokens is not None
+                                    else 0
+                                ),
+                                "output_tokens": (
+                                    usage.output_tokens
+                                    if usage.output_tokens is not None
+                                    else 0
+                                ),
+                                "total_tokens": (
+                                    usage.input_tokens + usage.output_tokens
+                                    if usage.input_tokens is not None
+                                    and usage.output_tokens is not None
+                                    else 0
+                                ),
                             }
-                            span.set_attribute("llm.token.counts", json.dumps(usage_dict))
+                            span.set_attribute(
+                                "llm.token.counts", json.dumps(usage_dict)
+                            )
                 span.set_status(StatusCode.OK)
                 span.end()
                 return result
@@ -218,6 +246,7 @@ def chat_create(original_method, version, tracer):
             span.set_status(Status(StatusCode.ERROR, str(error)))
             span.end()
             raise
+
     return traced_method
 
 
@@ -229,16 +258,12 @@ def chat_stream(original_method, version, tracer):
 
         message = kwargs.get("message", "")
         prompt_tokens = estimate_tokens(message)
-        prompts = json.dumps([
-            {
-                "role": "USER",
-                "content": message
-            }
-        ])
+        prompts = json.dumps([{"role": "USER", "content": message}])
         preamble = kwargs.get("preamble")
         if preamble:
             prompts = json.dumps(
-                [{"role": "system", "content": preamble}] + [{"role": "USER", "content": message}]
+                [{"role": "system", "content": preamble}]
+                + [{"role": "USER", "content": message}]
             )
 
         chat_history = kwargs.get("chat_history")
@@ -250,8 +275,10 @@ def chat_stream(original_method, version, tracer):
                             item.get("role") if item.get("role") is not None else "USER"
                         ),
                         "content": (
-                            item.get("message") if item.get("message") is not None else ""
-                        )
+                            item.get("message")
+                            if item.get("message") is not None
+                            else ""
+                        ),
                     }
                 }
                 for item in chat_history
@@ -268,10 +295,12 @@ def chat_stream(original_method, version, tracer):
             "langtrace.version": "1.0.0",
             "url.full": APIS["CHAT_STREAM"]["URL"],
             "llm.api": APIS["CHAT_STREAM"]["ENDPOINT"],
-            "llm.model": kwargs.get("model") if kwargs.get("model") is not None else "command-r",
-            "llm.stream": False,
+            "llm.model": (
+                kwargs.get("model") if kwargs.get("model") is not None else "command-r"
+            ),
+            "llm.stream": True,
             "llm.prompts": prompts,
-            **(extra_attributes if extra_attributes is not None else {})
+            **(extra_attributes if extra_attributes is not None else {}),
         }
 
         attributes = LLMSpanAttributes(**span_attributes)
@@ -306,9 +335,7 @@ def chat_stream(original_method, version, tracer):
             # stringify the list of objects
             attributes.llm_tool_results = json.dumps(kwargs.get("tool_results"))
 
-        span = tracer.start_span(
-            APIS["CHAT_CREATE"]["METHOD"], kind=SpanKind.CLIENT
-        )
+        span = tracer.start_span(APIS["CHAT_CREATE"]["METHOD"], kind=SpanKind.CLIENT)
         for field, value in attributes.model_dump(by_alias=True).items():
             if value is not None:
                 span.set_attribute(field, value)

{langtrace_python_sdk-1.3.1 → langtrace_python_sdk-1.3.3}/src/langtrace_python_sdk/instrumentation/llamaindex/instrumentation.py

@@ -10,7 +10,14 @@ from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
 from opentelemetry.trace import get_tracer
 from wrapt import wrap_function_wrapper
 
-from langtrace_python_sdk.instrumentation.llamaindex.patch import generic_patch
+from langtrace_python_sdk.instrumentation.llamaindex.patch import (
+    generic_patch,
+    async_generic_patch,
+)
+
+import logging
+
+logging.basicConfig(level=logging.FATAL)
 
 import logging