uncertainty-engine-types 0.0.9__tar.gz → 0.0.10__tar.gz

{uncertainty_engine_types-0.0.9 → uncertainty_engine_types-0.0.10}/PKG-INFO

@@ -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 → uncertainty_engine_types-0.0.10}/pyproject.toml

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

{uncertainty_engine_types-0.0.9 → uncertainty_engine_types-0.0.10}/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-0.0.9/uncertainty_engine_types/conversation.py → uncertainty_engine_types-0.0.10/uncertainty_engine_types/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-0.0.10/uncertainty_engine_types/message.py

@@ -0,0 +1,10 @@
+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.0.10/uncertainty_engine_types/version.py

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

uncertainty_engine_types-0.0.9/uncertainty_engine_types/message.py

@@ -1,30 +0,0 @@
-from datetime import datetime
-from typing import Dict, Literal, Sequence, Union
-
-from pydantic import BaseModel, model_validator
-
-
-Value = Union[str, float, int]
-StructuredOutput = Dict[str, Union[Value, Sequence[Value], Dict[str, Value]]]
-
-
-class Message(BaseModel):
-    role: Literal["instruction", "user", "engine"]
-    content: Union[str, StructuredOutput]
-    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-0.0.9/uncertainty_engine_types/version.py

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