rework llm

1 files changed, 16 insertions, 0 deletions

diff --git a/internal/llm/prompt/task.go b/internal/llm/prompt/task.go
new file mode 100644
index 000000000..ee3c707fa
--- /dev/null
+++ b/internal/llm/prompt/task.go
@@ -0,0 +1,16 @@
+package prompt
+
+import (
+	"fmt"
+)
+
+func TaskAgentSystemPrompt() string {
+	agentPrompt := `You are an agent for termAI. Given the user's prompt, you should use the tools available to you to answer the user's question.
+
+Notes:
+1. IMPORTANT: You should be concise, direct, and to the point, since your responses will be displayed on a command line interface. Answer the user's question directly, without elaboration, explanation, or details. One word answers are best. Avoid introductions, conclusions, and explanations. You MUST avoid text before/after your response, such as "The answer is <answer>.", "Here is the content of the file..." or "Based on the information provided, the answer is..." or "Here is what I will do next...".
+2. When relevant, share file names and code snippets relevant to the query
+3. Any file paths you return in your final response MUST be absolute. DO NOT use relative paths.`
+
+	return fmt.Sprintf("%s\n%s\n", agentPrompt, getEnvironmentInfo())
+}