uncertainty-engine-types 0.0.4__tar.gz → 0.0.6__tar.gz

{uncertainty_engine_types-0.0.4 → uncertainty_engine_types-0.0.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: uncertainty-engine-types
-Version: 0.0.4
+Version: 0.0.6
 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 → uncertainty_engine_types-0.0.6}/README.md

@@ -30,15 +30,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
@@ -48,10 +42,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
 
@@ -71,14 +63,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
 
@@ -92,15 +80,7 @@ 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 → uncertainty_engine_types-0.0.6}/pyproject.toml

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

{uncertainty_engine_types-0.0.4 → uncertainty_engine_types-0.0.6}/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-0.0.6/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-0.0.4 → uncertainty_engine_types-0.0.6}/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-0.0.4 → uncertainty_engine_types-0.0.6}/uncertainty_engine_types/node_info.py

@@ -5,12 +5,6 @@ from pydantic import BaseModel
 from .version import __version__
 
 
-class Versions(BaseModel):
-    base_image: str
-    types_lib: str = __version__
-    node: str
-
-
 class NodeInputInfo(BaseModel):
     type: str
     label: str
@@ -36,6 +30,9 @@ 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
+    version_types_lib: str = __version__
+    version_base_image: int
+    version_node: int

uncertainty_engine_types-0.0.6/uncertainty_engine_types/version.py

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

uncertainty_engine_types-0.0.4/uncertainty_engine_types/version.py

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