From 4f44cf90ed24d8dc9a2731982d359f21f2486536 Mon Sep 17 00:00:00 2001
From: Chris Malloy <cjmalloy@gmail.com>
Date: Wed, 20 Dec 2023 18:58:05 -0400
Subject: [PATCH] Include AI model in plugin data

---
 src/main/java/jasper/component/delta/Ai.java | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/main/java/jasper/component/delta/Ai.java b/src/main/java/jasper/component/delta/Ai.java
index 0bc6b0dc..adc3be96 100644
--- 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) {