kalshi-trading-bot-cli 2.1.0

package/src/utils/telemetry.ts

@@ -0,0 +1,103 @@
+import { loadConfig, saveConfig } from './config.js';
+
+const CLIENT_KEY = 'client-TbH2tfgwg2CYu87Y932Wj2CdNTfyy7303HFszn0YZny';
+const LOG_EVENT_URL = 'https://events.statsigapi.net/v1/log_event';
+
+let userId: string | null = null;
+let disabled = false;
+let eventQueue: StatsigEvent[] = [];
+
+interface StatsigEvent {
+  eventName: string;
+  value?: string;
+  metadata?: Record<string, string | number | boolean>;
+  time: number;
+  user: { userID: string };
+}
+
+function isEnabled(): boolean {
+  const val = process.env.TELEMETRY_ENABLED;
+  if (val === 'false' || val === '0') return false;
+  return true;
+}
+
+function getOrCreateAnonymousId(): string {
+  const config = loadConfig();
+  if (config.anonymousId) return config.anonymousId;
+  const id = crypto.randomUUID();
+  saveConfig({ ...config, anonymousId: id });
+  return id;
+}
+
+export async function initTelemetry(): Promise<void> {
+  if (disabled || userId) return;
+  if (!isEnabled()) {
+    disabled = true;
+    return;
+  }
+
+  try {
+    userId = getOrCreateAnonymousId();
+
+    // Intercept process.exit to flush telemetry before exiting.
+    // dispatch.ts calls process.exit() in many code paths, which would
+    // kill the process before events are sent.
+    const originalExit = process.exit;
+    process.exit = (async (code?: number) => {
+      await shutdownTelemetry();
+      originalExit(code as any);
+    }) as never;
+  } catch {
+    disabled = true;
+  }
+}
+
+export function trackEvent(
+  name: string,
+  metadata?: Record<string, string | number | boolean>,
+): void {
+  if (!userId || disabled) return;
+  try {
+    eventQueue.push({
+      eventName: name,
+      metadata,
+      time: Date.now(),
+      user: { userID: userId },
+    });
+  } catch {}
+}
+
+async function flushEvents(): Promise<void> {
+  if (eventQueue.length === 0 || !userId) return;
+  const events = eventQueue;
+  eventQueue = [];
+  try {
+    await fetch(LOG_EVENT_URL, {
+      method: 'POST',
+      headers: {
+        'Content-Type': 'application/json',
+        'statsig-api-key': CLIENT_KEY,
+        'statsig-sdk-type': 'js-mono',
+        'statsig-sdk-version': '1.0.0',
+      },
+      body: JSON.stringify({
+        events,
+        statsigMetadata: {
+          sdkType: 'js-mono',
+          sdkVersion: '1.0.0',
+        },
+      }),
+    });
+  } catch {}
+}
+
+export async function shutdownTelemetry(): Promise<void> {
+  if (disabled) return;
+  try {
+    await Promise.race([
+      flushEvents(),
+      new Promise((resolve) => setTimeout(resolve, 3000)),
+    ]);
+  } catch {}
+  userId = null;
+}

package/src/utils/text-navigation.ts

@@ -0,0 +1,81 @@
+/**
+ * Find the start position of the previous word from cursor position.
+ * Used for Option+Left (Mac) / Ctrl+Left (Windows) navigation.
+ */
+export function findPrevWordStart(text: string, pos: number): number {
+  if (pos <= 0) return 0;
+  let i = pos - 1;
+  // Skip non-word chars
+  while (i > 0 && !/\w/.test(text[i])) i--;
+  // Move to word start
+  while (i > 0 && /\w/.test(text[i - 1])) i--;
+  return i;
+}
+
+/**
+ * Find the end position of the next word from cursor position.
+ * Used for Option+Right (Mac) / Ctrl+Right (Windows) navigation.
+ */
+export function findNextWordEnd(text: string, pos: number): number {
+  const len = text.length;
+  if (pos >= len) return len;
+  let i = pos;
+  // Skip non-word chars
+  while (i < len && !/\w/.test(text[i])) i++;
+  // Move to word end
+  while (i < len && /\w/.test(text[i])) i++;
+  return i;
+}
+
+// ============================================================================
+// Multi-line cursor navigation utilities
+// ============================================================================
+
+/**
+ * Get the line number (0-indexed) and column from a cursor position.
+ */
+export function getLineAndColumn(text: string, pos: number): { line: number; column: number } {
+  const beforeCursor = text.slice(0, pos);
+  const lines = beforeCursor.split('\n');
+  return {
+    line: lines.length - 1,
+    column: lines[lines.length - 1].length,
+  };
+}
+
+/**
+ * Get cursor position from line number and column.
+ * Clamps column to the actual line length if it exceeds.
+ */
+export function getCursorPosition(text: string, line: number, column: number): number {
+  const lines = text.split('\n');
+  let pos = 0;
+  for (let i = 0; i < line && i < lines.length; i++) {
+    pos += lines[i].length + 1; // +1 for newline
+  }
+  const targetLine = lines[line] || '';
+  return pos + Math.min(column, targetLine.length);
+}
+
+/**
+ * Get the start position of the line containing the cursor.
+ */
+export function getLineStart(text: string, pos: number): number {
+  const lastNewline = text.lastIndexOf('\n', pos - 1);
+  return lastNewline + 1; // -1 + 1 = 0 if no newline found
+}
+
+/**
+ * Get the end position of the line containing the cursor (before the newline).
+ */
+export function getLineEnd(text: string, pos: number): number {
+  const nextNewline = text.indexOf('\n', pos);
+  return nextNewline === -1 ? text.length : nextNewline;
+}
+
+/**
+ * Get the total number of lines in the text.
+ */
+export function getLineCount(text: string): number {
+  return text.split('\n').length;
+}

package/src/utils/thinking-verbs.ts

@@ -0,0 +1,18 @@
+export const THINKING_VERBS = [
+  'Analyzing', 'Assessing', 'Brainstorming', 'Brewing',
+  'Calculating', 'Calibrating', 'Catalyzing', 'Cerebrating',
+  'Churning', 'Cogitating', 'Constructing', 'Crafting',
+  'Crunching', 'Deliberating', 'Distilling', 'Drafting',
+  'Engineering', 'Evaluating', 'Experimenting', 'Finessing',
+  'Formulating', 'Forging', 'Hatching', 'Hypothesizing',
+  'Ideating', 'Inventing', 'Marinating', 'Modeling',
+  'Mulling', 'Musing', 'Observing', 'Percolating',
+  'Pondering', 'Processing', 'Puzzling', 'Reviewing',
+  'Riffing', 'Ruminating', 'Sculpting', 'Simmering',
+  'Sketching', 'Synthesizing', 'Tinkering', 'Triangulating',
+  'Verifying', 'Whittling', 'Wrangling',
+] as const;
+
+export function getRandomThinkingVerb(): string {
+  return THINKING_VERBS[Math.floor(Math.random() * THINKING_VERBS.length)];
+}

package/src/utils/tokens.ts

@@ -0,0 +1,36 @@
+/**
+ * Token estimation utilities for context management.
+ * Used to prevent exceeding LLM context window limits.
+ */
+
+/**
+ * Rough token estimation based on character count.
+ * JSON is denser than prose, so we use ~3.5 chars per token.
+ * This is conservative - better to underestimate available space.
+ */
+export function estimateTokens(text: string): number {
+  return Math.ceil(text.length / 3.5);
+}
+
+/**
+ * Maximum token budget for context data in final answer generation.
+ * Conservative limit that leaves room for system prompt, query, and response.
+ */
+export const TOKEN_BUDGET = 150_000;
+
+// ============================================================================
+// Anthropic-style Context Management Constants
+// ============================================================================
+
+/**
+ * Token threshold at which context clearing is triggered.
+ * Matches Anthropic's default of 100k tokens.
+ * When estimated context exceeds this, oldest tool results are cleared.
+ */
+export const CONTEXT_THRESHOLD = 100_000;
+
+/**
+ * Number of most recent tool results to keep when clearing.
+ * Anthropic's default is 3, but we use 5 for slightly more context.
+ */
+export const KEEP_TOOL_USES = 5;

package/src/utils/tool-description.ts

@@ -0,0 +1,61 @@
+/**
+ * Generate a deterministic human-readable description of a tool call.
+ * Used for context compaction during the agent loop.
+ * 
+ * Examples:
+ * - "AAPL income statements (annual) - 5 periods"
+ * - '"bitcoin price" tavily search'
+ */
+export function getToolDescription(toolName: string, args: Record<string, unknown>): string {
+  const parts: string[] = [];
+  const usedKeys = new Set<string>();
+
+  // Add ticker if present (most common identifier)
+  if (args.ticker) {
+    parts.push(String(args.ticker).toUpperCase());
+    usedKeys.add('ticker');
+  }
+
+  // Add search query if present
+  if (args.query) {
+    parts.push(`"${args.query}"`);
+    usedKeys.add('query');
+  }
+
+  // Format tool name: get_income_statements -> income statements
+  const formattedToolName = toolName
+    .replace(/^get_/, '')
+    .replace(/^search_/, '')
+    .replace(/_/g, ' ');
+  parts.push(formattedToolName);
+
+  // Add period qualifier if present
+  if (args.period) {
+    parts.push(`(${args.period})`);
+    usedKeys.add('period');
+  }
+
+  // Add limit if present
+  if (args.limit && typeof args.limit === 'number') {
+    parts.push(`- ${args.limit} periods`);
+    usedKeys.add('limit');
+  }
+
+  // Add date range if present
+  if (args.start_date && args.end_date) {
+    parts.push(`from ${args.start_date} to ${args.end_date}`);
+    usedKeys.add('start_date');
+    usedKeys.add('end_date');
+  }
+
+  // Append any remaining args not explicitly handled
+  const remainingArgs = Object.entries(args)
+    .filter(([key]) => !usedKeys.has(key))
+    .map(([key, value]) => `${key}=${value}`);
+
+  if (remainingArgs.length > 0) {
+    parts.push(`[${remainingArgs.join(', ')}]`);
+  }
+
+  return parts.join(' ');
+}