Add log_argument

Signed-off-by: András Schmelczer <andras@schmelczer.dev>

good_ai/src/good_ai/good_ai/views/trace.py

@@ -1,16 +1,27 @@
-from typing import Any, List
+from typing import Any, Dict, List, Optional
 from uuid import uuid4
 
-from pydantic import BaseModel
+from pydantic import BaseModel, validator
 
 from .model import Model
 
 
 class Trace(BaseModel):
-    id = str(uuid4())
+    evaluation_id: Optional[str]
     created: str
     execution_time_ms: float
-    input: Any
+    logged_values: Dict[str, Any]
     models: List[Model]
     output: Any
     evaluation: Any = None
+
+    @validator("evaluation_id", always=True)
+    def validate_single_set(
+        cls, v: Optional[str], values: Dict[str, Any]
+    ) -> Optional[str]:
+        if not v:
+            return str(uuid4())
+        return v
+
+    def __hash__(self) -> int:
+        return hash((type(self),) + tuple(self.__dict__.values()))