diff --git a/flytekit/extend/backend/base_agent.py b/flytekit/extend/backend/base_agent.py index fda71020a6..e8ec18806e 100644 --- a/flytekit/extend/backend/base_agent.py +++ b/flytekit/extend/backend/base_agent.py @@ -120,7 +120,7 @@ class SyncAgentBase(AgentBase): @abstractmethod def do( - self, task_template: TaskTemplate, inputs: Optional[LiteralMap], output_prefix: Optional[str], **kwargs + self, task_template: TaskTemplate, output_prefix: str, inputs: Optional[LiteralMap] = None, **kwargs ) -> Resource: """ This is the method that the agent will run. @@ -263,8 +263,8 @@ async def _do( self: PythonTask, agent: SyncAgentBase, template: TaskTemplate, + output_prefix: str, inputs: Dict[str, Any] = None, - output_prefix: Optional[str] = None, ) -> Resource: try: ctx = FlyteContext.current_context() diff --git a/pyproject.toml b/pyproject.toml index 546d034530..126f05050a 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -23,7 +23,7 @@ dependencies = [ "flyteidl>=1.11.0b1", "fsspec>=2023.3.0", "gcsfs>=2023.3.0", - "googleapis-common-protos>=1.57,!=1.63.1", + "googleapis-common-protos>=1.57", "grpcio", "grpcio-status", "importlib-metadata",