@johpaz/hive-sdk 0.0.14 → 0.0.16

package/packages/core/src/gateway/server.ts

@@ -0,0 +1,139 @@
+/**
+ * Hive Gateway — simplified HTTP/WebSocket server for the SDK harness.
+ *
+ * Provides:
+ * - POST /chat — chat with an agent
+ * - GET /status — health check
+ * - WebSocket /ws — real-time streaming
+ */
+
+import { logger } from "../utils/logger";
+import { runAgent } from "../agent/AgentRunner";
+import type { MCPClientManager } from "../mcp/index";
+
+const log = logger.child("gateway");
+
+export interface GatewayConfig {
+  host?: string;
+  port?: number;
+  agentId?: string;
+  mcpManager?: MCPClientManager | null;
+}
+
+export async function startGateway(config: GatewayConfig = {}) {
+  const host = config.host ?? "127.0.0.1";
+  const port = config.port ?? 18790;
+  const agentId = config.agentId ?? "main";
+
+  log.info(`Starting gateway on ${host}:${port}`);
+
+  const server = Bun.serve({
+    hostname: host,
+    port,
+    fetch(req, server) {
+      const url = new URL(req.url);
+
+      // WebSocket upgrade
+      if (url.pathname === "/ws") {
+        const success = server.upgrade(req);
+        if (success) return undefined as any;
+      }
+
+      // Health check
+      if (url.pathname === "/status" && req.method === "GET") {
+        return Response.json({
+          status: "ok",
+          gateway: true,
+          agentId,
+          uptime: process.uptime(),
+        });
+      }
+
+      // Chat endpoint
+      if (url.pathname === "/chat" && req.method === "POST") {
+        return handleChat(req, agentId, config.mcpManager);
+      }
+
+      return new Response("Not Found", { status: 404 });
+    },
+    websocket: {
+      async message(ws, message) {
+        try {
+          const data = JSON.parse(String(message));
+          const threadId = data.threadId ?? crypto.randomUUID();
+
+          const stream = runAgent({
+            agentId,
+            userMessage: data.message ?? "",
+            threadId,
+            channel: "webchat",
+            mcpManager: config.mcpManager,
+          });
+
+          for await (const chunk of stream) {
+            ws.send(JSON.stringify(chunk));
+          }
+          ws.send(JSON.stringify({ done: true }));
+        } catch (err) {
+          log.error("WebSocket error:", err);
+          ws.send(JSON.stringify({ error: (err as Error).message }));
+        }
+      },
+      open(ws) {
+        log.info("WebSocket client connected");
+      },
+      close(ws, code, reason) {
+        log.info("WebSocket client disconnected");
+      },
+    },
+  });
+
+  log.info(`Gateway ready at http://${host}:${port}`);
+  return server;
+}
+
+async function handleChat(
+  req: Request,
+  agentId: string,
+  mcpManager?: MCPClientManager | null
+): Promise<Response> {
+  try {
+    const body = await req.json();
+    const message = body.message ?? "";
+    const threadId = body.threadId ?? crypto.randomUUID();
+
+    const stream = runAgent({
+      agentId,
+      userMessage: message,
+      threadId,
+      channel: "webchat",
+      mcpManager,
+    });
+
+    const chunks: any[] = [];
+    for await (const chunk of stream) {
+      chunks.push(chunk);
+    }
+
+    // Extract text from the last agent message
+    let responseText = "";
+    for (const chunk of chunks) {
+      if (chunk.agent?.messages) {
+        for (const msg of chunk.agent.messages) {
+          if (msg.content && typeof msg.content === "string") {
+            responseText = msg.content;
+          }
+        }
+      }
+    }
+
+    return Response.json({
+      response: responseText,
+      threadId,
+      chunks,
+    });
+  } catch (err) {
+    log.error("Chat error:", err);
+    return Response.json({ error: (err as Error).message }, { status: 500 });
+  }
+}

package/packages/core/src/heartbeat/index.ts

@@ -0,0 +1,157 @@
+import type { Config } from "../config/loader.ts";
+import { logger } from "../utils/logger.ts";
+
+export interface HealthStatus {
+  status: "healthy" | "degraded" | "unhealthy";
+  checks: Record<string, {
+    status: "ok" | "warning" | "error";
+    message?: string;
+    latency?: number;
+  }>;
+  uptime: number;
+  lastCheck: Date;
+}
+
+export interface HeartbeatOptions {
+  intervalMs?: number;
+  onHealthChange?: (status: HealthStatus) => void;
+}
+
+type HealthCheck = () => Promise<{
+  status: "ok" | "warning" | "error";
+  message?: string;
+  latency?: number;
+}>;
+
+export class Heartbeat {
+  private intervalMs: number;
+  private checks: Map<string, HealthCheck> = new Map();
+  private intervalId: Timer | null = null;
+  private startTime: Date;
+  private lastStatus: HealthStatus | null = null;
+  private onHealthChange?: (status: HealthStatus) => void;
+  private log = logger.child("heartbeat");
+
+  constructor(_config: Config, options: HeartbeatOptions = {}) {
+    this.intervalMs = options.intervalMs ?? 30000;
+    this.onHealthChange = options.onHealthChange;
+    this.startTime = new Date();
+  }
+
+  registerCheck(name: string, check: HealthCheck): void {
+    this.checks.set(name, check);
+    this.log.debug(`Registered health check: ${name}`);
+  }
+
+  removeCheck(name: string): boolean {
+    return this.checks.delete(name);
+  }
+
+  async runChecks(): Promise<HealthStatus> {
+    const checks: HealthStatus["checks"] = {};
+    let overallStatus: "healthy" | "degraded" | "unhealthy" = "healthy";
+
+    for (const [name, check] of this.checks) {
+      try {
+        const start = Date.now();
+        const result = await check();
+        const latency = Date.now() - start;
+
+        checks[name] = {
+          status: result.status,
+          message: result.message,
+          latency: result.latency ?? latency,
+        };
+
+        if (result.status === "warning" && overallStatus === "healthy") {
+          overallStatus = "degraded";
+        } else if (result.status === "error") {
+          overallStatus = "unhealthy";
+        }
+      } catch (error) {
+        checks[name] = {
+          status: "error",
+          message: (error as Error).message,
+        };
+        overallStatus = "unhealthy";
+      }
+    }
+
+    const status: HealthStatus = {
+      status: overallStatus,
+      checks,
+      uptime: Date.now() - this.startTime.getTime(),
+      lastCheck: new Date(),
+    };
+
+    const prevStatus = this.lastStatus?.status;
+    if (prevStatus && prevStatus !== overallStatus) {
+      this.log.info(`Health status changed: ${prevStatus} -> ${overallStatus}`);
+      this.onHealthChange?.(status);
+    }
+
+    this.lastStatus = status;
+    return status;
+  }
+
+  start(): void {
+    if (this.intervalId) {
+      this.log.warn("Heartbeat already running");
+      return;
+    }
+
+    this.runChecks();
+
+    this.intervalId = setInterval(async () => {
+      await this.runChecks();
+    }, this.intervalMs);
+
+    this.log.info(`Heartbeat started (interval: ${this.intervalMs}ms)`);
+  }
+
+  stop(): void {
+    if (this.intervalId) {
+      clearInterval(this.intervalId);
+      this.intervalId = null;
+      this.log.info("Heartbeat stopped");
+    }
+  }
+
+  getStatus(): HealthStatus | null {
+    return this.lastStatus;
+  }
+
+  isRunning(): boolean {
+    return this.intervalId !== null;
+  }
+}
+
+export function createHeartbeat(config: Config, options?: HeartbeatOptions): Heartbeat {
+  const heartbeat = new Heartbeat(config, options);
+
+  heartbeat.registerCheck("memory", async () => {
+    const memUsage = process.memoryUsage();
+    const heapUsedMB = Math.round(memUsage.heapUsed / 1024 / 1024);
+    const heapTotalMB = Math.round(memUsage.heapTotal / 1024 / 1024);
+    const ratio = memUsage.heapUsed / memUsage.heapTotal;
+
+    if (ratio > 0.9) {
+      return {
+        status: "error",
+        message: `Memory critically high: ${heapUsedMB}/${heapTotalMB}MB`,
+      };
+    } else if (ratio > 0.75) {
+      return {
+        status: "warning",
+        message: `Memory usage high: ${heapUsedMB}/${heapTotalMB}MB`,
+      };
+    }
+
+    return {
+      status: "ok",
+      message: `${heapUsedMB}/${heapTotalMB}MB used`,
+    };
+  });
+
+  return heartbeat;
+}

package/packages/core/src/index.ts

@@ -0,0 +1,81 @@
+// ─── API ─────────────────────────────────────────────────────────────────────
+export { createAgent } from "./api/index.ts";
+export type { AgentConfig, Agent, AgentEvent } from "./api/index.ts";
+
+// ─── Tools ───────────────────────────────────────────────────────────────────
+export { defineTool } from "./tools/ToolRegistry.ts";
+export type { ToolDefinition } from "./tools/ToolRegistry.ts";
+export { ToolRegistry } from "./tools/ToolRegistry.ts";
+export { ToolExecutor } from "./tools/ToolExecutor.ts";
+export type { ToolExecutionResult } from "./tools/ToolExecutor.ts";
+
+// ─── Skills ──────────────────────────────────────────────────────────────────
+export { defineSkill } from "./skills/defineSkill.ts";
+export type { SkillDefinition } from "./skills/defineSkill.ts";
+export { SkillLoader } from "./skills/index.ts";
+export type { Skill, SkillStep, OutputFormat, SkillsConfig } from "./skills/index.ts";
+
+// ─── Agent ───────────────────────────────────────────────────────────────────
+export { AgentLoop, runAgent, runAgentIsolated, buildAgentLoop, getAgentLoop, rebuildAgentLoop } from "./agent/index.ts";
+export type { AgentLoopOptions, StepEvent, StreamChunk } from "./agent/index.ts";
+export type { Tool, ToolParameter, ToolResult } from "./agent/index.ts";
+
+// ─── Swarm / Scheduler ───────────────────────────────────────────────────────
+export { DAGScheduler } from "./swarm/index.ts";
+export type { DAGSchedulerOptions, IAgentExecutor } from "./swarm/index.ts";
+export { TaskGraph, TaskNode } from "./swarm/index.ts";
+export type { TaskNodeConfig, NodeStatus, DAGResult, NodeSummary } from "./swarm/index.ts";
+export { CronScheduler } from "./scheduler/index.ts";
+export type { CronJob } from "./scheduler/index.ts";
+
+// ─── MCP ─────────────────────────────────────────────────────────────────────
+export { MCPClientManager } from "./mcp/index.ts";
+export type { MCPTool, MCPResource, MCPPrompt, MCPConfig, MCPServerConfig } from "./mcp/index.ts";
+
+// ─── Ethics ──────────────────────────────────────────────────────────────────
+export { EthicsGuard } from "./ethics/index.ts";
+export type { EthicsRule } from "./ethics/index.ts";
+
+// ─── Memory ──────────────────────────────────────────────────────────────────
+export { Scratchpad } from "./memory/index.ts";
+export type { IStorage } from "./memory/index.ts";
+
+// ─── Storage ─────────────────────────────────────────────────────────────────
+export { initializeDatabase, dbService } from "./storage/index.ts";
+
+// ─── Config ──────────────────────────────────────────────────────────────────
+export { loadConfig, loadEnv, getHiveDir } from "./config/index.ts";
+export type { Config } from "./config/index.ts";
+
+// ─── Gateway ─────────────────────────────────────────────────────────────────
+export { startGateway } from "./gateway/index.ts";
+export type { GatewayConfig } from "./gateway/index.ts";
+
+// ─── Channels ────────────────────────────────────────────────────────────────
+export { ChannelManager } from "./channels/manager.ts";
+export { BaseChannel } from "./channels/base.ts";
+export { TelegramChannel } from "./channels/telegram.ts";
+export { DiscordChannel } from "./channels/discord.ts";
+export { WhatsAppChannel } from "./channels/whatsapp.ts";
+export { SlackChannel } from "./channels/slack.ts";
+export { WebChatChannel } from "./channels/webchat.ts";
+
+// ─── Canvas ──────────────────────────────────────────────────────────────────
+export { CanvasManager } from "./canvas/CanvasManager.ts";
+export { emitCanvas, subscribeCanvas, unsubscribeCanvas } from "./canvas/emitter.ts";
+
+// ─── Tool Runtime ────────────────────────────────────────────────────────────
+export { executeToolBatch } from "./tool-runtime/index.ts";
+export type { ToolBatchResult, ExecuteToolBatchOptions } from "./tool-runtime/index.ts";
+
+// ─── Events ──────────────────────────────────────────────────────────────────
+export { eventBus } from "./events/event-bus.ts";
+export { agentBus } from "./events/agent-bus.ts";
+
+// ─── Workers ─────────────────────────────────────────────────────────────────
+export { createWorker, WorkerPool } from "./workers/index.ts";
+export type { WorkerConfig, WorkerInstance, WorkerChunk, WorkerPoolConfig, PoolTask, PoolTaskResult } from "./workers/index.ts";
+
+// ─── Utils ───────────────────────────────────────────────────────────────────
+export { logger } from "./utils/index.ts";
+export { retry } from "./utils/retry.ts";