@castlemilk/omega 0.1.9 → 0.3.0

package/dist/server.js

@@ -5,6 +5,9 @@ import express from "express";
 import cors from "cors";
 import path from "path";
 import { fileURLToPath } from "url";
+import { z } from "zod";
+import * as grpc from "@grpc/grpc-js";
+import * as protoLoader from "@grpc/proto-loader";
 
 // ../providers/src/openai.ts
 var DEFAULT_BASE_URL = "https://api.openai.com/v1";
@@ -41,7 +44,7 @@ var OpenAIProvider = class {
       })
     });
     if (!res.ok) {
-      throw new Error(`OpenAI request failed: ${res.status} ${res.statusText}`);
+      throw new Error(`OpenAI request failed: ${res.status.toString()} ${res.statusText}`);
     }
     const data = await res.json();
     return data.choices?.[0]?.message?.content ?? "";
@@ -86,7 +89,7 @@ var AnthropicProvider = class {
       body: JSON.stringify(body)
     });
     if (!res.ok) {
-      throw new Error(`Anthropic request failed: ${res.status} ${res.statusText}`);
+      throw new Error(`Anthropic request failed: ${res.status.toString()} ${res.statusText}`);
     }
     const data = await res.json();
     return data.content?.find((c) => c.type === "text")?.text ?? "";
@@ -130,7 +133,7 @@ var OllamaProvider = class {
       })
     });
     if (!res.ok) {
-      throw new Error(`Ollama request failed: ${res.status} ${res.statusText}`);
+      throw new Error(`Ollama request failed: ${res.status.toString()} ${res.statusText}`);
     }
     const data = await res.json();
     return data.message?.content ?? "";
@@ -174,7 +177,7 @@ var GeminiProvider = class {
       body: JSON.stringify(body)
     });
     if (!res.ok) {
-      throw new Error(`Gemini request failed: ${res.status} ${res.statusText}`);
+      throw new Error(`Gemini request failed: ${res.status.toString()} ${res.statusText}`);
     }
     const data = await res.json();
     return data.candidates?.[0]?.content?.parts?.map((p) => p.text).join("") ?? "";
@@ -186,7 +189,7 @@ var KimiProvider = class extends OpenAIProvider {
   constructor(config) {
     super({
       ...config,
-      baseUrl: config.baseUrl || "https://api.moonshot.ai/v1"
+      baseUrl: config.baseUrl ?? "https://api.moonshot.ai/v1"
     });
     this.config = config;
   }
@@ -212,7 +215,7 @@ function createProvider(config) {
     case "generic":
       return new GenericProvider(config);
     default:
-      throw new Error(`Unknown provider kind: ${config.kind}`);
+      throw new Error(`Unknown provider kind: ${String(config.kind)}`);
   }
 }
 
@@ -255,9 +258,11 @@ function selectProvider(configs, rules2, task) {
     return true;
   }).sort((a, b) => b.priority - a.priority);
   for (const rule of matchingRules) {
-    const provider = rule.then.provider ? enabled.find((cfg) => cfg.id === rule.then.provider || cfg.name === rule.then.provider) : enabled[0];
+    const provider = rule.then.provider ? enabled.find(
+      (cfg) => cfg.id === rule.then.provider || cfg.name === rule.then.provider
+    ) : enabled[0];
     if (provider) {
-      return { provider, model: rule.then.model || provider.defaultModel };
+      return { provider, model: rule.then.model ?? provider.defaultModel };
     }
   }
   let best;
@@ -275,6 +280,11 @@ function selectProvider(configs, rules2, task) {
 // src/server.ts
 var __filename = fileURLToPath(import.meta.url);
 var __dirname = path.dirname(__filename);
+function asyncHandler(fn) {
+  return (req, res, next) => {
+    Promise.resolve(fn(req, res, next)).catch(next);
+  };
+}
 var app = express();
 app.use(cors());
 app.use(express.json());
@@ -305,11 +315,44 @@ if (process.env.KIMI_API_KEY) {
 }
 var rules = [];
 function newId() {
-  return `${Date.now()}-${Math.random().toString(36).slice(2)}`;
+  return `${Date.now().toString()}-${Math.random().toString(36).slice(2)}`;
 }
 function now() {
   return /* @__PURE__ */ new Date();
 }
+function parseCapabilities(value) {
+  if (Array.isArray(value)) return value;
+  if (typeof value === "string") return JSON.parse(value);
+  return [];
+}
+var projectCreateSchema = z.object({
+  name: z.string().min(1),
+  path: z.string().min(1),
+  repoUrl: z.string().optional(),
+  description: z.string().optional(),
+  env: z.record(z.string()).optional()
+});
+var taskCreateSchema = z.object({
+  projectId: z.string().min(1),
+  title: z.string().min(1),
+  description: z.string().optional(),
+  complexity: z.enum(["simple", "medium", "complex"]).default("simple"),
+  tags: z.array(z.string()).default([])
+});
+var providerCreateSchema = z.object({
+  name: z.string().min(1),
+  kind: z.enum(["openai", "anthropic", "ollama", "gemini", "kimi", "generic"]),
+  baseUrl: z.string().optional(),
+  apiKey: z.string().optional(),
+  defaultModel: z.string().min(1),
+  capabilities: z.union([z.string(), z.array(z.any())]).optional(),
+  enabled: z.boolean().optional()
+});
+var routerSelectSchema = z.object({
+  title: z.string().min(1),
+  complexity: z.enum(["simple", "medium", "complex"]).default("simple"),
+  tags: z.array(z.string()).default([])
+});
 app.get("/projects", (_req, res) => {
   res.json(
     projects.map((p) => ({
@@ -319,14 +362,10 @@ app.get("/projects", (_req, res) => {
   );
 });
 app.post("/projects", (req, res) => {
-  const body = req.body;
+  const body = projectCreateSchema.parse(req.body);
   const project = {
     id: newId(),
-    name: body.name,
-    path: body.path,
-    repoUrl: body.repoUrl,
-    description: body.description,
-    env: body.env,
+    ...body,
     createdAt: now()
   };
   projects.push(project);
@@ -338,19 +377,15 @@ app.delete("/projects/:id", (req, res) => {
   res.status(204).send();
 });
 app.get("/tasks", (req, res) => {
-  const projectId = req.query.projectId;
+  const projectId = typeof req.query.projectId === "string" ? req.query.projectId : void 0;
   res.json(tasks.filter((t) => projectId ? t.projectId === projectId : true));
 });
 app.post("/tasks", (req, res) => {
-  const body = req.body;
+  const body = taskCreateSchema.parse(req.body);
   const task = {
     id: newId(),
-    projectId: body.projectId,
-    title: body.title,
-    description: body.description,
+    ...body,
     status: "todo",
-    complexity: body.complexity || "simple",
-    tags: body.tags || [],
     createdAt: now(),
     updatedAt: now()
   };
@@ -363,9 +398,12 @@ app.patch("/tasks/:id", (req, res) => {
   Object.assign(task, req.body, { updatedAt: now() });
   res.json(task);
 });
-app.post("/tasks/:id/run", async (req, res) => {
+app.post("/tasks/:id/run", asyncHandler(async (req, res) => {
   const task = tasks.find((t) => t.id === req.params.id);
-  if (!task) return res.status(404).json({ error: "Task not found" });
+  if (!task) {
+    res.status(404).json({ error: "Task not found" });
+    return;
+  }
   task.status = "in_progress";
   task.error = void 0;
   task.result = void 0;
@@ -373,7 +411,8 @@ app.post("/tasks/:id/run", async (req, res) => {
   if (!selection) {
     task.status = "failed";
     task.error = "No provider available for this task";
-    return res.json(task);
+    res.json(task);
+    return;
   }
   try {
     const provider = createProvider(selection.provider);
@@ -387,20 +426,16 @@ app.post("/tasks/:id/run", async (req, res) => {
     task.error = err instanceof Error ? err.message : String(err);
   }
   res.json(task);
-});
+}));
 app.get("/providers", (_req, res) => {
   res.json(providerConfigs);
 });
 app.post("/providers", (req, res) => {
-  const body = req.body;
+  const body = providerCreateSchema.parse(req.body);
   const cfg = {
     id: newId(),
-    name: body.name,
-    kind: body.kind,
-    baseUrl: body.baseUrl,
-    apiKey: body.apiKey,
-    defaultModel: body.defaultModel,
-    capabilities: Array.isArray(body.capabilities) ? body.capabilities : JSON.parse(body.capabilities || "[]"),
+    ...body,
+    capabilities: parseCapabilities(body.capabilities),
     enabled: body.enabled ?? true
   };
   providerConfigs.push(cfg);
@@ -417,14 +452,12 @@ app.delete("/providers/:id", (req, res) => {
   res.status(204).send();
 });
 app.post("/router/select", (req, res) => {
-  const body = req.body;
+  const body = routerSelectSchema.parse(req.body);
   const task = {
     id: "preview",
     projectId: "preview",
-    title: body.title,
+    ...body,
     status: "todo",
-    complexity: body.complexity || "simple",
-    tags: body.tags || [],
     createdAt: now(),
     updatedAt: now()
   };
@@ -437,7 +470,125 @@ app.use(express.static(webDir));
 app.get("*", (_req, res) => {
   res.sendFile(path.join(webDir, "index.html"));
 });
-var PORT = process.env.PORT || 4e3;
+var PORT = Number(process.env.PORT ?? 4e3);
 app.listen(PORT, () => {
-  console.log(`Omega harness server on http://localhost:${PORT}`);
+  console.log(`Omega harness server on http://localhost:${PORT.toString()}`);
+});
+var GRPC_PORT = Number(process.env.GRPC_PORT ?? 50051);
+var GRPC_PROTO_PATH = path.join(__dirname, "proto/tasks.proto");
+var grpcPackageDef = protoLoader.loadSync(GRPC_PROTO_PATH, {
+  keepCase: true,
+  longs: String,
+  enums: String,
+  defaults: true,
+  oneofs: true
+});
+var grpcProto = grpc.loadPackageDefinition(grpcPackageDef);
+function parseGrpcRequest(req) {
+  const r = req;
+  return {
+    project_id: typeof r.project_id === "string" ? r.project_id : "",
+    title: typeof r.title === "string" ? r.title : "",
+    description: typeof r.description === "string" ? r.description : void 0,
+    complexity: typeof r.complexity === "string" ? r.complexity : void 0,
+    tags: Array.isArray(r.tags) ? r.tags.filter((t) => typeof t === "string") : void 0,
+    auto_run: typeof r.auto_run === "boolean" ? r.auto_run : void 0
+  };
+}
+function isValidComplexity(value) {
+  return ["simple", "medium", "complex"].includes(value);
+}
+async function executeTask(task) {
+  task.status = "in_progress";
+  task.error = void 0;
+  task.result = void 0;
+  const selection = selectProvider(providerConfigs, rules, task);
+  if (!selection) {
+    task.status = "failed";
+    task.error = "No provider available for this task";
+    return;
+  }
+  try {
+    const provider = createProvider(selection.provider);
+    const prompt = [task.title, task.description].filter(Boolean).join("\n\n");
+    const result = await provider.send(prompt, { model: selection.model });
+    task.status = "done";
+    task.result = result;
+    task.assignedModel = { provider: selection.provider.name, model: selection.model };
+  } catch (err) {
+    task.status = "failed";
+    task.error = err instanceof Error ? err.message : String(err);
+  }
+}
+var grpcServer = new grpc.Server();
+grpcServer.addService(grpcProto.omega.TaskIngestion.service, {
+  submitTask: (call, callback) => {
+    void (async () => {
+      try {
+        const req = parseGrpcRequest(call.request);
+        if (!req.project_id || !req.title) {
+          callback(
+            { code: grpc.status.INVALID_ARGUMENT, message: "project_id and title are required" },
+            null
+          );
+          return;
+        }
+        const complexity = isValidComplexity(req.complexity ?? "") ? req.complexity : "simple";
+        const task = {
+          id: newId(),
+          projectId: req.project_id,
+          title: req.title,
+          description: req.description ?? void 0,
+          status: "todo",
+          complexity,
+          tags: req.tags ?? [],
+          createdAt: now(),
+          updatedAt: now()
+        };
+        tasks.push(task);
+        if (req.auto_run) {
+          await executeTask(task);
+        }
+        callback(null, { id: task.id, status: task.status, error: task.error ?? "" });
+      } catch (err) {
+        const message = err instanceof Error ? err.message : String(err);
+        callback({ code: grpc.status.INTERNAL, message }, null);
+      }
+    })();
+  },
+  streamTasks: (call) => {
+    const req = call.request;
+    const projectId = req.project_id;
+    let cancelled = false;
+    const interval = setInterval(() => {
+      if (cancelled) return;
+      const filtered = projectId ? tasks.filter((t) => t.projectId === projectId) : tasks;
+      for (const task of filtered.slice(-20)) {
+        call.write({
+          id: task.id,
+          title: task.title,
+          status: task.status,
+          provider: task.assignedModel?.provider ?? "",
+          model: task.assignedModel?.model ?? "",
+          result: task.result ?? "",
+          error: task.error ?? ""
+        });
+      }
+    }, 1e3);
+    call.on("cancelled", () => {
+      cancelled = true;
+      clearInterval(interval);
+    });
+    call.on("error", () => {
+      cancelled = true;
+      clearInterval(interval);
+    });
+  }
+});
+grpcServer.bindAsync(`0.0.0.0:${GRPC_PORT.toString()}`, grpc.ServerCredentials.createInsecure(), (err) => {
+  if (err) {
+    console.error("gRPC server failed to start:", err);
+    return;
+  }
+  console.log(`gRPC task ingestion on 0.0.0.0:${GRPC_PORT.toString()}`);
 });

package/dist/web/dist/App.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"App.d.ts","sourceRoot":"","sources":["../src/App.tsx"],"names":[],"mappings":"AAOA,iBAAS,GAAG,gCAoDX;AAED,eAAe,GAAG,CAAC"}
+{"version":3,"file":"App.d.ts","sourceRoot":"","sources":["../src/App.tsx"],"names":[],"mappings":"AAOA,iBAAS,GAAG,gCAsDX;AAED,eAAe,GAAG,CAAC"}

package/dist/web/dist/App.js

@@ -30,10 +30,12 @@ function App() {
         }
     }
     useEffect(() => {
-        loadProjects();
-        loadProviders();
+        void (async () => {
+            await loadProjects();
+            await loadProviders();
+        })();
     }, []);
-    return (_jsxs("div", { className: "flex h-screen w-screen overflow-hidden", children: [_jsx(ProjectSidebar, { projects: projects, selectedId: selectedProjectId, onSelect: setSelectedProjectId, onChange: loadProjects }), _jsx(TaskBoard, { projectId: selectedProjectId }), _jsxs("aside", { className: "w-80 bg-white border-l border-gray-200 h-screen overflow-y-auto", children: [error && (_jsx("div", { className: "p-3 bg-red-50 text-red-700 text-xs border-b border-red-100", children: error })), _jsx(ProviderSettings, { providers: providers, onChange: loadProviders }), _jsx(RouterPanel, {})] })] }));
+    return (_jsxs("div", { className: "flex h-screen w-screen overflow-hidden", children: [_jsx(ProjectSidebar, { projects: projects, selectedId: selectedProjectId, onSelect: setSelectedProjectId, onChange: () => { void loadProjects(); } }), _jsx(TaskBoard, { projectId: selectedProjectId }), _jsxs("aside", { className: "w-80 bg-white border-l border-gray-200 h-screen overflow-y-auto", children: [error && (_jsx("div", { className: "p-3 bg-red-50 text-red-700 text-xs border-b border-red-100", children: error })), _jsx(ProviderSettings, { providers: providers, onChange: () => { void loadProviders(); } }), _jsx(RouterPanel, {})] })] }));
 }
 export default App;
 //# sourceMappingURL=App.js.map

package/dist/web/dist/App.js.map

@@ -1 +1 @@
-{"version":3,"file":"App.js","sourceRoot":"","sources":["../src/App.tsx"],"names":[],"mappings":";AAAA,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AAC5C,OAAO,EAAE,GAAG,EAAE,MAAM,cAAc,CAAC;AACnC,OAAO,EAAE,cAAc,EAAgB,MAAM,gCAAgC,CAAC;AAC9E,OAAO,EAAE,SAAS,EAAE,MAAM,2BAA2B,CAAC;AACtD,OAAO,EAAE,gBAAgB,EAAiB,MAAM,kCAAkC,CAAC;AACnF,OAAO,EAAE,WAAW,EAAE,MAAM,6BAA6B,CAAC;AAE1D,SAAS,GAAG;IACV,MAAM,CAAC,QAAQ,EAAE,WAAW,CAAC,GAAG,QAAQ,CAAY,EAAE,CAAC,CAAC;IACxD,MAAM,CAAC,SAAS,EAAE,YAAY,CAAC,GAAG,QAAQ,CAAa,EAAE,CAAC,CAAC;IAC3D,MAAM,CAAC,iBAAiB,EAAE,oBAAoB,CAAC,GAAG,QAAQ,EAAU,CAAC;IACrE,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,QAAQ,CAAC,EAAE,CAAC,CAAC;IAEvC,KAAK,UAAU,YAAY;QACzB,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,WAAW,EAAE,CAAC;YACrC,WAAW,CAAC,IAAI,CAAC,CAAC;YAClB,QAAQ,CAAC,EAAE,CAAC,CAAC;QACf,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,QAAQ,CAAC,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;QAC7D,CAAC;IACH,CAAC;IAED,KAAK,UAAU,aAAa;QAC1B,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,YAAY,EAAE,CAAC;YACtC,YAAY,CAAC,IAAI,CAAC,CAAC;QACrB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,QAAQ,CAAC,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;QAC7D,CAAC;IACH,CAAC;IAED,SAAS,CAAC,GAAG,EAAE;QACb,YAAY,EAAE,CAAC;QACf,aAAa,EAAE,CAAC;IAClB,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,OAAO,CACL,eAAK,SAAS,EAAC,wCAAwC,aACrD,KAAC,cAAc,IACb,QAAQ,EAAE,QAAQ,EAClB,UAAU,EAAE,iBAAiB,EAC7B,QAAQ,EAAE,oBAAoB,EAC9B,QAAQ,EAAE,YAAY,GACtB,EAEF,KAAC,SAAS,IAAC,SAAS,EAAE,iBAAiB,GAAI,EAE3C,iBAAO,SAAS,EAAC,iEAAiE,aAC/E,KAAK,IAAI,CACR,cAAK,SAAS,EAAC,4DAA4D,YACxE,KAAK,GACF,CACP,EACD,KAAC,gBAAgB,IAAC,SAAS,EAAE,SAAS,EAAE,QAAQ,EAAE,aAAa,GAAI,EACnE,KAAC,WAAW,KAAG,IACT,IACJ,CACP,CAAC;AACJ,CAAC;AAED,eAAe,GAAG,CAAC"}
+{"version":3,"file":"App.js","sourceRoot":"","sources":["../src/App.tsx"],"names":[],"mappings":";AAAA,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AAC5C,OAAO,EAAE,GAAG,EAAE,MAAM,cAAc,CAAC;AACnC,OAAO,EAAE,cAAc,EAAgB,MAAM,gCAAgC,CAAC;AAC9E,OAAO,EAAE,SAAS,EAAE,MAAM,2BAA2B,CAAC;AACtD,OAAO,EAAE,gBAAgB,EAAiB,MAAM,kCAAkC,CAAC;AACnF,OAAO,EAAE,WAAW,EAAE,MAAM,6BAA6B,CAAC;AAE1D,SAAS,GAAG;IACV,MAAM,CAAC,QAAQ,EAAE,WAAW,CAAC,GAAG,QAAQ,CAAY,EAAE,CAAC,CAAC;IACxD,MAAM,CAAC,SAAS,EAAE,YAAY,CAAC,GAAG,QAAQ,CAAa,EAAE,CAAC,CAAC;IAC3D,MAAM,CAAC,iBAAiB,EAAE,oBAAoB,CAAC,GAAG,QAAQ,EAAU,CAAC;IACrE,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,QAAQ,CAAC,EAAE,CAAC,CAAC;IAEvC,KAAK,UAAU,YAAY;QACzB,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,WAAW,EAAE,CAAC;YACrC,WAAW,CAAC,IAAI,CAAC,CAAC;YAClB,QAAQ,CAAC,EAAE,CAAC,CAAC;QACf,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,QAAQ,CAAC,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;QAC7D,CAAC;IACH,CAAC;IAED,KAAK,UAAU,aAAa;QAC1B,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,YAAY,EAAE,CAAC;YACtC,YAAY,CAAC,IAAI,CAAC,CAAC;QACrB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,QAAQ,CAAC,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;QAC7D,CAAC;IACH,CAAC;IAED,SAAS,CAAC,GAAG,EAAE;QACb,KAAK,CAAC,KAAK,IAAI,EAAE;YACf,MAAM,YAAY,EAAE,CAAC;YACrB,MAAM,aAAa,EAAE,CAAC;QACxB,CAAC,CAAC,EAAE,CAAC;IACP,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,OAAO,CACL,eAAK,SAAS,EAAC,wCAAwC,aACrD,KAAC,cAAc,IACb,QAAQ,EAAE,QAAQ,EAClB,UAAU,EAAE,iBAAiB,EAC7B,QAAQ,EAAE,oBAAoB,EAC9B,QAAQ,EAAE,GAAG,EAAE,GAAG,KAAK,YAAY,EAAE,CAAC,CAAC,CAAC,GACxC,EAEF,KAAC,SAAS,IAAC,SAAS,EAAE,iBAAiB,GAAI,EAE3C,iBAAO,SAAS,EAAC,iEAAiE,aAC/E,KAAK,IAAI,CACR,cAAK,SAAS,EAAC,4DAA4D,YACxE,KAAK,GACF,CACP,EACD,KAAC,gBAAgB,IAAC,SAAS,EAAE,SAAS,EAAE,QAAQ,EAAE,GAAG,EAAE,GAAG,KAAK,aAAa,EAAE,CAAC,CAAC,CAAC,GAAI,EACrF,KAAC,WAAW,KAAG,IACT,IACJ,CACP,CAAC;AACJ,CAAC;AAED,eAAe,GAAG,CAAC"}