Wait for completedAt in remote dispatch polling

examples/human_in_the_loop.py

@@ -1,8 +1,9 @@
 import asyncio
 
-from narada import Agent, Narada, UserAbortedError
 from narada_core.actions.models import PromptForUserInputVariable
 
+from narada import Agent, Narada, UserAbortedError
+
 
 async def main() -> None:
     async with Narada() as narada:

packages/narada-core/src/narada_core/actions/models.py

@@ -433,6 +433,17 @@ class UserApprovalResponse(BaseModel):
     approved: bool
 
 
+ActiveInputAction = Annotated[
+    PromptForUserInputRequest | UserApprovalRequest,
+    Field(discriminator="name"),
+]
+
+
+class ActiveInputRequest(BaseModel):
+    input_id: str = Field(alias="inputId")
+    action: ActiveInputAction
+
+
 type ExtensionActionRequest = (
     AgenticSelectorRequest
     | AgenticMatchingSelectorsFinderRequest

packages/narada-core/src/narada_core/models.py

@@ -5,6 +5,8 @@
 
 from pydantic import BaseModel, Field
 
+from narada_core.actions.models import ActiveInputRequest
+
 
 class Agent(Enum):
     PRODUCTIVITY = 1
@@ -395,8 +397,19 @@ class PythonAgentRunTrace(TypedDict):
 type ActionTrace = OperatorActionTrace | APAActionTrace
 
 
+class TextResponseOutput(TypedDict):
+    type: Literal["text"]
+    content: str
+
+
+class StructuredResponseOutput(TypedDict, Generic[_MaybeStructuredOutput]):
+    type: Literal["structured"]
+    content: _MaybeStructuredOutput
+
+
 class ResponseContent(TypedDict, Generic[_MaybeStructuredOutput]):
     text: str
+    output: TextResponseOutput | StructuredResponseOutput[_MaybeStructuredOutput]
     structuredOutput: _MaybeStructuredOutput
     actionTrace: NotRequired[ActionTrace]
     workflowTrace: NotRequired[dict[str, Any]]
@@ -409,11 +422,22 @@ class Usage(TypedDict):
 
 class Response(TypedDict, Generic[_MaybeStructuredOutput]):
     requestId: str
-    status: Literal["success", "error"]
+    status: Literal["success", "error", "input-required"]
     response: ResponseContent[_MaybeStructuredOutput] | None
     createdAt: str
     completedAt: str | None
     usage: Usage
+    activeInputRequest: ActiveInputRequest | None
+
+
+class _RemoteDispatchPollResponse(TypedDict):
+    requestId: str
+    status: Literal["pending", "input-required", "success", "error"]
+    response: dict[str, Any] | None
+    createdAt: str
+    completedAt: str | None
+    usage: Usage | None
+    activeInputRequest: ActiveInputRequest | None
 
 
 class File(TypedDict):

packages/narada-core/src/narada_core/tracing/model.py

@@ -260,7 +260,7 @@ class PythonSubAgentCallEvent(BaseModel):
     ts_end: int
     agent_type: str
     prompt: str
-    status: Literal["success", "error", "timeout"]
+    status: Literal["success", "error", "timeout", "input-required"]
     request_id: str | None = None
     text: str | None = None
     error_message: str | None = None

packages/narada-pyodide/src/narada/__init__.py

@@ -1,4 +1,4 @@
-from narada_core.actions.models import CriticResult
+from narada_core.actions.models import ActiveInputRequest, CriticResult
 from narada_core.errors import (
     NaradaError,
     NaradaTimeoutError,
@@ -23,6 +23,7 @@
 
 __all__ = [
     "__version__",
+    "ActiveInputRequest",
     "Agent",
     "CloudBrowserWindow",
     "CriticConfig",

packages/narada-pyodide/src/narada/_trace.py

@@ -75,7 +75,7 @@ def dump_model(model: BaseModel) -> dict[str, Any]:
 # ---------------------------------------------------------------------------
 
 
-SubAgentCallStatus = Literal["success", "error", "timeout"]
+SubAgentCallStatus = Literal["success", "error", "timeout", "input-required"]
 ExtensionActionStatus = Literal["success", "error", "timeout"]
 SideEffectType = Literal["download_file", "render_html"]