uncertainty-engine-types 0.1.0__tar.gz → 0.2.0__tar.gz

{uncertainty_engine_types-0.1.0 → uncertainty_engine_types-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: uncertainty-engine-types
-Version: 0.1.0
+Version: 0.2.0
 Summary: Common type definitions for the Uncertainty Engine
 Author: Freddy Wordingham
 Author-email: freddy@digilab.ai

{uncertainty_engine_types-0.1.0 → uncertainty_engine_types-0.2.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uncertainty-engine-types"
-version = "0.1.0"
+version = "0.2.0"
 description = "Common type definitions for the Uncertainty Engine"
 authors = [
     { name = "Freddy Wordingham", email = "freddy@digilab.ai" },

uncertainty_engine_types-0.2.0/uncertainty_engine_types/message.py

@@ -0,0 +1,15 @@
+from datetime import datetime
+from typing import Any, Literal
+
+from pydantic import BaseModel, field_validator
+
+
+class Message(BaseModel):
+    role: Literal["instruction", "user", "engine"]
+    content: str
+    timestamp: datetime
+
+    @field_validator("content", mode="before")
+    @classmethod
+    def convert_content_to_string(cls, value: Any) -> str:
+        return str(value)

{uncertainty_engine_types-0.1.0 → uncertainty_engine_types-0.2.0}/uncertainty_engine_types/model_config.py

@@ -5,6 +5,8 @@ from pydantic import BaseModel
 
 class ModelConfig(BaseModel):
     input_variance: Optional[float] = None
+    input_retained_dimensions: Optional[int] = None
     output_variance: Optional[float] = None
+    output_retained_dimensions: Optional[int] = None
     model_type: Literal["SingleTaskGPTorch"] = "SingleTaskGPTorch"
     seed: Optional[int] = None

uncertainty_engine_types-0.2.0/uncertainty_engine_types/version.py

@@ -0,0 +1 @@
+__version__ = "0.2.0"

uncertainty_engine_types-0.1.0/uncertainty_engine_types/message.py

@@ -1,10 +0,0 @@
-from datetime import datetime
-from typing import Literal
-
-from pydantic import BaseModel
-
-
-class Message(BaseModel):
-    role: Literal["instruction", "user", "engine"]
-    content: str
-    timestamp: datetime

uncertainty_engine_types-0.1.0/uncertainty_engine_types/version.py

@@ -1 +0,0 @@
-__version__ = "0.1.0"