add prompt and context

Author: wadoon

keyext.llm/src/main/java/org/key_project/key/llm/LlmClient.java

@@ -1,19 +1,16 @@
 package org.key_project.key.llm;
 
 import com.google.gson.GsonBuilder;
-import org.apache.hc.client5.http.classic.methods.HttpGet;
 import org.apache.hc.client5.http.classic.methods.HttpPost;
 import org.apache.hc.client5.http.impl.classic.AbstractHttpClientResponseHandler;
 import org.apache.hc.client5.http.impl.classic.HttpClients;
 import org.apache.hc.core5.http.HttpEntity;
 import org.apache.hc.core5.http.ParseException;
 import org.apache.hc.core5.http.io.entity.EntityUtils;
 import org.apache.hc.core5.http.io.entity.StringEntity;
-import org.key_project.util.java.IOUtil;
-import org.slf4j.LoggerFactory;
 
 import java.io.IOException;
-import java.util.List;
+import java.util.ArrayList;
 import java.util.Map;
 import java.util.concurrent.Callable;
 
@@ -24,9 +21,13 @@
  */
 public class LlmClient implements Callable<Map<String, Object>> {
     private final LlmSession llmSession;
+    private final LlmContext context;
+    private final LlmContext.LlmMessage prompt;
 
-    public LlmClient(LlmSession llmSession) {
+    public LlmClient(LlmSession llmSession, LlmContext context, String message) {
         this.llmSession = llmSession;
+        this.context = context;
+        this.prompt = new LlmContext.LlmMessage("user", message);
     }
 
     @Override
@@ -38,13 +39,13 @@ public Map<String, Object> call() throws Exception {
         request.addHeader("Content-Type", "application/json");
         request.addHeader("Accept", "application/json");
 
+        var msg = new ArrayList<>(context.getMessages());
+        msg.add(prompt);
+
         var data = Map.of(
-                "model", "azure.gpt-4.1-mini",
-                "messages", List.of(
-                        createMessage("system", "Du bist ein hilfreicher Assistent am KIT."),
-                        createMessage("user", "Erkläre das Prinzip der Rayleigh-Streuung indrei Sätzen.")
-                )
-        );
+                "model", llmSession.getModel(),
+                "messages", msg);
+
         var gson = new GsonBuilder().create();
         var stringBody = gson.toJson(data);
         request.setEntity(new StringEntity(stringBody));
@@ -54,10 +55,6 @@ public Map<String, Object> call() throws Exception {
         }
     }
 
-    private Map<String, Object> createMessage(String role, String content) {
-        return Map.of("role", role, "content", content);
-    }
-
 
     private static class GsonHttpClientResponseHandler extends AbstractHttpClientResponseHandler<Map<String, Object>> {
         @Override

keyext.llm/src/main/java/org/key_project/key/llm/LlmContext.java

@@ -0,0 +1,24 @@
+package org.key_project.key.llm;
+
+import java.util.ArrayList;
+import java.util.List;
+
+/**
+ *
+ * @author Alexander Weigl
+ * @version 1 (11/19/25)
+ */
+public class LlmContext {
+    private final List<LlmMessage> messages = new ArrayList<>();
+
+    public void addMessage(LlmMessage message) {
+        messages.add(message);
+    }
+
+    public List<LlmMessage> getMessages() {
+        return messages;
+    }
+
+    public record LlmMessage(String role, String content) {
+    }
+}

keyext.llm/src/main/java/org/key_project/key/llm/LlmPrompt.java

@@ -51,11 +51,9 @@ public void keyTyped(KeyEvent e) {
                     var node = MainWindow.getInstance().getMediator().getSelectedNode();
 
                     LlmSession session = LlmUtils.getSession(proof);
-                    LlmClient client = new LlmClient(session);
-
                     var txt = txtInput.getText();
-                    addBox(txt);
-
+                    LlmClient client = new LlmClient(session, new LlmContext(), txt);
+                    addInput(txt);
                     txtInput.setText("");
 
                     var sw = new SwingWorker<Map<String, Object>, Void>() {
@@ -68,9 +66,9 @@ protected Map<String, Object> doInBackground() throws Exception {
                         protected void done() {
                             try {
                                 handle(resultNow());
-                            } catch (Exception ex) {
-                                LOGGER.error(ex.getMessage(), ex);
-                                handle(ex);
+                            } catch (IllegalStateException ex) {
+                                LOGGER.error("Exceptional case", exceptionNow());
+                                handle(exceptionNow());
                             }
                         }
                     };
@@ -88,6 +86,7 @@ public static class OutputBox<T> extends JPanel {
 
         public OutputBox(T userData) {
             this(userData, userData.toString());
+            output.add(menu);
         }
 
         public OutputBox(T userData, String text) {

keyext.llm/src/main/java/org/key_project/key/llm/LlmSession.java

@@ -6,6 +6,7 @@
  * @version 1 (11/18/25)
  */
 public class LlmSession {
+    private String model = "azure.gpt-4.1-mini";
     private String apiEndpoint;
     private String authToken;
 
@@ -29,4 +30,12 @@ public String getAuthToken() {
     public void setAuthToken(String authToken) {
         this.authToken = authToken;
     }
+
+    public String getModel() {
+        return model;
+    }
+
+    public void setModel(String model) {
+        this.model = model;
+    }
 }