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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: uncertainty-engine-types
-Version: 0.0.10
+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.10 → uncertainty_engine_types-0.0.11}/pyproject.toml

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

{uncertainty_engine_types-0.0.10 → uncertainty_engine_types-0.0.11}/uncertainty_engine_types/__init__.py

@@ -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
@@ -50,12 +53,15 @@ __all__ = [
     "MachineLearningModel",
     "Mesh",
     "Message",
+    "ModelConfig",
     "NodeElement",
     "NodeId",
     "NodeInfo",
     "NodeInputInfo",
     "NodeOutputInfo",
     "PDF",
+    "Prompt",
+    "ResourceID",
     "S3Storage",
     "SensorDesigner",
     "SourceHandle",

{uncertainty_engine_types-0.0.10 → uncertainty_engine_types-0.0.11}/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-0.0.11/uncertainty_engine_types/id.py

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

{uncertainty_engine_types-0.0.10 → uncertainty_engine_types-0.0.11}/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-0.0.10 → uncertainty_engine_types-0.0.11}/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-0.0.11/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-0.0.11/uncertainty_engine_types/prompt.py

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

uncertainty_engine_types-0.0.11/uncertainty_engine_types/version.py

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

uncertainty_engine_types-0.0.10/uncertainty_engine_types/version.py

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