pygeai 0.2.7b36__py3-none-any.whl → 0.2.7b38__py3-none-any.whl

pygeai/core/base/mappers.py

@@ -351,13 +351,22 @@ class ModelMapper:
     @classmethod
     def map_to_item(cls, data: dict) -> RequestItem:
         return RequestItem(
+            api_token=data.get('apiToken'),
             assistant=data.get('assistant'),
+            cost=data.get('cost'),
+            elapsed_time_ms=data.get('elapsedTimeMs'),
+            end_timestamp=data.get('endTimestamp'),
+            feedback=data.get('feedback'),
             intent=data.get('intent'),
-            timestamp=data.get('timestamp'),
+            module=data.get('module'),
             prompt=data.get('prompt'),
             output=data.get('output'),
             input_text=data.get('inputText'),
-            status=data.get('status')
+            rag_sources_consulted=data.get('ragSourcesConsulted'),
+            session_id=data.get('sessionId'),
+            start_timestamp=data.get('startTimestamp'),
+            status=data.get('status'),
+            timestamp=data.get('timestamp')
         )
 
     @classmethod

pygeai/core/models.py

@@ -1,6 +1,6 @@
 from datetime import datetime
 
-from pydantic import Field, field_validator
+from pydantic import Field, field_validator, model_validator
 from typing import Optional, Literal, Any, List, Union, Iterator
 
 from pygeai.core import CustomBaseModel
@@ -247,23 +247,32 @@ class Project(CustomBaseModel):
 
 class RequestItem(CustomBaseModel):
     """
-    {
-      "assistant": "string",
-      "intent": "string",
-      "timestamp": "string",
-      "prompt": "string",
-      "output": "string",
-      "inputText": "string",
-      "status": "string"
-    }
+    Represents a request item with metadata about an API interaction.
+    Follows a JSON schema with camelCase aliases for external compatibility.
     """
+    api_token: str = Field(..., alias="apiToken")
     assistant: str = Field(..., alias="assistant")
+    cost: float = Field(..., alias="cost")
+    elapsed_time_ms: int = Field(..., alias="elapsedTimeMs")
+    end_timestamp: datetime = Field(..., alias="endTimestamp")
+    feedback: Optional[str] = Field(None, alias="feedback")
     intent: Optional[str] = Field(None, alias="intent")
-    timestamp: str = Field(..., alias="timestamp")
+    module: str = Field(..., alias="module")
     prompt: Optional[str] = Field(None, alias="prompt")
     output: Optional[str] = Field(None, alias="output")
     input_text: Optional[str] = Field(None, alias="inputText")
+    rag_sources_consulted: Optional[str] = Field(None, alias="ragSourcesConsulted")
+    session_id: str = Field(..., alias="sessionId")
+    start_timestamp: datetime = Field(..., alias="startTimestamp")
     status: str = Field(..., alias="status")
+    timestamp: datetime = Field(..., alias="timestamp")
+
+    @model_validator(mode="after")
+    def validate_status(self):
+        valid_statuses = {"succeeded", "failed", "pending"}
+        if self.status not in valid_statuses:
+            raise ValueError(f"Status must be one of {valid_statuses}")
+        return self
 
     def to_dict(self):
         return self.model_dump(by_alias=True, exclude_none=True)