@aigne/cli 1.60.0-beta → 1.74.0-beta

package/dist/utils/agent-v1.d.mts

@@ -0,0 +1,138 @@
+//#region src/utils/agent-v1.d.ts
+declare function isV1Package(src: string): Promise<boolean>;
+declare function toAIGNEPackage(src: string, dst: string): Promise<void>;
+type Parameter = {
+  id: string;
+  key?: string;
+  hidden?: boolean;
+  required?: boolean;
+  placeholder?: string;
+} & ({
+  type?: "string" | "number" | "boolean" | "language";
+} | {
+  type?: "select";
+  options?: {
+    id: string;
+    label?: string;
+    value?: string;
+  }[];
+});
+interface VariableTypeBase {
+  id: string;
+  name?: string;
+  description?: string;
+  required?: boolean;
+  hidden?: boolean;
+}
+type OutputVariable = VariableTypeBase & ({
+  type?: undefined;
+} | {
+  type: "string";
+  defaultValue?: string;
+} | {
+  type: "number";
+  defaultValue?: number;
+} | {
+  type: "boolean";
+  defaultValue?: boolean;
+} | {
+  type: "object";
+  properties?: OutputVariable[];
+} | {
+  type: "array";
+  element?: OutputVariable;
+});
+type AgentV1 = {
+  id: string;
+  name?: string;
+  description?: string;
+  parameters?: Parameter[];
+  outputVariables?: OutputVariable[];
+} & ({
+  type: "prompt";
+  prompts?: Prompt[];
+  temperature?: number;
+  topP?: number;
+  presencePenalty?: number;
+  frequencyPenalty?: number;
+  maxTokens?: number;
+  model?: string;
+} | {
+  type: "router";
+  defaultToolId?: string;
+  prompt?: string;
+  decisionType?: "ai";
+  routes?: Tool[];
+  temperature?: number;
+  topP?: number;
+  presencePenalty?: number;
+  frequencyPenalty?: number;
+  maxTokens?: number;
+  model?: string;
+} | {
+  type: "image";
+  prompt?: string;
+  model?: string;
+  n?: number;
+  quality?: string;
+  style?: string;
+  size?: string;
+  modelSettings?: {
+    [key: string]: unknown;
+  };
+} | {
+  type: "api";
+  requestParameters?: {
+    id: string;
+    key?: string;
+    value?: string;
+  }[];
+  requestMethod?: string;
+  requestUrl?: string;
+  requestHeaders: {
+    id: string;
+    key?: string;
+    value?: string;
+  }[];
+} | {
+  type: "function";
+  code?: string;
+} | {
+  type: "callAgent";
+  agents?: Tool[];
+});
+interface ProjectDefinitionV1 {
+  project: {
+    name?: string;
+    description?: string;
+  };
+  agents: AgentV1[];
+}
+type Prompt = {
+  type: "message";
+  data: PromptMessage;
+  visibility?: "hidden";
+} | {
+  type: "executeBlock";
+  visibility?: "hidden";
+};
+type PromptMessage = {
+  id: string;
+  role: Role;
+  content?: string;
+  name?: string;
+};
+type Role = "system" | "user" | "assistant";
+type Tool = {
+  blockletDid?: string;
+  projectId?: string;
+  id: string;
+  from?: "assistant" | "blockletAPI" | "knowledge";
+  parameters?: {
+    [key: string]: unknown;
+  };
+  functionName?: string;
+};
+//#endregion
+export { AgentV1, OutputVariable, Parameter, ProjectDefinitionV1, Prompt, PromptMessage, Role, Tool, VariableTypeBase, isV1Package, toAIGNEPackage };
+//# sourceMappingURL=agent-v1.d.mts.map

package/dist/utils/agent-v1.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-v1.d.mts","names":[],"sources":["../../src/utils/agent-v1.ts"],"mappings":";iBAKsB,WAAA,CAAA,GAAA,WAAuB,OAAA;AAAA,iBASvB,cAAA,CAAA,GAAA,UAAA,GAAA,WAAuC,OAAA;AAAA,KA+OjD,SAAA;EAAA,EAAA;EAAA,GAAA;EAAA,MAAA;EAAA,QAAA;EAAA,WAAA;AAAA;EAAA,IAAA;AAAA;EAAA,IAAA;EAAA,OAAA;IAAA,EAAA;IAAA,KAAA;IAAA,KAAA;EAAA;AAAA;AAAA,UAoBK,gBAAA;EAAA,EAAA;EAAA,IAAA;EAAA,WAAA;EAAA,QAAA;EAAA,MAAA;AAAA;AAAA,KAQL,cAAA,GAAiB,gBAAA;EAAA,IAAA;AAAA;EAAA,IAAA;EAAA,YAAA;AAAA;EAAA,IAAA;EAAA,YAAA;AAAA;EAAA,IAAA;EAAA,YAAA;AAAA;EAAA,IAAA;EAAA,UAAA,GAmBR,cAAA;AAAA;EAAA,IAAA;EAAA,OAAA,GAIH,cAAA;AAAA;AAAA,KAIN,OAAA;EAAA,EAAA;EAAA,IAAA;EAAA,WAAA;EAAA,UAAA,GAIG,SAAA;EAAA,eAAA,GACK,cAAA;AAAA;EAAA,IAAA;EAAA,OAAA,GAIJ,MAAA;EAAA,WAAA;EAAA,IAAA;EAAA,eAAA;EAAA,gBAAA;EAAA,SAAA;EAAA,KAAA;AAAA;EAAA,IAAA;EAAA,aAAA;EAAA,MAAA;EAAA,YAAA;EAAA,MAAA,GAaD,IAAA;EAAA,WAAA;EAAA,IAAA;EAAA,eAAA;EAAA,gBAAA;EAAA,SAAA;EAAA,KAAA;AAAA;EAAA,IAAA;EAAA,MAAA;EAAA,KAAA;EAAA,CAAA;EAAA,OAAA;EAAA,KAAA;EAAA,IAAA;EAAA,aAAA;IAAA,CAAA,GAAA;EAAA;AAAA;EAAA,IAAA;EAAA,iBAAA;IAAA,EAAA;IAAA,GAAA;IAAA,KAAA;EAAA;EAAA,aAAA;EAAA,UAAA;EAAA,cAAA;IAAA,EAAA;IAAA,GAAA;IAAA,KAAA;EAAA;AAAA;EAAA,IAAA;EAAA,IAAA;AAAA;EAAA,IAAA;EAAA,MAAA,GAyCA,IAAA;AAAA;AAAA,UAIE,mBAAA;EAAA,OAAA;IAAA,IAAA;IAAA,WAAA;EAAA;EAAA,MAAA,EAKP,OAAA;AAAA;AAAA,KAGE,MAAA;EAAA,IAAA;EAAA,IAAA,EAGA,aAAA;EAAA,UAAA;AAAA;EAAA,IAAA;EAAA,UAAA;AAAA;AAAA,KAQA,aAAA;EAAA,EAAA;EAAA,IAAA,EAEJ,IAAA;EAAA,OAAA;EAAA,IAAA;AAAA;AAAA,KAKI,IAAA;AAAA,KAEA,IAAA;EAAA,WAAA;EAAA,SAAA;EAAA,EAAA;EAAA,IAAA;EAAA,UAAA;IAAA,CAAA,GAAA;EAAA;EAAA,YAAA;AAAA"}

package/dist/utils/agent-v1.mjs

@@ -0,0 +1,179 @@
+import { join } from "node:path";
+import { readFile, stat, writeFile } from "node:fs/promises";
+import { parse, stringify } from "yaml";
+import { glob } from "glob";
+
+//#region src/utils/agent-v1.ts
+async function isV1Package(src) {
+	return stat(join(src, "project.yaml")).then((res) => res.isFile()).catch((error) => {
+		if (error.code === "ENOENT") return false;
+		throw error;
+	});
+}
+async function toAIGNEPackage(src, dst) {
+	const definition = await loadAgentV1Package(src);
+	const aigne = {
+		chat_model: { name: "gpt-4o-mini" },
+		agents: [],
+		cli: { agents: [] }
+	};
+	for (const agent of definition.agents) {
+		const { content } = await assistantToAIGNEV2(agent, definition);
+		const filename = getAgentFilename(agent);
+		await writeFile(join(dst, filename), content);
+		aigne.agents.push(filename);
+		aigne.cli.agents.push(filename);
+	}
+	await writeFile(join(dst, "aigne.yaml"), stringify(aigne));
+}
+async function loadAgentV1Package(path$1) {
+	const agentFilePaths = await glob("prompts/**/*.yaml", { cwd: path$1 });
+	const definition = {
+		project: parse(await readFile(join(path$1, "project.yaml"), "utf8")),
+		agents: []
+	};
+	for (const filename of agentFilePaths) {
+		const agent = parse(await readFile(join(path$1, filename), "utf8"));
+		definition.agents.push(agent);
+	}
+	return definition;
+}
+function assistantToAIGNEV2(agent, project) {
+	const converter = AGENT_MAP[agent.type];
+	if (!converter) throw new Error(`Unsupported agent type: ${agent.type}`);
+	return converter(agent, project);
+}
+const AGENT_MAP = {
+	prompt: (agent) => {
+		if (agent.type !== "prompt") throw new Error(`Expected agent type 'prompt', but got '${agent.type}'`);
+		return { content: stringify({
+			name: agent.name || agent.id,
+			description: agent.description,
+			input_schema: convertInputSchema(agent),
+			output_schema: convertOutputSchema(agent),
+			instructions: agent.prompts?.filter((i) => i.visibility !== "hidden" && i.type === "message").map((i) => i.data.content).join("\n")
+		}) };
+	},
+	function: async (agent) => {
+		if (agent.type !== "function") throw new Error(`Expected agent type 'function', but got '${agent.type}'`);
+		return { content: await formatCode(`\
+export default async function agent({${(agent.parameters?.map((i) => i.key).filter(Boolean) ?? []).join(", ")}}) {
+${agent.code}
+}
+
+agent.agent_name = ${JSON.stringify(agent.name || agent.id)};
+
+agent.description = ${agent.description ? JSON.stringify(agent.description) : "undefined"};
+
+agent.input_schema = ${JSON.stringify(convertInputSchema(agent))};
+
+agent.output_schema = ${JSON.stringify(convertOutputSchema(agent))};
+`) };
+	},
+	router: (agent, project) => {
+		if (agent.type !== "router") throw new Error(`Expected agent type 'router', but got '${agent.type}'`);
+		return { content: stringify({
+			name: agent.name || agent.id,
+			description: agent.description,
+			instructions: agent.prompt,
+			input_schema: convertInputSchema(agent),
+			output_schema: convertOutputSchema(agent),
+			skills: agent.routes?.map((i) => {
+				const tool = project.agents.find((j) => j.id === i.id);
+				if (!tool) throw new Error(`Tool ${i.id} not found in project definition`);
+				return getAgentFilename(tool);
+			}),
+			tool_choice: "router"
+		}) };
+	}
+};
+function getAgentFilename(agent) {
+	switch (agent.type) {
+		case "prompt":
+		case "router": return `${agent.name || agent.id}.yaml`;
+		case "function": return `${agent.name || agent.id}.js`;
+		default: throw new Error(`Unsupported agent type: ${agent.type}`);
+	}
+}
+async function formatCode(code) {
+	const [prettier, typescriptPlugin, estreePlugin] = await Promise.all([
+		import("prettier"),
+		import("prettier/plugins/typescript"),
+		import("prettier/plugins/estree")
+	]);
+	return prettier.format(code, {
+		parser: "typescript",
+		plugins: [typescriptPlugin, estreePlugin.default],
+		printWidth: 120,
+		useTabs: false,
+		tabWidth: 2,
+		trailingComma: "es5",
+		bracketSameLine: true,
+		semi: true,
+		singleQuote: true
+	});
+}
+function convertInputSchema(agent) {
+	const parameters = (agent.parameters ?? []).filter((i) => !!i.key && !i.hidden);
+	const properties = parameters.map((i) => [i.key, parameterToJsonSchema(i)]);
+	return {
+		type: "object",
+		properties: Object.fromEntries(properties),
+		required: parameters.filter((i) => i.required).map((i) => i.key)
+	};
+}
+function parameterToJsonSchema(parameter) {
+	switch (parameter.type) {
+		case void 0:
+		case "string":
+		case "language": return {
+			type: "string",
+			description: parameter.placeholder
+		};
+		case "select": return {
+			type: "string",
+			enum: parameter.options?.map((i) => i.value),
+			description: parameter.placeholder
+		};
+		case "number":
+		case "boolean": return {
+			type: parameter.type,
+			description: parameter.placeholder
+		};
+		default: throw new Error(`Unsupported parameter type: ${parameter.type}`);
+	}
+}
+function convertOutputSchema(agent) {
+	const outputs = agent.outputVariables?.filter((i) => !!i.name && !i.hidden) ?? [];
+	const properties = outputs.map((i) => [i.name, variableTypeToJsonSchema(i)]).filter((i) => !!i[1]);
+	return {
+		type: "object",
+		properties: Object.fromEntries(properties),
+		required: outputs.filter((i) => i.required).map((i) => i.name)
+	};
+}
+function variableTypeToJsonSchema(output) {
+	if (output.type === "object") {
+		if (!output.properties) return void 0;
+		return {
+			type: "object",
+			properties: Object.fromEntries(output.properties.filter((i) => !!i.name).map((i) => [i.name, variableTypeToJsonSchema(i)]).filter((i) => !!i[1])),
+			required: output.properties.filter((i) => i.required && i.name).map((i) => i.name)
+		};
+	}
+	if (output.type === "array") {
+		if (!output.element) return void 0;
+		return {
+			type: "array",
+			items: variableTypeToJsonSchema(output.element)
+		};
+	}
+	return {
+		type: output.type || "string",
+		description: output.description
+	};
+}
+
+//#endregion
+export { isV1Package, toAIGNEPackage };
+//# sourceMappingURL=agent-v1.mjs.map

package/dist/utils/agent-v1.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-v1.mjs","names":["path"],"sources":["../../src/utils/agent-v1.ts"],"sourcesContent":["import { readFile, stat, writeFile } from \"node:fs/promises\";\nimport { join } from \"node:path\";\nimport { glob } from \"glob\";\nimport { parse, stringify } from \"yaml\";\n\nexport async function isV1Package(src: string) {\n  return stat(join(src, \"project.yaml\"))\n    .then((res) => res.isFile())\n    .catch((error) => {\n      if (error.code === \"ENOENT\") return false;\n      throw error;\n    });\n}\n\nexport async function toAIGNEPackage(src: string, dst: string) {\n  const definition = await loadAgentV1Package(src);\n\n  const aigne = {\n    chat_model: {\n      name: \"gpt-4o-mini\", // TODO: get from config\n    },\n    agents: <string[]>[],\n    cli: {\n      agents: <string[]>[],\n    },\n  };\n\n  for (const agent of definition.agents) {\n    const { content } = await assistantToAIGNEV2(agent, definition);\n    const filename = getAgentFilename(agent);\n    await writeFile(join(dst, filename), content);\n    aigne.agents.push(filename);\n    aigne.cli.agents.push(filename);\n  }\n\n  await writeFile(join(dst, \"aigne.yaml\"), stringify(aigne));\n}\n\nasync function loadAgentV1Package(path: string) {\n  const agentFilePaths = await glob(\"prompts/**/*.yaml\", {\n    cwd: path,\n  });\n\n  const project: ProjectDefinitionV1[\"project\"] = parse(\n    await readFile(join(path, \"project.yaml\"), \"utf8\"),\n  );\n\n  const definition: ProjectDefinitionV1 = {\n    project,\n    agents: [],\n  };\n\n  for (const filename of agentFilePaths) {\n    const agent = parse(await readFile(join(path, filename), \"utf8\"));\n    definition.agents.push(agent);\n  }\n\n  return definition;\n}\n\nfunction assistantToAIGNEV2(agent: AgentV1, project: ProjectDefinitionV1) {\n  const converter = AGENT_MAP[agent.type];\n\n  if (!converter) throw new Error(`Unsupported agent type: ${agent.type}`);\n\n  return converter(agent, project);\n}\n\nconst AGENT_MAP: {\n  [key: string]: (\n    agent: AgentV1,\n    project: ProjectDefinitionV1,\n  ) => Promise<{ content: string }> | { content: string };\n} = {\n  prompt: (agent: AgentV1) => {\n    if (agent.type !== \"prompt\")\n      throw new Error(`Expected agent type 'prompt', but got '${agent.type}'`);\n\n    const obj = {\n      name: agent.name || agent.id,\n      description: agent.description,\n      input_schema: convertInputSchema(agent),\n      output_schema: convertOutputSchema(agent),\n      instructions: agent.prompts\n        ?.filter(\n          (i): i is Extract<Prompt, { type: \"message\" }> =>\n            i.visibility !== \"hidden\" && i.type === \"message\",\n        )\n        .map((i) => i.data.content) // TODO: should support multiple messages with different roles\n        .join(\"\\n\"),\n    };\n\n    return {\n      content: stringify(obj),\n    };\n  },\n  function: async (agent: AgentV1) => {\n    if (agent.type !== \"function\")\n      throw new Error(`Expected agent type 'function', but got '${agent.type}'`);\n\n    const inputNames = agent.parameters?.map((i) => i.key).filter(Boolean) ?? [];\n\n    return {\n      content: await formatCode(`\\\nexport default async function agent({${inputNames.join(\", \")}}) {\n${agent.code}\n}\n\nagent.agent_name = ${JSON.stringify(agent.name || agent.id)};\n\nagent.description = ${agent.description ? JSON.stringify(agent.description) : \"undefined\"};\n\nagent.input_schema = ${JSON.stringify(convertInputSchema(agent))};\n\nagent.output_schema = ${JSON.stringify(convertOutputSchema(agent))};\n`),\n    };\n  },\n  router: (agent: AgentV1, project: ProjectDefinitionV1) => {\n    if (agent.type !== \"router\")\n      throw new Error(`Expected agent type 'router', but got '${agent.type}'`);\n\n    return {\n      content: stringify({\n        name: agent.name || agent.id,\n        description: agent.description,\n        instructions: agent.prompt,\n        input_schema: convertInputSchema(agent),\n        output_schema: convertOutputSchema(agent),\n        skills: agent.routes?.map((i) => {\n          const tool = project.agents.find((j) => j.id === i.id);\n          if (!tool) throw new Error(`Tool ${i.id} not found in project definition`);\n          return getAgentFilename(tool);\n        }),\n        tool_choice: \"router\",\n      }),\n    };\n  },\n};\n\nfunction getAgentFilename(agent: AgentV1) {\n  switch (agent.type) {\n    case \"prompt\":\n    case \"router\":\n      return `${agent.name || agent.id}.yaml`;\n    case \"function\":\n      return `${agent.name || agent.id}.js`;\n    default:\n      throw new Error(`Unsupported agent type: ${agent.type}`);\n  }\n}\n\nasync function formatCode(code: string) {\n  const [prettier, typescriptPlugin, estreePlugin] = await Promise.all([\n    import(\"prettier\"),\n    import(\"prettier/plugins/typescript\"),\n    import(\"prettier/plugins/estree\"),\n  ]);\n\n  return prettier.format(code, {\n    parser: \"typescript\",\n    plugins: [typescriptPlugin, estreePlugin.default],\n    printWidth: 120,\n    useTabs: false,\n    tabWidth: 2,\n    trailingComma: \"es5\",\n    bracketSameLine: true,\n    semi: true,\n    singleQuote: true,\n  });\n}\n\nfunction convertInputSchema(agent: AgentV1) {\n  const parameters = (agent.parameters ?? []).filter((i) => !!i.key && !i.hidden);\n\n  const properties = parameters.map((i) => [i.key, parameterToJsonSchema(i)]);\n\n  return {\n    type: \"object\",\n    properties: Object.fromEntries(properties),\n    required: parameters.filter((i) => i.required).map((i) => i.key),\n  };\n}\n\nfunction parameterToJsonSchema(parameter: Parameter): object | undefined {\n  switch (parameter.type) {\n    case undefined:\n    case \"string\":\n    case \"language\":\n      return {\n        type: \"string\",\n        description: parameter.placeholder,\n      };\n    case \"select\":\n      return {\n        type: \"string\",\n        enum: parameter.options?.map((i) => i.value),\n        description: parameter.placeholder,\n      };\n    case \"number\":\n    case \"boolean\":\n      return {\n        type: parameter.type,\n        description: parameter.placeholder,\n      };\n    default:\n      throw new Error(`Unsupported parameter type: ${(parameter as Record<string, string>).type}`);\n  }\n}\n\nfunction convertOutputSchema(agent: AgentV1) {\n  const outputs = agent.outputVariables?.filter((i) => !!i.name && !i.hidden) ?? [];\n  const properties = outputs\n    .map((i) => [i.name, variableTypeToJsonSchema(i)])\n    .filter((i) => !!i[1]);\n  return {\n    type: \"object\",\n    properties: Object.fromEntries(properties),\n    required: outputs.filter((i) => i.required).map((i) => i.name),\n  };\n}\n\nfunction variableTypeToJsonSchema(output: OutputVariable): object | undefined {\n  if (output.type === \"object\") {\n    if (!output.properties) return undefined;\n\n    return {\n      type: \"object\",\n      properties: Object.fromEntries(\n        output.properties\n          .filter((i): i is typeof i & { name: string } => !!i.name)\n          .map((i) => [i.name, variableTypeToJsonSchema(i)])\n          .filter((i) => !!i[1]),\n      ),\n      required: output.properties.filter((i) => i.required && i.name).map((i) => i.name),\n    };\n  }\n\n  if (output.type === \"array\") {\n    if (!output.element) return undefined;\n\n    return {\n      type: \"array\",\n      items: variableTypeToJsonSchema(output.element),\n    };\n  }\n\n  return {\n    type: output.type || \"string\",\n    description: output.description,\n  };\n}\n\nexport type Parameter = {\n  id: string;\n  key?: string;\n  hidden?: boolean;\n  required?: boolean;\n  placeholder?: string;\n} & (\n  | {\n      type?: \"string\" | \"number\" | \"boolean\" | \"language\";\n    }\n  | {\n      type?: \"select\";\n      options?: {\n        id: string;\n        label?: string;\n        value?: string;\n      }[];\n    }\n);\n\nexport interface VariableTypeBase {\n  id: string;\n  name?: string;\n  description?: string;\n  required?: boolean;\n  hidden?: boolean;\n}\n\nexport type OutputVariable = VariableTypeBase &\n  (\n    | {\n        type?: undefined;\n      }\n    | {\n        type: \"string\";\n        defaultValue?: string;\n      }\n    | {\n        type: \"number\";\n        defaultValue?: number;\n      }\n    | {\n        type: \"boolean\";\n        defaultValue?: boolean;\n      }\n    | {\n        type: \"object\";\n        properties?: OutputVariable[];\n      }\n    | {\n        type: \"array\";\n        element?: OutputVariable;\n      }\n  );\n\nexport type AgentV1 = {\n  id: string;\n  name?: string;\n  description?: string;\n  parameters?: Parameter[];\n  outputVariables?: OutputVariable[];\n} & (\n  | {\n      type: \"prompt\";\n      prompts?: Prompt[];\n      temperature?: number;\n      topP?: number;\n      presencePenalty?: number;\n      frequencyPenalty?: number;\n      maxTokens?: number;\n      model?: string;\n    }\n  | {\n      type: \"router\";\n      defaultToolId?: string;\n      prompt?: string;\n      decisionType?: \"ai\";\n      routes?: Tool[];\n      temperature?: number;\n      topP?: number;\n      presencePenalty?: number;\n      frequencyPenalty?: number;\n      maxTokens?: number;\n      model?: string;\n    }\n  | {\n      type: \"image\";\n      prompt?: string;\n      model?: string;\n      n?: number;\n      quality?: string;\n      style?: string;\n      size?: string;\n      modelSettings?: {\n        [key: string]: unknown;\n      };\n    }\n  | {\n      type: \"api\";\n      requestParameters?: {\n        id: string;\n        key?: string;\n        value?: string;\n      }[];\n      requestMethod?: string;\n      requestUrl?: string;\n      requestHeaders: {\n        id: string;\n        key?: string;\n        value?: string;\n      }[];\n    }\n  | {\n      type: \"function\";\n      code?: string;\n    }\n  | {\n      type: \"callAgent\";\n      agents?: Tool[];\n    }\n);\n\nexport interface ProjectDefinitionV1 {\n  project: {\n    name?: string;\n    description?: string;\n  };\n  agents: AgentV1[];\n}\n\nexport type Prompt =\n  | {\n      type: \"message\";\n      data: PromptMessage;\n      visibility?: \"hidden\";\n    }\n  | {\n      type: \"executeBlock\";\n      visibility?: \"hidden\";\n    };\n\nexport type PromptMessage = {\n  id: string;\n  role: Role;\n  content?: string;\n  name?: string;\n};\n\nexport type Role = \"system\" | \"user\" | \"assistant\";\n\nexport type Tool = {\n  blockletDid?: string;\n  projectId?: string;\n  id: string;\n  from?: \"assistant\" | \"blockletAPI\" | \"knowledge\";\n  parameters?: {\n    [key: string]: unknown;\n  };\n  functionName?: string;\n};\n"],"mappings":";;;;;;AAKA,eAAsB,YAAY,KAAa;AAC7C,QAAO,KAAK,KAAK,KAAK,eAAe,CAAC,CACnC,MAAM,QAAQ,IAAI,QAAQ,CAAC,CAC3B,OAAO,UAAU;AAChB,MAAI,MAAM,SAAS,SAAU,QAAO;AACpC,QAAM;GACN;;AAGN,eAAsB,eAAe,KAAa,KAAa;CAC7D,MAAM,aAAa,MAAM,mBAAmB,IAAI;CAEhD,MAAM,QAAQ;EACZ,YAAY,EACV,MAAM,eACP;EACD,QAAkB,EAAE;EACpB,KAAK,EACH,QAAkB,EAAE,EACrB;EACF;AAED,MAAK,MAAM,SAAS,WAAW,QAAQ;EACrC,MAAM,EAAE,YAAY,MAAM,mBAAmB,OAAO,WAAW;EAC/D,MAAM,WAAW,iBAAiB,MAAM;AACxC,QAAM,UAAU,KAAK,KAAK,SAAS,EAAE,QAAQ;AAC7C,QAAM,OAAO,KAAK,SAAS;AAC3B,QAAM,IAAI,OAAO,KAAK,SAAS;;AAGjC,OAAM,UAAU,KAAK,KAAK,aAAa,EAAE,UAAU,MAAM,CAAC;;AAG5D,eAAe,mBAAmB,QAAc;CAC9C,MAAM,iBAAiB,MAAM,KAAK,qBAAqB,EACrD,KAAKA,QACN,CAAC;CAMF,MAAM,aAAkC;EACtC,SAL8C,MAC9C,MAAM,SAAS,KAAKA,QAAM,eAAe,EAAE,OAAO,CACnD;EAIC,QAAQ,EAAE;EACX;AAED,MAAK,MAAM,YAAY,gBAAgB;EACrC,MAAM,QAAQ,MAAM,MAAM,SAAS,KAAKA,QAAM,SAAS,EAAE,OAAO,CAAC;AACjE,aAAW,OAAO,KAAK,MAAM;;AAG/B,QAAO;;AAGT,SAAS,mBAAmB,OAAgB,SAA8B;CACxE,MAAM,YAAY,UAAU,MAAM;AAElC,KAAI,CAAC,UAAW,OAAM,IAAI,MAAM,2BAA2B,MAAM,OAAO;AAExE,QAAO,UAAU,OAAO,QAAQ;;AAGlC,MAAM,YAKF;CACF,SAAS,UAAmB;AAC1B,MAAI,MAAM,SAAS,SACjB,OAAM,IAAI,MAAM,0CAA0C,MAAM,KAAK,GAAG;AAgB1E,SAAO,EACL,SAAS,UAfC;GACV,MAAM,MAAM,QAAQ,MAAM;GAC1B,aAAa,MAAM;GACnB,cAAc,mBAAmB,MAAM;GACvC,eAAe,oBAAoB,MAAM;GACzC,cAAc,MAAM,SAChB,QACC,MACC,EAAE,eAAe,YAAY,EAAE,SAAS,UAC3C,CACA,KAAK,MAAM,EAAE,KAAK,QAAQ,CAC1B,KAAK,KAAK;GACd,CAGwB,EACxB;;CAEH,UAAU,OAAO,UAAmB;AAClC,MAAI,MAAM,SAAS,WACjB,OAAM,IAAI,MAAM,4CAA4C,MAAM,KAAK,GAAG;AAI5E,SAAO,EACL,SAAS,MAAM,WAAW;wCAHT,MAAM,YAAY,KAAK,MAAM,EAAE,IAAI,CAAC,OAAO,QAAQ,IAAI,EAAE,EAI9B,KAAK,KAAK,CAAC;EAC3D,MAAM,KAAK;;;qBAGQ,KAAK,UAAU,MAAM,QAAQ,MAAM,GAAG,CAAC;;sBAEtC,MAAM,cAAc,KAAK,UAAU,MAAM,YAAY,GAAG,YAAY;;uBAEnE,KAAK,UAAU,mBAAmB,MAAM,CAAC,CAAC;;wBAEzC,KAAK,UAAU,oBAAoB,MAAM,CAAC,CAAC;EACjE,EACG;;CAEH,SAAS,OAAgB,YAAiC;AACxD,MAAI,MAAM,SAAS,SACjB,OAAM,IAAI,MAAM,0CAA0C,MAAM,KAAK,GAAG;AAE1E,SAAO,EACL,SAAS,UAAU;GACjB,MAAM,MAAM,QAAQ,MAAM;GAC1B,aAAa,MAAM;GACnB,cAAc,MAAM;GACpB,cAAc,mBAAmB,MAAM;GACvC,eAAe,oBAAoB,MAAM;GACzC,QAAQ,MAAM,QAAQ,KAAK,MAAM;IAC/B,MAAM,OAAO,QAAQ,OAAO,MAAM,MAAM,EAAE,OAAO,EAAE,GAAG;AACtD,QAAI,CAAC,KAAM,OAAM,IAAI,MAAM,QAAQ,EAAE,GAAG,kCAAkC;AAC1E,WAAO,iBAAiB,KAAK;KAC7B;GACF,aAAa;GACd,CAAC,EACH;;CAEJ;AAED,SAAS,iBAAiB,OAAgB;AACxC,SAAQ,MAAM,MAAd;EACE,KAAK;EACL,KAAK,SACH,QAAO,GAAG,MAAM,QAAQ,MAAM,GAAG;EACnC,KAAK,WACH,QAAO,GAAG,MAAM,QAAQ,MAAM,GAAG;EACnC,QACE,OAAM,IAAI,MAAM,2BAA2B,MAAM,OAAO;;;AAI9D,eAAe,WAAW,MAAc;CACtC,MAAM,CAAC,UAAU,kBAAkB,gBAAgB,MAAM,QAAQ,IAAI;EACnE,OAAO;EACP,OAAO;EACP,OAAO;EACR,CAAC;AAEF,QAAO,SAAS,OAAO,MAAM;EAC3B,QAAQ;EACR,SAAS,CAAC,kBAAkB,aAAa,QAAQ;EACjD,YAAY;EACZ,SAAS;EACT,UAAU;EACV,eAAe;EACf,iBAAiB;EACjB,MAAM;EACN,aAAa;EACd,CAAC;;AAGJ,SAAS,mBAAmB,OAAgB;CAC1C,MAAM,cAAc,MAAM,cAAc,EAAE,EAAE,QAAQ,MAAM,CAAC,CAAC,EAAE,OAAO,CAAC,EAAE,OAAO;CAE/E,MAAM,aAAa,WAAW,KAAK,MAAM,CAAC,EAAE,KAAK,sBAAsB,EAAE,CAAC,CAAC;AAE3E,QAAO;EACL,MAAM;EACN,YAAY,OAAO,YAAY,WAAW;EAC1C,UAAU,WAAW,QAAQ,MAAM,EAAE,SAAS,CAAC,KAAK,MAAM,EAAE,IAAI;EACjE;;AAGH,SAAS,sBAAsB,WAA0C;AACvE,SAAQ,UAAU,MAAlB;EACE,KAAK;EACL,KAAK;EACL,KAAK,WACH,QAAO;GACL,MAAM;GACN,aAAa,UAAU;GACxB;EACH,KAAK,SACH,QAAO;GACL,MAAM;GACN,MAAM,UAAU,SAAS,KAAK,MAAM,EAAE,MAAM;GAC5C,aAAa,UAAU;GACxB;EACH,KAAK;EACL,KAAK,UACH,QAAO;GACL,MAAM,UAAU;GAChB,aAAa,UAAU;GACxB;EACH,QACE,OAAM,IAAI,MAAM,+BAAgC,UAAqC,OAAO;;;AAIlG,SAAS,oBAAoB,OAAgB;CAC3C,MAAM,UAAU,MAAM,iBAAiB,QAAQ,MAAM,CAAC,CAAC,EAAE,QAAQ,CAAC,EAAE,OAAO,IAAI,EAAE;CACjF,MAAM,aAAa,QAChB,KAAK,MAAM,CAAC,EAAE,MAAM,yBAAyB,EAAE,CAAC,CAAC,CACjD,QAAQ,MAAM,CAAC,CAAC,EAAE,GAAG;AACxB,QAAO;EACL,MAAM;EACN,YAAY,OAAO,YAAY,WAAW;EAC1C,UAAU,QAAQ,QAAQ,MAAM,EAAE,SAAS,CAAC,KAAK,MAAM,EAAE,KAAK;EAC/D;;AAGH,SAAS,yBAAyB,QAA4C;AAC5E,KAAI,OAAO,SAAS,UAAU;AAC5B,MAAI,CAAC,OAAO,WAAY,QAAO;AAE/B,SAAO;GACL,MAAM;GACN,YAAY,OAAO,YACjB,OAAO,WACJ,QAAQ,MAAwC,CAAC,CAAC,EAAE,KAAK,CACzD,KAAK,MAAM,CAAC,EAAE,MAAM,yBAAyB,EAAE,CAAC,CAAC,CACjD,QAAQ,MAAM,CAAC,CAAC,EAAE,GAAG,CACzB;GACD,UAAU,OAAO,WAAW,QAAQ,MAAM,EAAE,YAAY,EAAE,KAAK,CAAC,KAAK,MAAM,EAAE,KAAK;GACnF;;AAGH,KAAI,OAAO,SAAS,SAAS;AAC3B,MAAI,CAAC,OAAO,QAAS,QAAO;AAE5B,SAAO;GACL,MAAM;GACN,OAAO,yBAAyB,OAAO,QAAQ;GAChD;;AAGH,QAAO;EACL,MAAM,OAAO,QAAQ;EACrB,aAAa,OAAO;EACrB"}

package/dist/utils/aigne-hub/constants.cjs

@@ -0,0 +1,22 @@
+const require_rolldown_runtime = require('../../_virtual/rolldown_runtime.cjs');
+let node_path = require("node:path");
+let node_os = require("node:os");
+
+//#region src/utils/aigne-hub/constants.ts
+const ACCESS_KEY_PREFIX = "/api/access-key";
+const WELLKNOWN_SERVICE_PATH_PREFIX = "/.well-known/service";
+const ACCESS_KEY_SESSION_API = `${ACCESS_KEY_PREFIX}/session`;
+const DEFAULT_AIGNE_HUB_MODEL = "openai/gpt-5-mini";
+const AIGNE_HUB_PROVIDER = "aignehub";
+const DEFAULT_AIGNE_HUB_PROVIDER_MODEL = `${AIGNE_HUB_PROVIDER}:${DEFAULT_AIGNE_HUB_MODEL}`;
+const isTest = process.env.CI || process.env.NODE_ENV === "test";
+const TEST_AIGNE_ENV_FILE = (0, node_path.join)((0, node_os.homedir)(), ".aigne", "test-aigne-hub-connected.yaml");
+const PROD_AIGNE_ENV_FILE = (0, node_path.join)((0, node_os.homedir)(), ".aigne", "aigne-hub-connected.yaml");
+const AIGNE_ENV_FILE = isTest ? TEST_AIGNE_ENV_FILE : PROD_AIGNE_ENV_FILE;
+
+//#endregion
+exports.ACCESS_KEY_SESSION_API = ACCESS_KEY_SESSION_API;
+exports.AIGNE_ENV_FILE = AIGNE_ENV_FILE;
+exports.AIGNE_HUB_PROVIDER = AIGNE_HUB_PROVIDER;
+exports.WELLKNOWN_SERVICE_PATH_PREFIX = WELLKNOWN_SERVICE_PATH_PREFIX;
+exports.isTest = isTest;

package/dist/utils/aigne-hub/constants.mjs

@@ -0,0 +1,18 @@
+import { join } from "node:path";
+import { homedir } from "node:os";
+
+//#region src/utils/aigne-hub/constants.ts
+const ACCESS_KEY_PREFIX = "/api/access-key";
+const WELLKNOWN_SERVICE_PATH_PREFIX = "/.well-known/service";
+const ACCESS_KEY_SESSION_API = `${ACCESS_KEY_PREFIX}/session`;
+const DEFAULT_AIGNE_HUB_MODEL = "openai/gpt-5-mini";
+const AIGNE_HUB_PROVIDER = "aignehub";
+const DEFAULT_AIGNE_HUB_PROVIDER_MODEL = `${AIGNE_HUB_PROVIDER}:${DEFAULT_AIGNE_HUB_MODEL}`;
+const isTest = process.env.CI || process.env.NODE_ENV === "test";
+const TEST_AIGNE_ENV_FILE = join(homedir(), ".aigne", "test-aigne-hub-connected.yaml");
+const PROD_AIGNE_ENV_FILE = join(homedir(), ".aigne", "aigne-hub-connected.yaml");
+const AIGNE_ENV_FILE = isTest ? TEST_AIGNE_ENV_FILE : PROD_AIGNE_ENV_FILE;
+
+//#endregion
+export { ACCESS_KEY_SESSION_API, AIGNE_ENV_FILE, AIGNE_HUB_PROVIDER, WELLKNOWN_SERVICE_PATH_PREFIX, isTest };
+//# sourceMappingURL=constants.mjs.map

package/dist/utils/aigne-hub/constants.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.mjs","names":[],"sources":["../../../src/utils/aigne-hub/constants.ts"],"sourcesContent":["import { homedir } from \"node:os\";\nimport { join } from \"node:path\";\n\nconst ACCESS_KEY_PREFIX = \"/api/access-key\";\nexport const WELLKNOWN_SERVICE_PATH_PREFIX = \"/.well-known/service\";\nexport const ACCESS_KEY_SESSION_API = `${ACCESS_KEY_PREFIX}/session`;\n\nconst DEFAULT_AIGNE_HUB_MODEL = \"openai/gpt-5-mini\";\nexport const AIGNE_HUB_PROVIDER = \"aignehub\";\nexport const DEFAULT_AIGNE_HUB_PROVIDER_MODEL = `${AIGNE_HUB_PROVIDER}:${DEFAULT_AIGNE_HUB_MODEL}`;\n\nexport const isTest = process.env.CI || process.env.NODE_ENV === \"test\";\nconst TEST_AIGNE_ENV_FILE = join(homedir(), \".aigne\", \"test-aigne-hub-connected.yaml\");\nconst PROD_AIGNE_ENV_FILE = join(homedir(), \".aigne\", \"aigne-hub-connected.yaml\");\nexport const AIGNE_ENV_FILE = isTest ? TEST_AIGNE_ENV_FILE : PROD_AIGNE_ENV_FILE;\n"],"mappings":";;;;AAGA,MAAM,oBAAoB;AAC1B,MAAa,gCAAgC;AAC7C,MAAa,yBAAyB,GAAG,kBAAkB;AAE3D,MAAM,0BAA0B;AAChC,MAAa,qBAAqB;AAClC,MAAa,mCAAmC,GAAG,mBAAmB,GAAG;AAEzE,MAAa,SAAS,QAAQ,IAAI,MAAM,QAAQ,IAAI,aAAa;AACjE,MAAM,sBAAsB,KAAK,SAAS,EAAE,UAAU,gCAAgC;AACtF,MAAM,sBAAsB,KAAK,SAAS,EAAE,UAAU,2BAA2B;AACjF,MAAa,iBAAiB,SAAS,sBAAsB"}

package/dist/utils/aigne-hub/credential.cjs

@@ -0,0 +1,179 @@
+const require_rolldown_runtime = require('../../_virtual/rolldown_runtime.cjs');
+const require_constants = require('./constants.cjs');
+const require_crypto = require('./crypto.cjs');
+const require_index = require('./store/index.cjs');
+let node_fs = require("node:fs");
+let node_path = require("node:path");
+let _aigne_core_utils_fetch = require("@aigne/core/utils/fetch");
+let ufo = require("ufo");
+let _aigne_aigne_hub = require("@aigne/aigne-hub");
+let inquirer = require("inquirer");
+inquirer = require_rolldown_runtime.__toESM(inquirer);
+let node_os = require("node:os");
+let open = require("open");
+open = require_rolldown_runtime.__toESM(open);
+let p_wait_for = require("p-wait-for");
+p_wait_for = require_rolldown_runtime.__toESM(p_wait_for);
+
+//#region src/utils/aigne-hub/credential.ts
+const request = async (config) => {
+	const headers = {};
+	if (config.requestCount !== void 0) headers["X-Request-Count"] = config.requestCount.toString();
+	return { data: await (await (0, _aigne_core_utils_fetch.fetch)(config.url, {
+		method: config.method || "GET",
+		headers
+	})).json() };
+};
+const fetchConfigs = async ({ connectUrl, sessionId, fetchInterval, fetchTimeout }) => {
+	const sessionURL = (0, ufo.withQuery)((0, ufo.joinURL)(connectUrl, require_constants.ACCESS_KEY_SESSION_API), { sid: sessionId });
+	let requestCount = 0;
+	const condition = async () => {
+		const { data: session$1 } = await request({
+			url: sessionURL,
+			requestCount
+		});
+		requestCount++;
+		return Boolean(session$1.accessKeyId && session$1.accessKeySecret);
+	};
+	await (0, p_wait_for.default)(condition, {
+		interval: fetchInterval,
+		timeout: fetchTimeout
+	});
+	const { data: session } = await request({
+		url: sessionURL,
+		requestCount
+	});
+	await request({
+		url: sessionURL,
+		method: "DELETE"
+	});
+	return {
+		...session,
+		accessKeyId: session.accessKeyId,
+		accessKeySecret: require_crypto.decrypt(session.accessKeySecret, session.accessKeyId, session.challenge)
+	};
+};
+function baseWrapSpinner(_, waiting) {
+	return waiting();
+}
+async function createConnect({ connectUrl, openPage, fetchInterval = 3 * 1e3, retry = 1500, source = "Blocklet CLI", connectAction = "connect-cli", wrapSpinner = baseWrapSpinner, closeOnSuccess, intervalFetchConfig, appName = "AIGNE CLI", appLogo = "https://www.aigne.io/favicon.ico?imageFilter=resize&w=32" }) {
+	const { data: session } = await request({
+		url: (0, ufo.joinURL)(connectUrl, require_constants.ACCESS_KEY_SESSION_API),
+		method: "POST"
+	});
+	const token = session.id;
+	const pageUrl = (0, ufo.withQuery)((0, ufo.joinURL)(connectUrl, connectAction), {
+		__token__: require_crypto.encodeEncryptionKey(token),
+		source,
+		closeOnSuccess,
+		cli: true,
+		appName: ` ${appName}`,
+		appLogo
+	});
+	openPage?.(pageUrl);
+	return await wrapSpinner(`Waiting for connection: ${connectUrl}`, async () => {
+		return await (intervalFetchConfig ?? fetchConfigs)({
+			connectUrl,
+			sessionId: token,
+			fetchTimeout: retry * fetchInterval,
+			fetchInterval: retry
+		});
+	});
+}
+async function connectToAIGNEHub(url) {
+	const { origin } = new URL(url);
+	const connectUrl = (0, ufo.joinURL)(origin, require_constants.WELLKNOWN_SERVICE_PATH_PREFIX);
+	const apiUrl = await (0, _aigne_aigne_hub.getAIGNEHubMountPoint)(url, _aigne_aigne_hub.AIGNE_HUB_BLOCKLET_DID);
+	const openFn = require_constants.isTest ? () => {} : open.default;
+	const accessKeyOptions = {
+		apiKey: (await createConnect({
+			connectUrl,
+			connectAction: "gen-simple-access-key",
+			source: `@aigne/cli connect to AIGNE hub`,
+			closeOnSuccess: true,
+			openPage: (pageUrl) => openFn(pageUrl)
+		})).accessKeySecret,
+		url: apiUrl
+	};
+	const secretStore = await require_index.default();
+	await secretStore.setKey(accessKeyOptions.url, accessKeyOptions.apiKey);
+	await secretStore.setDefault(accessKeyOptions.url);
+	return accessKeyOptions;
+}
+const checkConnectionStatus = async (host) => {
+	const env = await (await require_index.default()).getKey(host);
+	if (!env?.AIGNE_HUB_API_KEY) throw new Error("AIGNE_HUB_API_KEY key not found, need to login first");
+	return {
+		apiKey: env.AIGNE_HUB_API_KEY,
+		url: env.AIGNE_HUB_API_URL
+	};
+};
+async function loadAIGNEHubCredential(options) {
+	if (process.env.BLOCKLET_AIGNE_API_URL && process.env.BLOCKLET_AIGNE_API_PROVIDER) return void 0;
+	const aigneDir = (0, node_path.join)((0, node_os.homedir)(), ".aigne");
+	if (!(0, node_fs.existsSync)(aigneDir)) (0, node_fs.mkdirSync)(aigneDir, { recursive: true });
+	const defaultHost = await (await require_index.default()).getDefault();
+	const inquirerPrompt = options?.inquirerPromptFn ?? inquirer.default.prompt;
+	const configUrl = options?.aigneHubUrl || process.env.AIGNE_HUB_API_URL;
+	const url = configUrl || defaultHost?.AIGNE_HUB_API_URL || _aigne_aigne_hub.AIGNE_HUB_URL;
+	const connectUrl = (0, ufo.joinURL)(new URL(url).origin, require_constants.WELLKNOWN_SERVICE_PATH_PREFIX);
+	const { host } = new URL(url);
+	let credential = {};
+	try {
+		if (process.env.AIGNE_HUB_API_KEY) credential = {
+			apiKey: process.env.AIGNE_HUB_API_KEY,
+			url: process.env.AIGNE_HUB_API_URL || url
+		};
+		else credential = await checkConnectionStatus(host);
+	} catch (error) {
+		if (error instanceof Error && error.message.includes("login first")) {
+			let aigneHubUrl = connectUrl;
+			if (!configUrl) {
+				const { subscribe } = await inquirerPrompt({
+					type: "list",
+					name: "subscribe",
+					message: "No LLM API Keys or AIGNE Hub connections found. How would you like to proceed?",
+					choices: [
+						{
+							name: "Connect to the Arcblock official AIGNE Hub (recommended, free credits for new users)",
+							value: "official"
+						},
+						connectUrl.includes(_aigne_aigne_hub.AIGNE_HUB_URL) ? {
+							name: "Connect to your own AIGNE Hub instance (self-hosted)",
+							value: "custom"
+						} : null,
+						{
+							name: "Exit and configure my own LLM API Keys",
+							value: "manual"
+						}
+					].filter(Boolean),
+					default: "official"
+				});
+				if (subscribe === "custom") {
+					const { customUrl } = await inquirerPrompt({
+						type: "input",
+						name: "customUrl",
+						message: "Enter the URL of your AIGNE Hub:",
+						validate(input) {
+							try {
+								return new URL(input).protocol.startsWith("http") ? true : "Must be a valid URL with http or https";
+							} catch {
+								return "Invalid URL";
+							}
+						}
+					});
+					aigneHubUrl = customUrl;
+				} else if (subscribe === "manual") {
+					console.log("You chose to configure your own LLM API Keys. Please follow the instructions in the documentation: https://www.arcblock.io/docs/aigne-framework/models-configuration-9dc03e");
+					process.exit(0);
+				}
+			}
+			credential = await connectToAIGNEHub(aigneHubUrl);
+		} else throw error;
+	}
+	return credential;
+}
+
+//#endregion
+exports.connectToAIGNEHub = connectToAIGNEHub;
+exports.loadAIGNEHubCredential = loadAIGNEHubCredential;