added register secret flow

Author: NiveditJain

python-sdk/exospherehost/node/BaseNode.py

@@ -42,6 +42,12 @@ class Outputs(BaseModel):
         """
         pass
 
+    class Secrets(BaseModel):
+        """
+        Secrets schema for the node.
+        """
+        pass
+
     async def _execute(self, inputs: Inputs) -> Outputs | List[Outputs]:
         """
         Internal method to execute the node with validated inputs.

python-sdk/exospherehost/runtime.py

@@ -135,6 +135,9 @@ async def _register(self):
                         "namespace": self._namespace,
                         "inputs_schema": node.Inputs.model_json_schema(),
                         "outputs_schema": node.Outputs.model_json_schema(),
+                        "secrets": [
+                            secret_name for secret_name in node.Secrets.model_fields.keys()
+                        ]
                     } for node in self._nodes
                 ]
             }
@@ -246,9 +249,17 @@ def _validate_nodes(self):
             if not hasattr(node, "Outputs"):
                 errors.append(f"{node.__name__} does not have an Outputs class")
             if not issubclass(node.Inputs, BaseModel):
-                errors.append(f"{node.__name__} does not have an Inputs class that inherits from pydantic.BaseModel")
+                errors.append(f"{node.__name__} does not have an Inputs class that inherits from pydantic.BaseModel")                                          
             if not issubclass(node.Outputs, BaseModel):
                 errors.append(f"{node.__name__} does not have an Outputs class that inherits from pydantic.BaseModel")
+            if not hasattr(node, "Secrets"):
+                errors.append(f"{node.__name__} does not have an Secrets class")
+            if not issubclass(node.Secrets, BaseModel):
+                errors.append(f"{node.__name__} does not have an Secrets class that inherits from pydantic.BaseModel")
+            
+            for field_name, field_info in node.Secrets.model_fields.items():
+                if field_info.annotation != str:
+                    errors.append(f"{node.__name__}.Secrets field '{field_name}' must be of type str, got {field_info.annotation}")
 
         # Find nodes with the same __class__.__name__
         class_names = [node.__name__ for node in self._nodes]

state-manager/app/controller/register_nodes.py

@@ -30,7 +30,8 @@ async def register_nodes(namespace_name: str, body: RegisterNodesRequestModel, x
                         RegisteredNode.runtime_name: body.runtime_name,
                         RegisteredNode.runtime_namespace: namespace_name,
                         RegisteredNode.inputs_schema: node_data.inputs_schema, # type: ignore
-                        RegisteredNode.outputs_schema: node_data.outputs_schema # type: ignore
+                        RegisteredNode.outputs_schema: node_data.outputs_schema, # type: ignore
+                        RegisteredNode.secrets: node_data.secrets # type: ignore
                 }))
                 logger.info(f"Updated existing node {node_data.name} in namespace {namespace_name}", x_exosphere_request_id=x_exosphere_request_id)
 
@@ -42,7 +43,8 @@ async def register_nodes(namespace_name: str, body: RegisterNodesRequestModel, x
                     runtime_name=body.runtime_name,
                     runtime_namespace=namespace_name,
                     inputs_schema=node_data.inputs_schema,
-                    outputs_schema=node_data.outputs_schema
+                    outputs_schema=node_data.outputs_schema,
+                    secrets=node_data.secrets
                 )
                 await new_node.insert()
                 logger.info(f"Created new node {node_data.name} in namespace {namespace_name}", x_exosphere_request_id=x_exosphere_request_id)
@@ -51,7 +53,8 @@ async def register_nodes(namespace_name: str, body: RegisterNodesRequestModel, x
                 RegisteredNodeModel(
                     name=node_data.name,
                     inputs_schema=node_data.inputs_schema,
-                    outputs_schema=node_data.outputs_schema
+                    outputs_schema=node_data.outputs_schema,
+                    secrets=node_data.secrets
                 )
             )
 

state-manager/app/models/db/registered_node.py

@@ -9,4 +9,5 @@ class RegisteredNode(BaseDatabaseModel):
     runtime_name: str = Field(..., description="Name of the runtime that registered this node")
     runtime_namespace: str = Field(..., description="Namespace of the runtime that registered this node")
     inputs_schema: dict[str, Any] = Field(..., description="JSON schema for node inputs")
-    outputs_schema: dict[str, Any] = Field(..., description="JSON schema for node outputs") 
+    outputs_schema: dict[str, Any] = Field(..., description="JSON schema for node outputs") 
+    secrets: list[str] = Field(..., description="List of secrets that the node uses")

state-manager/app/models/register_nodes_request.py

@@ -6,8 +6,9 @@ class NodeRegistrationModel(BaseModel):
     name: str = Field(..., description="Unique name of the node")
     inputs_schema: dict[str, Any] = Field(..., description="JSON schema for node inputs")
     outputs_schema: dict[str, Any] = Field(..., description="JSON schema for node outputs")
+    secrets: List[str] = Field(..., description="List of secrets that the node uses")
 
 
 class RegisterNodesRequestModel(BaseModel):
     runtime_name: str = Field(..., description="Name of the runtime registering the nodes")
-    nodes: List[NodeRegistrationModel] = Field(..., description="List of nodes to register") 
+    nodes: List[NodeRegistrationModel] = Field(..., description="List of nodes to register")

state-manager/app/models/register_nodes_response.py

@@ -6,6 +6,7 @@ class RegisteredNodeModel(BaseModel):
     name: str = Field(..., description="Name of the registered node")
     inputs_schema: dict[str, Any] = Field(..., description="Inputs for the registered node")
     outputs_schema: dict[str, Any] = Field(..., description="Outputs for the registered node")
+    secrets: List[str] = Field(..., description="List of secrets that the node uses")
 
 
 class RegisterNodesResponseModel(BaseModel):