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

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/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/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.10"
+__version__ = "0.0.11"

{uncertainty_engine_types-0.0.10.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.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.dist-info → uncertainty_engine_types-0.0.11.dist-info}/RECORD

@@ -1,23 +1,26 @@
-uncertainty_engine_types/__init__.py,sha256=HYCR2VJzvv-F4EJJ_5AoMwJIt5WTK2k3VIandPGYWVI,1654
+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/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/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=-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,,
+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,,