Context items

Signed-off-by: worksofliam <[email protected]>

Author: worksofliam

src/aiProviders/copilot/index.ts

@@ -2,7 +2,7 @@ import * as vscode from "vscode";
 import {
   canTalkToDb,
 } from "../context";
-import { buildPrompt, Db2ContextItems } from "../prompt";
+import { getContextItems, Db2ContextItems } from "../prompt";
 import { registerSqlRunTool, RUN_SQL_TOOL_ID } from "./sqlTool";
 
 const CHAT_ID = `vscode-db2i.chat`;
@@ -82,7 +82,7 @@ export function activateChat(context: vscode.ExtensionContext) {
             });
           }
 
-          const contextItems = await buildPrompt(request.prompt, {
+          const contextItems = await getContextItems(request.prompt, {
             history,
             progress: stream.progress
           });

src/aiProviders/prompt.ts

@@ -7,6 +7,7 @@ import { buildSchemaDefinition, canTalkToDb, getContentItemsForRefs, getSqlConte
 import { DB2_SYSTEM_PROMPT } from "./prompts";
 
 export interface PromptOptions {
+  userInputNotContext?: boolean
   history?: Db2ContextItems[];
   progress?: (text: string) => void;
 }
@@ -24,7 +25,7 @@ export interface BuildResult {
   followUps: string[];
 }
 
-export async function buildPrompt(input: string, options: PromptOptions = {}): Promise<BuildResult> {
+export async function getContextItems(input: string, options: PromptOptions = {}): Promise<BuildResult> {
   const currentJob: JobInfo = JobManager.getSelection();
 
   let contextItems: Db2ContextItems[] = [];
@@ -123,12 +124,14 @@ export async function buildPrompt(input: string, options: PromptOptions = {}): P
       });
     }
 
-    contextItems.push({
-      name: `user prompt`,
-      content: input,
-      description: `user prompt`,
-      type: `user`
-    });
+    if (!options.userInputNotContext) {
+      contextItems.push({
+        name: `user prompt`,
+        content: input,
+        description: `user prompt`,
+        type: `user`
+      });
+    }
   }
 
   return {