@c4ccz/zero 1.0.0 → 1.0.1

package/packages/core/src/types.ts

@@ -0,0 +1,356 @@
+/**
+ * ZERO Core - Type Definitions
+ * Unified types used across all ZERO modules
+ */
+
+import { z } from "zod";
+
+// ============================================================================
+// Message Types
+// ============================================================================
+
+export type MessageRole = "user" | "assistant" | "system" | "tool";
+
+export interface TextContent {
+  type: "text";
+  text: string;
+}
+
+export interface ImageContent {
+  type: "image";
+  data: string;
+  mimeType: string;
+}
+
+export interface ToolCallContent {
+  type: "tool_call";
+  id: string;
+  name: string;
+  arguments: Record<string, unknown>;
+}
+
+export interface ToolResultContent {
+  type: "tool_result";
+  id: string;
+  name: string;
+  result: unknown;
+  error?: string;
+}
+
+export type MessageContent = TextContent | ImageContent | ToolCallContent | ToolResultContent;
+
+export interface Message {
+  id: string;
+  role: MessageRole;
+  content: MessageContent[];
+  timestamp: number;
+  metadata?: Record<string, unknown>;
+}
+
+// ============================================================================
+// Tool Types
+// ============================================================================
+
+export const ToolParameterSchema = z.object({
+  type: z.enum(["string", "number", "boolean", "array", "object"]),
+  description: z.string(),
+  required: z.boolean().optional(),
+  default: z.unknown().optional(),
+  enum: z.array(z.string()).optional(),
+  items: z.lazy(() => ToolParameterSchema).optional(),
+  properties: z.record(z.lazy(() => ToolParameterSchema)).optional(),
+});
+
+export type ToolParameter = z.infer<typeof ToolParameterSchema>;
+
+export interface ToolDefinition {
+  name: string;
+  description: string;
+  parameters: Record<string, ToolParameter>;
+  category: ToolCategory;
+  requiresApproval: boolean;
+  handler: (args: Record<string, unknown>, context: ToolContext) => Promise<ToolResult>;
+}
+
+export type ToolCategory =
+  | "file"
+  | "shell"
+  | "search"
+  | "web"
+  | "memory"
+  | "git"
+  | "system"
+  | "mcp";
+
+export interface ToolContext {
+  sessionId: string;
+  workingDirectory: string;
+  permissions: PermissionSet;
+  abortSignal?: AbortSignal;
+}
+
+export interface ToolResult {
+  success: boolean;
+  output: string;
+  data?: unknown;
+  error?: string;
+}
+
+// ============================================================================
+// Provider Types
+// ============================================================================
+
+export interface LLMProvider {
+  id: string;
+  name: string;
+  type: ProviderType;
+  models: ModelConfig[];
+  createClient(config: ProviderConfig): LLMClient;
+}
+
+export type ProviderType =
+  | "openai"
+  | "anthropic"
+  | "google"
+  | "azure"
+  | "ollama"
+  | "custom"
+  | "bedrock"
+  | "vertex";
+
+export interface ModelConfig {
+  id: string;
+  name: string;
+  contextWindow: number;
+  maxOutputTokens: number;
+  supportsImages: boolean;
+  supportsToolUse: boolean;
+  supportsStreaming: boolean;
+  costPerInputToken?: number;
+  costPerOutputToken?: number;
+}
+
+export interface ProviderConfig {
+  apiKey?: string;
+  baseUrl?: string;
+  model: string;
+  temperature?: number;
+  maxTokens?: number;
+  topP?: number;
+  headers?: Record<string, string>;
+}
+
+export interface LLMClient {
+  chat(messages: Message[], options?: ChatOptions): Promise<ChatResponse>;
+  stream(messages: Message[], options?: ChatOptions): AsyncIterable<StreamChunk>;
+}
+
+export interface ChatOptions {
+  tools?: ToolDefinition[];
+  systemPrompt?: string;
+  maxTokens?: number;
+  temperature?: number;
+  stopSequences?: string[];
+}
+
+export interface ChatResponse {
+  id: string;
+  content: MessageContent[];
+  usage: TokenUsage;
+  stopReason: StopReason;
+  model: string;
+}
+
+export interface TokenUsage {
+  inputTokens: number;
+  outputTokens: number;
+  cacheReadTokens?: number;
+  cacheWriteTokens?: number;
+  totalCost?: number;
+}
+
+export type StopReason = "end_turn" | "max_tokens" | "tool_use" | "stop_sequence";
+
+export type StreamChunk =
+  | { type: "text_delta"; text: string }
+  | { type: "tool_call_start"; id: string; name: string }
+  | { type: "tool_call_delta"; id: string; arguments: string }
+  | { type: "tool_call_end"; id: string }
+  | { type: "usage"; usage: TokenUsage }
+  | { type: "done"; stopReason: StopReason };
+
+// ============================================================================
+// Permission Types
+// ============================================================================
+
+export interface PermissionSet {
+  fileRead: PermissionLevel;
+  fileWrite: PermissionLevel;
+  shellExecution: PermissionLevel;
+  networkAccess: PermissionLevel;
+  mcpAccess: PermissionLevel;
+}
+
+export type PermissionLevel = "allow" | "ask" | "deny";
+
+export interface PermissionRequest {
+  type: "file_read" | "file_write" | "shell" | "network" | "mcp";
+  resource: string;
+  details?: string;
+}
+
+export type PermissionDecision = "allow" | "deny" | "allow_always";
+
+export type PermissionHandler = (request: PermissionRequest) => Promise<PermissionDecision>;
+
+// ============================================================================
+// Session Types
+// ============================================================================
+
+export interface Session {
+  id: string;
+  name: string;
+  createdAt: number;
+  updatedAt: number;
+  messages: Message[];
+  config: SessionConfig;
+  state: SessionState;
+}
+
+export interface SessionConfig {
+  provider: string;
+  model: string;
+  systemPrompt: string;
+  workingDirectory: string;
+  permissions: PermissionSet;
+  tools: string[];
+  mcpServers: MCPServerConfig[];
+}
+
+export interface SessionState {
+  status: "idle" | "running" | "waiting_approval" | "error" | "completed";
+  currentAgent?: string;
+  tokenUsage: TokenUsage;
+  error?: string;
+}
+
+// ============================================================================
+// MCP Types
+// ============================================================================
+
+export interface MCPServerConfig {
+  id: string;
+  name: string;
+  command: string;
+  args?: string[];
+  env?: Record<string, string>;
+  enabled: boolean;
+}
+
+export interface MCPTool {
+  name: string;
+  description: string;
+  inputSchema: Record<string, unknown>;
+  serverId: string;
+}
+
+export interface MCPResource {
+  uri: string;
+  name: string;
+  description?: string;
+  mimeType?: string;
+  serverId: string;
+}
+
+// ============================================================================
+// Agent Types
+// ============================================================================
+
+export interface AgentConfig {
+  id: string;
+  name: string;
+  description: string;
+  systemPrompt: string;
+  tools: string[];
+  maxTurns: number;
+  model: string;
+  subAgents?: string[];
+  hooks?: AgentHooks;
+}
+
+export interface AgentHooks {
+  onToolCall?: (tool: string, args: Record<string, unknown>) => Promise<boolean>;
+  onToolResult?: (tool: string, result: ToolResult) => Promise<void>;
+  onMessage?: (message: Message) => Promise<void>;
+  onError?: (error: Error) => Promise<void>;
+  onComplete?: (result: AgentResult) => Promise<void>;
+}
+
+export interface AgentResult {
+  success: boolean;
+  messages: Message[];
+  tokenUsage: TokenUsage;
+  duration: number;
+  error?: string;
+}
+
+// ============================================================================
+// Memory Types
+// ============================================================================
+
+export interface MemoryEntry {
+  id: string;
+  type: "fact" | "preference" | "context" | "file_context";
+  content: string;
+  relevance: number;
+  createdAt: number;
+  expiresAt?: number;
+  tags: string[];
+  metadata?: Record<string, unknown>;
+}
+
+export interface MemoryStore {
+  save(entry: MemoryEntry): Promise<void>;
+  search(query: string, limit?: number): Promise<MemoryEntry[]>;
+  delete(id: string): Promise<void>;
+  getAll(): Promise<MemoryEntry[]>;
+}
+
+// ============================================================================
+// Config Schema
+// ============================================================================
+
+export const ZeroConfigSchema = z.object({
+  version: z.literal(1),
+  defaultProvider: z.string(),
+  defaultModel: z.string(),
+  providers: z.record(z.object({
+    type: z.string(),
+    apiKey: z.string().optional(),
+    baseUrl: z.string().optional(),
+    models: z.array(z.string()).optional(),
+  })),
+  permissions: z.object({
+    fileRead: z.enum(["allow", "ask", "deny"]).default("allow"),
+    fileWrite: z.enum(["allow", "ask", "deny"]).default("ask"),
+    shellExecution: z.enum(["allow", "ask", "deny"]).default("ask"),
+    networkAccess: z.enum(["allow", "ask", "deny"]).default("ask"),
+    mcpAccess: z.enum(["allow", "ask", "deny"]).default("allow"),
+  }).default({}),
+  agents: z.record(z.object({
+    systemPrompt: z.string(),
+    tools: z.array(z.string()),
+    maxTurns: z.number().default(50),
+    model: z.string().optional(),
+  })).optional(),
+  mcpServers: z.array(z.object({
+    id: z.string(),
+    name: z.string(),
+    command: z.string(),
+    args: z.array(z.string()).optional(),
+    env: z.record(z.string()).optional(),
+    enabled: z.boolean().default(true),
+  })).optional(),
+});
+
+export type ZeroConfig = z.infer<typeof ZeroConfigSchema>;

package/packages/sdk/src/index.ts

@@ -0,0 +1,247 @@
+/**
+ * ZERO SDK - Programmatic API
+ *
+ * Use ZERO programmatically in your own applications.
+ *
+ * @example
+ * ```typescript
+ * import { Zero } from "@zero/sdk";
+ *
+ * const zero = new Zero({
+ *   provider: "openai",
+ *   model: "gpt-4o",
+ *   apiKey: process.env.OPENAI_API_KEY,
+ * });
+ *
+ * const result = await zero.chat("Create a hello world in Python");
+ * console.log(result.text);
+ * ```
+ */
+
+import {
+  AgentEngine,
+  AgentRegistry,
+  DEFAULT_AGENTS,
+  ProviderRegistry,
+  ToolRegistry,
+  SessionManager,
+  FileMemoryStore,
+  ContextManager,
+  MCPManager,
+  createAutoPermissionHandler,
+  getAllBuiltinTools,
+} from "@zero/core";
+import type {
+  AgentResult,
+  Message,
+  ProviderConfig,
+  ToolDefinition,
+  AgentConfig,
+  MCPServerConfig,
+  TokenUsage,
+} from "@zero/core";
+
+export interface ZeroOptions {
+  provider?: string;
+  model?: string;
+  apiKey?: string;
+  baseUrl?: string;
+  workingDirectory?: string;
+  systemPrompt?: string;
+  autoApprove?: boolean;
+  mcpServers?: MCPServerConfig[];
+  customTools?: ToolDefinition[];
+  customAgents?: AgentConfig[];
+}
+
+export interface ZeroChatResult {
+  text: string;
+  messages: Message[];
+  usage: TokenUsage;
+  duration: number;
+  success: boolean;
+  error?: string;
+}
+
+export class Zero {
+  private providerRegistry: ProviderRegistry;
+  private toolRegistry: ToolRegistry;
+  private agentRegistry: AgentRegistry;
+  private sessionManager: SessionManager;
+  private memory: FileMemoryStore;
+  private contextManager: ContextManager;
+  private mcpManager: MCPManager;
+  private options: ZeroOptions;
+  private initialized = false;
+
+  constructor(options: ZeroOptions = {}) {
+    this.options = {
+      provider: options.provider || "openai",
+      model: options.model || "gpt-4o",
+      workingDirectory: options.workingDirectory || process.cwd(),
+      autoApprove: options.autoApprove ?? false,
+      ...options,
+    };
+
+    this.providerRegistry = new ProviderRegistry();
+    this.toolRegistry = new ToolRegistry();
+    this.agentRegistry = null as any;
+    this.sessionManager = null as any;
+    this.memory = null as any;
+    this.contextManager = null as any;
+    this.mcpManager = new MCPManager();
+  }
+
+  /**
+   * Initialize ZERO (must be called before use)
+   */
+  async init(): Promise<void> {
+    if (this.initialized) return;
+
+    // Load providers
+    this.providerRegistry.loadBuiltins();
+
+    // Load tools
+    for (const tool of getAllBuiltinTools()) {
+      this.toolRegistry.register(tool);
+    }
+
+    // Add custom tools
+    if (this.options.customTools) {
+      for (const tool of this.options.customTools) {
+        this.toolRegistry.register(tool);
+      }
+    }
+
+    // Initialize session manager
+    const tmpDir = `/tmp/.zero-${Date.now()}`;
+    this.sessionManager = new SessionManager({
+      storagePath: `${tmpDir}/sessions`,
+      defaultConfig: {
+        provider: this.options.provider,
+        model: this.options.model,
+        workingDirectory: this.options.workingDirectory,
+      },
+    });
+
+    // Initialize memory
+    this.memory = new FileMemoryStore(`${tmpDir}/memory`);
+    await this.memory.init();
+    this.contextManager = new ContextManager(this.memory);
+
+    // Initialize agent registry
+    const session = await this.sessionManager.create("SDK Session");
+    this.agentRegistry = new AgentRegistry({
+      clientFactory: (model) => {
+        const modelId = model === "default" ? this.options.model! : model;
+        return this.providerRegistry.createClient(this.options.provider!, {
+          model: modelId,
+          apiKey: this.options.apiKey,
+          baseUrl: this.options.baseUrl,
+        });
+      },
+      tools: this.toolRegistry.getMap(),
+      permissionHandler: this.options.autoApprove
+        ? async () => "allow"
+        : createAutoPermissionHandler(),
+      workingDirectory: this.options.workingDirectory!,
+      sessionId: session.id,
+    });
+
+    // Register agents
+    const agents = [...DEFAULT_AGENTS, ...(this.options.customAgents || [])];
+    for (const agent of agents) {
+      this.agentRegistry.register(agent);
+    }
+
+    // Connect MCP servers
+    if (this.options.mcpServers) {
+      for (const server of this.options.mcpServers) {
+        await this.mcpManager.connect(server);
+      }
+    }
+
+    this.initialized = true;
+  }
+
+  /**
+   * Send a chat message and get a response
+   */
+  async chat(prompt: string, options?: { agent?: string }): Promise<ZeroChatResult> {
+    if (!this.initialized) await this.init();
+
+    const agentId = options?.agent || "coder";
+    const engine = this.agentRegistry.createEngine(agentId);
+
+    // Add memory context
+    const context = await this.contextManager.buildContext(prompt);
+    const fullPrompt = context ? `${prompt}\n\n${context}` : prompt;
+
+    const result = await engine.run(fullPrompt);
+
+    // Extract text from response
+    const text = result.messages
+      .filter((m) => m.role === "assistant")
+      .flatMap((m) => m.content)
+      .filter((c) => c.type === "text")
+      .map((c) => (c as any).text)
+      .join("\n");
+
+    return {
+      text,
+      messages: result.messages,
+      usage: result.tokenUsage,
+      duration: result.duration,
+      success: result.success,
+      error: result.error,
+    };
+  }
+
+  /**
+   * Stream a chat response
+   */
+  async *stream(prompt: string): AsyncGenerator<string> {
+    if (!this.initialized) await this.init();
+
+    const client = this.providerRegistry.createClient(this.options.provider!, {
+      model: this.options.model!,
+      apiKey: this.options.apiKey,
+      baseUrl: this.options.baseUrl,
+    });
+
+    const messages: Message[] = [
+      {
+        id: `msg_${Date.now()}`,
+        role: "user",
+        content: [{ type: "text", text: prompt }],
+        timestamp: Date.now(),
+      },
+    ];
+
+    for await (const chunk of client.stream(messages, {
+      systemPrompt: DEFAULT_AGENTS[0].systemPrompt,
+    })) {
+      if (chunk.type === "text_delta") {
+        yield chunk.text;
+      }
+    }
+  }
+
+  /**
+   * Clean up resources
+   */
+  async close(): Promise<void> {
+    await this.mcpManager.disconnectAll();
+  }
+}
+
+// Re-export types
+export type {
+  AgentResult,
+  Message,
+  ProviderConfig,
+  ToolDefinition,
+  AgentConfig,
+  MCPServerConfig,
+  TokenUsage,
+};