uncertainty-engine-types 0.0.9__py3-none-any.whl → 0.0.10__py3-none-any.whl

uncertainty_engine_types/__init__.py

@@ -1,6 +1,6 @@
 from . import utils
+from .chat_history import ChatHistory
 from .context import Context
-from .conversation import Conversation
 from .dataset import CSVDataset
 from .embeddings import TextEmbeddingsConfig, TextEmbeddingsProvider
 from .execution_error import ExecutionError
@@ -32,8 +32,8 @@ from .version import __version__
 
 __all__ = [
     "__version__",
+    "ChatHistory",
     "Context",
-    "Conversation",
     "CSVDataset",
     "Document",
     "ExecutionError",

uncertainty_engine_types/{conversation.py → chat_history.py}

@@ -3,5 +3,5 @@ from pydantic import BaseModel
 from .message import Message
 
 
-class Conversation(BaseModel):
+class ChatHistory(BaseModel):
     messages: list[Message]

uncertainty_engine_types/message.py

@@ -1,30 +1,10 @@
 from datetime import datetime
-from typing import Dict, Literal, Sequence, Union
+from typing import Literal
 
-from pydantic import BaseModel, model_validator
-
-
-Value = Union[str, float, int]
-StructuredOutput = Dict[str, Union[Value, Sequence[Value], Dict[str, Value]]]
+from pydantic import BaseModel
 
 
 class Message(BaseModel):
     role: Literal["instruction", "user", "engine"]
-    content: Union[str, StructuredOutput]
+    content: str
     timestamp: datetime
-
-    @model_validator(mode="before")
-    @classmethod
-    def validate(cls, values: Dict) -> Dict:
-        role = values.get("role")
-        content = values.get("content")
-
-        if role in ["instruction", "user"]:
-            if not isinstance(content, str):
-                raise ValueError(f"Content must be a string when role is {role}")
-        elif role == "engine":
-            if not isinstance(content, StructuredOutput):
-                raise ValueError(
-                    "Content must be a structured output when role is engine"
-                )
-        return values

uncertainty_engine_types/version.py

@@ -1 +1 @@
-__version__ = "0.0.9"
+__version__ = "0.0.10"

{uncertainty_engine_types-0.0.9.dist-info → uncertainty_engine_types-0.0.10.dist-info}/METADATA

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

{uncertainty_engine_types-0.0.9.dist-info → uncertainty_engine_types-0.0.10.dist-info}/RECORD

@@ -1,6 +1,6 @@
-uncertainty_engine_types/__init__.py,sha256=G8H0XonAfxXoXlgeQK7FsxuMRgXosOZfm-jp6P5aHu0,1656
+uncertainty_engine_types/__init__.py,sha256=HYCR2VJzvv-F4EJJ_5AoMwJIt5WTK2k3VIandPGYWVI,1654
+uncertainty_engine_types/chat_history.py,sha256=OY1fZXP7_AtPKEmgjxh60PlbEGtQWJWafZMgs2Ec0BU,121
 uncertainty_engine_types/context.py,sha256=k9rlArn-L2x1P5vWy5khfuqJW5tJOqr3NlhubZCbyG4,209
-uncertainty_engine_types/conversation.py,sha256=uDHTr4uPzjkcflhyFAanfiQ1KbEIs4C9jogrwzOEt_s,122
 uncertainty_engine_types/dataset.py,sha256=sDpQu5X3KxJ1lNkZOwhppJ2SY0Cv19cbcYfGpp_hyUQ,75
 uncertainty_engine_types/embeddings.py,sha256=PmbaT7A7IuQ1UbGVNIg4usNnDbEgWiXM87gQPOeaff8,954
 uncertainty_engine_types/execution_error.py,sha256=tvIBuZPM8UhFUERHCEqoW8blAYYuBq1ZqidO0i_BCGs,105
@@ -9,7 +9,7 @@ uncertainty_engine_types/graph.py,sha256=ii2YT2jxoRB7UhcceI2k_ArKmlUXEwBCRBpsEUj
 uncertainty_engine_types/handle.py,sha256=buX_xEdBwT7vZRs5uCEPB6VCbSWIFbCE-NYIL4fZSjg,942
 uncertainty_engine_types/job.py,sha256=eYDe-MW4s-kXGL1Ke1VxdwoduW66orJAa0L1XwGdYi0,347
 uncertainty_engine_types/llm.py,sha256=kuhTlkTIgE8oIYLofr7MOcz6S5qiN0w5FxfNcT05-E4,910
-uncertainty_engine_types/message.py,sha256=KncFJQHL_ko8nfOuso_eh7i2gB0y4nXI6gBW9U8lAEY,971
+uncertainty_engine_types/message.py,sha256=5IY-S6Ipt6oYmnVSkRVv3QLSH-lKQvpxp_qF5YGSbok,209
 uncertainty_engine_types/model.py,sha256=O9E_7DE9AKEc1o2VnhpUyl3Quh4sGdV43gqDJwk-y68,196
 uncertainty_engine_types/node_info.py,sha256=2p87fCdX_CrdT2ZODtGJ-mR1Mvg46WhaE1GMddaEVak,800
 uncertainty_engine_types/sensor_designer.py,sha256=hr3ek4_dRjRK0-78uaT6h8-bGUCm7Mfs6mxJSWxE64c,80
@@ -17,7 +17,7 @@ uncertainty_engine_types/sql.py,sha256=SBzmgMEZ-sa-OBRoZbmKiOqddop4zJixgxx-JCus9
 uncertainty_engine_types/token.py,sha256=4tQQkvl-zsYoVk8ZEx0cB2JiU0VDRJ6uUe76XBXEpBY,178
 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=46Yjk3fz9o8aTN8E95McnzpJcjGzVJmHmQqUZ5mXzfc,22
-uncertainty_engine_types-0.0.9.dist-info/METADATA,sha256=nAO5QC9gAFMHlvPhlPUR5jGij0OSIYic8VZ7R5eiITA,2815
-uncertainty_engine_types-0.0.9.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-uncertainty_engine_types-0.0.9.dist-info/RECORD,,
+uncertainty_engine_types/version.py,sha256=-nNlMKS9nph3FR78_ZG9RGKrbxseeNp2K6nMr0pVGaU,23
+uncertainty_engine_types-0.0.10.dist-info/METADATA,sha256=YKiP4rjDbXahzA9ulm2B532Tp7XwD5Vl4FyDrYn9vhU,2816
+uncertainty_engine_types-0.0.10.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+uncertainty_engine_types-0.0.10.dist-info/RECORD,,