omnikey-cli 1.0.19 → 1.0.20

package/backend-dist/agent/agentPrompts.js

@@ -26,14 +26,16 @@ ${hasTaskInstructions
 - If a request needs BOTH machine data AND web search: emit a \`<shell_script>\` first → wait for \`TERMINAL OUTPUT:\` → then call the web tool with concrete values. Never use placeholders like "my IP" in a web query.
 
 **Incoming message tags:**
-- \`TERMINAL OUTPUT:\` — stdout/stderr from a prior script. Use it to continue reasoning or emit a follow-up.
+- \`TERMINAL OUTPUT:\` — stdout/stderr from a prior script. Analyze it immediately and respond with EITHER a follow-up \`<shell_script>\` (if more data is needed) OR a \`<final_answer>\` (if you have enough to conclude). You MUST pick one — never respond with plain text.
 - \`COMMAND ERROR:\` — script failed. Diagnose and emit a corrected \`<shell_script>\` or explain in \`<final_answer>\`.
 - No prefix — direct user message; treat as the primary request.
 
 **Response format — every response must be exactly one of:**
-1. \`<shell_script>...</shell_script>\` — to run commands.
+1. \`<shell_script>...</shell_script>\` — to run commands and gather more data.
 2. A \`web_search\` or \`web_fetch\` tool call — to fetch web context (use native tool calling, not XML tags).
-3. \`<final_answer>...</final_answer>\` — when done.
+3. \`<final_answer>...</final_answer>\` — your conclusion once you have enough information.
+
+**Critical rule:** After receiving \`TERMINAL OUTPUT:\` you MUST immediately produce either \`<shell_script>\` or \`<final_answer>\`. Never output raw text, markdown, or any other format. If the terminal output contains enough information to answer the user's request, output \`<final_answer>\` right away.
 
 No plain text, reasoning, or other tags outside these blocks. Never wrap in additional XML/JSON.
 

package/backend-dist/agent/agentServer.js

@@ -123,6 +123,9 @@ async function runToolLoop(initialResult, session, sessionId, send, log, tools,
     log.info('Finished reasoning and tool calls: ', {
         reason: result.finish_reason,
     });
+    if (result.assistantMessage) {
+        session.history.push(result.assistantMessage);
+    }
     return result;
 }
 function buildAvailableTools() {
@@ -254,7 +257,12 @@ async function authenticateFromAuthHeader(authHeader, log) {
     }
 }
 function createUserContent(content, hasStoredPrompt) {
-    return hasStoredPrompt ? content.replace(/@omniAgent/g, '').trim() : content;
+    return hasStoredPrompt
+        ? content
+            .toLowerCase()
+            .replace(/@omniagent/g, '')
+            .trim()
+        : content;
 }
 async function runAgentTurn(sessionId, subscription, clientMessage, send, log) {
     const { sessionState: session, hasStoredPrompt } = await getOrCreateSession(sessionId, subscription, clientMessage.platform, log);
@@ -293,6 +301,10 @@ async function runAgentTurn(sessionId, subscription, clientMessage, send, log) {
     });
     const isAssistance = isTerminalOutput || isErrorFlag;
     if (!clientMessage?.is_web_call) {
+        // Terminal output and command errors are always user-role messages — they
+        // represent environment feedback that the agent must reason about next.
+        // Pushing them as 'assistant' would create two consecutive assistant turns
+        // which breaks most LLM APIs and prevents the model from processing the output.
         session.history.push({
             role: 'user',
             content: isAssistance
@@ -306,7 +318,7 @@ async function runAgentTurn(sessionId, subscription, clientMessage, send, log) {
     const tools = isFinalTurn ? undefined : buildAvailableTools();
     const recordUsage = async (result) => {
         const usage = result.usage;
-        if (!usage || !subscription.id)
+        if (!usage || !subscription.id || config_1.config.isSelfHosted)
             return;
         try {
             await subscriptionUsage_1.SubscriptionUsage.create({
@@ -367,6 +379,13 @@ async function runAgentTurn(sessionId, subscription, clientMessage, send, log) {
                 sessionMessages.delete(sessionId);
                 return;
             }
+            await runAgentTurn(sessionId, subscription, {
+                sender: 'agent',
+                session_id: sessionId,
+                content: '',
+                is_web_call: true,
+            }, send, logger_1.logger);
+            return;
         }
         // Ensure that a proper <final_answer> block is produced for the
         // desktop clients once we reach the final turn. If the model did
@@ -410,6 +429,31 @@ async function runAgentTurn(sessionId, subscription, clientMessage, send, log) {
             });
             sessionMessages.delete(sessionId);
         }
+        else if (content) {
+            // Fallback: the LLM returned content without any recognized tag and it
+            // is not the final turn (e.g. plain-text conclusion after terminal
+            // output). Treat it as a final answer so the client is never left
+            // hanging.
+            log.info('Agent returned untagged content on a non-final turn; treating as final answer', {
+                sessionId,
+                subscriptionId: subscription.id,
+                turn: session.turns,
+            });
+            session.history.push({ role: 'assistant', content });
+            send({
+                session_id: sessionId,
+                sender: 'agent',
+                content: `<final_answer>\n${content}\n</final_answer>`,
+            });
+            sessionMessages.delete(sessionId);
+        }
+        else {
+            log.warn('Agent returned empty content with no recognized tags; sending error', {
+                sessionId,
+            });
+            sendFinalAnswer(send, sessionId, 'The agent returned an empty response. Please try again.', true);
+            sessionMessages.delete(sessionId);
+        }
     }
     catch (err) {
         log.error('Agent LLM call failed', { error: err });

package/backend-dist/featureRoutes.js

@@ -118,7 +118,7 @@ async function enhanceText(logger, text, cmd, subscription) {
         const { rawResponse, usage, model } = result;
         // Record token usage for this subscription and model, if usage
         // data is available and we know which subscription made the call.
-        if (usage && subscription.id) {
+        if (usage && subscription.id && !config_1.config.isSelfHosted) {
             try {
                 await subscriptionUsage_1.SubscriptionUsage.create({
                     subscriptionId: subscription.id,
@@ -197,7 +197,7 @@ async function streamEnhanceResponse(res, text, cmd) {
             return;
         }
         const { usage, model } = result;
-        if (usage && subscription.id) {
+        if (usage && subscription.id && !config_1.config.isSelfHosted) {
             try {
                 await subscriptionUsage_1.SubscriptionUsage.create({
                     subscriptionId: subscription.id,

package/package.json

@@ -4,7 +4,7 @@
     "access": "public",
     "registry": "https://registry.npmjs.org/"
   },
-  "version": "1.0.19",
+  "version": "1.0.20",
   "description": "CLI for onboarding users to Omnikey AI and configuring OPENAI_API_KEY. Use Yarn for install/build.",
   "engines": {
     "node": ">=14.0.0",