langtrace-python-sdk 2.0.4__py3-none-any.whl → 2.0.5__py3-none-any.whl

examples/openai_example/tool_calling_streaming.py

@@ -9,7 +9,8 @@ from langtrace_python_sdk import langtrace
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(write_to_langtrace_cloud=False)
+langtrace.init(write_spans_to_console=True)
+
 
 client = OpenAI()
 
@@ -21,7 +22,9 @@ def get_current_weather(location, unit="fahrenheit"):
     if "tokyo" in location.lower():
         return json.dumps({"location": "Tokyo", "temperature": "10", "unit": unit})
     elif "san francisco" in location.lower():
-        return json.dumps({"location": "San Francisco", "temperature": "72", "unit": unit})
+        return json.dumps(
+            {"location": "San Francisco", "temperature": "72", "unit": unit}
+        )
     elif "paris" in location.lower():
         return json.dumps({"location": "Paris", "temperature": "22", "unit": unit})
     else:
@@ -42,7 +45,12 @@ def get_current_time(location):
 
 def run_conversation():
     # Step 1: send the conversation and available functions to the model
-    messages = [{"role": "user", "content": "What's the weather like in San Francisco, Tokyo, and Paris?"}]
+    messages = [
+        {
+            "role": "user",
+            "content": "What's the weather like in San Francisco, Tokyo, and Paris?",
+        }
+    ]
     tools = [
         {
             "type": "function",
@@ -78,7 +86,7 @@ def run_conversation():
                     "required": ["location"],
                 },
             },
-        }
+        },
     ]
     response = client.chat.completions.create(
         model="gpt-4",
@@ -95,29 +103,34 @@ def run_conversation():
     name = ""
     arguments = ""
     for chunk in response:
-        if chunk.choices[0].delta is not None and chunk.choices[0].delta.tool_calls is not None:
+        if (
+            chunk.choices[0].delta is not None
+            and chunk.choices[0].delta.tool_calls is not None
+        ):
             for choice in chunk.choices:
                 for tool_call in choice.delta.tool_calls:
                     if tool_call.id and id != tool_call.id:
                         id = tool_call.id if tool_call.id else ""
-                        name = tool_call.function.name if tool_call.function and tool_call.function.name else ""
+                        name = (
+                            tool_call.function.name
+                            if tool_call.function and tool_call.function.name
+                            else ""
+                        )
                         tool_call_dict[name] = {
                             "id": id,
-                            "function": {
-                                "name": name,
-                                "arguments": arguments
-                            },
-                            "type": "function"
+                            "function": {"name": name, "arguments": arguments},
+                            "type": "function",
                         }
-                    arguments += tool_call.function.arguments if tool_call.function and tool_call.function.arguments else ""
+                    arguments += (
+                        tool_call.function.arguments
+                        if tool_call.function and tool_call.function.arguments
+                        else ""
+                    )
                 if name != "":
                     tool_call_dict[name] = {
                         "id": id,
-                        "function": {
-                            "name": name,
-                            "arguments": arguments
-                        },
-                        "type": "function"
+                        "function": {"name": name, "arguments": arguments},
+                        "type": "function",
                     }
     for key, value in tool_call_dict.items():
         tool_calls.append(value)
@@ -133,9 +146,9 @@ def run_conversation():
         # messages.append(response_message)  # extend conversation with assistant's reply
         # Step 4: send the info for each function call and function response to the model
         for tool_call in tool_calls:
-            function_name = tool_call['function']['name']
+            function_name = tool_call["function"]["name"]
             function_to_call = available_functions[function_name]
-            function_args = json.loads(tool_call['function']['arguments'])
+            function_args = json.loads(tool_call["function"]["arguments"])
             function_response = function_to_call(
                 location=function_args.get("location"),
                 unit=function_args.get("unit"),
@@ -143,10 +156,7 @@ def run_conversation():
             func_res = json.loads(function_response)
             content = f"Use the below information to answer the user's question: The current weather in {func_res['location']} is {func_res['temperature']} degrees {func_res['unit']}"
             messages.append(
-                {
-                    "role": "system",
-                    "content": content
-                }
+                {"role": "system", "content": content}
             )  # extend conversation with function response
         print(messages)
         second_response = client.chat.completions.create(
@@ -158,10 +168,13 @@ def run_conversation():
         for chunk in second_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 "")
+                    (
+                        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 second_response
+        # return second_response

examples/perplexity_example/basic.py

@@ -8,7 +8,7 @@ from langtrace_python_sdk.utils.with_root_span import (
 
 # _ = load_dotenv(find_dotenv())
 
-langtrace.init(write_to_langtrace_cloud=False)
+langtrace.init(write_spans_to_console=True)
 client = OpenAI(base_url="https://api.perplexity.ai", api_key="PPLX_API_KEY")
 
 

examples/pinecone_example/basic.py

@@ -11,7 +11,10 @@ from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(write_to_langtrace_cloud=False)
+langtrace.init(
+    write_spans_to_console=True,
+    disable_instrumentations={"all_except": ["pinecone", "openai"]},
+)
 
 client = OpenAI()
 pinecone = Pinecone()

examples/qdrant_example/basic.py

@@ -8,7 +8,7 @@ from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(write_to_langtrace_cloud=False)
+langtrace.init()
 
 
 @with_langtrace_root_span()
@@ -16,10 +16,13 @@ def basic():
     client = QdrantClient(":memory:")
     cohere_client = cohere.Client()
 
-    client.create_collection(collection_name="MyCollection4", vectors_config=VectorParams(
-        size=1024,
-        distance=Distance.COSINE,
-    ))
+    client.create_collection(
+        collection_name="MyCollection4",
+        vectors_config=VectorParams(
+            size=1024,
+            distance=Distance.COSINE,
+        ),
+    )
 
     client.upsert(
         collection_name="MyCollection4",
@@ -40,7 +43,6 @@ def basic():
             input_type="search_query",  # Input type for search queries
             texts=["Which database is written in Rust?"],
         ).embeddings[0],
-
     )
     print(answer[0])
 

langtrace_python_sdk/constants/instrumentation/groq.py

@@ -1,5 +1,3 @@
-
-
 APIS = {
     "CHAT_COMPLETION": {
         "METHOD": "groq.chat.completions.create",

langtrace_python_sdk/extensions/langtrace_exporter.py

@@ -25,18 +25,16 @@ class LangTraceExporter(SpanExporter):
     **Attributes:**
 
     * `api_key` (str): An API key to authenticate with the LangTrace collector (required).
-    * `write_to_remote_url` (bool): A flag indicating whether to send spans to the remote URL (defaults to False).
 
     **Methods:**
 
-    * `__init__(api_key: str = None, url: str = None, write_to_remote_url: bool = False) -> None`:
+    * `__init__(api_key: str = None, url: str = None) -> None`:
         - Initializes a `LangTraceExporter` instance.
         - Retrieves the API key and URL from environment variables if not provided explicitly.
-        - Raises a `ValueError` if the API key is missing or the URL is missing when `write_to_remote_url` is True.
+        - Raises a `ValueError` if the API key is missing.
     * `export(self, spans: typing.Sequence[ReadableSpan]) -> SpanExportResult`:
         - Exports a batch of `opentelemetry.trace.Span` objects to LangTrace.
         - Converts each span into a dictionary representation including trace ID, instrumentation library, events, dropped data counts, duration, and other span attributes.
-        - If `write_to_remote_url` is False, returns `SpanExportResult.SUCCESS` without sending data.
         - Otherwise, sends the data to the configured URL using a POST request with JSON data and the API key in the header.
         - Returns `SpanExportResult.SUCCESS` on successful export or `SpanExportResult.FAILURE` on errors.
     * `shutdown(self) -> None`:
@@ -44,23 +42,20 @@ class LangTraceExporter(SpanExporter):
 
     **Raises:**
 
-    * `ValueError`: If the API key is not provided or the URL is missing when `write_to_remote_url` is True.
+    * `ValueError`: If the API key is not provided.
     """
 
     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:
         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
 
-        if self.write_to_remote_url and not self.api_key:
+        if not self.api_key:
             raise ValueError("No API key provided")
 
     def export(self, spans: typing.Sequence[ReadableSpan]) -> SpanExportResult:
@@ -86,9 +81,6 @@ class LangTraceExporter(SpanExporter):
             for span in spans
         ]
 
-        if not self.write_to_remote_url:
-            return
-
         # Send data to remote URL
         try:
             requests.post(

langtrace_python_sdk/instrumentation/anthropic/instrumentation.py

@@ -22,8 +22,7 @@ from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
 from opentelemetry.trace import get_tracer
 from wrapt import wrap_function_wrapper
 
-from langtrace_python_sdk.instrumentation.anthropic.patch import \
-    messages_create
+from langtrace_python_sdk.instrumentation.anthropic.patch import messages_create
 
 logging.basicConfig(level=logging.FATAL)
 

langtrace_python_sdk/instrumentation/anthropic/patch.py

@@ -23,7 +23,9 @@ from opentelemetry.trace.status import Status, StatusCode
 
 from langtrace_python_sdk.constants.instrumentation.anthropic import APIS
 from langtrace_python_sdk.constants.instrumentation.common import (
-    LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY, SERVICE_PROVIDERS)
+    LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY,
+    SERVICE_PROVIDERS,
+)
 
 
 def messages_create(original_method, version, tracer):
@@ -58,7 +60,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 {})
+            **(extra_attributes if extra_attributes is not None else {}),
         }
 
         attributes = LLMSpanAttributes(**span_attributes)
@@ -85,7 +87,10 @@ def messages_create(original_method, version, tracer):
             result = wrapped(*args, **kwargs)
             if kwargs.get("stream") is False:
                 if hasattr(result, "content") and result.content is not None:
-                    span.set_attribute("llm.model", result.model if result.model else kwargs.get("model"))
+                    span.set_attribute(
+                        "llm.model",
+                        result.model if result.model else kwargs.get("model"),
+                    )
                     span.set_attribute(
                         "llm.responses",
                         json.dumps(
@@ -140,7 +145,12 @@ def messages_create(original_method, version, tracer):
         output_tokens = 0
         try:
             for chunk in result:
-                if hasattr(chunk, "message") and chunk.message is not None and hasattr(chunk.message, "model") and chunk.message.model is not None:
+                if (
+                    hasattr(chunk, "message")
+                    and chunk.message is not None
+                    and hasattr(chunk.message, "model")
+                    and chunk.message.model is not None
+                ):
                     span.set_attribute("llm.model", chunk.message.model)
                 content = ""
                 if hasattr(chunk, "delta") and chunk.delta is not None:

langtrace_python_sdk/instrumentation/chroma/patch.py

@@ -21,7 +21,9 @@ from opentelemetry.trace.status import Status, StatusCode
 
 from langtrace_python_sdk.constants.instrumentation.chroma import APIS
 from langtrace_python_sdk.constants.instrumentation.common import (
-    LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY, SERVICE_PROVIDERS)
+    LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY,
+    SERVICE_PROVIDERS,
+)
 
 
 def collection_patch(method, version, tracer):
@@ -42,7 +44,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 {})
+            **(extra_attributes if extra_attributes is not None else {}),
         }
 
         if hasattr(instance, "name") and instance.name is not None:

langtrace_python_sdk/instrumentation/cohere/instrumentation.py

@@ -21,9 +21,12 @@ from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
 from opentelemetry.trace import get_tracer
 from wrapt import wrap_function_wrapper
 
-from langtrace_python_sdk.instrumentation.cohere.patch import (chat_create,
-                                                               chat_stream,
-                                                               embed, rerank)
+from langtrace_python_sdk.instrumentation.cohere.patch import (
+    chat_create,
+    chat_stream,
+    embed,
+    rerank,
+)
 
 
 class CohereInstrumentation(BaseInstrumentor):

langtrace_python_sdk/instrumentation/groq/instrumentation.py

@@ -23,7 +23,9 @@ from opentelemetry.trace import get_tracer
 from wrapt import wrap_function_wrapper
 
 from langtrace_python_sdk.instrumentation.groq.patch import (
-    async_chat_completions_create, chat_completions_create)
+    async_chat_completions_create,
+    chat_completions_create,
+)
 
 logging.basicConfig(level=logging.FATAL)
 

langtrace_python_sdk/instrumentation/groq/patch.py

@@ -22,10 +22,11 @@ from opentelemetry.trace import SpanKind
 from opentelemetry.trace.status import Status, StatusCode
 
 from langtrace_python_sdk.constants.instrumentation.common import (
-    LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY, SERVICE_PROVIDERS)
+    LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY,
+    SERVICE_PROVIDERS,
+)
 from langtrace_python_sdk.constants.instrumentation.groq 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 chat_completions_create(original_method, version, tracer):
@@ -136,7 +137,8 @@ def chat_completions_create(original_method, version, tracer):
                                 if "content_filter_results" in choice
                                 else {}
                             ),
-                        } for choice in result.choices
+                        }
+                        for choice in result.choices
                     ]
                     span.set_attribute("llm.responses", json.dumps(responses))
                 else:
@@ -224,16 +226,22 @@ def chat_completions_create(original_method, version, tracer):
                     elif tool_calls:
                         for choice in chunk.choices:
                             tool_call = ""
-                            if (choice.delta and choice.delta.tool_calls is not None):
+                            if choice.delta and choice.delta.tool_calls is not None:
                                 toolcalls = choice.delta.tool_calls
                                 content = []
                                 for tool_call in toolcalls:
-                                    if tool_call and tool_call.function is not None and tool_call.function.arguments is not None:
+                                    if (
+                                        tool_call
+                                        and tool_call.function is not None
+                                        and tool_call.function.arguments is not None
+                                    ):
                                         token_counts = estimate_tokens(
                                             tool_call.function.arguments
                                         )
                                         completion_tokens += token_counts
-                                        content = content + [tool_call.function.arguments]
+                                        content = content + [
+                                            tool_call.function.arguments
+                                        ]
                                     else:
                                         content = content + []
                 else:
@@ -389,7 +397,8 @@ def async_chat_completions_create(original_method, version, tracer):
                                 if "content_filter_results" in choice
                                 else {}
                             ),
-                        } for choice in result.choices
+                        }
+                        for choice in result.choices
                     ]
                     span.set_attribute("llm.responses", json.dumps(responses))
                 else:
@@ -477,16 +486,22 @@ def async_chat_completions_create(original_method, version, tracer):
                     elif tool_calls:
                         for choice in chunk.choices:
                             tool_call = ""
-                            if (choice.delta and choice.delta.tool_calls is not None):
+                            if choice.delta and choice.delta.tool_calls is not None:
                                 toolcalls = choice.delta.tool_calls
                                 content = []
                                 for tool_call in toolcalls:
-                                    if tool_call and tool_call.function is not None and tool_call.function.arguments is not None:
+                                    if (
+                                        tool_call
+                                        and tool_call.function is not None
+                                        and tool_call.function.arguments is not None
+                                    ):
                                         token_counts = estimate_tokens(
                                             tool_call.function.arguments
                                         )
                                         completion_tokens += token_counts
-                                        content = content + [tool_call.function.arguments]
+                                        content = content + [
+                                            tool_call.function.arguments
+                                        ]
                                     else:
                                         content = content + []
                 else:

langtrace_python_sdk/instrumentation/langchain/patch.py

@@ -22,7 +22,9 @@ from opentelemetry.trace import SpanKind, StatusCode
 from opentelemetry.trace.status import Status
 
 from langtrace_python_sdk.constants.instrumentation.common import (
-    LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY, SERVICE_PROVIDERS)
+    LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY,
+    SERVICE_PROVIDERS,
+)
 
 
 def generic_patch(
@@ -43,7 +45,7 @@ def generic_patch(
             "langtrace.service.version": version,
             "langtrace.version": "1.0.0",
             "langchain.task.name": task,
-            **(extra_attributes if extra_attributes is not None else {})
+            **(extra_attributes if extra_attributes is not None else {}),
         }
 
         if len(args) > 0 and trace_input:

langtrace_python_sdk/instrumentation/langchain_community/instrumentation.py

@@ -22,8 +22,7 @@ from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
 from opentelemetry.trace import get_tracer
 from wrapt import wrap_function_wrapper
 
-from langtrace_python_sdk.instrumentation.langchain_community.patch import \
-    generic_patch
+from langtrace_python_sdk.instrumentation.langchain_community.patch import generic_patch
 
 
 def patch_module_classes(

langtrace_python_sdk/instrumentation/langchain_community/patch.py

@@ -14,7 +14,6 @@ See the License for the specific language governing permissions and
 limitations under the License.
 """
 
-
 import json
 
 from langtrace.trace_attributes import FrameworkSpanAttributes
@@ -23,7 +22,9 @@ from opentelemetry.trace import SpanKind
 from opentelemetry.trace.status import Status, StatusCode
 
 from langtrace_python_sdk.constants.instrumentation.common import (
-    LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY, SERVICE_PROVIDERS)
+    LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY,
+    SERVICE_PROVIDERS,
+)
 
 
 def generic_patch(
@@ -40,7 +41,7 @@ def generic_patch(
             "langtrace.service.version": version,
             "langtrace.version": "1.0.0",
             "langchain.task.name": task,
-            **(extra_attributes if extra_attributes is not None else {})
+            **(extra_attributes if extra_attributes is not None else {}),
         }
 
         if trace_input and len(args) > 0:

langtrace_python_sdk/instrumentation/langchain_core/instrumentation.py

@@ -23,7 +23,9 @@ from opentelemetry.trace import get_tracer
 from wrapt import wrap_function_wrapper
 
 from langtrace_python_sdk.instrumentation.langchain_core.patch import (
-    generic_patch, runnable_patch)
+    generic_patch,
+    runnable_patch,
+)
 
 
 # pylint: disable=dangerous-default-value

langtrace_python_sdk/instrumentation/langchain_core/patch.py

@@ -22,7 +22,9 @@ from opentelemetry.trace import SpanKind, StatusCode
 from opentelemetry.trace.status import Status
 
 from langtrace_python_sdk.constants.instrumentation.common import (
-    LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY, SERVICE_PROVIDERS)
+    LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY,
+    SERVICE_PROVIDERS,
+)
 
 
 def generic_patch(
@@ -49,7 +51,7 @@ def generic_patch(
             "langtrace.service.version": version,
             "langtrace.version": "1.0.0",
             "langchain.task.name": task,
-            **(extra_attributes if extra_attributes is not None else {})
+            **(extra_attributes if extra_attributes is not None else {}),
         }
 
         if len(args) > 0 and trace_input:

langtrace_python_sdk/instrumentation/langgraph/instrumentation.py

@@ -22,8 +22,7 @@ from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
 from opentelemetry.trace import get_tracer
 from wrapt import wrap_function_wrapper
 
-from langtrace_python_sdk.instrumentation.langgraph.patch import \
-    patch_graph_methods
+from langtrace_python_sdk.instrumentation.langgraph.patch import patch_graph_methods
 
 
 class LanggraphInstrumentation(BaseInstrumentor):
@@ -41,24 +40,34 @@ class LanggraphInstrumentation(BaseInstrumentor):
 
         # List of modules to patch, with their corresponding patch names
         modules_to_patch = [
-            ("langgraph.graph.graph", ["add_node", "add_edge", "set_entry_point", "set_finish_point", "add_conditional_edges"]),
+            (
+                "langgraph.graph.graph",
+                [
+                    "add_node",
+                    "add_edge",
+                    "set_entry_point",
+                    "set_finish_point",
+                    "add_conditional_edges",
+                ],
+            ),
         ]
 
         for module_name, methods in modules_to_patch:
             module = importlib.import_module(module_name)
             for name, obj in inspect.getmembers(
                 module,
-                lambda member: inspect.isclass(member) and member.__module__ == module.__name__,
+                lambda member: inspect.isclass(member)
+                and member.__module__ == module.__name__,
             ):
-                for method_name, _ in inspect.getmembers(obj, predicate=inspect.isfunction):
+                for method_name, _ in inspect.getmembers(
+                    obj, predicate=inspect.isfunction
+                ):
                     if method_name in methods:
                         module = f"{name}.{method_name}"
                         wrap_function_wrapper(
                             module_name,
                             module,
-                            patch_graph_methods(
-                                module, tracer, version
-                            ),
+                            patch_graph_methods(module, tracer, version),
                         )
 
     def _uninstrument(self, **kwargs):