1
1
# Copyright AGNTCY Contributors (https://github.com/agntcy)
2
2
# SPDX-License-Identifier: Apache-2.0
3
- from typing import Any , Optional
3
+ from typing import Any , Optional , Dict
4
4
from langchain_core .runnables import RunnableConfig
5
5
from langgraph .utils .runnable import RunnableCallable
6
6
from agntcy_acp import (
@@ -87,7 +87,7 @@ def _extract_config(self, config: Any):
87
87
88
88
return self .configType .model_validate (config )
89
89
90
- def _set_output (self , state : Any , output : Any ):
90
+ def _set_output (self , state : Any , output : Dict [ str , Any ] ):
91
91
output_parent = state
92
92
for el in self .outputPath .split ("." )[:- 1 ]:
93
93
output_parent = getattr (output_parent , el )
@@ -107,8 +107,7 @@ def _prepare_run_create(self, state: Any, config: RunnableConfig) -> RunCreateSt
107
107
108
108
def _handle_run_output (self , state : Any , run_output : RunOutput ):
109
109
if isinstance (run_output .actual_instance , RunResult ):
110
- run_result : RunResult = run_output .actual_instance
111
- self ._set_output (state , run_result .values )
110
+ self ._set_output (state , run_output .to_dict ())
112
111
elif isinstance (run_output .actual_instance , RunError ):
113
112
run_error : RunError = run_output .actual_instance
114
113
raise ACPRunException (f"Run Failed: { run_error } " )
0 commit comments