@bpmnkit/casen-worker-ai 0.1.0

package/src/operations/extract.ts

@@ -0,0 +1,80 @@
+import type { WorkerJob, WorkerJobResult } from "@bpmnkit/cli-sdk"
+import type { AiWorkerConfig } from "../config.js"
+import { RetryableError, callLlm, parseJsonResponse } from "../llm.js"
+
+interface ExtractResponse {
+	extracted: Record<string, unknown>
+	missingFields: string[]
+}
+
+const SYSTEM_PROMPT = `You are a structured data extraction assistant.
+The user will provide unstructured text and a list of fields to extract.
+Respond ONLY with valid JSON matching this exact shape:
+{ "extracted": { "<field>": <value or null>, ... }, "missingFields": ["<fields not found>"] }
+Use null for fields you cannot confidently determine. List those field names in missingFields.
+Do not include any prose outside the JSON object.`
+
+/**
+ * Extracts structured fields from job.variables.input.
+ *
+ * Required job variables:
+ * - input: string — the unstructured text
+ * - fields: string[] — field names to extract
+ *
+ * Optional:
+ * - schema: Record<string, string> — type hints per field, e.g. { amount: "number", date: "ISO 8601 string" }
+ *
+ * Output variables: extracted (object), missingFields (array), aiModel, processedAt
+ */
+export async function extract(job: WorkerJob, config: AiWorkerConfig): Promise<WorkerJobResult> {
+	const input = String(job.variables.input ?? "")
+	const fields = job.variables.fields
+	if (!Array.isArray(fields) || fields.length === 0) {
+		return {
+			outcome: "error",
+			errorCode: "AI_INVALID_INPUT",
+			errorMessage: 'Job variable "fields" must be a non-empty array of field name strings.',
+		}
+	}
+	const schema = job.variables.schema
+	const schemaHint =
+		schema && typeof schema === "object" && !Array.isArray(schema)
+			? `\nType hints: ${JSON.stringify(schema)}`
+			: ""
+	const userMessage = `Fields to extract: ${JSON.stringify(fields)}${schemaHint}\n\nText:\n${input}`
+
+	let raw: string
+	try {
+		raw = await callLlm(SYSTEM_PROMPT, userMessage, config)
+	} catch (err) {
+		if (err instanceof RetryableError) {
+			return { outcome: "fail", errorMessage: err.message, retries: 2, retryBackOff: 30_000 }
+		}
+		return {
+			outcome: "error",
+			errorCode: "AI_API_ERROR",
+			errorMessage: err instanceof Error ? err.message : String(err),
+		}
+	}
+
+	let parsed: ExtractResponse
+	try {
+		parsed = parseJsonResponse<ExtractResponse>(raw)
+	} catch {
+		return {
+			outcome: "error",
+			errorCode: "AI_PARSE_ERROR",
+			errorMessage: `Model returned non-JSON response: ${raw.slice(0, 200)}`,
+		}
+	}
+
+	return {
+		outcome: "complete",
+		variables: {
+			extracted: parsed.extracted,
+			missingFields: parsed.missingFields ?? [],
+			aiModel: config.model,
+			processedAt: new Date().toISOString(),
+		},
+	}
+}

package/src/operations/summarize.test.ts

@@ -0,0 +1,48 @@
+import { describe, expect, it, vi } from "vitest"
+import type { AiWorkerConfig } from "../config.js"
+import * as llmModule from "../llm.js"
+import { summarize } from "./summarize.js"
+
+const cfg: AiWorkerConfig = {
+	apiKey: "test",
+	model: "claude-test",
+	maxTokens: 512,
+	timeoutMs: 5000,
+}
+
+function makeJob(variables: Record<string, unknown>) {
+	return {
+		jobKey: "1",
+		processDefinitionId: "p1",
+		elementId: "e1",
+		processInstanceKey: "i1",
+		variables,
+	}
+}
+
+describe("summarize", () => {
+	it("returns complete with summary and wordCount", async () => {
+		vi.spyOn(llmModule, "callLlm").mockResolvedValue(
+			JSON.stringify({ summary: "A short summary.", wordCount: 3 }),
+		)
+		const result = await summarize(makeJob({ input: "Long text here..." }), cfg)
+		expect(result.outcome).toBe("complete")
+		if (result.outcome === "complete") {
+			expect(result.variables.summary).toBe("A short summary.")
+			expect(result.variables.wordCount).toBe(3)
+		}
+	})
+
+	it("returns error on parse failure", async () => {
+		vi.spyOn(llmModule, "callLlm").mockResolvedValue("Here is the summary: blah blah.")
+		const result = await summarize(makeJob({ input: "text" }), cfg)
+		expect(result.outcome).toBe("error")
+		if (result.outcome === "error") expect(result.errorCode).toBe("AI_PARSE_ERROR")
+	})
+
+	it("returns fail on RetryableError", async () => {
+		vi.spyOn(llmModule, "callLlm").mockRejectedValue(new llmModule.RetryableError("timeout"))
+		const result = await summarize(makeJob({ input: "text" }), cfg)
+		expect(result.outcome).toBe("fail")
+	})
+})

package/src/operations/summarize.ts

@@ -0,0 +1,68 @@
+import type { WorkerJob, WorkerJobResult } from "@bpmnkit/cli-sdk"
+import type { AiWorkerConfig } from "../config.js"
+import { RetryableError, callLlm, parseJsonResponse } from "../llm.js"
+
+interface SummarizeResponse {
+	summary: string
+	wordCount: number
+}
+
+const SYSTEM_PROMPT = `You are a text summarization assistant.
+The user will provide text to summarize along with a target length and style.
+Respond ONLY with valid JSON matching this exact shape:
+{ "summary": "<the summary text>", "wordCount": <integer> }
+Do not include any prose outside the JSON object.`
+
+/**
+ * Summarizes job.variables.input.
+ *
+ * Required job variables:
+ * - input: string — the text to summarize
+ *
+ * Optional:
+ * - maxWords: number — target word count (default: 100)
+ * - style: "bullet" | "paragraph" — output style (default: "paragraph")
+ *
+ * Output variables: summary, wordCount, aiModel, processedAt
+ */
+export async function summarize(job: WorkerJob, config: AiWorkerConfig): Promise<WorkerJobResult> {
+	const input = String(job.variables.input ?? "")
+	const maxWords = Number(job.variables.maxWords ?? 100)
+	const style = String(job.variables.style ?? "paragraph")
+	const userMessage = `Summarize the following text in ${style} style, targeting ${maxWords} words:\n\n${input}`
+
+	let raw: string
+	try {
+		raw = await callLlm(SYSTEM_PROMPT, userMessage, config)
+	} catch (err) {
+		if (err instanceof RetryableError) {
+			return { outcome: "fail", errorMessage: err.message, retries: 2, retryBackOff: 30_000 }
+		}
+		return {
+			outcome: "error",
+			errorCode: "AI_API_ERROR",
+			errorMessage: err instanceof Error ? err.message : String(err),
+		}
+	}
+
+	let parsed: SummarizeResponse
+	try {
+		parsed = parseJsonResponse<SummarizeResponse>(raw)
+	} catch {
+		return {
+			outcome: "error",
+			errorCode: "AI_PARSE_ERROR",
+			errorMessage: `Model returned non-JSON response: ${raw.slice(0, 200)}`,
+		}
+	}
+
+	return {
+		outcome: "complete",
+		variables: {
+			summary: parsed.summary,
+			wordCount: parsed.wordCount,
+			aiModel: config.model,
+			processedAt: new Date().toISOString(),
+		},
+	}
+}

package/tsconfig.json

@@ -0,0 +1,10 @@
+{
+	"extends": "../../tsconfig.base.json",
+	"compilerOptions": {
+		"composite": true,
+		"outDir": "dist",
+		"rootDir": "src",
+		"types": ["node"]
+	},
+	"include": ["src"]
+}