Skip to content

Commit e28f3c7

Browse files
committed
fix: remove stale type
1 parent a03d97f commit e28f3c7

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

agntcy_acp/langgraph/acp_node.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
from langchain_core.runnables import RunnableConfig
55
from langgraph.utils.runnable import RunnableCallable
66
from agntcy_acp import ACPClient, ApiClient, AsyncACPClient, AsyncApiClient, ApiClientConfiguration
7-
from agntcy_acp.models import RunCreateStateless, RunWaitResponse
7+
from agntcy_acp.models import RunCreateStateless
88
import logging
99

1010
logger = logging.getLogger(__name__)
@@ -107,15 +107,15 @@ def invoke(self, state: Any, config: RunnableConfig) -> Any:
107107
run_create = self._prepare_run_create(state, config)
108108
with ApiClient(configuration=self.clientConfig) as api_client:
109109
acp_client = ACPClient(api_client=api_client)
110-
run_output: RunWaitResponse = acp_client.create_and_wait_for_stateless_run_output(run_create)
110+
run_output = acp_client.create_and_wait_for_stateless_run_output(run_create)
111111

112112
return self._set_output(state, run_output.output)
113113

114114
async def ainvoke(self, state: Any, config: RunnableConfig) -> Any:
115115
run_create = self._prepare_run_create(state, config)
116116
async with AsyncApiClient(configuration=self.clientConfig) as api_client:
117117
acp_client = AsyncACPClient(api_client=api_client)
118-
run_output: RunWaitResponse = await acp_client.create_and_wait_for_stateless_run_output(run_create)
118+
run_output = await acp_client.create_and_wait_for_stateless_run_output(run_create)
119119

120120
return self._set_output(state, run_output.output)
121121

0 commit comments

Comments
 (0)