task-o-matic-core 0.1.0

package/src/test/utils/model-parser.test.ts

@@ -0,0 +1,67 @@
+import assert from "assert";
+import { parseModelString } from "../../utils/model-parser";
+
+describe("Model Parser", () => {
+  it("should parse simple model string", () => {
+    const result = parseModelString("gpt-4o");
+    assert.deepStrictEqual(result, {
+      provider: undefined,
+      model: "gpt-4o",
+      reasoning: undefined,
+    });
+  });
+
+  it("should parse provider and model", () => {
+    const result = parseModelString("openai:gpt-4o");
+    assert.deepStrictEqual(result, {
+      provider: "openai",
+      model: "gpt-4o",
+      reasoning: undefined,
+    });
+  });
+
+  it("should parse model with reasoning default budget", () => {
+    const result = parseModelString("gpt-4o;reasoning");
+    assert.deepStrictEqual(result, {
+      provider: undefined,
+      model: "gpt-4o",
+      reasoning: "2000",
+    });
+  });
+
+  it("should parse model with explicit reasoning budget", () => {
+    const result = parseModelString("gpt-4o;reasoning=5000");
+    assert.deepStrictEqual(result, {
+      provider: undefined,
+      model: "gpt-4o",
+      reasoning: "5000",
+    });
+  });
+
+  it("should parse provider, model, and reasoning", () => {
+    const result = parseModelString("anthropic:claude-3-5-sonnet;reasoning=1000");
+    assert.deepStrictEqual(result, {
+      provider: "anthropic",
+      model: "claude-3-5-sonnet",
+      reasoning: "1000",
+    });
+  });
+
+  it("should treat unknown provider as part of model name", () => {
+    const result = parseModelString("unknown:model");
+    assert.deepStrictEqual(result, {
+      provider: undefined,
+      model: "unknown:model",
+      reasoning: undefined,
+    });
+  });
+
+  it("should handle complex model names", () => {
+    const result = parseModelString("google/gemini-pro-1.5");
+    assert.deepStrictEqual(result, {
+      provider: undefined,
+      model: "google/gemini-pro-1.5",
+      reasoning: undefined,
+    });
+  });
+});

package/src/test/validation.test.ts

@@ -0,0 +1,19 @@
+import assert from "assert";
+import { isValidAIProvider } from "../lib/validation";
+
+describe("Validation Utils", () => {
+  describe("isValidAIProvider", () => {
+    it("should return true for valid providers", () => {
+      assert.strictEqual(isValidAIProvider("openrouter"), true);
+      assert.strictEqual(isValidAIProvider("openai"), true);
+      assert.strictEqual(isValidAIProvider("anthropic"), true);
+      assert.strictEqual(isValidAIProvider("custom"), true);
+    });
+
+    it("should return false for invalid providers", () => {
+      assert.strictEqual(isValidAIProvider("invalid"), false);
+      assert.strictEqual(isValidAIProvider(""), false);
+      assert.strictEqual(isValidAIProvider("gpt-4"), false);
+    });
+  });
+});

package/src/types/callbacks.ts

@@ -0,0 +1,14 @@
+export type ProgressEvent =
+  | { type: "started"; message: string }
+  | { type: "progress"; message: string; current?: number; total?: number }
+  | { type: "stream-chunk"; text: string }
+  | { type: "reasoning-chunk"; text: string }
+  | { type: "completed"; message: string }
+  | { type: "info"; message: string }
+  | { type: "warning"; message: string };
+
+export interface ProgressCallback {
+  onProgress?: (event: ProgressEvent) => void;
+  onComplete?: (result: any) => void;
+  onError?: (error: Error) => void;
+}