@workglow/ai 0.2.4 → 0.2.6

package/dist/bun.js

@@ -1145,7 +1145,7 @@ class AiTask extends Task {
     if (modelTaskProperties.length > 0) {
       const modelRepo = registry.get(MODEL_REPOSITORY);
       const taskModels = await modelRepo.findModelsByTask(this.type) ?? [];
-      for (const [key, propSchema] of modelTaskProperties) {
+      for (const [key] of modelTaskProperties) {
         const requestedModel = input[key];
         if (typeof requestedModel === "string") {
           const found = taskModels?.find((m) => m.model_id === requestedModel);
@@ -3206,6 +3206,8 @@ var inputSchema7 = {
       description: "The document ID"
     },
     documentTree: {
+      type: "object",
+      additionalProperties: true,
       title: "Document Tree",
       description: "The hierarchical document tree to enrich"
     },
@@ -3429,6 +3431,7 @@ Workflow14.prototype.documentEnricher = CreateWorkflow14(DocumentEnricherTask);
 
 // src/task/DocumentUpsertTask.ts
 import { Document, TypeKnowledgeBase as TypeKnowledgeBase5 } from "@workglow/knowledge-base";
+import { DocumentMetadataSchema } from "@workglow/knowledge-base";
 import { CreateWorkflow as CreateWorkflow15, Task as Task10, Workflow as Workflow15 } from "@workglow/task-graph";
 var inputSchema8 = {
   type: "object",
@@ -3449,10 +3452,16 @@ var inputSchema8 = {
     title: {
       type: "string",
       title: "Title",
-      description: "Human-readable title stored in the document metadata"
+      description: "Optional human-readable title. If provided, overrides metadata.title. " + "Either this or metadata.title must be supplied."
+    },
+    metadata: {
+      ...DocumentMetadataSchema,
+      required: [],
+      title: "Metadata",
+      description: "Optional document metadata. May contain `title` (unless the top-level " + "`title` input is also provided), `sourceUri`, `createdAt`, and any " + "additional caller-defined fields (the schema is open)."
     }
   },
-  required: ["knowledgeBase", "doc_id", "documentTree", "title"],
+  required: ["knowledgeBase", "doc_id", "documentTree"],
   additionalProperties: false
 };
 var outputSchema8 = {
@@ -3481,10 +3490,17 @@ class DocumentUpsertTask extends Task10 {
     return outputSchema8;
   }
   async execute(input, context) {
-    const { knowledgeBase, doc_id, documentTree, title } = input;
+    const { knowledgeBase, doc_id, documentTree, title, metadata } = input;
     const kb = knowledgeBase;
+    const merged = {
+      ...metadata ?? {},
+      ...title !== undefined ? { title } : {}
+    };
+    if (!merged.title) {
+      throw new Error("DocumentUpsertTask: title is required \u2014 provide it via the 'title' input or 'metadata.title'");
+    }
     await context.updateProgress(1, "Upserting document");
-    const document = new Document(documentTree, { title }, [], doc_id);
+    const document = new Document(documentTree, merged, [], doc_id);
     const stored = await kb.upsertDocument(document);
     return {
       doc_id: stored.doc_id ?? doc_id
@@ -4194,6 +4210,8 @@ var inputSchema9 = {
       description: "The ID of the document"
     },
     documentTree: {
+      type: "object",
+      additionalProperties: true,
       title: "Document Tree",
       description: "The hierarchical document tree to chunk"
     },
@@ -5635,7 +5653,6 @@ class RerankerTask extends Task15 {
   static title = "Reranker";
   static description = "Rerank retrieved chunks to improve relevance";
   static cacheable = true;
-  resolvedCrossEncoderModel;
   static inputSchema() {
     return inputSchema12;
   }
@@ -5807,8 +5824,10 @@ var outputSchema13 = {
       description: "Generated or provided document ID"
     },
     documentTree: {
+      type: "object",
       title: "Document Tree",
-      description: "Parsed hierarchical document tree"
+      description: "Parsed hierarchical document tree",
+      additionalProperties: true
     },
     nodeCount: {
       type: "number",
@@ -5834,7 +5853,7 @@ class StructuralParserTask extends Task16 {
   }
   async execute(input, context) {
     const { text, title, format = "auto", sourceUri, doc_id: providedDocId } = input;
-    const doc_id = providedDocId || uuid42();
+    const doc_id = providedDocId || sourceUri || uuid42();
     let documentTree;
     if (format === "markdown") {
       documentTree = await StructuralParser.parseMarkdown(doc_id, text, title);
@@ -7650,4 +7669,4 @@ export {
   AgentInputSchema
 };
 
-//# debugId=A0AE35A82942CF9964756E2164756E21
+//# debugId=3A8ABD53CB039BD464756E2164756E21