langtrace-python-sdk 2.1.21__py3-none-any.whl → 2.1.22__py3-none-any.whl

examples/crewai_example/basic.py

@@ -0,0 +1,70 @@
+import os
+
+os.environ["OPENAI_MODEL_NAME"] = "gpt-3.5-turbo"
+os.environ["SERPER_API_KEY"] = ""  # serper.dev API key
+from langtrace_python_sdk import langtrace
+from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
+from crewai import Crew, Process
+from crewai import Task
+from crewai import Agent
+from crewai_tools import SerperDevTool
+from crewai_tools import YoutubeVideoSearchTool
+
+langtrace.init()
+
+search_tool = SerperDevTool()
+
+# Targeted search within a specific Youtube video's content
+youtube_tool = YoutubeVideoSearchTool(
+    youtube_video_url="https://www.youtube.com/watch?v=blqIZGXWUpU"
+)
+
+# Creating a senior researcher agent with memory and verbose mode
+researcher = Agent(
+    role="Senior Researcher",
+    goal="Uncover groundbreaking technologies in {topic}",
+    verbose=True,
+    memory=True,
+    backstory=(
+        "Driven by curiosity, you're at the forefront of"
+        "innovation, eager to explore and share knowledge that could change"
+        "the world."
+    ),
+    tools=[youtube_tool],
+)
+
+# Research task
+research_task = Task(
+    description=(
+        "Do a {topic} of the given youtube video."
+        "Focus on identifying the overall narrative."
+        "Your final report should clearly articulate the key points."
+    ),
+    expected_output="10 key points from the shared video.",
+    tools=[youtube_tool],
+    agent=researcher,
+    callback="research_callback",  # Example of task callback
+    human_input=True,
+)
+
+
+# Forming the tech-focused crew with some enhanced configurations
+crew = Crew(
+    agents=[researcher],
+    tasks=[research_task],
+    process=Process.sequential,  # Optional: Sequential task execution is default
+    memory=False,
+    cache=False,
+    max_rpm=20,
+)
+
+# Starting the task execution process with enhanced feedback
+
+
+@with_langtrace_root_span("Crew")
+def test_crew():
+    result = crew.kickoff(inputs={"topic": "summary"})
+    return result
+
+
+test_crew()

langtrace_python_sdk/constants/instrumentation/common.py

@@ -11,6 +11,7 @@ SERVICE_PROVIDERS = {
     "ANTHROPIC": "Anthropic",
     "AZURE": "Azure",
     "CHROMA": "Chroma",
+    "CREWAI": "CrewAI",
     "DSPY": "DSPy",
     "GROQ": "Groq",
     "LANGCHAIN": "Langchain",

langtrace_python_sdk/instrumentation/__init__.py

@@ -1,6 +1,7 @@
 from .anthropic import AnthropicInstrumentation
 from .chroma import ChromaInstrumentation
 from .cohere import CohereInstrumentation
+from .crewai import CrewAIInstrumentation
 from .groq import GroqInstrumentation
 from .langchain import LangchainInstrumentation
 from .langchain_community import LangchainCommunityInstrumentation
@@ -12,12 +13,13 @@ from .pinecone import PineconeInstrumentation
 from .qdrant import QdrantInstrumentation
 from .weaviate import WeaviateInstrumentation
 from .ollama import OllamaInstrumentor
-from .dspy import DspyInstrumentor
+from .dspy import DspyInstrumentation
 
 __all__ = [
     "AnthropicInstrumentation",
     "ChromaInstrumentation",
     "CohereInstrumentation",
+    "CrewAIInstrumentation",
     "GroqInstrumentation",
     "LangchainInstrumentation",
     "LangchainCommunityInstrumentation",
@@ -29,5 +31,5 @@ __all__ = [
     "QdrantInstrumentation",
     "WeaviateInstrumentation",
     "OllamaInstrumentor",
-    "DspyInstrumentor",
+    "DspyInstrumentation",
 ]

langtrace_python_sdk/instrumentation/crewai/__init__.py

@@ -0,0 +1,3 @@
+from .instrumentation import CrewAIInstrumentation
+
+__all__ = ["CrewAIInstrumentation"]

langtrace_python_sdk/instrumentation/crewai/instrumentation.py

@@ -0,0 +1,53 @@
+"""
+Copyright (c) 2024 Scale3 Labs
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+"""
+
+from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
+from opentelemetry.trace import get_tracer
+from wrapt import wrap_function_wrapper as _W
+from typing import Collection
+from importlib_metadata import version as v
+from .patch import patch_crew
+
+
+class CrewAIInstrumentation(BaseInstrumentor):
+    """
+    The CrewAIInstrumentation class represents the CrewAI instrumentation"""
+
+    def instrumentation_dependencies(self) -> Collection[str]:
+        return ["crewai >= 0.32.0"]
+
+    def _instrument(self, **kwargs):
+        tracer_provider = kwargs.get("tracer_provider")
+        tracer = get_tracer(__name__, "", tracer_provider)
+        version = v("crewai")
+        _W(
+            "crewai.crew",
+            "Crew.kickoff",
+            patch_crew("Crew.kickoff", version, tracer),
+        )
+        _W(
+            "crewai.agent",
+            "Agent.execute_task",
+            patch_crew("Agent.execute_task", version, tracer),
+        )
+        _W(
+            "crewai.task",
+            "Task.execute",
+            patch_crew("Task.execute", version, tracer),
+        )
+
+    def _uninstrument(self, **kwargs):
+        pass

langtrace_python_sdk/instrumentation/crewai/patch.py

@@ -0,0 +1,173 @@
+import json
+from importlib_metadata import version as v
+from langtrace_python_sdk.constants import LANGTRACE_SDK_NAME
+from langtrace_python_sdk.utils import set_span_attribute
+from langtrace_python_sdk.utils.silently_fail import silently_fail
+from langtrace_python_sdk.constants.instrumentation.common import (
+    LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY,
+    SERVICE_PROVIDERS,
+)
+from opentelemetry import baggage
+from langtrace.trace_attributes import FrameworkSpanAttributes
+from opentelemetry.trace import SpanKind
+from opentelemetry.trace.status import Status, StatusCode
+
+
+crew_properties = {
+    "tasks": "object",
+    "agents": "object",
+    "cache": "bool",
+    "process": "object",
+    "verbose": "bool",
+    "memory": "bool",
+    "embedder": "json",
+    "full_output": "bool",
+    "manager_llm": "object",
+    "manager_agent": "object",
+    "manager_callbacks": "object",
+    "function_calling_llm": "object",
+    "config": "json",
+    "id": "object",
+    "max_rpm": "int",
+    "share_crew": "bool",
+    "step_callback": "object",
+    "task_callback": "object",
+    "prompt_file": "object",
+    "output_log_file": "object",
+}
+
+task_properties = {
+    "id": "object",
+    "used_tools": "int",
+    "tools_errors": "int",
+    "delegations": "int",
+    "i18n": "object",
+    "thread": "object",
+    "prompt_context": "object",
+    "description": "str",
+    "expected_output": "str",
+    "config": "object",
+    "callback": "str",
+    "agent": "object",
+    "context": "object",
+    "async_execution": "bool",
+    "output_json": "object",
+    "output_pydantic": "object",
+    "output_file": "object",
+    "output": "object",
+    "tools": "object",
+    "human_input": "bool",
+}
+
+agent_properties = {
+    "formatting_errors": "int",
+    "id": "object",
+    "role": "str",
+    "goal": "str",
+    "backstory": "str",
+    "cache": "bool",
+    "config": "object",
+    "max_rpm": "int",
+    "verbose": "bool",
+    "allow_delegation": "bool",
+    "tools": "object",
+    "max_iter": "int",
+    "max_execution_time": "object",
+    "agent_executor": "object",
+    "tools_handler": "object",
+    "force_answer_max_iterations": "int",
+    "crew": "object",
+    "cache_handler": "object",
+    "step_callback": "object",
+    "i18n": "object",
+    "llm": "object",
+    "function_calling_llm": "object",
+    "callbacks": "object",
+    "system_template": "object",
+    "prompt_template": "object",
+    "response_template": "object",
+}
+
+
+def patch_crew(operation_name, version, tracer):
+    def traced_method(wrapped, instance, args, kwargs):
+
+        service_provider = SERVICE_PROVIDERS["CREWAI"]
+        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": "framework",
+            "langtrace.service.version": version,
+            "langtrace.version": v(LANGTRACE_SDK_NAME),
+            **(extra_attributes if extra_attributes is not None else {}),
+        }
+
+        crew_config = {}
+        for key, value in instance.__dict__.items():
+            if instance.__class__.__name__ == "Crew":
+                if key in crew_properties and value is not None:
+                    if crew_properties[key] == "json":
+                        crew_config[key] = json.dumps(value)
+                    elif crew_properties[key] == "object":
+                        crew_config[key] = str(value)
+                    else:
+                        crew_config[key] = value
+            elif instance.__class__.__name__ == "Agent":
+                if key in agent_properties and value is not None:
+                    if agent_properties[key] == "json":
+                        crew_config[key] = json.dumps(value)
+                    elif agent_properties[key] == "object":
+                        crew_config[key] = str(value)
+                    else:
+                        crew_config[key] = value
+            elif instance.__class__.__name__ == "Task":
+                if key in task_properties and value is not None:
+                    if task_properties[key] == "json":
+                        crew_config[key] = json.dumps(value)
+                    elif task_properties[key] == "object":
+                        crew_config[key] = str(value)
+                    else:
+                        crew_config[key] = value
+        if crew_config:
+            if instance.__class__.__name__ == "Crew":
+                if "inputs" in kwargs and kwargs["inputs"]:
+                    crew_config["inputs"] = json.dumps(kwargs["inputs"])
+                span_attributes["crewai.crew.config"] = json.dumps(crew_config)
+            elif instance.__class__.__name__ == "Agent":
+                if "context" in kwargs and kwargs["context"]:
+                    crew_config["context"] = json.dumps(kwargs["context"])
+                span_attributes["crewai.agent.config"] = json.dumps(crew_config)
+            elif instance.__class__.__name__ == "Task":
+                span_attributes["crewai.task.config"] = json.dumps(crew_config)
+
+        attributes = FrameworkSpanAttributes(**span_attributes)
+
+        with tracer.start_as_current_span(operation_name, kind=SpanKind.CLIENT) as span:
+            _set_input_attributes(span, kwargs, attributes)
+
+            try:
+                result = wrapped(*args, **kwargs)
+                if result:
+                    span.set_status(Status(StatusCode.OK))
+
+                span.end()
+                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
+
+
+@silently_fail
+def _set_input_attributes(span, kwargs, attributes):
+    for field, value in attributes.model_dump(by_alias=True).items():
+        set_span_attribute(span, field, value)

langtrace_python_sdk/instrumentation/dspy/__init__.py

@@ -1,3 +1,3 @@
-from .instrumentation import DspyInstrumentor
+from .instrumentation import DspyInstrumentation
 
-__all__ = ["DspyInstrumentor"]
+__all__ = ["DspyInstrumentation"]

langtrace_python_sdk/instrumentation/dspy/instrumentation.py

@@ -22,7 +22,7 @@ from importlib_metadata import version as v
 from .patch import patch_bootstrapfewshot_optimizer, patch_signature, patch_evaluate
 
 
-class DspyInstrumentor(BaseInstrumentor):
+class DspyInstrumentation(BaseInstrumentor):
     """
     The DspyInstrumentor class represents the DSPy instrumentation"""
 

langtrace_python_sdk/instrumentation/dspy/patch.py

@@ -37,24 +37,24 @@ def patch_bootstrapfewshot_optimizer(operation_name, version, tracer):
                     "signature": str(args[0].prog.signature) if args[0].prog.signature else None,
                 }
                 span_attributes["dspy.optimizer.module.prog"] = json.dumps(prog)
-        if instance.metric:
+        if "metric" in instance and instance.metric:
             span_attributes["dspy.optimizer.metric"] = instance.metric.__name__
         if kwargs.get("trainset") and len(kwargs.get("trainset")) > 0:
             span_attributes["dspy.optimizer.trainset"] = str(kwargs.get("trainset"))
         config = {}
-        if instance.metric_threshold:
+        if "metric_threshold" in instance and instance.metric_threshold:
             config["metric_threshold"] = instance.metric_threshold
-        if instance.teacher_settings:
+        if "teacher_settings" in instance and instance.teacher_settings:
             config["teacher_settings"] = instance.teacher_settings
-        if instance.max_bootstrapped_demos:
+        if "max_bootstrapped_demos" in instance and instance.max_bootstrapped_demos:
             config["max_bootstrapped_demos"] = instance.max_bootstrapped_demos
-        if instance.max_labeled_demos:
+        if "max_labeled_demos" in instance and instance.max_labeled_demos:
             config["max_labeled_demos"] = instance.max_labeled_demos
-        if instance.max_rounds:
+        if "max_rounds" in instance and instance.max_rounds:
             config["max_rounds"] = instance.max_rounds
-        if instance.max_errors:
+        if "max_errors" in instance and instance.max_errors:
             config["max_errors"] = instance.max_errors
-        if instance.error_count:
+        if "error_count" in instance and instance.error_count:
             config["error_count"] = instance.error_count
         if config and len(config) > 0:
             span_attributes["dspy.optimizer.config"] = json.dumps(config)
@@ -149,25 +149,25 @@ def patch_evaluate(operation_name, version, tracer):
             **(extra_attributes if extra_attributes is not None else {}),
         }
 
-        if instance.devset is not None:
+        if "devset" in instance and instance.devset is not None:
             span_attributes["dspy.evaluate.devset"] = str(instance.devset)
-        if instance.display is not None:
+        if "display" in instance and instance.display is not None:
             span_attributes["dspy.evaluate.display"] = str(instance.display)
-        if instance.num_threads is not None:
+        if "num_threads" in instance and instance.num_threads is not None:
             span_attributes["dspy.evaluate.num_threads"] = str(instance.num_threads)
-        if instance.return_outputs is not None:
+        if "return_outputs" in instance and instance.return_outputs is not None:
             span_attributes["dspy.evaluate.return_outputs"] = str(instance.return_outputs)
-        if instance.display_table is not None:
+        if "display_table" in instance and instance.display_table is not None:
             span_attributes["dspy.evaluate.display_table"] = str(instance.display_table)
-        if instance.display_progress is not None:
+        if "display_progress" in instance and instance.display_progress is not None:
             span_attributes["dspy.evaluate.display_progress"] = str(instance.display_progress)
-        if instance.metric is not None:
+        if "metric" in instance and instance.metric is not None:
             span_attributes["dspy.evaluate.metric"] = instance.metric.__name__
-        if instance.error_count is not None:
+        if "error_count" in instance and instance.error_count is not None:
             span_attributes["dspy.evaluate.error_count"] = str(instance.error_count)
-        if instance.error_lock is not None:
+        if "error_lock" in instance and instance.error_lock is not None:
             span_attributes["dspy.evaluate.error_lock"] = str(instance.error_lock)
-        if instance.max_errors is not None:
+        if "max_errors" in instance and instance.max_errors is not None:
             span_attributes["dspy.evaluate.max_errors"] = str(instance.max_errors)
         if args and len(args) > 0:
             span_attributes["dspy.evaluate.args"] = str(args)

langtrace_python_sdk/instrumentation/openai/patch.py

@@ -275,6 +275,110 @@ def images_edit(original_method, version, tracer):
     return traced_method
 
 
+class StreamWrapper:
+    def __init__(self, stream, span, prompt_tokens, function_call=False, tool_calls=False):
+        self.stream = stream
+        self.span = span
+        self.prompt_tokens = prompt_tokens
+        self.function_call = function_call
+        self.tool_calls = tool_calls
+        self.result_content = []
+        self.completion_tokens = 0
+
+    def __enter__(self):
+        self.span.add_event(Event.STREAM_START.value)
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        self.span.add_event(Event.STREAM_END.value)
+        self.span.set_attribute(
+            "llm.token.counts",
+            json.dumps(
+                {
+                    "input_tokens": self.prompt_tokens,
+                    "output_tokens": self.completion_tokens,
+                    "total_tokens": self.prompt_tokens + self.completion_tokens,
+                }
+            ),
+        )
+        self.span.set_attribute(
+            "llm.responses",
+            json.dumps(
+                [
+                    {
+                        "role": "assistant",
+                        "content": "".join(self.result_content),
+                    }
+                ]
+            ),
+        )
+        self.span.set_status(StatusCode.OK)
+        self.span.end()
+
+    def __iter__(self):
+        return self
+
+    def __next__(self):
+        try:
+            chunk = next(self.stream)
+            self.process_chunk(chunk)
+            return chunk
+        except StopIteration:
+            raise
+
+    def process_chunk(self, chunk):
+        if hasattr(chunk, "model") and chunk.model is not None:
+            self.span.set_attribute("llm.model", chunk.model)
+        if hasattr(chunk, "choices") and chunk.choices is not None:
+            content = []
+            if not self.function_call and not self.tool_calls:
+                for choice in chunk.choices:
+                    if choice.delta and choice.delta.content is not None:
+                        token_counts = estimate_tokens(choice.delta.content)
+                        self.completion_tokens += token_counts
+                        content = [choice.delta.content]
+            elif self.function_call:
+                for choice in chunk.choices:
+                    if (
+                        choice.delta
+                        and choice.delta.function_call is not None
+                        and choice.delta.function_call.arguments is not None
+                    ):
+                        token_counts = estimate_tokens(
+                            choice.delta.function_call.arguments
+                        )
+                        self.completion_tokens += token_counts
+                        content = [choice.delta.function_call.arguments]
+            elif self.tool_calls:
+                for choice in chunk.choices:
+                    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
+                            ):
+                                token_counts = estimate_tokens(
+                                    tool_call.function.arguments
+                                )
+                                self.completion_tokens += token_counts
+                                content.append(tool_call.function.arguments)
+            self.span.add_event(
+                Event.STREAM_OUTPUT.value,
+                {
+                    "response": (
+                        "".join(content)
+                        if len(content) > 0 and content[0] is not None
+                        else ""
+                    )
+                },
+            )
+            if content:
+                self.result_content.append(content[0])
+
+
 def chat_completions_create(original_method, version, tracer):
     """Wrap the `create` method of the `ChatCompletion` class to trace it."""
 
@@ -427,7 +531,7 @@ def chat_completions_create(original_method, version, tracer):
                             json.dumps(function), kwargs.get("model")
                         )
 
-                return handle_streaming_response(
+                return StreamWrapper(
                     result,
                     span,
                     prompt_tokens,
@@ -441,98 +545,6 @@ def chat_completions_create(original_method, version, tracer):
             span.end()
             raise
 
-    def handle_streaming_response(
-        result, span, prompt_tokens, function_call=False, tool_calls=False
-    ):
-        """Process and yield streaming response chunks."""
-        result_content = []
-        span.add_event(Event.STREAM_START.value)
-        completion_tokens = 0
-        try:
-            for chunk in result:
-                if hasattr(chunk, "model") and chunk.model is not None:
-                    span.set_attribute("llm.model", chunk.model)
-                if hasattr(chunk, "choices") and chunk.choices is not None:
-                    if not function_call and not tool_calls:
-                        for choice in chunk.choices:
-                            if choice.delta and choice.delta.content is not None:
-                                token_counts = estimate_tokens(choice.delta.content)
-                                completion_tokens += token_counts
-                                content = [choice.delta.content]
-                    elif function_call:
-                        for choice in chunk.choices:
-                            if (
-                                choice.delta
-                                and choice.delta.function_call is not None
-                                and choice.delta.function_call.arguments is not None
-                            ):
-                                token_counts = estimate_tokens(
-                                    choice.delta.function_call.arguments
-                                )
-                                completion_tokens += token_counts
-                                content = [choice.delta.function_call.arguments]
-                    elif tool_calls:
-                        for choice in chunk.choices:
-                            tool_call = ""
-                            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
-                                    ):
-                                        token_counts = estimate_tokens(
-                                            tool_call.function.arguments
-                                        )
-                                        completion_tokens += token_counts
-                                        content = content + [
-                                            tool_call.function.arguments
-                                        ]
-                                    else:
-                                        content = content + []
-                else:
-                    content = []
-                span.add_event(
-                    Event.STREAM_OUTPUT.value,
-                    {
-                        "response": (
-                            "".join(content)
-                            if len(content) > 0 and content[0] is not None
-                            else ""
-                        )
-                    },
-                )
-                result_content.append(content[0] if len(content) > 0 else "")
-                yield chunk
-        finally:
-            # Finalize span after processing all chunks
-            span.add_event(Event.STREAM_END.value)
-            span.set_attribute(
-                "llm.token.counts",
-                json.dumps(
-                    {
-                        "input_tokens": prompt_tokens,
-                        "output_tokens": completion_tokens,
-                        "total_tokens": prompt_tokens + completion_tokens,
-                    }
-                ),
-            )
-            span.set_attribute(
-                "llm.responses",
-                json.dumps(
-                    [
-                        {
-                            "role": "assistant",
-                            "content": "".join(result_content),
-                        }
-                    ]
-                ),
-            )
-            span.set_status(StatusCode.OK)
-            span.end()
-
     # return the wrapped method
     return traced_method
 
@@ -687,7 +699,7 @@ def async_chat_completions_create(original_method, version, tracer):
                             json.dumps(function), kwargs.get("model")
                         )
 
-                return ahandle_streaming_response(
+                return StreamWrapper(
                     result,
                     span,
                     prompt_tokens,
@@ -701,99 +713,6 @@ def async_chat_completions_create(original_method, version, tracer):
             span.end()
             raise
 
-    async def ahandle_streaming_response(
-        result, span, prompt_tokens, function_call=False, tool_calls=False
-    ):
-        """Process and yield streaming response chunks."""
-        result_content = []
-        span.add_event(Event.STREAM_START.value)
-        completion_tokens = 0
-        try:
-            content = []
-            async for chunk in result:
-                if hasattr(chunk, "model") and chunk.model is not None:
-                    span.set_attribute("llm.model", chunk.model)
-                if hasattr(chunk, "choices") and chunk.choices is not None:
-                    if not function_call and not tool_calls:
-                        for choice in chunk.choices:
-                            if choice.delta and choice.delta.content is not None:
-                                token_counts = estimate_tokens(choice.delta.content)
-                                completion_tokens += token_counts
-                                content = [choice.delta.content]
-                    elif function_call:
-                        for choice in chunk.choices:
-                            if (
-                                choice.delta
-                                and choice.delta.function_call
-                                and choice.delta.function_call.arguments is not None
-                            ):
-                                token_counts = estimate_tokens(
-                                    choice.delta.function_call.arguments
-                                )
-                                completion_tokens += token_counts
-                                content = [choice.delta.function_call.arguments]
-                    elif tool_calls:
-                        for choice in chunk.choices:
-                            tool_call = ""
-                            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
-                                    ):
-                                        token_counts = estimate_tokens(
-                                            tool_call.function.arguments
-                                        )
-                                        completion_tokens += token_counts
-                                        content = content + [
-                                            tool_call.function.arguments
-                                        ]
-                                    else:
-                                        content = content + []
-                else:
-                    content = []
-                span.add_event(
-                    Event.STREAM_OUTPUT.value,
-                    {
-                        "response": (
-                            "".join(content)
-                            if len(content) > 0 and content[0] is not None
-                            else ""
-                        )
-                    },
-                )
-                result_content.append(content[0] if len(content) > 0 else "")
-                yield chunk
-        finally:
-            # Finalize span after processing all chunks
-            span.add_event(Event.STREAM_END.value)
-            span.set_attribute(
-                "llm.token.counts",
-                json.dumps(
-                    {
-                        "input_tokens": prompt_tokens,
-                        "output_tokens": completion_tokens,
-                        "total_tokens": prompt_tokens + completion_tokens,
-                    }
-                ),
-            )
-            span.set_attribute(
-                "llm.responses",
-                json.dumps(
-                    [
-                        {
-                            "role": "assistant",
-                            "content": "".join(result_content),
-                        }
-                    ]
-                ),
-            )
-            span.set_status(StatusCode.OK)
-            span.end()
-
     # return the wrapped method
     return traced_method
 

langtrace_python_sdk/langtrace.py

@@ -40,6 +40,7 @@ from langtrace_python_sdk.instrumentation import (
     AnthropicInstrumentation,
     ChromaInstrumentation,
     CohereInstrumentation,
+    CrewAIInstrumentation,
     GroqInstrumentation,
     LangchainInstrumentation,
     LangchainCommunityInstrumentation,
@@ -51,7 +52,7 @@ from langtrace_python_sdk.instrumentation import (
     QdrantInstrumentation,
     WeaviateInstrumentation,
     OllamaInstrumentor,
-    DspyInstrumentor,
+    DspyInstrumentation,
 )
 from opentelemetry.instrumentation.sqlalchemy import SQLAlchemyInstrumentor
 from colorama import Fore
@@ -109,7 +110,8 @@ def init(
         "weaviate": WeaviateInstrumentation(),
         "sqlalchemy": SQLAlchemyInstrumentor(),
         "ollama": OllamaInstrumentor(),
-        "dspy": DspyInstrumentor(),
+        "dspy": DspyInstrumentation(),
+        "crewai": CrewAIInstrumentation(),
     }
 
     init_instrumentations(disable_instrumentations, all_instrumentations)

langtrace_python_sdk/version.py

@@ -1 +1 @@
-__version__ = "2.1.21"
+__version__ = "2.1.22"

{langtrace_python_sdk-2.1.21.dist-info → langtrace_python_sdk-2.1.22.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: langtrace-python-sdk
-Version: 2.1.21
+Version: 2.1.22
 Summary: Python SDK for LangTrace
 Project-URL: Homepage, https://github.com/Scale3-Labs/langtrace-python-sdk
 Author-email: Scale3 Labs <engineering@scale3labs.com>

{langtrace_python_sdk-2.1.21.dist-info → langtrace_python_sdk-2.1.22.dist-info}/RECORD

@@ -9,6 +9,7 @@ examples/cohere_example/chat_stream.py,sha256=daWGrj2yucN9sWKGiPzTw18clh23Grv-MI
 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/crewai_example/basic.py,sha256=PBu4f8yQfZO1L_22UDm_ReU9lnEcycjZcGuy5UpgDJM,1948
 examples/dspy_example/math_problems_cot.py,sha256=Z98nB6myt8WJse2dWS6Ap7CFUhC27lBNb37R1Gg80VQ,1282
 examples/dspy_example/program_of_thought_basic.py,sha256=oEbtJdeKENMUbex25-zyStWwurRWW6OdP0KDs-jUkko,984
 examples/dspy_example/quiz_gen.py,sha256=OyGhepeX8meKOtLdmlYUjMD2ECk-ZQuQXUZif1hFQY4,3371
@@ -49,15 +50,15 @@ examples/qdrant_example/basic.py,sha256=DCMjHSuBZKkhEjCkwy5d5La9WMyW0lCWqtcZWiFC
 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=VZM6i71NR7pBQK6XvJWRelknuTYUhqwqE7PlicKa5Wg,1166
-langtrace_python_sdk/langtrace.py,sha256=enOnfpDPoFibbUzUEY2vV8S3No0dySdXxo3E76rZ8mA,7410
-langtrace_python_sdk/version.py,sha256=msdDHhj5VY39vNBuQYgH15DYZ-b9Z3koedzF3TN1kW8,23
+langtrace_python_sdk/langtrace.py,sha256=pG_dWzzQxUP8r5SAMUwRScb6IopINcda1CZvJERjXBo,7486
+langtrace_python_sdk/version.py,sha256=8X_w-VRCyU1lQMWejIqkf5tmhnrgO7csLi27xV88LJs,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
 langtrace_python_sdk/constants/instrumentation/anthropic.py,sha256=YX3llt3zwDY6XrYk3CB8WEVqgrzRXEw_ffyk56JoF3k,126
 langtrace_python_sdk/constants/instrumentation/chroma.py,sha256=hiPGYdHS0Yj4Kh3eaYBbuCAl_swqIygu80yFqkOgdak,955
 langtrace_python_sdk/constants/instrumentation/cohere.py,sha256=tf9sDfb5K3qOAHChEE5o8eYWPZ1io58VsOjZDCZPxfw,577
-langtrace_python_sdk/constants/instrumentation/common.py,sha256=kqs3ST0RTcXpQKPSQ6R6hoBabQsLGPNB_rvHlQXsn80,826
+langtrace_python_sdk/constants/instrumentation/common.py,sha256=KodH_uGGjWGGP8rqTi7Ua-osjUwtPKslx69DJRbDiT4,850
 langtrace_python_sdk/constants/instrumentation/groq.py,sha256=VFXmIl4aqGY_fS0PAmjPj_Qm7Tibxbx7Ur_e7rQpqXc,134
 langtrace_python_sdk/constants/instrumentation/ollama.py,sha256=zFfSUKX5v1c612doKSxsmIwQeeQxSPkFp_ZzjWQSPNE,142
 langtrace_python_sdk/constants/instrumentation/openai.py,sha256=uEOH5UXapU2DSf2AdgXTRhhJEHGWXUNFkUGD5QafflM,1164
@@ -67,7 +68,7 @@ langtrace_python_sdk/constants/instrumentation/weaviate.py,sha256=Iytf2OpB_irZYE
 langtrace_python_sdk/extensions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/extensions/langtrace_exporter.py,sha256=gWVRU2DlB4xjZ4ww7M63DaLiAN5zQ2k1HPrythmjEdo,4202
 langtrace_python_sdk/extensions/langtrace_filesystem.py,sha256=qpnkpkuTZ2yhGgpBK64QJLt0T1iL-1zpEMPz4quJ_ng,6925
-langtrace_python_sdk/instrumentation/__init__.py,sha256=y5ftlcrAjSHvoQ_v8ffDlsP5la3SG-B4PKpPZIeoTds,1166
+langtrace_python_sdk/instrumentation/__init__.py,sha256=BYTzQSmHDbKAbtV98UDjxzV1UoNq_wtsWV7V4PZPzbc,1243
 langtrace_python_sdk/instrumentation/anthropic/__init__.py,sha256=donrurJAGYlxrSRA3BIf76jGeUcAx9Tq8CVpah68S0Y,101
 langtrace_python_sdk/instrumentation/anthropic/instrumentation.py,sha256=-srgE8qumAn0ulQYZxMa8ch-9IBH0XgBW_rfEnGk6LI,1684
 langtrace_python_sdk/instrumentation/anthropic/patch.py,sha256=52GRkW7fhF2VB9JvYGMaPwOvWH3Q5rsIEaVlnFm_Zz0,8548
@@ -77,9 +78,12 @@ langtrace_python_sdk/instrumentation/chroma/patch.py,sha256=HhIkXGbVfzpEB4edxzlG
 langtrace_python_sdk/instrumentation/cohere/__init__.py,sha256=sGUSLdTUyYf36Tm6L5jQflhzCqvmWrhnBOMYHjvp6Hs,95
 langtrace_python_sdk/instrumentation/cohere/instrumentation.py,sha256=YQFHZIBd7SSPD4b6Va-ZR0thf_AuBCqj5yzHLHJVWnM,2121
 langtrace_python_sdk/instrumentation/cohere/patch.py,sha256=uGjZm_c1-xpZilHD8ljGr4AqQB3UNU04SKl9s2alGzs,26726
-langtrace_python_sdk/instrumentation/dspy/__init__.py,sha256=qTdAqLOuLX7dsiD_pdqe23pz9Ehit4aC5771WJe2Q8I,78
-langtrace_python_sdk/instrumentation/dspy/instrumentation.py,sha256=fvnTpDoMlCzpUyhlZefUgUO1yd50dH27UPDQ6EBcrJA,2920
-langtrace_python_sdk/instrumentation/dspy/patch.py,sha256=8VrY3Et0zQRagYkSU6jIv6tvY6PpvVTDRRcWtgz3QEw,8662
+langtrace_python_sdk/instrumentation/crewai/__init__.py,sha256=_UBKfvQv7l0g2_wnmA5F6CdSAFH0atNOVPd49zsN3aM,88
+langtrace_python_sdk/instrumentation/crewai/instrumentation.py,sha256=W8PLTLzgEdrEx1DCo79wNs9xcuWK0YuxEICLavOESDw,1715
+langtrace_python_sdk/instrumentation/crewai/patch.py,sha256=Vnpip9Pbk4UFbTFHoUrHtAnDgsaihwSvZBgtUeOtLr8,6109
+langtrace_python_sdk/instrumentation/dspy/__init__.py,sha256=tM1srfi_QgyCzrde4izojMrRq2Wm7Dj5QUvVQXIJzkk,84
+langtrace_python_sdk/instrumentation/dspy/instrumentation.py,sha256=Y-_qcH5jCT4TbGvci6Iw0_OWUUlCmovMxnysj6NCnXI,2923
+langtrace_python_sdk/instrumentation/dspy/patch.py,sha256=0zBlrzvGmQEhJpb44L8fkTIhMH5i5pWOkYoSH9e8D6U,9216
 langtrace_python_sdk/instrumentation/groq/__init__.py,sha256=ZXeq_nrej6Lm_uoMFEg8wbSejhjB2UJ5IoHQBPc2-C0,91
 langtrace_python_sdk/instrumentation/groq/instrumentation.py,sha256=Ttf07XVKhdYY1_fqJc7QWiSdmgEhEVyQB_3Az2_wqYo,1832
 langtrace_python_sdk/instrumentation/groq/patch.py,sha256=EKH9tjoDbwWavIAwUadaR5tFiy-S69h1IxxzH_1SSRg,26354
@@ -103,7 +107,7 @@ langtrace_python_sdk/instrumentation/ollama/instrumentation.py,sha256=jdsvkqUJAA
 langtrace_python_sdk/instrumentation/ollama/patch.py,sha256=CGLgt0qZg3WDr6XLn32qqs4D9USdMmeTmJxctbAsDeM,7908
 langtrace_python_sdk/instrumentation/openai/__init__.py,sha256=VPHRNCQEdkizIVP2d0Uw_a7t8XOTSTprEIB8oboJFbs,95
 langtrace_python_sdk/instrumentation/openai/instrumentation.py,sha256=A0BJHRLcZ74TNVg6I0I9M5YWvSpAtXwMmME6N5CEQ_M,2945
-langtrace_python_sdk/instrumentation/openai/patch.py,sha256=uUoeeZMAlH8PTZyIObjAI8kU2H5iHuZbPF09Gk70IKo,41103
+langtrace_python_sdk/instrumentation/openai/patch.py,sha256=-eKopJp6m56mg4553wrDr0ibGoCPpYH2iyNMh1lX9yc,36526
 langtrace_python_sdk/instrumentation/pinecone/__init__.py,sha256=DzXyGh9_MGWveJvXULkFwdkf7PbG2s3bAWtT1Dmz7Ok,99
 langtrace_python_sdk/instrumentation/pinecone/instrumentation.py,sha256=HDXkRITrVPwdQEoOYJOfMzZE_2-vDDvuqHTlD8W1lQw,1845
 langtrace_python_sdk/instrumentation/pinecone/patch.py,sha256=KiIRRz8kk47FllFT746Cb_w6F6M60AN_pcsguD979E4,5172
@@ -159,8 +163,8 @@ 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.21.dist-info/METADATA,sha256=y6OnKJCGYcnpezk9jVMO1Y-EBummvLJ92jRUOa1M4hE,13161
-langtrace_python_sdk-2.1.21.dist-info/WHEEL,sha256=zEMcRr9Kr03x1ozGwg5v9NQBKn3kndp6LSoSlVg-jhU,87
-langtrace_python_sdk-2.1.21.dist-info/entry_points.txt,sha256=1_b9-qvf2fE7uQNZcbUei9vLpFZBbbh9LrtGw95ssAo,70
-langtrace_python_sdk-2.1.21.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-langtrace_python_sdk-2.1.21.dist-info/RECORD,,
+langtrace_python_sdk-2.1.22.dist-info/METADATA,sha256=OBp1W_aK6Om6TkkaFfhhyo-7tSDZ3TC49Hdbs6KdFR0,13161
+langtrace_python_sdk-2.1.22.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+langtrace_python_sdk-2.1.22.dist-info/entry_points.txt,sha256=1_b9-qvf2fE7uQNZcbUei9vLpFZBbbh9LrtGw95ssAo,70
+langtrace_python_sdk-2.1.22.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+langtrace_python_sdk-2.1.22.dist-info/RECORD,,

{langtrace_python_sdk-2.1.21.dist-info → langtrace_python_sdk-2.1.22.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: hatchling 1.24.2
+Generator: hatchling 1.25.0
 Root-Is-Purelib: true
 Tag: py3-none-any