@quantish/agent 0.1.50 → 0.1.52

package/dist/index.js

@@ -3579,195 +3579,22 @@ function createLLMProvider(config) {
 }
 
 // src/agent/loop.ts
-var MAX_TOOL_RESULT_CHARS = 8e3;
-function truncateToolResult(result, toolName) {
-  const resultStr = JSON.stringify(result);
-  if (resultStr.length <= MAX_TOOL_RESULT_CHARS) {
-    return result;
-  }
-  if (typeof result === "object" && result !== null) {
-    const obj = result;
-    if (Array.isArray(obj.content) && obj.content.length > 0) {
-      const firstContent = obj.content[0];
-      if (firstContent?.type === "text" && typeof firstContent.text === "string") {
-        try {
-          const innerData = JSON.parse(firstContent.text);
-          const truncatedInner = truncateDataObject(innerData);
-          return {
-            content: [{
-              type: "text",
-              text: JSON.stringify(truncatedInner)
-            }]
-          };
-        } catch {
-          const truncatedText = firstContent.text.length > MAX_TOOL_RESULT_CHARS ? firstContent.text.substring(0, MAX_TOOL_RESULT_CHARS) + "... [truncated]" : firstContent.text;
-          return {
-            content: [{
-              type: "text",
-              text: truncatedText
-            }]
-          };
-        }
-      }
-    }
-  }
-  if (Array.isArray(result)) {
-    return truncateArray(result);
-  }
-  if (typeof result === "object" && result !== null) {
-    return truncateDataObject(result);
-  }
-  if (typeof result === "string" && result.length > MAX_TOOL_RESULT_CHARS) {
-    return result.substring(0, MAX_TOOL_RESULT_CHARS) + "... [truncated]";
-  }
-  return result;
-}
-function truncateArray(arr) {
-  const MAX_ITEMS = 5;
-  const truncated = arr.slice(0, MAX_ITEMS).map(
-    (item) => typeof item === "object" && item !== null ? truncateObject(item) : item
-  );
-  return {
-    _truncated: arr.length > MAX_ITEMS,
-    _originalCount: arr.length,
-    _note: arr.length > MAX_ITEMS ? `Showing ${MAX_ITEMS} of ${arr.length} items.` : void 0,
-    items: truncated
-  };
-}
-function truncateDataObject(obj) {
-  const truncated = {};
-  const MAX_ARRAY_ITEMS = 5;
-  for (const [key, value] of Object.entries(obj)) {
-    if (Array.isArray(value)) {
-      if (value.length > MAX_ARRAY_ITEMS) {
-        truncated[key] = value.slice(0, MAX_ARRAY_ITEMS).map(
-          (item) => typeof item === "object" && item !== null ? truncateObject(item) : item
-        );
-        truncated[`_${key}Count`] = value.length;
-        truncated["_truncated"] = true;
-      } else {
-        truncated[key] = value.map(
-          (item) => typeof item === "object" && item !== null ? truncateObject(item) : item
-        );
-      }
-    } else if (typeof value === "object" && value !== null) {
-      truncated[key] = truncateObject(value);
-    } else if (typeof value === "string" && value.length > 500) {
-      truncated[key] = value.substring(0, 500) + "...";
-    } else {
-      truncated[key] = value;
-    }
-  }
-  return truncated;
-}
-var ACTIONABLE_FIELDS = /* @__PURE__ */ new Set([
-  // Market identifiers (required for trading)
-  "conditionId",
-  "tokenId",
-  "marketId",
-  "id",
-  "ticker",
-  // Token info (required for order placement)  
-  "token_id",
-  "clobTokenIds",
-  "tokens",
-  // Pricing (required for trading decisions)
-  "price",
-  "probability",
-  "outcomePrices",
-  "bestBid",
-  "bestAsk",
-  // Market identity (for user understanding)
-  "title",
-  "question",
-  "slug",
-  "outcome",
-  "name",
-  // Status info (affects tradability)
-  "active",
-  "closed",
-  "status",
-  "endDate",
-  // Platform (for multi-platform support)
-  "platform",
-  // Nested structures containing price data (Discovery MCP response)
-  "markets",
-  "outcomes"
-]);
-var SUMMARY_FIELDS = /* @__PURE__ */ new Set([
-  "volume",
-  "liquidity",
-  "volume24hr"
-]);
-var DROP_FIELDS = /* @__PURE__ */ new Set([
-  "description",
-  "rules",
-  "resolutionSource",
-  "image",
-  "icon",
-  "createdAt",
-  "updatedAt",
-  "lastTradePrice",
-  "spread",
-  "acceptingOrders",
-  "acceptingOrdersTimestamp",
-  "minimum_tick_size",
-  "minimum_order_size",
-  "maker_base_fee",
-  "taker_base_fee",
-  "neg_risk",
-  "neg_risk_market_id",
-  "neg_risk_request_id",
-  "notifications_enabled",
-  "is_50_50_outcome",
-  "game_start_time",
-  "seconds_delay",
-  "icon",
-  "fpmm",
-  "rewards",
-  "competitive"
-]);
-function truncateObject(obj) {
-  const truncated = {};
-  for (const [key, value] of Object.entries(obj)) {
-    if (DROP_FIELDS.has(key)) continue;
-    if (ACTIONABLE_FIELDS.has(key)) {
-      if (typeof value === "string" && value.length > 150) {
-        truncated[key] = value.substring(0, 150) + "...";
-      } else if (Array.isArray(value)) {
-        truncated[key] = value.slice(0, 10).map((item) => {
-          if (typeof item === "object" && item !== null) {
-            return extractTokenInfo(item);
-          }
-          return item;
-        });
-      } else {
-        truncated[key] = value;
-      }
-      continue;
-    }
-    if (SUMMARY_FIELDS.has(key)) {
-      if (typeof value === "number" || typeof value === "string") {
-        truncated[key] = value;
-      }
-      continue;
-    }
-    if (typeof value !== "object" && JSON.stringify(truncated).length < 800) {
-      truncated[key] = value;
-    }
-  }
-  return truncated;
-}
-function extractTokenInfo(token) {
-  return {
-    token_id: token.token_id ?? token.tokenId,
-    outcome: token.outcome ?? token.name,
-    price: token.price ?? token.probability
-  };
-}
 var DEFAULT_SYSTEM_PROMPT = `You are Quantish, an AI trading agent for prediction markets (Polymarket, Kalshi).
 
-You have tools to search markets and place trades. When showing market data, display ALL relevant information from the response including prices/probabilities.
+## CRITICAL: Market Display Rules
+
+When showing market search results, ALWAYS include:
+- Market title
+- Platform  
+- **Price/Probability** (REQUIRED - never omit this)
+- Market ID
+
+Format market tables like this:
+| Market | Platform | Price | ID |
+|--------|----------|-------|-----|
+| Example market | Polymarket | Yes 45\xA2 / No 55\xA2 | 12345 |
+
+The price data is in the tool result - extract and display it.
 
 ## Building Applications
 
@@ -3808,6 +3635,12 @@ var Agent = class _Agent {
     cost: { inputCost: 0, outputCost: 0, cacheWriteCost: 0, cacheReadCost: 0, totalCost: 0 },
     sessionCost: 0
   };
+  // Sliding window context management
+  conversationSummary = null;
+  toolHistory = [];
+  exchanges = [];
+  static MAX_TOOL_HISTORY = 10;
+  static MAX_EXCHANGES = 5;
   constructor(config) {
     this.config = {
       enableLocalTools: true,
@@ -3869,11 +3702,8 @@ var Agent = class _Agent {
    * Get or create the LLM provider instance
    */
   async getOrCreateProvider() {
-    if (this.llmProvider) {
-      return this.llmProvider;
-    }
     const allTools = await this.getAllTools();
-    const systemPrompt = this.config.systemPrompt ?? DEFAULT_SYSTEM_PROMPT;
+    const systemPrompt = this.buildSystemContext();
     const defaultModel = this.config.provider === "openrouter" ? "z-ai/glm-4.7" : DEFAULT_MODEL;
     const model = this.config.model ?? defaultModel;
     const maxTokens = this.config.maxTokens ?? 8192;
@@ -3895,14 +3725,9 @@ var Agent = class _Agent {
     const maxIterations = this.config.maxIterations ?? 200;
     const useStreaming = this.config.streaming ?? true;
     const provider = await this.getOrCreateProvider();
-    const contextMessage = `[Working directory: ${this.workingDirectory}]
-
-${userMessage}`;
-    this.conversationHistory.push({
-      role: "user",
-      content: contextMessage
-    });
+    const messages = this.buildSlimHistory(userMessage);
     this.clearToolCallLoopTracking();
+    let currentTurnMessages = [...messages];
     const toolCalls = [];
     let iterations = 0;
     let finalText = "";
@@ -3916,7 +3741,7 @@ ${userMessage}`;
       this.config.onStreamStart?.();
       let response;
       if (useStreaming) {
-        response = await provider.streamChat(this.conversationHistory, {
+        response = await provider.streamChat(currentTurnMessages, {
           onText: (text) => {
             finalText += text;
             this.config.onText?.(text, false);
@@ -3932,7 +3757,7 @@ ${userMessage}`;
           this.config.onText?.("", true);
         }
       } else {
-        response = await provider.chat(this.conversationHistory);
+        response = await provider.chat(currentTurnMessages);
         if (response.text) {
           finalText += response.text;
           this.config.onText?.(response.text, true);
@@ -3958,10 +3783,6 @@ ${userMessage}`;
         });
       }
       if (response.toolCalls.length === 0) {
-        this.conversationHistory.push({
-          role: "assistant",
-          content: responseContent
-        });
         break;
       }
       const toolResults = [];
@@ -3973,6 +3794,7 @@ ${userMessage}`;
         );
         const success2 = !(result && typeof result === "object" && "error" in result);
         this.config.onToolResult?.(toolCall2.name, result, success2);
+        this.addToolHistory(toolCall2.name, toolCall2.input, success2);
         toolCalls.push({
           name: toolCall2.name,
           input: toolCall2.input,
@@ -3985,19 +3807,21 @@ ${userMessage}`;
           content: JSON.stringify(result)
         });
       }
-      this.conversationHistory.push({
+      currentTurnMessages.push({
         role: "assistant",
         content: responseContent
       });
-      this.conversationHistory.push({
+      currentTurnMessages.push({
         role: "user",
         content: toolResults
       });
-      this.truncateLastToolResults();
       if (response.stopReason === "end_turn" && response.toolCalls.length === 0) {
         break;
       }
     }
+    if (finalText.trim()) {
+      this.storeTextExchange(userMessage, finalText.trim());
+    }
     return {
       text: finalText,
       toolCalls,
@@ -4086,17 +3910,11 @@ ${userMessage}`;
     const maxIterations = this.config.maxIterations ?? 15;
     const model = this.config.model ?? "claude-sonnet-4-5-20250929";
     const maxTokens = this.config.maxTokens ?? 8192;
-    const systemPrompt = this.config.systemPrompt ?? DEFAULT_SYSTEM_PROMPT;
+    const systemPrompt = this.buildSystemContext();
     const useStreaming = this.config.streaming ?? true;
     const allTools = await this.getAllTools();
     const contextManagement = this.config.contextEditing && this.config.contextEditing.length > 0 ? { edits: this.config.contextEditing } : void 0;
-    const contextMessage = `[Working directory: ${this.workingDirectory}]
-
-${userMessage}`;
-    this.conversationHistory.push({
-      role: "user",
-      content: contextMessage
-    });
+    let currentTurnMessages = this.buildSlimHistory(userMessage);
     this.clearToolCallLoopTracking();
     const toolCalls = [];
     let iterations = 0;
@@ -4126,7 +3944,7 @@ ${userMessage}`;
           max_tokens: maxTokens,
           system: systemWithCache,
           tools: allTools,
-          messages: this.conversationHistory
+          messages: currentTurnMessages
         };
         if (contextManagement) {
           streamOptions.context_management = contextManagement;
@@ -4168,7 +3986,7 @@ ${userMessage}`;
           max_tokens: maxTokens,
           system: systemWithCache,
           tools: allTools,
-          messages: this.conversationHistory
+          messages: currentTurnMessages
         };
         if (contextManagement) {
           createOptions.context_management = contextManagement;
@@ -4189,10 +4007,6 @@ ${userMessage}`;
       }
       this.config.onStreamEnd?.();
       if (toolUses.length === 0) {
-        this.conversationHistory.push({
-          role: "assistant",
-          content: responseContent
-        });
         break;
       }
       const toolResults = [];
@@ -4204,6 +4018,7 @@ ${userMessage}`;
         );
         const success2 = !(result && typeof result === "object" && "error" in result);
         this.config.onToolResult?.(toolUse.name, result, success2);
+        this.addToolHistory(toolUse.name, toolUse.input, success2);
         toolCalls.push({
           name: toolUse.name,
           input: toolUse.input,
@@ -4216,19 +4031,21 @@ ${userMessage}`;
           content: JSON.stringify(result)
         });
       }
-      this.conversationHistory.push({
+      currentTurnMessages.push({
         role: "assistant",
         content: responseContent
       });
-      this.conversationHistory.push({
+      currentTurnMessages.push({
         role: "user",
         content: toolResults
       });
-      this.truncateLastToolResults();
       if (response.stop_reason === "end_turn" && toolUses.length === 0) {
         break;
       }
     }
+    if (finalText.trim()) {
+      this.storeTextExchange(userMessage, finalText.trim());
+    }
     return {
       text: finalText,
       toolCalls,
@@ -4241,6 +4058,9 @@ ${userMessage}`;
    */
   clearHistory() {
     this.conversationHistory = [];
+    this.conversationSummary = null;
+    this.toolHistory = [];
+    this.exchanges = [];
   }
   /**
    * Get current conversation history
@@ -4248,6 +4068,126 @@ ${userMessage}`;
   getHistory() {
     return [...this.conversationHistory];
   }
+  /**
+   * Extract primary input from tool arguments for compact history.
+   * Returns the most relevant parameter value, truncated if needed.
+   */
+  extractPrimaryInput(input) {
+    const primaryKeys = ["query", "path", "command", "marketId", "content", "url", "pattern", "ticker"];
+    for (const key of primaryKeys) {
+      if (input[key] && typeof input[key] === "string") {
+        const val = input[key];
+        return val.length > 40 ? val.slice(0, 40) + "..." : val;
+      }
+    }
+    for (const val of Object.values(input)) {
+      if (typeof val === "string" && val.length > 0) {
+        return val.length > 40 ? val.slice(0, 40) + "..." : val;
+      }
+    }
+    const firstKey = Object.keys(input)[0];
+    if (firstKey) {
+      const val = String(input[firstKey]);
+      return val.length > 40 ? val.slice(0, 40) + "..." : val;
+    }
+    return "(no input)";
+  }
+  /**
+   * Add a tool call to history after execution.
+   * Keeps only the last 10 entries.
+   */
+  addToolHistory(tool, input, success2) {
+    this.toolHistory.push({
+      tool,
+      primaryInput: this.extractPrimaryInput(input),
+      success: success2,
+      timestamp: Date.now()
+    });
+    if (this.toolHistory.length > _Agent.MAX_TOOL_HISTORY) {
+      this.toolHistory = this.toolHistory.slice(-_Agent.MAX_TOOL_HISTORY);
+    }
+  }
+  /**
+   * Format tool history for context injection.
+   * Simple, clean format without emojis.
+   */
+  formatToolHistory() {
+    if (this.toolHistory.length === 0) return "";
+    const lines = this.toolHistory.map((t) => {
+      const status = t.success ? "ok" : "failed";
+      return `- ${t.tool}: "${t.primaryInput}" - ${status}`;
+    });
+    return "Recent actions:\n" + lines.join("\n");
+  }
+  /**
+   * Add a user/model exchange to history.
+   * If we exceed max exchanges, compact older ones first.
+   * @deprecated Use storeTextExchange instead
+   */
+  /**
+   * Build the full system context including tool history and summary.
+   */
+  buildSystemContext() {
+    const basePrompt = this.config.systemPrompt ?? DEFAULT_SYSTEM_PROMPT;
+    const parts = [basePrompt];
+    const toolHistoryStr = this.formatToolHistory();
+    if (toolHistoryStr) {
+      parts.push(toolHistoryStr);
+    }
+    if (this.conversationSummary) {
+      parts.push(`Previous context:
+${this.conversationSummary}`);
+    }
+    return parts.join("\n\n");
+  }
+  /**
+   * Build messages array from exchanges for API call.
+   * Converts stored exchanges to MessageParam format.
+   */
+  buildMessagesFromExchanges() {
+    const messages = [];
+    for (const exchange of this.exchanges) {
+      messages.push({ role: "user", content: exchange.userMessage });
+      messages.push({ role: "assistant", content: exchange.assistantResponse });
+    }
+    return messages;
+  }
+  /**
+   * Build slim history for API call: last 2 text exchanges + current user message.
+   * NO tool calls, NO tool results - just text.
+   */
+  buildSlimHistory(currentUserMessage) {
+    const messages = [];
+    const recentExchanges = this.exchanges.slice(-2);
+    for (const exchange of recentExchanges) {
+      messages.push({ role: "user", content: exchange.userMessage });
+      messages.push({ role: "assistant", content: exchange.assistantResponse });
+    }
+    messages.push({ role: "user", content: currentUserMessage });
+    return messages;
+  }
+  /**
+   * Store a text-only exchange (no tool calls).
+   * Keeps only last 2 exchanges for context.
+   */
+  storeTextExchange(userMessage, assistantResponse) {
+    this.exchanges.push({
+      userMessage,
+      assistantResponse,
+      timestamp: Date.now()
+    });
+    if (this.exchanges.length > 2) {
+      this.exchanges = this.exchanges.slice(-2);
+    }
+  }
+  /**
+   * Extract final text response from assistant content blocks.
+   * Filters out tool_use blocks, returns only text.
+   */
+  extractTextResponse(content) {
+    const textBlocks = content.filter((block) => block.type === "text");
+    return textBlocks.map((block) => block.text).join("\n").trim();
+  }
   /**
    * Set working directory
    */
@@ -4260,52 +4200,6 @@ ${userMessage}`;
   getWorkingDirectory() {
     return this.workingDirectory;
   }
-  /**
-   * Truncate tool results in the last message of conversation history.
-   * 
-   * This is called AFTER Claude has seen the full tool results and responded.
-   * We then replace the full results with truncated versions to save context
-   * on future turns. This way:
-   * - Current turn: Claude sees full data, can display everything to user
-   * - Future turns: Only actionable data (IDs, prices) is in context
-   */
-  truncateLastToolResults() {
-    for (let i = this.conversationHistory.length - 1; i >= 0; i--) {
-      const message = this.conversationHistory[i];
-      if (message.role === "user" && Array.isArray(message.content)) {
-        const toolResults = message.content.filter(
-          (block) => block.type === "tool_result"
-        );
-        if (toolResults.length > 0) {
-          const truncatedContent = message.content.map((block) => {
-            if (block.type === "tool_result" && typeof block.content === "string") {
-              try {
-                const fullResult = JSON.parse(block.content);
-                const truncatedResult = truncateToolResult(fullResult, "unknown");
-                return {
-                  ...block,
-                  content: JSON.stringify(truncatedResult)
-                };
-              } catch {
-                if (block.content.length > MAX_TOOL_RESULT_CHARS) {
-                  return {
-                    ...block,
-                    content: block.content.substring(0, MAX_TOOL_RESULT_CHARS) + "... [truncated for context]"
-                  };
-                }
-              }
-            }
-            return block;
-          });
-          this.conversationHistory[i] = {
-            ...message,
-            content: truncatedContent
-          };
-          break;
-        }
-      }
-    }
-  }
   /**
    * Update cumulative token usage from API response
    * @param usage - Token counts from the API response

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@quantish/agent",
-  "version": "0.1.50",
+  "version": "0.1.52",
   "description": "AI-powered agent for trading on Polymarket and Kalshi",
   "type": "module",
   "bin": {