uncertainty-engine-types 0.0.9__py3-none-any.whl → 0.0.11__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
@@ -19,11 +19,14 @@ from .file import (
 )
 from .graph import Graph, NodeElement, NodeId, SourceHandle, TargetHandle
 from .handle import Handle
+from .id import ResourceID
 from .job import JobInfo, JobStatus
 from .llm import LLMConfig, LLMProvider
 from .message import Message
 from .model import MachineLearningModel
+from .model_config import ModelConfig
 from .node_info import NodeInfo, NodeInputInfo, NodeOutputInfo
+from .prompt import Prompt
 from .sensor_designer import SensorDesigner
 from .sql import SQLConfig, SQLKind
 from .token import Token
@@ -32,8 +35,8 @@ from .version import __version__
 
 __all__ = [
     "__version__",
+    "ChatHistory",
     "Context",
-    "Conversation",
     "CSVDataset",
     "Document",
     "ExecutionError",
@@ -50,12 +53,15 @@ __all__ = [
     "MachineLearningModel",
     "Mesh",
     "Message",
+    "ModelConfig",
     "NodeElement",
     "NodeId",
     "NodeInfo",
     "NodeInputInfo",
     "NodeOutputInfo",
     "PDF",
+    "Prompt",
+    "ResourceID",
     "S3Storage",
     "SensorDesigner",
     "SourceHandle",

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/embeddings.py

@@ -1,10 +1,10 @@
-from enum import StrEnum
+from enum import Enum
 from typing import Optional
 
 from pydantic import BaseModel, model_validator
 
 
-class TextEmbeddingsProvider(StrEnum):
+class TextEmbeddingsProvider(Enum):
     OPENAI = "openai"
     OLLAMA = "ollama"
 
@@ -23,9 +23,11 @@ class TextEmbeddingsConfig(BaseModel):
     @classmethod
     def check_provider(cls, values):
         provider = values.get("provider")
-        if provider == TextEmbeddingsProvider.OLLAMA and not values.get("ollama_url"):
+        if provider == TextEmbeddingsProvider.OLLAMA.value and not values.get(
+            "ollama_url"
+        ):
             raise ValueError("ollama_url must be provided for 'ollama' provider.")
-        if provider == TextEmbeddingsProvider.OPENAI and not values.get(
+        if provider == TextEmbeddingsProvider.OPENAI.value and not values.get(
             "openai_api_key"
         ):
             raise ValueError("openai_api_key must be provided for 'openai' provider.")

uncertainty_engine_types/id.py

@@ -0,0 +1,5 @@
+from pydantic import BaseModel
+
+
+class ResourceID(BaseModel):
+    id: str

uncertainty_engine_types/job.py

@@ -1,10 +1,10 @@
-from enum import StrEnum
+from enum import Enum
 from typing import Optional
 
 from pydantic import BaseModel
 
 
-class JobStatus(StrEnum):
+class JobStatus(Enum):
     PENDING = "pending"
     RUNNING = "running"
     COMPLETED = "completed"

uncertainty_engine_types/llm.py

@@ -1,10 +1,10 @@
-from enum import StrEnum
+from enum import Enum
 from typing import Optional
 
 from pydantic import BaseModel, model_validator
 
 
-class LLMProvider(StrEnum):
+class LLMProvider(Enum):
     OPENAI = "openai"
     OLLAMA = "ollama"
 
@@ -24,8 +24,8 @@ class LLMConfig(BaseModel):
     @classmethod
     def check_provider(cls, values):
         provider = values.get("provider")
-        if provider == LLMProvider.OLLAMA and not values.get("ollama_url"):
+        if provider == LLMProvider.OLLAMA.value and not values.get("ollama_url"):
             raise ValueError("ollama_url must be provided for 'ollama' provider.")
-        if provider == LLMProvider.OPENAI and not values.get("openai_api_key"):
+        if provider == LLMProvider.OPENAI.value and not values.get("openai_api_key"):
             raise ValueError("openai_api_key must be provided for 'openai' provider.")
         return values

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/model_config.py

@@ -0,0 +1,11 @@
+from typing import Literal, Optional
+
+from pydantic import BaseModel
+
+
+class ModelConfig(BaseModel):
+    train_test_ratio: float = 1.0
+    input_variance: Optional[float] = None
+    output_variance: Optional[float] = None
+    model_type: Literal["SingleTaskGPTorch"] = "SingleTaskGPTorch"
+    seed: Optional[int] = None

uncertainty_engine_types/prompt.py

@@ -0,0 +1,5 @@
+from pydantic import BaseModel
+
+
+class Prompt(BaseModel):
+    prompt: str

uncertainty_engine_types/version.py

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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: uncertainty-engine-types
-Version: 0.0.9
+Version: 0.0.11
 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.11.dist-info}/RECORD

@@ -1,23 +1,26 @@
-uncertainty_engine_types/__init__.py,sha256=G8H0XonAfxXoXlgeQK7FsxuMRgXosOZfm-jp6P5aHu0,1656
+uncertainty_engine_types/__init__.py,sha256=_C9BE7MJkVmE1Ky2AVhSrcAro-QPf7dsNsbQTUcxZJc,1797
+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/embeddings.py,sha256=sNDWjpuqmzOefKyQX--9EpdeTEyI4kQeSH4IELVUT7g,982
 uncertainty_engine_types/execution_error.py,sha256=tvIBuZPM8UhFUERHCEqoW8blAYYuBq1ZqidO0i_BCGs,105
 uncertainty_engine_types/file.py,sha256=J202bhOIVnvvvAG54sXyYF8fwjZGTUP3097ARnRnb3Q,579
 uncertainty_engine_types/graph.py,sha256=ii2YT2jxoRB7UhcceI2k_ArKmlUXEwBCRBpsEUjL2Qg,265
 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/id.py,sha256=Wpco2lLEsx6sdfdRJcPGl8GQlVa13dflWApaRt6dTYI,74
+uncertainty_engine_types/job.py,sha256=C3dS2NWL3-Nw2jb5x5VqGNMPWuYcKhIc219K4cNpBMw,341
+uncertainty_engine_types/llm.py,sha256=Ae7dw1R5RKLJHFjNwBzynLd-ddikKP0NsvHvoMdmnLM,916
+uncertainty_engine_types/message.py,sha256=5IY-S6Ipt6oYmnVSkRVv3QLSH-lKQvpxp_qF5YGSbok,209
 uncertainty_engine_types/model.py,sha256=O9E_7DE9AKEc1o2VnhpUyl3Quh4sGdV43gqDJwk-y68,196
+uncertainty_engine_types/model_config.py,sha256=Ghii7gk3zRhydGQJJJlzYRmMFW2cHBfwGj9FPNHEhCQ,320
 uncertainty_engine_types/node_info.py,sha256=2p87fCdX_CrdT2ZODtGJ-mR1Mvg46WhaE1GMddaEVak,800
+uncertainty_engine_types/prompt.py,sha256=l__qXytAapKg1Hoaj3RSVYN3rhy638nuUZIS-se0eMc,74
 uncertainty_engine_types/sensor_designer.py,sha256=hr3ek4_dRjRK0-78uaT6h8-bGUCm7Mfs6mxJSWxE64c,80
 uncertainty_engine_types/sql.py,sha256=SBzmgMEZ-sa-OBRoZbmKiOqddop4zJixgxx-JCus9fY,298
 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=OaIl7v-6zEWEY90Jlh6yoBjO3zWX1oX7RsvqrK3o1TU,23
+uncertainty_engine_types-0.0.11.dist-info/METADATA,sha256=6VC6b9iAOQDiRv0NO9FP2RGl59TbWMnSiY5zob1TDic,2816
+uncertainty_engine_types-0.0.11.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+uncertainty_engine_types-0.0.11.dist-info/RECORD,,