joeyutong commented on code in PR #924:
URL: https://github.com/apache/flink-agents/pull/924#discussion_r3763922566
##########
plan/src/main/java/org/apache/flink/agents/plan/actions/ChatModelAction.java:
##########
@@ -365,19 +367,35 @@ public ChatMessage call() throws Exception {
return chatModel.chat(messages, promptArgs, Map.of());
}
};
+ Map<String, Object> llmMetadata =
+ chatModel.getModel() == null ? Map.of() : Map.of(MODEL,
chatModel.getModel());
Review Comment:
Agreed. I added `LLMExecutionMetadataKeys.MODEL` and the matching Python
constant, and use them specifically for `entityMetadata.model`. The existing
Java `MODEL` constant now remains scoped to the ToolRequest context, where it
stores the ChatModel Resource name. The wire field remains `model`.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
