|
4 | 4 | from langchain_core.runnables import RunnableConfig |
5 | 5 | from langgraph.utils.runnable import RunnableCallable |
6 | 6 | from agntcy_acp import ACPClient, ApiClient, AsyncACPClient, AsyncApiClient, ApiClientConfiguration |
7 | | -from agntcy_acp.models import RunCreateStateless, RunWaitResponse |
| 7 | +from agntcy_acp.models import RunCreateStateless |
8 | 8 | import logging |
9 | 9 |
|
10 | 10 | logger = logging.getLogger(__name__) |
@@ -107,15 +107,15 @@ def invoke(self, state: Any, config: RunnableConfig) -> Any: |
107 | 107 | run_create = self._prepare_run_create(state, config) |
108 | 108 | with ApiClient(configuration=self.clientConfig) as api_client: |
109 | 109 | 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) |
111 | 111 |
|
112 | 112 | return self._set_output(state, run_output.output) |
113 | 113 |
|
114 | 114 | async def ainvoke(self, state: Any, config: RunnableConfig) -> Any: |
115 | 115 | run_create = self._prepare_run_create(state, config) |
116 | 116 | async with AsyncApiClient(configuration=self.clientConfig) as api_client: |
117 | 117 | 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) |
119 | 119 |
|
120 | 120 | return self._set_output(state, run_output.output) |
121 | 121 |
|
|
0 commit comments