Move files

src/great_ai/views/trace.py

@@ -1,89 +0,0 @@
-from pprint import pformat
-from typing import Any, Dict, Generic, List, Optional, TypeVar
-from uuid import uuid4
-
-from pydantic import BaseModel, Extra, validator
-
-from ..helper import HashableBaseModel
-from .model import Model
-
-T = TypeVar("T")
-
-
-class Trace(Generic[T], HashableBaseModel):
-    trace_id: Optional[str]
-    created: str
-    original_execution_time_ms: float
-    logged_values: Dict[str, Any]
-    models: List[Model]
-    exception: Optional[str]
-    output: T
-    feedback: Any = None
-    tags: List[str]
-
-    class Config:
-        extra = Extra.ignore
-
-    @validator("trace_id", always=True)
-    def generate_id(cls, v: Optional[str], values: Dict[str, Any]) -> Optional[str]:
-        if not v:
-            return str(uuid4())
-        return v
-
-    @property
-    def input(self) -> Any:
-        return (
-            self.logged_values["input"]
-            if list(self.logged_values.keys()) == ["input"]
-            else self.logged_values
-        )
-
-    @property
-    def models_flat(self) -> str:
-        return ", ".join(f"{m.key}:{m.version}" for m in self.models)
-
-    @property
-    def output_flat(self) -> str:
-        return pformat(self.output, indent=2, compact=True)
-
-    @property
-    def exception_flat(self) -> str:
-        return (
-            "null"
-            if self.exception is None
-            else pformat(self.exception, indent=2, compact=True)
-        )
-
-    @property
-    def feedback_flat(self) -> str:
-        return (
-            "null"
-            if self.feedback is None
-            else pformat(self.feedback, indent=2, compact=True)
-        )
-
-    @property
-    def tags_flat(self) -> str:
-        return ",\n".join(self.tags)
-
-    def to_flat_dict(self, include_original: bool = True) -> Dict[str, Any]:
-        return {
-            **(
-                self.dict()
-                if include_original
-                else {
-                    "trace_id": self.trace_id,
-                    "created": self.created,
-                    "original_execution_time_ms": self.original_execution_time_ms,
-                }
-            ),
-            **{
-                k: v.dict() if isinstance(v, BaseModel) else v
-                for k, v in self.logged_values.items()
-            },
-            "models_flat": self.models_flat,
-            "exception_flat": self.exception_flat,
-            "output_flat": self.output_flat,
-            "feedback_flat": self.feedback_flat,
-            "tags_flat": self.tags_flat,
-        }