diragent 0.1.0

package/dist/index.d.ts

@@ -0,0 +1,256 @@
+export { ServerOptions, startServer } from './server/index.js';
+import { z } from 'zod';
+import { EventEmitter } from 'events';
+import { ChildProcess } from 'child_process';
+import { Logger } from 'pino';
+import Database from 'better-sqlite3';
+
+declare const AgentTemplateSchema: z.ZodObject<{
+    driver: z.ZodString;
+    model: z.ZodOptional<z.ZodString>;
+    maxTokens: z.ZodOptional<z.ZodNumber>;
+    command: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
+    env: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>;
+}, "strip", z.ZodTypeAny, {
+    driver: string;
+    model?: string | undefined;
+    maxTokens?: number | undefined;
+    command?: string[] | undefined;
+    env?: Record<string, string> | undefined;
+}, {
+    driver: string;
+    model?: string | undefined;
+    maxTokens?: number | undefined;
+    command?: string[] | undefined;
+    env?: Record<string, string> | undefined;
+}>;
+declare const ConfigSchema: z.ZodObject<{
+    version: z.ZodDefault<z.ZodString>;
+    server: z.ZodDefault<z.ZodObject<{
+        port: z.ZodDefault<z.ZodNumber>;
+        host: z.ZodDefault<z.ZodString>;
+    }, "strip", z.ZodTypeAny, {
+        port: number;
+        host: string;
+    }, {
+        port?: number | undefined;
+        host?: string | undefined;
+    }>>;
+    auth: z.ZodDefault<z.ZodObject<{
+        enabled: z.ZodDefault<z.ZodBoolean>;
+        adminToken: z.ZodOptional<z.ZodString>;
+        apiKeys: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
+    }, "strip", z.ZodTypeAny, {
+        enabled: boolean;
+        adminToken?: string | undefined;
+        apiKeys?: string[] | undefined;
+    }, {
+        enabled?: boolean | undefined;
+        adminToken?: string | undefined;
+        apiKeys?: string[] | undefined;
+    }>>;
+    agents: z.ZodDefault<z.ZodObject<{
+        maxConcurrent: z.ZodDefault<z.ZodNumber>;
+        defaultTimeout: z.ZodDefault<z.ZodNumber>;
+        templates: z.ZodDefault<z.ZodRecord<z.ZodString, z.ZodObject<{
+            driver: z.ZodString;
+            model: z.ZodOptional<z.ZodString>;
+            maxTokens: z.ZodOptional<z.ZodNumber>;
+            command: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
+            env: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>;
+        }, "strip", z.ZodTypeAny, {
+            driver: string;
+            model?: string | undefined;
+            maxTokens?: number | undefined;
+            command?: string[] | undefined;
+            env?: Record<string, string> | undefined;
+        }, {
+            driver: string;
+            model?: string | undefined;
+            maxTokens?: number | undefined;
+            command?: string[] | undefined;
+            env?: Record<string, string> | undefined;
+        }>>>;
+    }, "strip", z.ZodTypeAny, {
+        maxConcurrent: number;
+        defaultTimeout: number;
+        templates: Record<string, {
+            driver: string;
+            model?: string | undefined;
+            maxTokens?: number | undefined;
+            command?: string[] | undefined;
+            env?: Record<string, string> | undefined;
+        }>;
+    }, {
+        maxConcurrent?: number | undefined;
+        defaultTimeout?: number | undefined;
+        templates?: Record<string, {
+            driver: string;
+            model?: string | undefined;
+            maxTokens?: number | undefined;
+            command?: string[] | undefined;
+            env?: Record<string, string> | undefined;
+        }> | undefined;
+    }>>;
+    logging: z.ZodDefault<z.ZodObject<{
+        level: z.ZodDefault<z.ZodEnum<["debug", "info", "warn", "error"]>>;
+        format: z.ZodDefault<z.ZodEnum<["pretty", "json"]>>;
+        file: z.ZodOptional<z.ZodString>;
+    }, "strip", z.ZodTypeAny, {
+        level: "debug" | "info" | "warn" | "error";
+        format: "pretty" | "json";
+        file?: string | undefined;
+    }, {
+        level?: "debug" | "info" | "warn" | "error" | undefined;
+        format?: "pretty" | "json" | undefined;
+        file?: string | undefined;
+    }>>;
+    database: z.ZodDefault<z.ZodObject<{
+        path: z.ZodDefault<z.ZodString>;
+    }, "strip", z.ZodTypeAny, {
+        path: string;
+    }, {
+        path?: string | undefined;
+    }>>;
+}, "strip", z.ZodTypeAny, {
+    version: string;
+    server: {
+        port: number;
+        host: string;
+    };
+    auth: {
+        enabled: boolean;
+        adminToken?: string | undefined;
+        apiKeys?: string[] | undefined;
+    };
+    agents: {
+        maxConcurrent: number;
+        defaultTimeout: number;
+        templates: Record<string, {
+            driver: string;
+            model?: string | undefined;
+            maxTokens?: number | undefined;
+            command?: string[] | undefined;
+            env?: Record<string, string> | undefined;
+        }>;
+    };
+    logging: {
+        level: "debug" | "info" | "warn" | "error";
+        format: "pretty" | "json";
+        file?: string | undefined;
+    };
+    database: {
+        path: string;
+    };
+}, {
+    version?: string | undefined;
+    server?: {
+        port?: number | undefined;
+        host?: string | undefined;
+    } | undefined;
+    auth?: {
+        enabled?: boolean | undefined;
+        adminToken?: string | undefined;
+        apiKeys?: string[] | undefined;
+    } | undefined;
+    agents?: {
+        maxConcurrent?: number | undefined;
+        defaultTimeout?: number | undefined;
+        templates?: Record<string, {
+            driver: string;
+            model?: string | undefined;
+            maxTokens?: number | undefined;
+            command?: string[] | undefined;
+            env?: Record<string, string> | undefined;
+        }> | undefined;
+    } | undefined;
+    logging?: {
+        level?: "debug" | "info" | "warn" | "error" | undefined;
+        format?: "pretty" | "json" | undefined;
+        file?: string | undefined;
+    } | undefined;
+    database?: {
+        path?: string | undefined;
+    } | undefined;
+}>;
+type DirigentConfig = z.infer<typeof ConfigSchema>;
+type AgentTemplate = z.infer<typeof AgentTemplateSchema>;
+declare function loadConfig(configPath?: string): DirigentConfig;
+declare function validateAuth(token: string, config: DirigentConfig): boolean;
+
+interface Agent {
+    id: string;
+    name: string;
+    template: string;
+    status: 'created' | 'starting' | 'running' | 'idle' | 'stopping' | 'stopped' | 'error';
+    workspace?: string;
+    model?: string;
+    pid?: number;
+    process?: ChildProcess;
+    currentTask?: string;
+    createdAt: number;
+    startedAt?: number;
+    stoppedAt?: number;
+    error?: string;
+}
+interface ManagerOptions {
+    dataDir: string;
+    config: DirigentConfig;
+    logger: Logger;
+}
+declare class AgentManager extends EventEmitter {
+    private agents;
+    private processes;
+    private dataDir;
+    private config;
+    private logger;
+    constructor(options: ManagerOptions);
+    private loadAgents;
+    spawn(options: {
+        template: string;
+        name?: string;
+        workspace?: string;
+        task?: string;
+        model?: string;
+    }): Promise<Agent>;
+    private startAgent;
+    private buildCommand;
+    stop(id: string, force?: boolean): Promise<void>;
+    stopAll(): Promise<void>;
+    send(id: string, message: string): Promise<void>;
+    get(id: string): Agent | undefined;
+    list(includeStopped?: boolean): Agent[];
+    log(agentId: string, level: string, message: string): void;
+    getLogs(agentId: string, limit?: number): any[];
+    getStats(): {
+        total: number;
+        running: number;
+        idle: number;
+        stopped: number;
+        error: number;
+    };
+}
+
+declare function initDatabase(path: string): Database.Database;
+declare function getDatabase(): Database.Database | null;
+declare function insertAgent(agent: {
+    id: string;
+    name: string;
+    template: string;
+    workspace?: string;
+    model?: string;
+    config?: object;
+}): void;
+declare function updateAgent(id: string, updates: Record<string, any>): void;
+declare function getAgent(id: string): any;
+declare function getAgents(includesStopped?: boolean): any[];
+declare function insertLog(log: {
+    agentId: string;
+    level: string;
+    message: string;
+    metadata?: object;
+}): void;
+declare function getLogs(agentId: string, limit?: number): any[];
+declare function audit(action: string, actor: string | null, targetType: string | null, targetId: string | null, details?: object): void;
+
+export { type Agent, AgentManager, type AgentTemplate, type DirigentConfig, audit, getAgent, getAgents, getDatabase, getLogs, initDatabase, insertAgent, insertLog, loadConfig, updateAgent, validateAuth };