@faapi/faapi 6.30.0 → 6.31.0

package/dist/cli/index.js

@@ -3237,9 +3237,20 @@ function extractAgentMetadata(program, filePath, pathMeta) {
     agents,
     model,
     maxTurns,
-    inputDescription
+    inputDescription,
+    inputTypeName: findInputDeclaration(sourceFile) ? "Input" : void 0
   };
 }
+function findInputDeclaration(sourceFile) {
+  let found = false;
+  ts11.forEachChild(sourceFile, (node) => {
+    if (found) return;
+    if ((ts11.isInterfaceDeclaration(node) || ts11.isTypeAliasDeclaration(node)) && node.name.text === "Input") {
+      found = true;
+    }
+  });
+  return found;
+}
 function findRunExportOwner(sourceFile) {
   let result = null;
   ts11.forEachChild(sourceFile, (node) => {
@@ -3464,6 +3475,7 @@ function serializeAgents(agents, dist = "dist") {
     model: a.model,
     maxTurns: a.maxTurns,
     inputDescription: a.inputDescription,
+    inputTypeName: a.inputTypeName,
     filePath: toProdFilePath(a.filePath, dist)
   }));
 }
@@ -3484,7 +3496,8 @@ function hydrateAgents(manifest) {
     agents: a.agents ?? void 0,
     model: a.model ?? void 0,
     maxTurns: a.maxTurns ?? void 0,
-    inputDescription: a.inputDescription ?? void 0
+    inputDescription: a.inputDescription ?? void 0,
+    inputTypeName: a.inputTypeName ?? void 0
   }));
 }
 function validateAgentList(metadata) {
@@ -3530,8 +3543,66 @@ async function generateAgentArtifacts(agents, rootDir, dist) {
   const serialized = serializeAgents(metadata, dist);
   const agentsPath = path15.resolve(rootDir, dist, AGENTS_FILE);
   await writeAgentsModule(serialized, agentsPath);
+  await generateAgentZodArtifacts(metadata, programByFile, rootDir, dist);
   return metadata;
 }
+async function generateAgentZodArtifacts(metadata, programByFile, rootDir, dist) {
+  const declared = metadata.filter((a) => a.inputTypeName);
+  if (declared.length === 0) return;
+  const resolversByFile = /* @__PURE__ */ new Map();
+  const sources = [];
+  for (const agent of declared) {
+    const absPath = path15.resolve(rootDir, agent.filePath);
+    const program = programByFile.get(absPath);
+    if (!program) {
+      throw new Error(
+        `agent "${agent.name}" \u6E90\u6587\u4EF6\u4E0D\u5728 Program \u4E2D: ${agent.filePath}\u2014\u2014\u65E0\u6CD5\u751F\u6210\u5165\u53C2 schema`
+      );
+    }
+    if (!resolversByFile.has(absPath)) {
+      resolversByFile.set(absPath, createLazyTypeResolver(program, absPath));
+    }
+    const typeInfo = extractTypeInfo(program, absPath, agent.inputTypeName);
+    sources.push({
+      name: agent.name,
+      filePath: absPath,
+      schemaName: agent.inputTypeName,
+      typeInfo
+    });
+  }
+  await generateZodArtifacts(sources, {
+    generateFileSource: generateAgentSchemaFileSource,
+    resolversByFile,
+    rootDir,
+    dist
+  });
+}
+function generateAgentSchemaFileSource(sources, resolveType, helpersImportPath) {
+  const lines = ["import { z } from 'zod';"];
+  const schemaBlocks = [];
+  for (const source of sources) {
+    if (!source.typeInfo) continue;
+    const block = [`// agent ${source.name} \u2192 ${source.schemaName}`];
+    const schemaCode = generateZodSchemaSource(
+      source.typeInfo,
+      resolveType,
+      source.schemaName,
+      false
+    ).replace(/^import \{ z \} from 'zod';\s*\n\s*\n/, "");
+    block.push(schemaCode);
+    block.push("");
+    schemaBlocks.push(block.join("\n"));
+  }
+  const allSchemaCode = schemaBlocks.join("\n");
+  if (helpersImportPath && usesCoerceHelpers(allSchemaCode)) {
+    lines.push(
+      `import { coerceNumber, coerceBoolean, coerceMap, coerceSet } from '${helpersImportPath}';`
+    );
+  }
+  lines.push("");
+  lines.push(...schemaBlocks);
+  return lines.join("\n").replace(/\n+$/, "\n");
+}
 var AGENTS_FILE;
 var init_generateAgentArtifacts = __esm({
   "src/cli/generateAgentArtifacts.ts"() {
@@ -3540,6 +3611,9 @@ var init_generateAgentArtifacts = __esm({
     init_createProgram();
     init_prodPaths();
     init_atomicWrite();
+    init_extractHandlerTypes();
+    init_generateZodSchema();
+    init_generateZodArtifacts();
     AGENTS_FILE = "faapi-agents.js";
   }
 });