Skip to content

Commit 55fdc03

Browse files
committed
fix: get values for output because typing does not work.
1 parent 823eda4 commit 55fdc03

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

agntcy_acp/langgraph/acp_node.py

+3-4
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
# Copyright AGNTCY Contributors (https://github.com/agntcy)
22
# SPDX-License-Identifier: Apache-2.0
3-
from typing import Any, Optional
3+
from typing import Any, Optional, Dict
44
from langchain_core.runnables import RunnableConfig
55
from langgraph.utils.runnable import RunnableCallable
66
from agntcy_acp import (
@@ -87,7 +87,7 @@ def _extract_config(self, config: Any):
8787

8888
return self.configType.model_validate(config)
8989

90-
def _set_output(self, state: Any, output: Any):
90+
def _set_output(self, state: Any, output: Dict[str, Any]):
9191
output_parent = state
9292
for el in self.outputPath.split(".")[:-1]:
9393
output_parent = getattr(output_parent, el)
@@ -107,8 +107,7 @@ def _prepare_run_create(self, state: Any, config: RunnableConfig) -> RunCreateSt
107107

108108
def _handle_run_output(self, state: Any, run_output: RunOutput):
109109
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())
112111
elif isinstance(run_output.actual_instance, RunError):
113112
run_error: RunError = run_output.actual_instance
114113
raise ACPRunException(f"Run Failed: {run_error}")

0 commit comments

Comments
 (0)