pulse-coder-engine 0.0.1-alpha.1 → 0.0.1-alpha.2

package/src/context/index.ts

@@ -1,134 +0,0 @@
-import { pruneMessages, type ModelMessage } from "ai";
-import { summarizeMessages } from "../ai";
-import {
-  COMPACT_TRIGGER,
-  COMPACT_TARGET,
-  KEEP_LAST_TURNS,
-} from "../config/index";
-import type { Context } from "../shared/types";
-
-type CompactResult = {
-  didCompact: boolean;
-  reason?: string;
-  newMessages?: ModelMessage[];
-};
-
-const ensureSummaryPrefix = (summary: string): string => {
-  const trimmed = summary.trim();
-  if (trimmed.length === 0) {
-    return '';
-  }
-  if (trimmed.startsWith('[COMPACTED_CONTEXT]')) {
-    return trimmed;
-  }
-  return `[COMPACTED_CONTEXT]\n${trimmed}`;
-};
-
-const safeStringify = (value: unknown): string => {
-  try {
-    return JSON.stringify(value);
-  } catch {
-    return String(value);
-  }
-};
-
-const estimateTokens = (messages: ModelMessage[]): number => {
-  let totalChars = 0;
-  for (const message of messages) {
-    totalChars += message.role.length;
-    if (typeof message.content === 'string') {
-      totalChars += message.content.length;
-    } else {
-      totalChars += safeStringify(message.content).length;
-    }
-  }
-  return Math.ceil(totalChars / 4);
-};
-
-const splitByTurns = (messages: ModelMessage[], keepLastTurns: number) => {
-  const userIndices: number[] = [];
-  messages.forEach((message, index) => {
-    if (message.role === 'user') {
-      userIndices.push(index);
-    }
-  });
-
-  if (userIndices.length <= keepLastTurns) {
-    return { oldMessages: [], recentMessages: messages };
-  }
-
-  const cutIndex = userIndices[userIndices.length - keepLastTurns];
-  return {
-    oldMessages: messages.slice(0, cutIndex),
-    recentMessages: messages.slice(cutIndex),
-  };
-};
-
-const takeLastTurns = (messages: ModelMessage[], keepLastTurns: number): ModelMessage[] => {
-  const userIndices: number[] = [];
-  messages.forEach((message, index) => {
-    if (message.role === 'user') {
-      userIndices.push(index);
-    }
-  });
-
-  if (userIndices.length === 0) {
-    return messages;
-  }
-
-  if (userIndices.length <= keepLastTurns) {
-    return messages;
-  }
-
-  const startIndex = userIndices[userIndices.length - keepLastTurns];
-  return messages.slice(startIndex);
-};
-
-export const maybeCompactContext = async (
-  context: Context,
-  options?: { force?: boolean }
-): Promise<CompactResult> => {
-  const { messages } = context;
-  if (messages.length === 0) {
-    return { didCompact: false };
-  }
-
-  const estimatedTokens = estimateTokens(messages);
-  if (!options?.force && estimatedTokens < COMPACT_TRIGGER) {
-    return { didCompact: false };
-  }
-
-  const { oldMessages, recentMessages } = splitByTurns(messages, KEEP_LAST_TURNS);
-  if (oldMessages.length === 0) {
-    return { didCompact: false };
-  }
-
-  try {
-    const summary = await summarizeMessages(oldMessages);
-    const summaryText = ensureSummaryPrefix(summary);
-    if (!summaryText) {
-      throw new Error('Empty summary result');
-    }
-
-    const nextMessages: ModelMessage[] = [
-      { role: 'assistant', content: summaryText },
-      ...recentMessages,
-    ];
-
-    if (estimateTokens(nextMessages) > COMPACT_TARGET) {
-      const newMessages = takeLastTurns(messages, KEEP_LAST_TURNS);
-      return { didCompact: true, reason: 'summary-too-large', newMessages };
-    }
-
-    return { didCompact: true, newMessages: nextMessages };
-  } catch (error) {
-    const pruned = pruneMessages({
-      messages,
-      reasoning: 'all',
-      toolCalls: 'all',
-      emptyMessages: 'remove',
-    });
-    const newMessages = takeLastTurns(pruned, KEEP_LAST_TURNS);
-    return { didCompact: true, reason: 'fallback', newMessages };
-  }
-};

package/src/core/loop.ts

@@ -1,147 +0,0 @@
-import { ToolSet, type StepResult, type ModelMessage } from "ai";
-import type { Context, ClarificationRequest, Tool } from "../shared/types";
-import { streamTextAI } from "../ai";
-import { maybeCompactContext } from "../context";
-import {
-  MAX_COMPACTION_ATTEMPTS,
-  MAX_ERROR_COUNT,
-  MAX_STEPS
-} from "../config/index.js";
-
-export interface LoopOptions {
-  onText?: (delta: string) => void;
-  onToolCall?: (toolCall: any) => void;
-  onToolResult?: (toolResult: any) => void;
-  onStepFinish?: (step: StepResult<any>) => void;
-  onClarificationRequest?: (request: ClarificationRequest) => Promise<string>;
-  onCompacted?: (newMessages: ModelMessage[]) => void;
-  onResponse?: (messages: StepResult<ToolSet>['response']['messages']) => void;
-  abortSignal?: AbortSignal;
-
-  tools?: Record<string, Tool>; // 允许传入工具覆盖默认工具
-}
-
-export async function loop(context: Context, options?: LoopOptions): Promise<string> {
-  let errorCount = 0;
-  let totalSteps = 0;
-  let compactionAttempts = 0;
-
-  while (true) {
-    try {
-      if (compactionAttempts < MAX_COMPACTION_ATTEMPTS) {
-        const { didCompact, newMessages } = await maybeCompactContext(context);
-        if (didCompact) {
-          compactionAttempts++;
-
-          if (newMessages) {
-            options?.onCompacted?.(newMessages)
-          }
-          continue;
-        }
-      }
-
-      const tools = options?.tools || {}; // 允许传入工具覆盖默认工具
-
-      // Prepare tool execution context
-      const toolExecutionContext = {
-        onClarificationRequest: options?.onClarificationRequest,
-        abortSignal: options?.abortSignal
-      };
-
-      const result = streamTextAI(context.messages, tools, {
-        abortSignal: options?.abortSignal,
-        toolExecutionContext,
-        onStepFinish: (step) => {
-          options?.onStepFinish?.(step);
-        },
-        onChunk: ({ chunk }) => {
-          if (chunk.type === 'text-delta') {
-            options?.onText?.(chunk.text);
-          }
-          if (chunk.type === 'tool-call') {
-            options?.onToolCall?.(chunk);
-          }
-          if (chunk.type === 'tool-result') {
-            options?.onToolResult?.(chunk);
-          }
-        },
-      });
-
-      const [text, steps, finishReason] = await Promise.all([
-        result.text,
-        result.steps,
-        result.finishReason,
-      ]);
-
-      totalSteps += steps.length;
-
-      for (const step of steps) {
-        if (step.response?.messages?.length) {
-          const messages = [...step.response.messages];
-          options?.onResponse?.(messages);
-        }
-      }
-
-      if (finishReason === 'stop') {
-        return text || 'Task completed.';
-      }
-
-      if (finishReason === 'length') {
-        if (compactionAttempts < MAX_COMPACTION_ATTEMPTS) {
-          const { didCompact, newMessages } = await maybeCompactContext(context, { force: true });
-          if (didCompact) {
-            compactionAttempts++;
-            if (newMessages) {
-              options?.onCompacted?.(newMessages)
-            }
-            continue;
-          }
-        }
-        return text || 'Context limit reached.';
-      }
-
-      if (finishReason === 'content-filter') {
-        return text || 'Content filtered.';
-      }
-
-      if (finishReason === 'error') {
-        return text || 'Task failed.';
-      }
-
-      if (finishReason === 'tool-calls') {
-        if (totalSteps >= MAX_STEPS) {
-          return text || 'Max steps reached, task may be incomplete.';
-        }
-        continue;
-      }
-
-      return text || 'Task completed.';
-    } catch (error: any) {
-      if (options?.abortSignal?.aborted || error?.name === 'AbortError') {
-        return 'Request aborted.';
-      }
-
-      errorCount++;
-      if (errorCount >= MAX_ERROR_COUNT) {
-        return `Failed after ${errorCount} errors: ${error?.message ?? String(error)}`;
-      }
-
-      if (isRetryableError(error)) {
-        const delay = Math.min(2000 * Math.pow(2, errorCount - 1), 30000);
-        await sleep(delay);
-        continue;
-      }
-
-      return `Error: ${error?.message ?? String(error)}`;
-    }
-  }
-}
-
-function isRetryableError(error: any): boolean {
-  const status = error?.status ?? error?.statusCode;
-  return status === 429 || status === 500 || status === 502 || status === 503;
-}
-
-function sleep(ms: number): Promise<void> {
-  return new Promise((resolve) => setTimeout(resolve, ms));
-}

package/src/index.ts

@@ -1,17 +0,0 @@
-export { Engine } from './Engine.js';
-export { Engine as PulseAgent } from './Engine.js'; // 添加 PulseAgent 别名
-
-// 插件系统导出
-export type { EnginePlugin, EnginePluginContext } from './plugin/EnginePlugin.js';
-export type { UserConfigPlugin, UserConfigPluginLoadOptions } from './plugin/UserConfigPlugin.js';
-export { PluginManager } from './plugin/PluginManager.js';
-
-// 内置插件导出
-export { builtInPlugins, builtInMCPPlugin, builtInSkillsPlugin, BuiltInSkillRegistry } from './built-in/index.js';
-
-// 原有导出保持不变
-export * from './shared/types.js';
-export { loop } from './core/loop.js';
-export { streamTextAI } from './ai/index.js';
-export { maybeCompactContext } from './context/index.js';
-export * from './tools/index.js';

package/src/plugin/EnginePlugin.ts

@@ -1,60 +0,0 @@
-import type { Tool } from 'ai';
-import type { EventEmitter } from 'events';
-
-// 修复：确保类型正确导出
-export interface EnginePlugin {
-  name: string;
-  version: string;
-  dependencies?: string[];
-
-  // 生命周期钩子 - 添加类型标注
-  beforeInitialize?(context: EnginePluginContext): Promise<void>;
-  initialize(context: EnginePluginContext): Promise<void>;
-  afterInitialize?(context: EnginePluginContext): Promise<void>;
-
-  // 清理钩子
-  destroy?(context: EnginePluginContext): Promise<void>;
-}
-
-export interface EnginePluginContext {
-  registerTool(name: string, tool: Tool): void;
-  registerTools(tools: Record<string, Tool>): void;
-  getTool(name: string): Tool | undefined;
-
-  registerProtocol(name: string, handler: ProtocolHandler): void;
-  getProtocol(name: string): ProtocolHandler | undefined;
-
-  registerService<T>(name: string, service: T): void;
-  getService<T>(name: string): T | undefined;
-
-  getConfig<T>(key: string): T | undefined;
-  setConfig<T>(key: string, value: T): void;
-
-  events: EventEmitter;
-
-  logger: {
-    debug(message: string, meta?: any): void;
-    info(message: string, meta?: any): void;
-    warn(message: string, meta?: any): void;
-    error(message: string, error?: Error, meta?: any): void;
-  };
-}
-
-export interface ProtocolHandler {
-  name: string;
-  handle(message: any): Promise<any>;
-}
-
-export interface EnginePluginLoadOptions {
-  plugins?: EnginePlugin[];
-  dirs?: string[];
-  scan?: boolean;
-}
-
-export const DEFAULT_ENGINE_PLUGIN_DIRS = [
-  '.pulse-coder/engine-plugins',
-  '.coder/engine-plugins',
-  '~/.pulse-coder/engine-plugins',
-  '~/.coder/engine-plugins',
-  './plugins/engine'
-];