ci fix and workflow methods

Author: nikhil-zlai

.github/workflows/test_python.yaml

@@ -60,6 +60,8 @@ jobs:
           thrift --gen py -out api/py/ api/thrift/common.thrift
           thrift --gen py -out api/py/ api/thrift/observability.thrift
           thrift --gen py -out api/py/ api/thrift/api.thrift
+          thrift --gen py -out api/py/ api/thrift/lineage.thrift
+          thrift --gen py -out api/py/ api/thrift/orchestration.thrift
           cd api/py
           pip3 install -r requirements/dev.txt
           pip3 install tox

api/py/ai/chronon/cli/plan/controller_iface.py

@@ -1,12 +1,15 @@
 from abc import ABC, abstractmethod
-from typing import Dict, List
+from typing import Dict, List, Optional
 
+from ai.chronon.cli.plan.physical_graph import PhysicalGraph
 from ai.chronon.cli.plan.physical_index import PhysicalNode
 
 
-class OrchestratorIface(ABC):
-    def __init__(self):
-        pass
+class ControllerIface(ABC):
+    """
+    Class used to make the rest of the planner code agnostic to the underlying orchestrator.
+    Mainly used to mock out the orchestrator for testing.
+    """
 
     @abstractmethod
     def fetch_missing_confs(self, node_to_hash: Dict[str, str]) -> List[str]:
@@ -15,3 +18,28 @@ def fetch_missing_confs(self, node_to_hash: Dict[str, str]) -> List[str]:
     @abstractmethod
     def upload_conf(self, name: str, hash: str, content: str) -> None:
         pass
+
+    @abstractmethod
+    def create_workflow(
+        self, physical_graph: PhysicalGraph, start_date: str, end_date: str
+    ) -> str:
+        """
+        Submit a physical graph to the orchestrator and return workflow id
+        """
+        pass
+
+    @abstractmethod
+    def get_workflow_status(self, workflow_id: str) -> str:
+        """
+        Get the status of a workflow
+        """
+        pass
+
+    @abstractmethod
+    def get_active_workflows(
+        self, branch: Optional[str] = None, user: Optional[str] = None
+    ) -> List[str]:
+        """
+        List all active workflows
+        """
+        pass