fix chat history in copilot

Author: ajshedivy

package.json

@@ -58,6 +58,10 @@
           {
             "name": "activity",
             "description": "Summarize the activity on the system"
+          },
+          {
+            "name": "set-schema",
+            "description": "Set the current schema"
           }
         ]
       }

src/aiProviders/copilot/chat.ts

@@ -10,8 +10,10 @@ import {
   refsToMarkdown,
 } from "../context";
 import { chatRequest } from "./send";
+import { JobManager } from "../../config";
 
 const CHAT_ID = `vscode-db2i.chat`;
+let usingSchema = getDefaultSchema();
 
 interface IDB2ChatResult extends vscode.ChatResult {
   metadata: {
@@ -36,7 +38,6 @@ export function activateChat(context: vscode.ExtensionContext) {
     let messages: vscode.LanguageModelChatMessage[];
 
     if (canTalkToDb()) {
-      const usingSchema = getDefaultSchema();
 
       switch (request.command) {
         case `activity`:
@@ -59,6 +60,21 @@ export function activateChat(context: vscode.ExtensionContext) {
 
           return { metadata: { command: "activity" } };
 
+        case `set-schema`:
+          stream.progress(`Setting Current Schema for SQL Job`);
+          const newSchema = request.prompt.split(' ')[0];
+          if (newSchema) {
+            const curJob = JobManager.getSelection();
+            if (curJob) {
+              const result = await curJob.job.setCurrentSchema(newSchema);
+              if (result) {
+                stream.progress(`Set Current Schema: ${newSchema}✅`);
+                usingSchema = newSchema;
+              }
+            }
+            return;
+          } 
+
         default:
           stream.progress(
             `Getting information from ${Statement.prettyName(usingSchema)}...`
@@ -82,18 +98,20 @@ export function activateChat(context: vscode.ExtensionContext) {
           ];
 
           if (context.history.length > 0) {
-            messages.push(...context.history.map(h => {
-              if ('prompt' in h) {
-                return vscode.LanguageModelChatMessage.Assistant(h.prompt);
-              } else {
-                return vscode.LanguageModelChatMessage.Assistant(
-                  h.response.filter(r => 'value' in r.value).map(r => r.value.value).join(`\n\n`)
-                );
-              }
-            }));
-
-            messages = messages.filter(m => m.content.trim().length > 0);
-          }
+            const historyMessages = context.history.map(h => {
+                if ('prompt' in h) {
+                    return vscode.LanguageModelChatMessage.Assistant(h.prompt);
+                } else {
+                    const responseContent = h.response
+                        .filter(r => r.value instanceof vscode.MarkdownString)
+                        .map(r => (r.value as vscode.MarkdownString).value)
+                        .join('\n\n');
+                    return vscode.LanguageModelChatMessage.Assistant(responseContent);
+                }
+            });
+        
+            messages.push(...historyMessages);
+        }
 
           if (Object.keys(refs).length > 0) {
             messages.push(

src/connection/sqlJob.ts

@@ -4,6 +4,7 @@ import { SelfValue } from "../views/jobManager/selfCodes/nodes";
 import { SQLJob } from "@ibm/mapepire-js";
 import { ConnectionResult, JobStatus, QueryResult, ServerRequest, ServerResponse } from "@ibm/mapepire-js/dist/src/types";
 import { JobLogEntry } from "./types";
+import Statement from "../database/statement";
 
 const DB2I_VERSION = (process.env[`DB2I_VERSION`] || `<version unknown>`) + ((process.env.DEV) ? ``:`-dev`);
 
@@ -168,6 +169,19 @@ export class OldSQLJob extends SQLJob {
       throw e;
     }
   }
+  
+  async setCurrentSchema(schema: string): Promise<QueryResult<any>> {
+    if (schema) {
+      const upperSchema = Statement.delimName(schema, true);
+      const result = await this.execute(`set current schema = ?`, {parameters: [upperSchema]});
+      if (result.success) {
+        this.options.libraries[0] = upperSchema;
+      }
+
+      return result;
+  
+    }
+  }
 
   getJobLog(): Promise<QueryResult<JobLogEntry>> {
     return this.query<JobLogEntry>(`select * from table(qsys2.joblog_info('*')) a`).execute();