uncertainty-engine-types 0.0.4__py3-none-any.whl → 0.0.5__py3-none-any.whl

uncertainty_engine_types/__init__.py

@@ -3,9 +3,10 @@ from .conversation import Conversation
 from .execution_error import ExecutionError
 from .graph import Graph, NodeElement, NodeId, SourceHandle, TargetHandle
 from .handle import Handle
+from .job import JobInfo, JobStatus
 from .llm import LLMConfig, LLMProvider
 from .message import Message
-from .model import TwinLabModel
+from .model import MachineLearningModel
 from .node_info import NodeInfo, NodeInputInfo, NodeOutputInfo, Versions
 from .sensor_designer import SensorDesigner
 from .sql import SQLConfig, SQLKind
@@ -22,8 +23,11 @@ __all__ = [
     "ExecutionError",
     "Graph",
     "Handle",
+    "JobInfo",
+    "JobStatus",
     "LLMConfig",
     "LLMProvider",
+    "MachineLearningModel",
     "Message",
     "NodeElement",
     "NodeId",
@@ -37,7 +41,6 @@ __all__ = [
     "TabularData",
     "TargetHandle",
     "Token",
-    "TwinLabModel",
     "VectorStoreConfig",
     "VectorStoreProvider",
     "Versions",

uncertainty_engine_types/job.py

@@ -0,0 +1,18 @@
+from enum import StrEnum
+from typing import Optional
+
+from pydantic import BaseModel
+
+
+class JobStatus(StrEnum):
+    PENDING = "pending"
+    RUNNING = "running"
+    COMPLETED = "completed"
+    FAILED = "failed"
+
+
+class JobInfo(BaseModel):
+    status: JobStatus
+    message: Optional[str] = None
+    inputs: dict
+    outputs: Optional[dict] = None

uncertainty_engine_types/model.py

@@ -1,7 +1,7 @@
 from pydantic import BaseModel, ConfigDict
 
 
-class TwinLabModel(BaseModel):
+class MachineLearningModel(BaseModel):
     model_type: str
     config: dict
     metadata: dict

uncertainty_engine_types/node_info.py

@@ -36,6 +36,7 @@ class NodeInfo(BaseModel):
     cost: int
     inputs: dict[str, NodeInputInfo]
     outputs: dict[str, NodeOutputInfo] = {}
+    load_balancer_url: Optional[str] = None
     queue_url: Optional[str] = None
     cache_url: Optional[str] = None
     versions: Versions

uncertainty_engine_types/version.py

@@ -1 +1 @@
-__version__ = "0.0.4"
+__version__ = "0.0.5"

{uncertainty_engine_types-0.0.4.dist-info → uncertainty_engine_types-0.0.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: uncertainty-engine-types
-Version: 0.0.4
+Version: 0.0.5
 Summary: Common type definitions for the Uncertainty Engine
 Author: Freddy Wordingham
 Author-email: freddy@digilab.ai
@@ -45,15 +45,9 @@ This library should be used by other packages to ensure consistency in the types
 
 ### Language Learning Models (LLMs)
 
-- **LLM**  
-  Abstract base class for language learning models.
-- **OpenAILLM**  
-  LLM implementation using OpenAI.
-- **OllamaLLM**  
-  LLM implementation using Ollama.
 - **LLMProvider**  
   Enum listing supported LLM providers.
-- **LLMManager**  
+- **LLMConfig**  
   Manages connections to LLMs based on the chosen provider and configuration.
 
 ### Messaging
@@ -63,10 +57,8 @@ This library should be used by other packages to ensure consistency in the types
 
 ### TwinLab Models
 
-- **TwinLabModel**  
+- **MachineLearningModel**  
   Represents a model configuration including metadata.
-- **save_model**  
-  Function to persist a model configuration.
 
 ### Node Metadata
 
@@ -86,14 +78,10 @@ This library should be used by other packages to ensure consistency in the types
 
 ### SQL Database Types
 
-- **SQLDatabase**  
-  Abstract base class for executing SQL queries.
-- **PostgreSQL**  
-  Implementation of SQLDatabase for PostgreSQL.
 - **SQLKind**  
   Enum listing supported SQL database types.
-- **SQLManager**  
-  Manages connections and operations for SQL databases.
+- **SQLConfig**  
+  Configures connections and operations for SQL databases.
 
 ### Tabular Data
 
@@ -107,16 +95,8 @@ This library should be used by other packages to ensure consistency in the types
 
 ### Vector Stores
 
-- **VectorStoreConnection**  
-  Abstract base class for vector store operations.
-- **WeaviateVectorStoreConnection**  
-  Implements a connection to a Weaviate vector store.
 - **VectorStoreProvider**  
   Enum for supported vector store providers.
-- **VectorStoreManager**  
-  Manages connections to vector stores.
-- **get_persistent_vector_store**  
-  Function to establish a persistent connection to a Weaviate vector store.
-- **get_embedding_function**  
-  Retrieves an embedding function based on configuration, supporting both HuggingFace and OpenAI options.
+- **VectorStoreConfig**  
+  Configures connections to vector stores.
 

{uncertainty_engine_types-0.0.4.dist-info → uncertainty_engine_types-0.0.5.dist-info}/RECORD

@@ -1,19 +1,20 @@
-uncertainty_engine_types/__init__.py,sha256=AbQ7nlKUqTMJ7MuFWBLrUGPnnz-KeNxjRjw_mlud6wU,1104
+uncertainty_engine_types/__init__.py,sha256=6pB7JPLiwVbNJJ65-SxaK_-Uj8Kc5kKlj_VTwYt6weM,1188
 uncertainty_engine_types/context.py,sha256=k9rlArn-L2x1P5vWy5khfuqJW5tJOqr3NlhubZCbyG4,209
 uncertainty_engine_types/conversation.py,sha256=uDHTr4uPzjkcflhyFAanfiQ1KbEIs4C9jogrwzOEt_s,122
 uncertainty_engine_types/execution_error.py,sha256=tvIBuZPM8UhFUERHCEqoW8blAYYuBq1ZqidO0i_BCGs,105
 uncertainty_engine_types/graph.py,sha256=ii2YT2jxoRB7UhcceI2k_ArKmlUXEwBCRBpsEUjL2Qg,265
 uncertainty_engine_types/handle.py,sha256=DEu3aCwJsOuinQ2Gxdy6dZbwnCHAXkplIT4WNL6o7oQ,444
+uncertainty_engine_types/job.py,sha256=eYDe-MW4s-kXGL1Ke1VxdwoduW66orJAa0L1XwGdYi0,347
 uncertainty_engine_types/llm.py,sha256=oivoY3aDv5Rsk8FWA-vOPEFfK9a6-SqSyu01AlEuO7o,406
 uncertainty_engine_types/message.py,sha256=KncFJQHL_ko8nfOuso_eh7i2gB0y4nXI6gBW9U8lAEY,971
-uncertainty_engine_types/model.py,sha256=sVP4e25uV114XOGbcDxvvngcI6bMXRaM3zLRLKP30C0,188
-uncertainty_engine_types/node_info.py,sha256=5_t18hKJ4ZOhB-64DzPzr7Teq9S1HeB7VXHL2J8UVeA,784
+uncertainty_engine_types/model.py,sha256=O9E_7DE9AKEc1o2VnhpUyl3Quh4sGdV43gqDJwk-y68,196
+uncertainty_engine_types/node_info.py,sha256=XJGOlxaTI7pyuKc3zQYuQf_uL6orYMe8AzaWnMO9ME8,828
 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/tabular_data.py,sha256=i_1LfgQqxDanPRWuDkP3Hd8-9eVet6FbPIlPZlAi1n0,76
 uncertainty_engine_types/token.py,sha256=4tQQkvl-zsYoVk8ZEx0cB2JiU0VDRJ6uUe76XBXEpBY,178
 uncertainty_engine_types/vector_store.py,sha256=9fYPJ04jWcy2DruyUSjiKQAgmqq-wgeAi5dBIrAOm30,392
-uncertainty_engine_types/version.py,sha256=1mptEzQihbdyqqzMgdns_j5ZGK9gz7hR2bsgA_TnjO4,22
-uncertainty_engine_types-0.0.4.dist-info/METADATA,sha256=f2-J06g-f0RHmgs30Q0GEzxlrCQMG5OOxt0B19YFzZs,3596
-uncertainty_engine_types-0.0.4.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-uncertainty_engine_types-0.0.4.dist-info/RECORD,,
+uncertainty_engine_types/version.py,sha256=S7u1lbuWmM3A3ajykBialmPoJUK6Jg-WmNqM-9OZFdk,22
+uncertainty_engine_types-0.0.5.dist-info/METADATA,sha256=SPui8CAUqXbPD3NgU6x1EjHRbMS4OO4g3jYlIS19ics,2815
+uncertainty_engine_types-0.0.5.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+uncertainty_engine_types-0.0.5.dist-info/RECORD,,