sam-coder-cli 1.0.44 → 1.0.45

package/bin/agi-cli.js

@@ -15,8 +15,42 @@ let OPENROUTER_API_KEY;
 let MODEL = 'deepseek/deepseek-chat-v3-0324:free';
 let API_BASE_URL = 'https://openrouter.ai/api/v1';
 
+// Tool/Function definitions for the AI
+// Knowledge management functions
+const KNOWLEDGE_FILE = path.join(os.homedir(), '.sam-coder-knowledge.json');
+
+async function loadKnowledge() {
+  try {
+    const data = await fs.readFile(KNOWLEDGE_FILE, 'utf8');
+    return JSON.parse(data);
+  } catch (error) {
+    // If file doesn't exist, return empty knowledge
+    return {};
+  }
+}
+
+async function saveKnowledge(knowledge) {
+  await fs.writeFile(KNOWLEDGE_FILE, JSON.stringify(knowledge, null, 2), 'utf8');
+}
+
 // Tool/Function definitions for the AI
 const tools = [
+  {
+    type: 'function',
+    function: {
+      name: 'addKnowledge',
+      description: 'Add or update knowledge in the knowledge base',
+      parameters: {
+        type: 'object',
+        properties: {
+          key: { type: 'string', description: 'Key/identifier for the knowledge' },
+          value: { type: 'string', description: 'The knowledge content' },
+          description: { type: 'string', description: 'Description of what this knowledge is about' }
+        },
+        required: ['key', 'value', 'description']
+      }
+    }
+  },
   {
     type: 'function',
     function: {
@@ -190,12 +224,19 @@ const tools = [
 // System prompt for the AI Assistant when using tool calling
 const TOOL_CALLING_PROMPT = `You are a helpful AI assistant with agency capabilities. You can perform actions on the user's system using the provided tools.
 
+KNOWLEDGE MANAGEMENT:
+- You can store and retrieve knowledge using the 'addKnowledge' tool.
+- Use this to remember user preferences, project details, or any other important information.
+- The knowledge will persist across sessions.
+- Example: If the user mentions they prefer Python for all projects, store this preference.
+
 TOOLS AVAILABLE:
-1. readFile - Read the contents of a file
-2. writeFile - Write content to a file
-3. editFile - Edit specific parts of a file
-4. runCommand - Execute a shell command
-5. searchFiles - Search for files using a glob pattern
+1. addKnowledge - Add or update knowledge in the knowledge base
+2. readFile - Read the contents of a file
+3. writeFile - Write content to a file
+4. editFile - Edit specific parts of a file
+5. runCommand - Execute a shell command
+6. searchFiles - Search for files using a glob pattern
 
 ENVIRONMENT:
 - OS: ${process.platform}
@@ -355,6 +396,28 @@ const agentUtils = {
   }
 };
 
+const toolHandlers = {
+  addKnowledge: async ({ key, value, description }) => {
+    try {
+      const knowledge = await loadKnowledge();
+      knowledge[key] = { 
+        value, 
+        description,
+        timestamp: new Date().toISOString()
+      };
+      await saveKnowledge(knowledge);
+      return { success: true, message: `Knowledge '${key}' added successfully` };
+    } catch (error) {
+      return { success: false, error: error.message };
+    }
+  },
+  readFile,
+  writeFile,
+  editFile,
+  runCommand,
+  searchFiles
+};
+
 // Extract JSON from markdown code blocks
 function extractJsonFromMarkdown(text) {
   // Try to find a markdown code block with JSON content
@@ -428,7 +491,29 @@ async function callOpenRouter(messages, currentModel, useJson = false) {
 // Process a query with tool calling
 async function processQueryWithTools(query, conversation = [], currentModel) {
   const userMessage = { role: 'user', content: query };
-  const messages = [...conversation, userMessage];
+  let messages = [...conversation, userMessage];
+
+  // Add system message with knowledge if this is the first message
+  if (conversation.length === 0) {
+    // Load knowledge and include it in the system prompt
+    const knowledge = await loadKnowledge();
+    let systemPrompt = TOOL_CALLING_PROMPT;
+    
+    // Add knowledge to the system prompt if any exists
+    const knowledgeEntries = Object.entries(knowledge);
+    if (knowledgeEntries.length > 0) {
+      systemPrompt += '\n\nCURRENT KNOWLEDGE BASE:\n';
+      systemPrompt += knowledgeEntries.map(([key, { value, description }]) => 
+        `- ${key}: ${value} (${description})`
+      ).join('\n');
+    }
+    
+    // Add system message at the beginning
+    messages = [
+      { role: 'system', content: systemPrompt },
+      ...messages
+    ];
+  }
 
   ui.startThinking();
 
@@ -471,12 +556,20 @@ async function processQueryWithTools(query, conversation = [], currentModel) {
 async function handleToolCalls(toolCalls, messages) {
   const results = [];
   
+  // Load knowledge at the start of processing tool calls
+  const knowledge = await loadKnowledge();
+  
   for (const toolCall of toolCalls) {
     const functionName = toolCall.function.name;
     let args;
     
     try {
       args = JSON.parse(toolCall.function.arguments);
+      
+      // If we have knowledge that might be relevant, include it in the context
+      if (knowledge.defaultLanguage && !args.language) {
+        args.language = knowledge.defaultLanguage.value;
+      }
     } catch (error) {
       console.error('❌ Failed to parse tool arguments:', error);
       results.push({
@@ -647,18 +740,28 @@ async function chat(rl, useToolCalling, initialModel) {
     if (input.toLowerCase() === '/setup') {
       await runSetup(rl, true);
       console.log('\nSetup complete. Please restart the application to apply changes.');
-      rl.close();
+      rl.prompt();
       return;
     }
-
-    if (input.toLowerCase() === 'exit') {
-      rl.close();
+    
+    if (input.toLowerCase() === '/eraseknowledge') {
+      try {
+        await fs.unlink(KNOWLEDGE_FILE);
+        console.log('✅ All knowledge has been erased.');
+      } catch (error) {
+        if (error.code === 'ENOENT') {
+          console.log('ℹ️ No knowledge to erase.');
+        } else {
+          console.error('❌ Failed to erase knowledge:', error.message);
+        }
+      }
+      rl.prompt();
       return;
     }
     
     const result = useToolCalling 
-      ? await processQueryWithTools(input, conversation, currentModel)
-      : await processQuery(input, conversation, currentModel);
+      ? await processQueryWithTools(query, conversation, currentModel)
+      : await processQuery(query, conversation, currentModel);
     ui.stopThinking();
     ui.showResponse(result.response);
     

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "sam-coder-cli",
-  "version": "1.0.44",
+  "version": "1.0.45",
   "description": "SAM-CODER: An animated command-line AI assistant with agency capabilities.",
   "main": "bin/agi-cli.js",
   "bin": {
@@ -24,6 +24,7 @@
     "ws": "^8.18.3"
   },
   "devDependencies": {
+    "@types/vscode": "^1.102.0",
     "eslint": "^7.27.0"
   },
   "engines": {

package/bin/agi-cli.js.new

@@ -1,328 +0,0 @@
-#!/usr/bin/env node
-
-const readline = require('readline');
-const path = require('path');
-const fs = require('fs').promises;
-const { exec } = require('child_process');
-const util = require('util');
-const execAsync = util.promisify(exec);
-
-// Configuration
-const MODEL = 'deepseek/deepseek-chat-v3-0324:free';
-const API_BASE_URL = 'https://openrouter.ai/api/v1';
-
-// System prompt matching the VS Code extension
-const SYSTEM_PROMPT = `You are a VS Code AI Assistant with agency capabilities. You can perform actions on the user's workspace.
-
-ENVIRONMENT CONTEXT:
-- OS: ${process.platform}
-- Working Directory: ${process.cwd()}
-
-When you need to perform actions, respond with JSON in the following format:
-\`\`\`json
-{
-  "thoughts": "Your reasoning about what needs to be done",
-  "actions": [
-    {
-      "type": "read|write|search|command|analyze|execute|stop",
-      "data": { ... action specific data ... }
-    }
-  ]
-}
-\`\`\`
-
-Action types and their data:
-- read: { "path": "relative/or/absolute/path" }
-- write: { "path": "relative/or/absolute/path", "content": "file content" }
-- search: { "type": "files", "pattern": "glob pattern" } or { "type": "text", "text": "search text" }
-- command: { "command": "command string to execute in terminal" }
-- execute: { "language": "js|python|bash|...", "code": "code to execute" }
-- analyze: { "code": "code to analyze", "question": "what you want to analyze" }
-- browse: { "query": "search query", "numResults": 5 } (free web search using DuckDuckGo, optional numResults)
-- edit: {
-    "path": "relative/or/absolute/path",
-    "edits": {
-      "operations": [
-        { "type": "replace", "startLine": 10, "endLine": 15, "newText": "new code here" },
-        { "type": "replace", "pattern": "oldFunction\\(\\)", "replacement": "newFunction()", "flags": "g" },
-        { "type": "insert", "line": 20, "text": "new line of code here" },
-        { "type": "insert", "position": "start", "text": "// Header comment" },
-        { "type": "insert", "position": "end", "text": "// Footer comment" },
-        { "type": "delete", "startLine": 25, "endLine": 30 }
-      ]
-    }
-  } (edit specific parts of an existing file)
-- stop: {} (use this to indicate you're done with the task and no more actions are needed)
-
-By default, you will continue to take actions in a loop until you decide to stop with the 'stop' action type.
-Always wrap your JSON in markdown code blocks with the json language specifier.
-When executing code or commands that might be potentially harmful, explain what the code does before executing it.`;
-
-// Agent utilities
-const agentUtils = {
-  async readFile(filePath) {
-    try {
-      const content = await fs.readFile(filePath, 'utf-8');
-      return content;
-    } catch (error) {
-      throw new Error(`Failed to read file ${filePath}: ${error.message}`);
-    }
-  },
-
-  async writeFile(filePath, content) {
-    try {
-      await fs.writeFile(filePath, content, 'utf-8');
-      return `Successfully wrote to ${filePath}`;
-    } catch (error) {
-      throw new Error(`Failed to write to file ${filePath}: ${error.message}`);
-    }
-  },
-
-  async runCommand(command) {
-    try {
-      const { stdout, stderr } = await execAsync(command, { cwd: process.cwd() });
-      if (stderr) {
-        console.error('Command stderr:', stderr);
-      }
-      return stdout || 'Command executed successfully (no output)';
-    } catch (error) {
-      throw new Error(`Command failed: ${error.message}`);
-    }
-  },
-
-  async searchFiles(pattern) {
-    try {
-      const { stdout } = await execAsync(`find . -name "${pattern}"`, { cwd: process.cwd() });
-      return stdout || 'No files found';
-    } catch (error) {
-      throw new Error(`Search failed: ${error.message}`);
-    }
-  }
-};
-
-// Extract JSON from markdown code blocks
-function extractJsonFromMarkdown(text) {
-  // Try to find a markdown code block with JSON content
-  const codeBlockRegex = /```json\s*([\s\S]*?)\s*```/;
-  const match = text.match(codeBlockRegex);
-  
-  if (match) {
-    try {
-      return JSON.parse(match[1]);
-    } catch (error) {
-      console.error('Error parsing JSON from markdown:', error);
-    }
-  }
-  
-  // If no code block, try to parse the entire text as JSON
-  try {
-    return JSON.parse(text);
-  } catch (error) {
-    console.error('Error parsing JSON:', error);
-    return null;
-  }
-}
-
-// Call OpenRouter API
-async function callOpenRouter(messages) {
-  const apiKey = process.env.OPENROUTER_API_KEY;
-  
-  if (!apiKey) {
-    throw new Error('OPENROUTER_API_KEY environment variable is not set');
-  }
-
-  try {
-    const response = await fetch(API_BASE_URL + '/chat/completions', {
-      method: 'POST',
-      headers: {
-        'Content-Type': 'application/json',
-        'Authorization': `Bearer ${apiKey}`,
-        'HTTP-Referer': 'https://github.com/yourusername/agi-cli'
-      },
-      body: JSON.stringify({
-        model: MODEL,
-        messages: messages
-      })
-    });
-
-    if (!response.ok) {
-      const error = await response.json();
-      throw new Error(`API error: ${error.error?.message || response.statusText}`);
-    }
-
-    return await response.json();
-  } catch (error) {
-    console.error('API call failed:', error);
-    throw new Error(`Failed to call OpenRouter API: ${error.message}`);
-  }
-}
-
-// Process a query with action handling
-async function processQuery(query, conversation = []) {
-  try {
-    // Add user message to conversation
-    const userMessage = { role: 'user', content: query };
-    const messages = [...conversation, userMessage];
-
-    // Add system message if this is the first message
-    if (conversation.length === 0) {
-      messages.unshift({
-        role: 'system',
-        content: SYSTEM_PROMPT
-      });
-    }
-
-    let shouldContinue = true;
-    let iteration = 0;
-    const maxIterations = 10; // Prevent infinite loops
-    let finalResponse = '';
-
-    while (shouldContinue && iteration < maxIterations) {
-      iteration++;
-      console.log('🤖 Thinking...');
-      
-      const response = await callOpenRouter(messages);
-      const assistantMessage = response.choices[0].message;
-      
-      // Add assistant's message to the conversation
-      messages.push(assistantMessage);
-      
-      // Check if the response contains actions
-      const actionData = extractJsonFromMarkdown(assistantMessage.content);
-      
-      if (actionData && actionData.actions && Array.isArray(actionData.actions)) {
-        console.log(`🔧 Processing ${actionData.actions.length} actions...`);
-        if (actionData.thoughts) {
-          console.log(`💭 ${actionData.thoughts}`);
-        }
-        
-        const actionResults = [];
-        
-        for (const action of actionData.actions) {
-          console.log(`🛠️  Executing action: ${action.type}`);
-          
-          // Handle stop action
-          if (action.type === 'stop') {
-            console.log('🛑 Stop action received, ending action processing');
-            shouldContinue = false;
-            finalResponse = 'Task completed successfully.';
-            break;
-          }
-          
-          try {
-            let result;
-            
-            switch (action.type) {
-              case 'read':
-                result = await agentUtils.readFile(action.data.path);
-                break;
-                
-              case 'write':
-                result = await agentUtils.writeFile(action.data.path, action.data.content);
-                break;
-                
-              case 'command':
-                result = await agentUtils.runCommand(action.data.command);
-                break;
-                
-              case 'search':
-                if (action.data.type === 'files') {
-                  result = await agentUtils.searchFiles(action.data.pattern);
-                } else {
-                  result = 'Text search not yet implemented';
-                }
-                break;
-                
-              default:
-                result = `Action type '${action.type}' is not supported yet.`;
-            }
-            
-            actionResults.push({
-              type: action.type,
-              success: true,
-              result: result
-            });
-            
-            console.log(`✅ Action ${action.type} completed successfully`);
-            
-          } catch (error) {
-            console.error(`❌ Action ${action.type} failed:`, error);
-            actionResults.push({
-              type: action.type,
-              success: false,
-              error: error.message
-            });
-          }
-        }
-        
-        // Add action results to the conversation
-        messages.push({
-          role: 'system',
-          content: `Action results:\n\`\`\`json\n${JSON.stringify(actionResults, null, 2)}\n\`\`\`\n` +
-          `Based on these results, determine what to do next. You can:\n` +
-          `1. Continue with more actions by returning a new JSON with "actions" array\n` +
-          `2. Stop the iteration by including an action with "type": "stop" if the task is completed\n` +
-          `3. Provide a final response to the user with your findings`
-        });
-        
-      } else {
-        // No actions, this is a regular response
-        shouldContinue = false;
-        finalResponse = assistantMessage.content;
-      }
-    }
-    
-    // If we hit max iterations, add a note
-    if (iteration >= maxIterations) {
-      finalResponse += '\n\n⚠️ Reached maximum number of iterations. Stopping execution.';
-    }
-    
-    return {
-      response: finalResponse,
-      conversation: messages
-    };
-    
-  } catch (error) {
-    console.error('Error processing query:', error);
-    return {
-      response: `Error: ${error.message}`,
-      conversation
-    };
-  }
-}
-
-// Main chat loop
-async function chat() {
-  const conversation = [];
-  console.log('Welcome to AGI-CLI. Type your message, or "exit" to quit.');
-  
-  const rl = readline.createInterface({
-    input: process.stdin,
-    output: process.stdout,
-    prompt: '> '
-  });
-  
-  rl.prompt();
-  
-  rl.on('line', async (input) => {
-    if (input.toLowerCase() === 'exit') {
-      rl.close();
-      return;
-    }
-    
-    const result = await processQuery(input, conversation);
-    console.log(result.response);
-    
-    // Update conversation with the full context
-    conversation.length = 0; // Clear the array
-    result.conversation.forEach(msg => conversation.push(msg));
-    
-    rl.prompt();
-  }).on('close', () => {
-    console.log('Goodbye!');
-    process.exit(0);
-  });
-}
-
-// Start the chat
-chat().catch(console.error);