Include AI model in plugin data

cjmalloy · Dec 20, 2023 · 4f44cf9 · 4f44cf9
1 parent 4d001fb
commit 4f44cf9
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/main/java/jasper/component/delta/Ai.java b/src/main/java/jasper/component/delta/Ai.java
@@ -4,6 +4,8 @@
 import com.fasterxml.jackson.core.type.TypeReference;
 import com.fasterxml.jackson.databind.JsonNode;
 import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import com.fasterxml.jackson.databind.node.TextNode;
 import com.theokanning.openai.OpenAiHttpException;
 import com.theokanning.openai.completion.chat.ChatCompletionChoice;
 import com.theokanning.openai.completion.chat.ChatMessage;
@@ -135,7 +137,9 @@ public void run(Ref ref) throws JsonProcessingException {
 					response.setComment(reply);
 					response.setTags(new ArrayList<>(List.of("plugin/debug", "+plugin/openai")));
 				}
-				response.setPlugin("+plugin/openai", objectMapper.convertValue(res.getUsage(), JsonNode.class));
+				var responsePlugin = objectMapper.convertValue(res.getUsage(), ObjectNode.class);
+				responsePlugin.set("model", TextNode.valueOf(config.model));
+				response.setPlugin("+plugin/openai", responsePlugin);
 				break;
 			} catch (Exception e) {
 				if (e instanceof OpenAiHttpException o) {