Skip to content

Commit 330264c

Browse files
declan-scaleclaude
andcommitted
fix: import missing RPC response types in agents resource
The custom create_task/cancel_task/send_message/send_event helpers reference CreateTaskResponse, CancelTaskResponse, SendMessageResponse, SendMessageStreamResponse, and SendEventResponse but only AgentRpcResponse was imported, causing NameError at runtime. Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
1 parent 09b2002 commit 330264c

2 files changed

Lines changed: 16 additions & 2 deletions

File tree

src/agentex/resources/agents/agents.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,14 @@
3838
)
3939
from ...types.agent import Agent
4040
from ..._base_client import make_request_options
41-
from ...types.agent_rpc_response import AgentRpcResponse
41+
from ...types.agent_rpc_response import (
42+
AgentRpcResponse,
43+
CancelTaskResponse,
44+
CreateTaskResponse,
45+
SendEventResponse,
46+
SendMessageResponse,
47+
SendMessageStreamResponse,
48+
)
4249
from ...types.agent_list_response import AgentListResponse
4350
from ...types.shared.delete_response import DeleteResponse
4451

src/agentex/types/agent_rpc_response.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,14 @@
1111
from .agent_rpc_result import AgentRpcResult
1212
from .task_message_update import TaskMessageUpdate
1313

14-
__all__ = ["AgentRpcResponse"]
14+
__all__ = [
15+
"AgentRpcResponse",
16+
"CancelTaskResponse",
17+
"CreateTaskResponse",
18+
"SendEventResponse",
19+
"SendMessageResponse",
20+
"SendMessageStreamResponse",
21+
]
1522

1623

1724
class BaseAgentRpcResponse(BaseModel):

0 commit comments

Comments
 (0)