uncertainty-engine-types 0.1.0__py3-none-any.whl → 0.3.0__py3-none-any.whl

uncertainty_engine_types/job.py

@@ -17,3 +17,4 @@ class JobInfo(BaseModel):
     message: Optional[str] = None
     inputs: dict
     outputs: Optional[dict] = None
+    progress: "None | str | dict[str, JobInfo]" = None

uncertainty_engine_types/message.py

@@ -1,10 +1,15 @@
 from datetime import datetime
-from typing import Literal
+from typing import Any, Literal
 
-from pydantic import BaseModel
+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/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/version.py

@@ -1 +1 @@
-__version__ = "0.1.0"
+__version__ = "0.3.0"

{uncertainty_engine_types-0.1.0.dist-info → uncertainty_engine_types-0.3.0.dist-info}/METADATA

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

{uncertainty_engine_types-0.1.0.dist-info → uncertainty_engine_types-0.3.0.dist-info}/RECORD

@@ -8,11 +8,11 @@ uncertainty_engine_types/file.py,sha256=J202bhOIVnvvvAG54sXyYF8fwjZGTUP3097ARnRn
 uncertainty_engine_types/graph.py,sha256=ii2YT2jxoRB7UhcceI2k_ArKmlUXEwBCRBpsEUjL2Qg,265
 uncertainty_engine_types/handle.py,sha256=buX_xEdBwT7vZRs5uCEPB6VCbSWIFbCE-NYIL4fZSjg,942
 uncertainty_engine_types/id.py,sha256=Wpco2lLEsx6sdfdRJcPGl8GQlVa13dflWApaRt6dTYI,74
-uncertainty_engine_types/job.py,sha256=J_7tlWufZprOlePIEfBmir-5ZY9YXHmmALK_Ykj0ubs,369
+uncertainty_engine_types/job.py,sha256=mKNeZ9Ay7SdO7VH9kYS74U7ptcrmQIXyrGVgiE7MRz8,424
 uncertainty_engine_types/llm.py,sha256=Ae7dw1R5RKLJHFjNwBzynLd-ddikKP0NsvHvoMdmnLM,916
-uncertainty_engine_types/message.py,sha256=5IY-S6Ipt6oYmnVSkRVv3QLSH-lKQvpxp_qF5YGSbok,209
+uncertainty_engine_types/message.py,sha256=sX1tq_BbZC9r9-Vf0A3WHiOFRE2hAAlY-7rD52kbsoU,381
 uncertainty_engine_types/model.py,sha256=O9E_7DE9AKEc1o2VnhpUyl3Quh4sGdV43gqDJwk-y68,196
-uncertainty_engine_types/model_config.py,sha256=AeN5oR06uZJMGoAJoaI_Ix4nCy5nNcU2Wh69ijP39rQ,286
+uncertainty_engine_types/model_config.py,sha256=ctHEHuVsaSW9vcC7teeku_EEslBye82Eu5Ig0KbUPvU,391
 uncertainty_engine_types/node_info.py,sha256=XHONP6rNbAcLOHFxi-NPl1-UrKAH9P8C1yUr_rDiyCg,1362
 uncertainty_engine_types/prompt.py,sha256=l__qXytAapKg1Hoaj3RSVYN3rhy638nuUZIS-se0eMc,74
 uncertainty_engine_types/sensor_designer.py,sha256=hr3ek4_dRjRK0-78uaT6h8-bGUCm7Mfs6mxJSWxE64c,80
@@ -21,7 +21,7 @@ uncertainty_engine_types/token.py,sha256=4tQQkvl-zsYoVk8ZEx0cB2JiU0VDRJ6uUe76XBX
 uncertainty_engine_types/uncertainty_plot.py,sha256=kJr0SuJ6JeTxaf2adpDNWqx7vVLZRpKG8tFbdx90pas,547
 uncertainty_engine_types/utils.py,sha256=72QVig8Kb5uIR-e1nofm-3x9CouebdQJIruDbq-aIn0,271
 uncertainty_engine_types/vector_store.py,sha256=9fYPJ04jWcy2DruyUSjiKQAgmqq-wgeAi5dBIrAOm30,392
-uncertainty_engine_types/version.py,sha256=kUR5RAFc7HCeiqdlX36dZOHkUI5wI6V_43RpEcD8b-0,22
-uncertainty_engine_types-0.1.0.dist-info/METADATA,sha256=jvQ6vZ_5TG74GjT0maweL1klrkSWR9iIYgUE8zP0qgs,2815
-uncertainty_engine_types-0.1.0.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-uncertainty_engine_types-0.1.0.dist-info/RECORD,,
+uncertainty_engine_types/version.py,sha256=VrXpHDu3erkzwl_WXrqINBm9xWkcyUy53IQOj042dOs,22
+uncertainty_engine_types-0.3.0.dist-info/METADATA,sha256=iU0Tkdsa9XSI9CqS8-EjHEQ8pargEcnDDbH_yxuWZPs,2815
+uncertainty_engine_types-0.3.0.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+uncertainty_engine_types-0.3.0.dist-info/RECORD,,