File tree Expand file tree Collapse file tree 2 files changed +5
-5
lines changed
spring-ai-alibaba-graph-core/src
main/java/com/alibaba/cloud/ai/graph/agent
test/java/com/alibaba/cloud/ai/graph/agent Expand file tree Collapse file tree 2 files changed +5
-5
lines changed Original file line number Diff line number Diff line change @@ -444,8 +444,8 @@ public Builder postToolHook(NodeAction postToolHook) {
444
444
return this ;
445
445
}
446
446
447
- public Builder llmInputMessagesKey (String llmInputMessagesKey ) {
448
- this .inputKey = llmInputMessagesKey ;
447
+ public Builder inputKey (String inputKey ) {
448
+ this .inputKey = inputKey ;
449
449
return this ;
450
450
}
451
451
Original file line number Diff line number Diff line change @@ -130,7 +130,7 @@ public void testReactAgentWithPreLlmHook() throws Exception {
130
130
.name ("weather_agent" )
131
131
.model (chatModel )
132
132
.tools (List .of (toolCallback ))
133
- .llmInputMessagesKey ("llm_input_messages" )
133
+ .inputKey ("llm_input_messages" )
134
134
.preLlmHook (state -> {
135
135
if (!state .value ("messages" ).isPresent ()) {
136
136
return Map .of ();
@@ -178,7 +178,7 @@ public void testReactAgentWithPostLlmHook() throws Exception {
178
178
ReactAgent agent = ReactAgent .builder ()
179
179
.name ("dataAgent" )
180
180
.model (chatModel )
181
- .llmInputMessagesKey ("llm_input_messages" )
181
+ .inputKey ("llm_input_messages" )
182
182
.tools (List .of (toolCallback ))
183
183
.postLlmHook (state -> {
184
184
@@ -212,7 +212,7 @@ public void testReactAgentWithPreToolHook() throws Exception {
212
212
.name ("dataAgent" )
213
213
.model (chatModel )
214
214
.tools (List .of (toolCallback ))
215
- .llmInputMessagesKey ("llm_input_messages" )
215
+ .inputKey ("llm_input_messages" )
216
216
.preToolHook (state -> {
217
217
// 在preToolHook中获取最新的时间戳 传给toolCall保证时效性
218
218
long currentTimestamp = System .currentTimeMillis ();
You can’t perform that action at this time.
0 commit comments