deepeval 3.6.7__py3-none-any.whl → 3.6.9__py3-none-any.whl

deepeval/integrations/llama_index/agent/patched.py

@@ -1,68 +0,0 @@
-from typing import TypeVar, cast, Optional, List
-from pydantic import Field
-from deepeval.metrics import BaseMetric
-from deepeval.telemetry import capture_tracing_integration
-
-try:
-    from llama_index.core.agent.workflow import (
-        FunctionAgent,
-        ReActAgent,
-        CodeActAgent,
-    )
-
-    is_llama_index_installed = True
-except:
-    is_llama_index_installed = False
-
-
-def is_llama_index_agent_installed():
-    if not is_llama_index_installed:
-        raise ImportError(
-            "llama-index is neccesary for this functionality. Please install it with `pip install llama-index` or with package manager of choice."
-        )
-
-
-T = TypeVar("T", bound=type)
-
-
-def with_metrics(cls: T) -> T:
-    class SubClassWithMetric(cls):  # type: ignore
-        metric_collection: Optional[str] = Field(default=None)
-        metrics: Optional[List[BaseMetric]] = Field(default_factory=list)
-
-    SubClassWithMetric.__name__ = cls.__name__
-    SubClassWithMetric.__qualname__ = cls.__qualname__
-    return cast(T, SubClassWithMetric)
-
-
-@with_metrics
-class FunctionAgent(FunctionAgent):
-    def __init__(self, *args, metric_collection=None, metrics=None, **kwargs):
-        with capture_tracing_integration(
-            "llama_index.agent.patched.FunctionAgent"
-        ):
-            super().__init__(*args, **kwargs)
-            self.metric_collection = metric_collection
-            self.metrics = metrics
-
-
-@with_metrics
-class ReActAgent(ReActAgent):
-    def __init__(self, *args, metric_collection=None, metrics=None, **kwargs):
-        with capture_tracing_integration(
-            "llama_index.agent.patched.ReActAgent"
-        ):
-            super().__init__(*args, **kwargs)
-            self.metric_collection = metric_collection
-            self.metrics = metrics
-
-
-@with_metrics
-class CodeActAgent(CodeActAgent):
-    def __init__(self, *args, metric_collection=None, metrics=None, **kwargs):
-        with capture_tracing_integration(
-            "llama_index.agent.patched.CodeActAgent"
-        ):
-            super().__init__(*args, **kwargs)
-            self.metric_collection = metric_collection
-            self.metrics = metrics

deepeval/tracing/message_types/__init__.py

@@ -1,10 +0,0 @@
-from .messages import TextMessage, ToolCallMessage
-from .tools import BaseTool, ToolSchema, ToolOutput
-
-__all__ = [
-    "BaseTool",
-    "TextMessage",
-    "ToolCallMessage",
-    "ToolSchema",
-    "ToolOutput",
-]

deepeval/tracing/message_types/base.py

@@ -1,6 +0,0 @@
-from typing import Literal
-from pydantic import BaseModel
-
-
-class BaseMessage(BaseModel):
-    role: Literal["user", "assistant"]

deepeval/tracing/message_types/messages.py

@@ -1,14 +0,0 @@
-from typing import Literal, Dict, Any
-from .base import BaseMessage
-
-
-class TextMessage(BaseMessage):
-    type: Literal["text", "thinking"]
-    content: str
-
-
-class ToolCallMessage(BaseMessage):
-    """This is a message for tool calls in response.choices[0].message.tool_calls"""
-
-    name: str
-    args: Dict[str, Any]

deepeval/tracing/message_types/tools.py

@@ -1,18 +0,0 @@
-from typing import Any, Optional, Dict
-from pydantic import BaseModel
-
-
-class BaseTool(BaseModel):
-    name: str
-    description: Optional[str] = None
-
-
-class ToolSchema(BaseTool):
-    parameters: Dict[str, Any]
-    is_called: Optional[bool] = False
-
-
-class ToolOutput(BaseTool):
-    """Output of the tool function"""
-
-    output: Any