Improved prompt for extractDBKeywords

Author: preeesha

ai-assistant/src/core/prompt/contents/extractDBKeywords.ts

@@ -1,29 +1,23 @@
 export const PROMPT_EXTRACT_DB_KEYWORDS = `
-You are an expert in extracting and predicting the possible keywords from the user's query about a codebase. The user might ask questions like:
-1. "Find the codebase for the user query CRC_TABLE in the main.ts"
-2. "Where is the function calculateChecksum defined in the repository?"
-3. "Which file contains the class definition for UserProfile?"
-4. "Show me the implementation of the handleSubmit method in the formHandler.js file."
-5. "Where is the API_BASE_URL constant declared?"
-6. "Find the test cases for the loginUser function."
+You are an expert in understanding and answering questions of user when given a proper context of the codebase.
 
-You have to answer these by predicting the possible keywords from the user's query that might be referenced in the codebase. For eg:
-- CRC_TABLE, main.ts
-- calculateChecksum, repository
-- UserProfile
-- handleSubmit, formHandler.js
-- API_BASE_URL
-- loginUser
+INPUT: User's text query in either natural language or code format.
 
-But you have to output them the following way:
-- <ANSWER>[CRC_TABLE,main.ts]</ANSWER>
-- <ANSWER>[calculateChecksum,repository]</ANSWER>
-- <ANSWER>[UserProfile]</ANSWER>
-- <ANSWER>[handleSubmit,formHandler.js]</ANSWER>
-- <ANSWER>[API_BASE_URL]</ANSWER>
-- <ANSWER>[loginUser]</ANSWER>
+TASKS:
+- NEVER USE backslash in any of the keywords even if it's part of the query.
+- Extract the possible keywords from the user's query.
+- Query the database to find the nodes names of which are similar to what user has requested.
+- If you are unable to extract the keywords, return an empty array.
+- EXTRACT THE KEYWORDS IN SUCH A WAY THAT EACH STRING AS ONE WORD ELEMENT ONLY.
+
+EXAMPLE:
+- INPUT:
+    "Find the codebase for the user query CRC_TABLE in the main.ts'
+- OUTPUT:
+    <ANSWER_START>CRC_TABLE, main.ts<ANSWER_END>
 
 RULES:
 - DO NOT REPEAT THE KEYWORDS MULTIPLE TIMES.
+- DO NOT SAY ANYTHING ELSE APART FROM THE PROVIDED OUTPUT FORMAT.
 - STRICTLY, do not make anything other than the answer to the user's query.
 `;

ai-assistant/src/core/prompt/prompt.factory.ts

@@ -1,7 +1,6 @@
 import { PROMPT_ASK_COMMAND } from "./contents/ask";
 import { PROMPT_DIAGRAM_COMMAND } from "./contents/diagram";
 import { PROMPT_DOCUMENT_COMMAND } from "./contents/document";
-import { PROMPT_EXTRACT_DB_KEYWORDS } from "./contents/extractDBKeywords";
 import { PROMPT_STYLEGUIDE_COMMAND } from "./contents/styleguide";
 import { PROMPT_SUGGEST_COMMAND } from "./contents/suggest";
 import { PROMPT_TRANSLATE_COMMAND } from "./contents/translate";
@@ -11,8 +10,6 @@ import { Prompt } from "./prompt";
 export namespace PromptFactory {
     export function makeDBKeywordQueryPrompt(query: string): Prompt {
         const prompt = new Prompt();
-
-        prompt.pushSystem(PROMPT_EXTRACT_DB_KEYWORDS);
         prompt.pushAssistant(
             "Sure, I will strictly follow my instructions. I will provide the answer the above specified format only."
         );
@@ -25,7 +22,7 @@ export namespace PromptFactory {
         prompt.pushAssistant(
             "Yeah sure. I understand this codebase very well and I am able to extract the possible keywords from the user's query. If I can't find the keywords, I'll return an empty array."
         );
-        prompt.pushSystem("<ANSWER>");
+        prompt.pushUser(query);
 
         return prompt;
     }