Changed outputs to list (#131)

* changed outputs to support list

* fixed CI for tests for python api server

* added None check

Author: NiveditJain

.github/workflows/ci.yml .github/workflows/test-api-server.yml.github/workflows/ci.yml renamed to .github/workflows/test-api-server.yml

@@ -3,8 +3,12 @@ name: Python API Server Tests
 on:
   push:
     branches: [main]
+    paths:
+      - 'api-server/**'
   pull_request:
     branches: [main]
+    paths:
+      - 'api-server/**'
 
 jobs:
   test:

python-sdk/exospherehost/node/BaseNode.py

@@ -1,5 +1,5 @@
 from abc import ABC, abstractmethod
-from typing import Optional, Any
+from typing import Optional, Any, List
 
 
 class BaseNode(ABC):
@@ -8,7 +8,7 @@ def __init__(self, unique_name: Optional[str] = None):
         self.unique_name: Optional[str] = unique_name
 
     @abstractmethod
-    async def execute(self, inputs: dict[str, Any]) -> dict[str, Any]:
+    async def execute(self, inputs: dict[str, Any]) -> dict[str, Any] | List[dict[str, Any]]:
         pass
 
     def get_unique_name(self) -> str:

python-sdk/exospherehost/runtime.py

@@ -69,7 +69,7 @@ async def _enqueue(self):
 
             await sleep(self._poll_interval)
 
-    async def _notify_executed(self, state_id: str, outputs: dict[str, Any]):
+    async def _notify_executed(self, state_id: str, outputs: List[dict[str, Any]]):
         async with ClientSession() as session:
             endpoint = self._get_executed_endpoint(state_id)
             body = {"outputs": outputs}
@@ -112,7 +112,15 @@ async def _worker(self):
             try:
                 node = self._node_mapping[state["node_name"]]
                 outputs = await node.execute(state["inputs"]) # type: ignore
+
+                if outputs is None:
+                    outputs = []
+
+                if isinstance(outputs, dict):
+                    outputs = [outputs]
+
                 await self._notify_executed(state["state_id"], outputs)
+                
             except Exception as e:
                 await self._notify_errored(state["state_id"], str(e))
 

state-manager/app/models/executed_models.py

@@ -1,9 +1,9 @@
 from pydantic import BaseModel, Field
-from typing import Any
+from typing import Any, List
 from .state_status_enum import StateStatusEnum
 
 class ExecutedRequestModel(BaseModel):
-    outputs: dict[str, Any] = Field(..., description="Outputs of the state")
+    outputs: List[dict[str, Any]] = Field(..., description="Outputs of the state")
 
 
 class ExecutedResponseModel(BaseModel):