@pikaa-ai/pikaa 0.2.2 → 0.2.4

package/dist/index.js

@@ -152,6 +152,7 @@ class DefaultModelClientSession {
       model,
       messages,
       stream: true,
+      stream_options: { include_usage: true },
       temperature: params.temperature ?? 0.2
     };
     if (toolsPayload && toolsPayload.length > 0) {
@@ -200,6 +201,7 @@ class DefaultModelClientSession {
     const reader = response.body.getReader();
     const decoder = new TextDecoder;
     let buffer = "";
+    let usageMetrics = {};
     const pendingToolCalls = new Map;
     try {
       while (true) {
@@ -229,11 +231,23 @@ class DefaultModelClientSession {
               };
             }
             pendingToolCalls.clear();
-            yield { type: "done" };
+            yield {
+              type: "done",
+              inputTokens: usageMetrics.inputTokens,
+              outputTokens: usageMetrics.outputTokens,
+              totalTokens: usageMetrics.totalTokens
+            };
             return;
           }
           try {
             const data = JSON.parse(dataStr);
+            if (data.usage) {
+              usageMetrics = {
+                inputTokens: data.usage.prompt_tokens,
+                outputTokens: data.usage.completion_tokens,
+                totalTokens: data.usage.total_tokens
+              };
+            }
             const choice = data.choices?.[0];
             if (!choice)
               continue;
@@ -439,7 +453,7 @@ class ExecPolicy {
   }
   initDefaultRules() {
     this.addRule(/^(git\s+(status|log|diff|branch|show|rev-parse))/i, "allow", "Safe git query");
-    this.addRule(/^(ls|dir|cat|grep|rg|find|pwd|echo|head|tail|wc|which|where)\b/i, "allow", "Safe read-only shell command");
+    this.addRule(/^(ls|dir|cat|type|grep|rg|find|pwd|echo|head|tail|wc|which|where)\b/i, "allow", "Safe read-only shell command");
     this.addRule(/^(bun\s+(test|--version|-v)|npm\s+(test|--version|-v)|node\s+-v)\b/i, "allow", "Testing & runtime check");
     this.addRule(/^(rm|del|rmdir|format|mkfs)\b/i, "prompt", "Destructive file removal");
     this.addRule(/^(git\s+(push|reset\s+--hard|clean\s+-fd|rebase))\b/i, "prompt", "Destructive git operation");
@@ -489,7 +503,8 @@ function createShellTool(policy = new ExecPolicy) {
       if (!command) {
         return { output: "Error: 'command' argument cannot be empty", isError: true };
       }
-      const policyDecision = policy.evaluate(command);
+      const activePolicy = ctx.execPolicy || policy;
+      const policyDecision = activePolicy.evaluate(command);
       if (policyDecision.decision === "deny") {
         return {
           output: `Error: Command execution denied by policy: ${policyDecision.reason}`,
@@ -1423,12 +1438,16 @@ function buildSystemPrompt(params) {
   const sections = [];
   sections.push(params.basePrompt || "You are Groupy, an expert autonomous AI coding assistant. You think step-by-step, act surgically, and write clean, correct code.");
   sections.push([
-    "## Operational Guidelines",
-    "1. Think before acting. Understand the task and the code it touches before making edits.",
-    "2. Simplicity first: write the minimum code that solves the problem. No unnecessary abstractions.",
-    "3. Use `apply_patch` for surgical code replacements rather than overwriting whole files.",
-    "4. Execute shell commands with `shell` to inspect environment, run tests, or build targets.",
-    "5. Always verify your changes with tests or execution checks."
+    "## Editing Constraints & Guidelines",
+    "- Use `apply_patch` for surgical single-file edits. TargetContent must match existing file content exactly.",
+    "- Use `write_file` for creating new files or when completely replacing the full content of a file.",
+    "- Use `read_file` to inspect files and `grep_search` / `find_files` to discover symbols and locate files across the project.",
+    "- NEVER create temporary scripts, scratch files, or chunk files (e.g. `_tmp_*.ps1`, `_tmp_*.txt`, `split_*.py`) in the workspace to manipulate, split, or read files.",
+    "- NEVER execute shell or PowerShell scripts as a workaround for reading, writing, or editing text files.",
+    "- Use the `shell` tool ONLY for running tests, build targets, package installations, or checking environment/git status.",
+    "- You may be in a dirty git worktree. NEVER revert existing changes made by the user.",
+    "- NEVER use destructive commands like `git reset --hard` or `git checkout --`.",
+    "- Be concise, direct, and act surgically. Write clean, correct code with minimal necessary modifications."
   ].join(`
 `));
   if (params.memoriesPrompt) {
@@ -1550,6 +1569,8 @@ async function runTurn(session, turnContext, input) {
     skillsPrompt
   });
   let iteration = 0;
+  let accumulatedInputTokens = 0;
+  let accumulatedOutputTokens = 0;
   const clientSession = session.modelClient.newSession();
   try {
     while (iteration < turnContext.maxIterations) {
@@ -1559,6 +1580,8 @@ async function runTurn(session, turnContext, input) {
       iteration++;
       let currentAgentText = "";
       const toolCallRequests = [];
+      let iterInputTokens = Math.ceil((effectiveSystemPrompt.length + JSON.stringify(session.getHistory()).length) / 4);
+      let iterOutputTokens = 0;
       const stream = clientSession.stream({
         model: turnContext.model,
         systemPrompt: effectiveSystemPrompt,
@@ -1585,10 +1608,20 @@ async function runTurn(session, turnContext, input) {
           });
         } else if (chunk.type === "tool_call") {
           toolCallRequests.push(chunk);
+        } else if (chunk.type === "done") {
+          if (chunk.inputTokens !== undefined)
+            iterInputTokens = chunk.inputTokens;
+          if (chunk.outputTokens !== undefined)
+            iterOutputTokens = chunk.outputTokens;
         } else if (chunk.type === "error") {
           throw chunk.error;
         }
       }
+      if (iterOutputTokens === 0) {
+        iterOutputTokens = Math.ceil((currentAgentText.length + JSON.stringify(toolCallRequests).length) / 4);
+      }
+      accumulatedInputTokens += iterInputTokens;
+      accumulatedOutputTokens += iterOutputTokens;
       if (currentAgentText.trim()) {
         const agentItem = {
           id: `msg_agent_${Date.now()}`,
@@ -1629,6 +1662,7 @@ async function runTurn(session, turnContext, input) {
             cwd: turnContext.environment.cwd,
             turnId,
             signal,
+            execPolicy: session.execPolicy,
             requestApproval: async (description, command) => {
               const approvalId = `appr_${Date.now()}_${Math.random().toString(36).slice(2, 6)}`;
               return session.requestApproval({
@@ -1666,9 +1700,16 @@ async function runTurn(session, turnContext, input) {
       }
       break;
     }
+    const totalContextTokens = estimateTotalTokens(session.getHistory()) + Math.ceil(effectiveSystemPrompt.length / 4);
+    const maxContextTokens = 128000;
     session.emitEvent({
       type: "TurnCompleted",
-      turnId
+      turnId,
+      inputTokens: accumulatedInputTokens,
+      outputTokens: accumulatedOutputTokens,
+      totalTokens: accumulatedInputTokens + accumulatedOutputTokens,
+      contextTokens: totalContextTokens,
+      maxContextTokens
     });
   } catch (error) {
     const isAborted = error instanceof TurnAbortedError || signal.aborted;
@@ -1762,6 +1803,7 @@ class Session {
   tools;
   skillsLoader;
   memoryStore;
+  execPolicy;
   history = [];
   activeTurn = null;
   status = "idle";
@@ -1779,6 +1821,7 @@ class Session {
     this.tools = options.tools || new ToolRouter;
     this.skillsLoader = options.skillsLoader;
     this.memoryStore = options.memoryStore;
+    this.execPolicy = options.execPolicy || new ExecPolicy;
     this.history = options.initialHistory ? [...options.initialHistory] : [];
     if (options.onEvent) {
       this.eventListeners.push(options.onEvent);
@@ -3987,7 +4030,7 @@ class AuthClient {
                 </head>
                 <body>
                   <div class="box">
-                    <h1>\u26A1 Authentication Successful!</h1>
+                    <h1>Authentication Successful!</h1>
                     <p>You have successfully logged in to Groupy CLI. You can close this window and return to your terminal.</p>
                   </div>
                 </body>

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pikaa-ai/pikaa",
-  "version": "0.2.2",
+  "version": "0.2.4",
   "description": "PIKAA CLI - AI coding agent that runs locally in your terminal.",
   "main": "./dist/index.js",
   "module": "./dist/index.js",