nolo-cli 0.1.13 → 0.1.14

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (320) hide show
  1. package/README.md +9 -2
  2. package/agent-runtime/hostAdapter.ts +53 -0
  3. package/agent-runtime/index.ts +28 -0
  4. package/agent-runtime/localLoop.ts +62 -0
  5. package/agent-runtime/runtimeDecision.ts +70 -0
  6. package/agent-runtime/types.ts +87 -0
  7. package/agentRuntimeCommands.ts +139 -22
  8. package/agentRuntimeLocal.ts +7 -0
  9. package/ai/agent/_executeModel.ts +118 -0
  10. package/ai/agent/agentSlice.ts +544 -1
  11. package/ai/agent/appWorkingMemory.ts +126 -0
  12. package/ai/agent/avatarUtils.ts +24 -0
  13. package/ai/agent/buildEditingContext.ts +373 -0
  14. package/ai/agent/buildSystemPrompt.ts +532 -0
  15. package/ai/agent/cleanAgentMessages.ts +140 -0
  16. package/ai/agent/cliChatClient.ts +119 -0
  17. package/ai/agent/contextCompiler.ts +107 -0
  18. package/ai/agent/contextLayerContract.ts +44 -0
  19. package/ai/agent/createAgentSchema.ts +234 -0
  20. package/ai/agent/executeToolCall.ts +58 -0
  21. package/ai/agent/fetchAgentContexts.ts +42 -0
  22. package/ai/agent/generatePrompt.ts +3 -0
  23. package/ai/agent/getFullChatContextKeys.ts +168 -0
  24. package/ai/agent/hooks/fetchPublicAgents.ts +133 -0
  25. package/ai/agent/hooks/useAgentConfig.ts +61 -0
  26. package/ai/agent/hooks/useAgentDialog.ts +35 -0
  27. package/ai/agent/hooks/useAgentFormValidation.ts +202 -0
  28. package/ai/agent/hooks/usePublicAgents.ts +473 -0
  29. package/ai/agent/persistMessageWithFixedId.ts +37 -0
  30. package/ai/agent/planSlice.ts +259 -0
  31. package/ai/agent/referenceUtils.ts +229 -0
  32. package/ai/agent/runAgentBackground.ts +238 -0
  33. package/ai/agent/runAgentClientLoop.ts +138 -0
  34. package/ai/agent/runtimeGuidance.ts +97 -0
  35. package/ai/agent/runtimeServerBase.ts +37 -0
  36. package/ai/agent/server/fetchPublicAgents.ts +128 -0
  37. package/ai/agent/startParallelAgentStreams.ts +424 -0
  38. package/ai/agent/startupProtocol.ts +53 -0
  39. package/ai/agent/streamAgentChatTurn.ts +1299 -0
  40. package/ai/agent/streamAgentChatTurnUtils.ts +738 -0
  41. package/ai/agent/types.ts +71 -0
  42. package/ai/agent/utils/imageOutput.ts +39 -0
  43. package/ai/agent/utils/publicImageAgentMode.ts +26 -0
  44. package/ai/agent/utils/sortUtils.ts +250 -0
  45. package/ai/agent/web/referencePickerUtils.ts +146 -0
  46. package/ai/ai.locale.ts +1083 -0
  47. package/ai/chat/accumulateToolCallChunks.ts +95 -0
  48. package/ai/chat/fetchUtils.native.ts +276 -0
  49. package/ai/chat/fetchUtils.ts +153 -0
  50. package/ai/chat/inlineImageUrlsForCustomProvider.ts +117 -0
  51. package/ai/chat/parseApiError.ts +64 -0
  52. package/ai/chat/parseMultilineSSE.ts +95 -0
  53. package/ai/chat/sendOpenAICompletionsRequest.native.ts +682 -0
  54. package/ai/chat/sendOpenAICompletionsRequest.ts +712 -0
  55. package/ai/chat/sendOpenAIResponseRequest.ts +512 -0
  56. package/ai/chat/shouldUseServerProxy.ts +18 -0
  57. package/ai/chat/sseClient.native.ts +91 -0
  58. package/ai/chat/sseClient.ts +67 -0
  59. package/ai/chat/streamReader.native.ts +31 -0
  60. package/ai/chat/streamReader.ts +62 -0
  61. package/ai/chat/updateTotalUsage.ts +72 -0
  62. package/ai/context/buildReferenceContext.ts +437 -0
  63. package/ai/context/calculateContextUsage.ts +133 -0
  64. package/ai/context/retention.ts +165 -0
  65. package/ai/context/tokenUtils.ts +78 -0
  66. package/ai/index.ts +1 -1
  67. package/ai/llm/agentCapabilities.ts +74 -0
  68. package/ai/llm/calculateGeminiImageTokens.ts +57 -0
  69. package/ai/llm/deepinfra.ts +28 -0
  70. package/ai/llm/fireworks.ts +68 -0
  71. package/ai/llm/generateRequestBody.ts +165 -0
  72. package/ai/llm/getModelContextWindow.ts +84 -0
  73. package/ai/llm/getNoloKey.ts +37 -0
  74. package/ai/llm/getPricing.ts +232 -0
  75. package/ai/llm/hooks/useModelPricing.ts +75 -0
  76. package/ai/llm/imagePricing.ts +66 -0
  77. package/ai/llm/isResponseAPIModel.ts +13 -0
  78. package/ai/llm/kimi.ts +18 -0
  79. package/ai/llm/mimo.ts +71 -0
  80. package/ai/llm/mistral.ts +22 -0
  81. package/ai/llm/modelAvatar.ts +427 -0
  82. package/ai/llm/models.ts +45 -0
  83. package/ai/llm/openrouterModels.ts +141 -0
  84. package/ai/llm/providers.ts +307 -0
  85. package/ai/llm/reasoningModels.ts +28 -0
  86. package/ai/llm/types.ts +59 -0
  87. package/ai/llm/usageRequestOptions.ts +59 -0
  88. package/ai/memory/capture.ts +148 -0
  89. package/ai/memory/consolidate.ts +104 -0
  90. package/ai/memory/delete.ts +147 -0
  91. package/ai/memory/overlay.ts +84 -0
  92. package/ai/memory/query.ts +38 -0
  93. package/ai/memory/queryShared.ts +160 -0
  94. package/ai/memory/rank.ts +105 -0
  95. package/ai/memory/recentRelationshipRecap.ts +247 -0
  96. package/ai/memory/remember.ts +167 -0
  97. package/ai/memory/runtime.ts +76 -0
  98. package/ai/memory/store.ts +20 -0
  99. package/ai/memory/storeShared.ts +76 -0
  100. package/ai/memory/types.ts +46 -0
  101. package/ai/memory/understanding.ts +349 -0
  102. package/ai/memory/understandingGreeting.ts +264 -0
  103. package/ai/messages/type.ts +20 -0
  104. package/ai/policy/personalizationDialog.ts +333 -0
  105. package/ai/policy/runtimePolicy.ts +440 -0
  106. package/ai/policy/selfUpdateFields.ts +48 -0
  107. package/ai/policy/types.ts +64 -0
  108. package/ai/skills/referenceRuntime.ts +274 -0
  109. package/ai/skills/skillDiagnostics.ts +251 -0
  110. package/ai/skills/skillDocBuilder.ts +139 -0
  111. package/ai/skills/skillDocProtocol.ts +434 -0
  112. package/ai/skills/skillReferenceSummary.ts +63 -0
  113. package/ai/skills/skillSummaryMarker.ts +26 -0
  114. package/ai/token/calculatePrice.ts +546 -0
  115. package/ai/token/db.ts +98 -0
  116. package/ai/token/externalToolCost.ts +321 -0
  117. package/ai/token/hooks/useRecords.ts +65 -0
  118. package/ai/token/missingUsageEstimate.ts +42 -0
  119. package/ai/token/modelUsageQuery.ts +252 -0
  120. package/ai/token/normalizeUsage.ts +84 -0
  121. package/ai/token/openaiImageGenerationUsage.ts +56 -0
  122. package/ai/token/prepareTokenUsageData.ts +88 -0
  123. package/ai/token/query.ts +88 -0
  124. package/ai/token/queryUserTokens.ts +59 -0
  125. package/ai/token/resolveBillingTarget.ts +52 -0
  126. package/ai/token/saveTokenRecord.ts +53 -0
  127. package/ai/token/serverDialogProjection.ts +78 -0
  128. package/ai/token/serverTokenWriter.ts +143 -0
  129. package/ai/token/stats.ts +21 -0
  130. package/ai/token/tokenThunks.ts +24 -0
  131. package/ai/token/types.ts +93 -0
  132. package/ai/tools/agent/agentTools.ts +176 -0
  133. package/ai/tools/agent/agentUpdateShared.ts +311 -0
  134. package/ai/tools/agent/callAgentTool.ts +139 -0
  135. package/ai/tools/agent/createAgentTool.ts +512 -0
  136. package/ai/tools/agent/createDialogTool.ts +69 -0
  137. package/ai/tools/agent/createSkillAgentTool.ts +62 -0
  138. package/ai/tools/agent/parallelBudget.ts +221 -0
  139. package/ai/tools/agent/presets/appBuilderPreset.ts +147 -0
  140. package/ai/tools/agent/runLlmTool.ts +96 -0
  141. package/ai/tools/agent/runStreamingAgentTool.ts +73 -0
  142. package/ai/tools/agent/skillAgentArgs.ts +106 -0
  143. package/ai/tools/agent/skillAgentPreset.ts +89 -0
  144. package/ai/tools/agent/streamParallelAgentsTool.ts +122 -0
  145. package/ai/tools/agent/updateAgentTool.ts +96 -0
  146. package/ai/tools/agent/updateSelfTool.ts +113 -0
  147. package/ai/tools/amazonProductScraperTool.ts +86 -0
  148. package/ai/tools/apifyActorClient.ts +45 -0
  149. package/ai/tools/appEditGuard.ts +372 -0
  150. package/ai/tools/appReadSnapshot.ts +153 -0
  151. package/ai/tools/appTools.ts +1549 -0
  152. package/ai/tools/applyEditTool.ts +256 -0
  153. package/ai/tools/applyLineEditsTool.ts +312 -0
  154. package/ai/tools/browserTools/click.ts +33 -0
  155. package/ai/tools/browserTools/closeSession.ts +29 -0
  156. package/ai/tools/browserTools/common.ts +27 -0
  157. package/ai/tools/browserTools/openSession.ts +48 -0
  158. package/ai/tools/browserTools/readContent.ts +38 -0
  159. package/ai/tools/browserTools/selectOption.ts +46 -0
  160. package/ai/tools/browserTools/typeText.ts +42 -0
  161. package/ai/tools/category/createCategoryTool.ts +66 -0
  162. package/ai/tools/category/queryContentsByCategoryTool.ts +69 -0
  163. package/ai/tools/category/updateContentCategoryTool.ts +75 -0
  164. package/ai/tools/cfBrowserTools.ts +319 -0
  165. package/ai/tools/cfSpeechToTextTool.ts +49 -0
  166. package/ai/tools/checkEnvTool.ts +65 -0
  167. package/ai/tools/cloudflareCrawlTool.ts +289 -0
  168. package/ai/tools/codeSearchTool.ts +111 -0
  169. package/ai/tools/codeTools.ts +101 -0
  170. package/ai/tools/createDocTool.ts +132 -0
  171. package/ai/tools/createPlanTool.ts +999 -0
  172. package/ai/tools/createSkillDocTool.ts +155 -0
  173. package/ai/tools/createWorkflowTool.ts +154 -0
  174. package/ai/tools/deepseekOcrTool.ts +34 -0
  175. package/ai/tools/delayTool.ts +31 -0
  176. package/ai/tools/deleteSpacesTool.ts +325 -0
  177. package/ai/tools/deleteSpacesToolModel.ts +159 -0
  178. package/ai/tools/devReloadUtils.ts +29 -0
  179. package/ai/tools/dialogMessageSearch.ts +137 -0
  180. package/ai/tools/doctorSkillTool.ts +72 -0
  181. package/ai/tools/ecommerceScraperTool.ts +86 -0
  182. package/ai/tools/emailTools.ts +549 -0
  183. package/ai/tools/evalSkillTool.ts +92 -0
  184. package/ai/tools/exaSearchTool.ts +64 -0
  185. package/ai/tools/execBashTool.ts +379 -0
  186. package/ai/tools/executeSqlTool.ts +192 -0
  187. package/ai/tools/fetchWebpageSupport.ts +309 -0
  188. package/ai/tools/fetchWebpageTool.ts +84 -0
  189. package/ai/tools/geminiImagePreviewTool.ts +361 -0
  190. package/ai/tools/generateDocxTool.ts +215 -0
  191. package/ai/tools/googleSearchScraperTool.ts +106 -0
  192. package/ai/tools/importDataTool.ts +133 -0
  193. package/ai/tools/importSkillTool.ts +162 -0
  194. package/ai/tools/index.ts +1927 -0
  195. package/ai/tools/listFilesTool.ts +82 -0
  196. package/ai/tools/listUserSpacesTool.ts +113 -0
  197. package/ai/tools/modelUsageTools.ts +199 -0
  198. package/ai/tools/olmOcrTool.ts +34 -0
  199. package/ai/tools/openaiImageTool.ts +267 -0
  200. package/ai/tools/prepareTools.ts +23 -0
  201. package/ai/tools/readDocTool.ts +84 -0
  202. package/ai/tools/readFileTool.ts +211 -0
  203. package/ai/tools/readTool.ts +163 -0
  204. package/ai/tools/readXPostTool.ts +233 -0
  205. package/ai/tools/rememberMemoryTool.ts +84 -0
  206. package/ai/tools/remotionVideoTool.ts +151 -0
  207. package/ai/tools/searchDialogMessagesTool.ts +222 -0
  208. package/ai/tools/searchRepoTool.ts +115 -0
  209. package/ai/tools/searchWorkspaceTool.ts +259 -0
  210. package/ai/tools/skillFollowup.ts +86 -0
  211. package/ai/tools/surfWeatherTool.ts +169 -0
  212. package/ai/tools/table/addTableRowTool.ts +217 -0
  213. package/ai/tools/table/createTableTool.ts +315 -0
  214. package/ai/tools/table/rowTools.ts +366 -0
  215. package/ai/tools/table/schemaTools.ts +244 -0
  216. package/ai/tools/table/shareTableTool.ts +148 -0
  217. package/ai/tools/table/toolShared.ts +129 -0
  218. package/ai/tools/toolApiClient.ts +198 -0
  219. package/ai/tools/toolNameAliases.ts +57 -0
  220. package/ai/tools/toolResultError.ts +42 -0
  221. package/ai/tools/toolRunSlice.ts +303 -0
  222. package/ai/tools/toolSchemaCompatibility.ts +53 -0
  223. package/ai/tools/toolVisibility.ts +4 -0
  224. package/ai/tools/types.ts +20 -0
  225. package/ai/tools/uiAskChoiceTool.ts +104 -0
  226. package/ai/tools/updateContentTitleTool.ts +84 -0
  227. package/ai/tools/updateDocTool.ts +105 -0
  228. package/ai/tools/updateUserPreferenceProfileTool.ts +145 -0
  229. package/ai/tools/whisperTool.ts +77 -0
  230. package/ai/tools/writeFileTool.ts +210 -0
  231. package/ai/tools/youtubeScraperTool.ts +116 -0
  232. package/ai/tools/ziweiChartTool.ts +678 -0
  233. package/ai/types.ts +55 -0
  234. package/ai/workflow/workflowExecutor.ts +323 -0
  235. package/ai/workflow/workflowSlice.ts +73 -0
  236. package/ai/workflow/workflowTypes.ts +106 -0
  237. package/client/agentRun.test.ts +240 -0
  238. package/client/agentRun.ts +182 -19
  239. package/client/compactDialog.test.ts +238 -0
  240. package/client/localRuntimeAdapter.test.ts +135 -0
  241. package/client/localRuntimeAdapter.ts +244 -0
  242. package/client/profileConfig.test.ts +40 -0
  243. package/client/streamingOutput.test.ts +22 -0
  244. package/client/streamingOutput.ts +38 -0
  245. package/commandRegistry.ts +9 -2
  246. package/connector-experimental/index.ts +5 -0
  247. package/database/actions/cacheMergedUserData.ts +64 -0
  248. package/database/actions/common.ts +242 -0
  249. package/database/actions/deleteFile.ts +40 -0
  250. package/database/actions/fetchUserData.ts +16 -0
  251. package/database/actions/fileContent.ts +125 -0
  252. package/database/actions/patch.ts +155 -0
  253. package/database/actions/read.ts +337 -0
  254. package/database/actions/readAndWait.ts +224 -0
  255. package/database/actions/readRequestManager.ts +120 -0
  256. package/database/actions/remove.ts +94 -0
  257. package/database/actions/replication.ts +366 -0
  258. package/database/actions/upload.ts +174 -0
  259. package/database/actions/upsert.ts +56 -0
  260. package/database/actions/write.ts +126 -0
  261. package/database/client/db.native.ts +73 -0
  262. package/database/client/db.ts +51 -0
  263. package/database/client/fetchUserData.ts +61 -0
  264. package/database/client/handleError.ts +19 -0
  265. package/database/client/queryRequest.ts +21 -0
  266. package/database/config.ts +21 -0
  267. package/database/dbActionThunks.ts +1 -0
  268. package/database/dbSlice.ts +149 -0
  269. package/database/email.ts +42 -0
  270. package/database/fileRing.ts +51 -0
  271. package/database/fileSharding.ts +70 -0
  272. package/database/fileStorage.native.ts +92 -0
  273. package/database/fileStorage.ts +232 -0
  274. package/database/fileUrl.ts +34 -0
  275. package/database/hooks/useUserData.ts +489 -0
  276. package/database/index.ts +1 -0
  277. package/database/keys.ts +765 -0
  278. package/database/queryPrefixes.ts +14 -0
  279. package/database/requests.ts +443 -0
  280. package/database/runtimeServerContext.ts +35 -0
  281. package/database/server/MemoryDB.ts +76 -0
  282. package/database/server/actorAccess.ts +76 -0
  283. package/database/server/agentDelegation.ts +124 -0
  284. package/database/server/coreDataOwnership.ts +13 -0
  285. package/database/server/coreDataProxy.ts +76 -0
  286. package/database/server/cybotReadonly.ts +18 -0
  287. package/database/server/dataHandlers.ts +111 -0
  288. package/database/server/db.ts +118 -0
  289. package/database/server/dbPath.ts +20 -0
  290. package/database/server/delete.ts +499 -0
  291. package/database/server/emailRepository.ts +1480 -0
  292. package/database/server/ensureDbOpen.ts +12 -0
  293. package/database/server/fileRead.ts +337 -0
  294. package/database/server/fileService.ts +436 -0
  295. package/database/server/handleTransaction.ts +86 -0
  296. package/database/server/patch.ts +282 -0
  297. package/database/server/query.ts +138 -0
  298. package/database/server/read.ts +325 -0
  299. package/database/server/resourceAccess.ts +211 -0
  300. package/database/server/routes.ts +110 -0
  301. package/database/server/spaceMemberAuthority.ts +67 -0
  302. package/database/server/upload.ts +159 -0
  303. package/database/server/write.ts +494 -0
  304. package/database/server/writeAuthority.ts +133 -0
  305. package/database/sqliteDb.ts +46 -0
  306. package/database/table/deleteTable.ts +120 -0
  307. package/database/tenantPlacement.ts +57 -0
  308. package/database/tombstones.ts +52 -0
  309. package/database/userDataLoadDecision.ts +17 -0
  310. package/database/userDataMerge.ts +95 -0
  311. package/database/userPreferenceRegister.ts +108 -0
  312. package/database/utils/dbPath.ts +47 -0
  313. package/database/utils/ulid.native.ts +6 -0
  314. package/database/utils/ulid.ts +1 -0
  315. package/index.ts +25 -15
  316. package/localRuntimeDb.ts +28 -0
  317. package/package.json +16 -4
  318. package/runtimeModeArgs.ts +33 -0
  319. package/tui/readlineWorkspace.ts +1 -0
  320. package/tui/session.ts +22 -0
@@ -0,0 +1,71 @@
1
+ // 文件路径: ai/agent/types.ts
2
+
3
+ export interface AgentRuntimeOptions {
4
+ parallelSessionId?: string;
5
+ parallelBranchId?: string;
6
+ parallelLabel?: string;
7
+ parallelIndex?: number;
8
+
9
+ /**
10
+ * 在本次调用中额外开放给 LLM 的工具名称。
11
+ * 会与 Agent 本身的 tools 数组合并后再传给 LLM。
12
+ * 名字要和 toolRegistry 中的 schema.name 对齐,例如 "addTableRow"。
13
+ */
14
+ extraTools?: string[];
15
+
16
+ /**
17
+ * 当前调用关注的“编辑目标”,用于后续构造 editingContext。
18
+ * 这里只定义接口,不涉及任何持久化或具体实现。
19
+ */
20
+ editingTarget?: {
21
+ /**
22
+ * 语义标签,比如: "table" | "page" | "article" | "code" 等。
23
+ * 未来可以扩展,但不需要改 Prompt 结构。
24
+ */
25
+ kind: string;
26
+ /**
27
+ * 可选:具体对象的 key,例如表的 dbKey、页面的 pageKey。
28
+ * 如果不传,后面可以从全局 state 中推断。
29
+ */
30
+ key?: string;
31
+ /**
32
+ * 可选:用于补充当前编辑对象的人类可读标题。
33
+ * 适用于对象不在本地 Redux 中、但调用方已知元数据的场景。
34
+ */
35
+ title?: string;
36
+ /**
37
+ * 可选:调用方预构造的一段简短说明,供 prompt 构造时使用。
38
+ */
39
+ summary?: string;
40
+ /**
41
+ * 可选:与当前编辑目标相关的轻量元数据。
42
+ * 例如应用的 framework、url、fileNames 等。
43
+ */
44
+ metadata?: Record<string, unknown>;
45
+ };
46
+
47
+ /**
48
+ * 本轮调用的图片生成配置,仅作用于这一轮请求。
49
+ * 会在调用前与 Agent / Dialog 的默认配置合并后传给 LLM。
50
+ */
51
+ imageConfigOverride?: {
52
+ /** 是否在本轮启用图片输出(不写或 true 表示按默认;false 可强制关闭) */
53
+ enabled?: boolean;
54
+
55
+ /** 本轮期望的宽高比,不指定则使用上层(Dialog / Agent / 模型)的默认值 */
56
+ aspectRatio?:
57
+ | "1:1"
58
+ | "2:3"
59
+ | "3:2"
60
+ | "3:4"
61
+ | "4:3"
62
+ | "4:5"
63
+ | "5:4"
64
+ | "9:16"
65
+ | "16:9"
66
+ | "21:9";
67
+
68
+ /** 本轮期望的分辨率大小,不指定则使用上层默认值 */
69
+ imageSize?: "1K" | "2K" | "4K";
70
+ };
71
+ }
@@ -0,0 +1,39 @@
1
+ import type { Agent } from "app/types";
2
+ import { getModelInfo } from "ai/llm/getModelContextWindow";
3
+ import { getPublicImageAgentMode } from "./publicImageAgentMode";
4
+
5
+ const LEGACY_IMAGE_MODELS = new Set([
6
+ "google/gemini-3-image-preview",
7
+ "google/gemini-3-pro-image-preview",
8
+ "google/gemini-3.1-flash-image-preview",
9
+ "google/gemini-2.5-flash-image",
10
+ ]);
11
+
12
+ export function supportsImageGeneration(agent: Agent): boolean {
13
+ if ((agent as any).hasImageOutput === true) return true;
14
+
15
+ // Move continuous-agent check before model registry early return
16
+ if (getPublicImageAgentMode(agent as any) === "continuous") return true;
17
+
18
+ const modelName = String(agent.model || "").trim();
19
+ if (!modelName) return false;
20
+
21
+ const modelInfo = getModelInfo(modelName);
22
+ if (modelInfo) {
23
+ return !!(modelInfo.hasImageOutput ?? (modelInfo as any).supportsImageOutput);
24
+ }
25
+
26
+ const normalized = modelName.toLowerCase();
27
+ if (LEGACY_IMAGE_MODELS.has(normalized)) return true;
28
+
29
+ if ((agent as any).imageConfig?.enabled && agent.hasVision) return true;
30
+
31
+ if (getPublicImageAgentMode(agent as any) === "continuous") return true;
32
+
33
+ return (
34
+ normalized.includes("image-preview") ||
35
+ normalized.includes("flash-image") ||
36
+ normalized.includes("seedream") ||
37
+ normalized.includes("flux.")
38
+ );
39
+ }
@@ -0,0 +1,26 @@
1
+ import type { PublicImageAgentMode } from "app/types";
2
+
3
+ export type { PublicImageAgentMode };
4
+
5
+ export const getPublicImageAgentMode = (
6
+ agent?: Partial<{
7
+ imageWorkflow: PublicImageAgentMode;
8
+ provider: string;
9
+ model: string;
10
+ imageConfig: { enabled?: boolean } | null;
11
+ }>
12
+ ): PublicImageAgentMode | null => {
13
+ if (!agent) return null;
14
+ if (agent.imageWorkflow) return agent.imageWorkflow;
15
+ return null;
16
+ };
17
+
18
+ export const getPublicImageAgentDefaultProfile = (mode: PublicImageAgentMode) => {
19
+ if (mode === "generate") {
20
+ return { quality: "medium", size: "1024x1024", outputFormat: "png" } as const;
21
+ }
22
+ if (mode === "edit") {
23
+ return { quality: "medium", size: "auto", outputFormat: "png" } as const;
24
+ }
25
+ return { quality: "low", size: "auto", outputFormat: "png" } as const;
26
+ };
@@ -0,0 +1,250 @@
1
+ // packages/ai/agent/utils/sortUtils.ts
2
+ import type { Agent } from "app/types";
3
+
4
+ export interface SortMeta {
5
+ createdAtMs: number;
6
+ publishedAtMs: number;
7
+ outputPriceNum: number;
8
+ useCount: number;
9
+ rating: number;
10
+ favoriteCount: number;
11
+ completenessScore: number;
12
+ recommendedScore: number;
13
+ }
14
+
15
+ export type MAgent = Agent & { __sort?: SortMeta };
16
+
17
+ export const RECOMMENDED_COMPLETENESS_WEIGHTS = {
18
+ name: 0.1,
19
+ introduction: 0.45,
20
+ cover: 0.2,
21
+ tags: 0.15,
22
+ tools: 0.1,
23
+ greeting: 0.1,
24
+ } as const;
25
+
26
+ export const RECOMMENDED_FRESHNESS_BUCKETS = [
27
+ { maxAgeDays: 3, score: 1 },
28
+ { maxAgeDays: 7, score: 0.7 },
29
+ { maxAgeDays: 21, score: 0.35 },
30
+ { maxAgeDays: 45, score: 0.15 },
31
+ ] as const;
32
+
33
+ export const RECOMMENDED_COLD_START_BUCKETS = [
34
+ { maxFavoriteCount: 0, maxAgeDays: 7, score: 0.75 },
35
+ { maxFavoriteCount: 2, maxAgeDays: 14, score: 0.35 },
36
+ ] as const;
37
+
38
+ export const RECOMMENDED_SCORE_WEIGHTS = {
39
+ favoriteLogWeight: 4,
40
+ freshnessWeight: 2,
41
+ completenessWeight: 1.5,
42
+ } as const;
43
+
44
+ export function toNumber(n: unknown, fallback: number) {
45
+ const v = typeof n === "number" ? n : parseFloat(String(n));
46
+ return Number.isFinite(v) ? v : fallback;
47
+ }
48
+
49
+ export function toTimeMs(t: unknown) {
50
+ if (typeof t === "number") return t;
51
+ const n = Date.parse(String(t ?? 0));
52
+ return Number.isFinite(n) ? n : 0;
53
+ }
54
+
55
+ export function firstFiniteNumber(...values: unknown[]): number | undefined {
56
+ for (const value of values) {
57
+ const parsed = typeof value === "number" ? value : parseFloat(String(value));
58
+ if (Number.isFinite(parsed)) return parsed;
59
+ }
60
+ return undefined;
61
+ }
62
+
63
+ function hasNonEmptyText(value: unknown) {
64
+ return typeof value === "string" && value.trim().length > 0;
65
+ }
66
+
67
+ function getCompletenessScore(agent?: Agent) {
68
+ if (!agent) return 0;
69
+
70
+ let score = 0;
71
+ if (hasNonEmptyText(agent.name)) score += RECOMMENDED_COMPLETENESS_WEIGHTS.name;
72
+ if (hasNonEmptyText(agent.introduction)) {
73
+ score += RECOMMENDED_COMPLETENESS_WEIGHTS.introduction;
74
+ }
75
+ if (hasNonEmptyText(agent.cover)) score += RECOMMENDED_COMPLETENESS_WEIGHTS.cover;
76
+ if (Array.isArray(agent.tags) && agent.tags.length > 0) {
77
+ score += RECOMMENDED_COMPLETENESS_WEIGHTS.tags;
78
+ }
79
+ if (Array.isArray(agent.tools) && agent.tools.length > 0) {
80
+ score += RECOMMENDED_COMPLETENESS_WEIGHTS.tools;
81
+ }
82
+ if (
83
+ hasNonEmptyText(agent.greeting) ||
84
+ (agent.greeting &&
85
+ typeof agent.greeting === "object" &&
86
+ hasNonEmptyText((agent.greeting as { text?: string }).text))
87
+ ) {
88
+ score += RECOMMENDED_COMPLETENESS_WEIGHTS.greeting;
89
+ }
90
+ return score;
91
+ }
92
+
93
+ function getFreshnessScore(ageDays: number) {
94
+ for (const bucket of RECOMMENDED_FRESHNESS_BUCKETS) {
95
+ if (ageDays <= bucket.maxAgeDays) return bucket.score;
96
+ }
97
+ return 0;
98
+ }
99
+
100
+ function getColdStartBoost(ageDays: number, favoriteCount: number) {
101
+ for (const bucket of RECOMMENDED_COLD_START_BUCKETS) {
102
+ if (favoriteCount <= bucket.maxFavoriteCount && ageDays <= bucket.maxAgeDays) {
103
+ return bucket.score;
104
+ }
105
+ }
106
+ return 0;
107
+ }
108
+
109
+ export function getRecommendedScore({
110
+ favoriteCount,
111
+ publishedAtMs,
112
+ completenessScore,
113
+ }: {
114
+ favoriteCount: number;
115
+ publishedAtMs: number;
116
+ completenessScore: number;
117
+ }) {
118
+ const ageMs = Math.max(0, Date.now() - publishedAtMs);
119
+ const ageDays = ageMs / (24 * 60 * 60 * 1000);
120
+ const freshnessScore = getFreshnessScore(ageDays);
121
+ const coldStartBoost = getColdStartBoost(ageDays, favoriteCount);
122
+
123
+ // Intentionally exclude useCount/dialogCount here until those signals are
124
+ // written consistently across the public-agent pipeline.
125
+ return Math.log1p(Math.max(0, favoriteCount)) *
126
+ RECOMMENDED_SCORE_WEIGHTS.favoriteLogWeight +
127
+ freshnessScore * RECOMMENDED_SCORE_WEIGHTS.freshnessWeight +
128
+ completenessScore * RECOMMENDED_SCORE_WEIGHTS.completenessWeight +
129
+ coldStartBoost;
130
+ }
131
+
132
+ /**
133
+ * 根据本地与远程条目构建排序元信息
134
+ */
135
+ export function buildSortMeta(local?: Agent, remote?: Agent): SortMeta {
136
+ const createdAtMs = Math.max(
137
+ toTimeMs(remote?.updatedAt ?? remote?.createdAt),
138
+ toTimeMs(local?.updatedAt ?? local?.createdAt)
139
+ );
140
+ const publishedAtMs = Math.max(
141
+ toTimeMs(remote?.createdAt),
142
+ toTimeMs(local?.createdAt)
143
+ );
144
+
145
+ const outputPriceNum = firstFiniteNumber(
146
+ (remote as any)?.outputPrice,
147
+ (local as any)?.outputPrice
148
+ );
149
+
150
+ const useCount = firstFiniteNumber(
151
+ (remote as any)?.metrics?.useCount,
152
+ (local as any)?.metrics?.useCount,
153
+ (remote as any)?.dialogCount,
154
+ (local as any)?.dialogCount
155
+ );
156
+
157
+ const rating = firstFiniteNumber(
158
+ (remote as any)?.metrics?.rating,
159
+ (local as any)?.metrics?.rating,
160
+ (remote as any)?.messageCount,
161
+ (local as any)?.messageCount
162
+ );
163
+
164
+ const favoriteCount = firstFiniteNumber(
165
+ (remote as any)?.metrics?.favoriteCount,
166
+ (local as any)?.metrics?.favoriteCount
167
+ );
168
+
169
+ const completenessScore = Math.max(
170
+ getCompletenessScore(local),
171
+ getCompletenessScore(remote)
172
+ );
173
+ const normalizedFavoriteCount = Number.isFinite(favoriteCount) ? favoriteCount : 0;
174
+ const normalizedPublishedAtMs = publishedAtMs || createdAtMs;
175
+ const recommendedScore = getRecommendedScore({
176
+ favoriteCount: normalizedFavoriteCount,
177
+ publishedAtMs: normalizedPublishedAtMs,
178
+ completenessScore,
179
+ });
180
+
181
+ return {
182
+ createdAtMs,
183
+ publishedAtMs: normalizedPublishedAtMs,
184
+ outputPriceNum: Number.isFinite(outputPriceNum)
185
+ ? outputPriceNum
186
+ : Number.POSITIVE_INFINITY,
187
+ useCount: Number.isFinite(useCount) ? useCount : 0,
188
+ rating: Number.isFinite(rating) ? rating : 0,
189
+ favoriteCount: normalizedFavoriteCount,
190
+ completenessScore,
191
+ recommendedScore,
192
+ };
193
+ }
194
+
195
+ /**
196
+ * 统一的智能体排序逻辑
197
+ */
198
+ export const sortAgents = (
199
+ agents: Agent[],
200
+ sortBy:
201
+ | "recommended"
202
+ | "newest"
203
+ | "popular"
204
+ | "rating"
205
+ | "outputPriceAsc"
206
+ | "outputPriceDesc"
207
+ | "favorite"
208
+ | string
209
+ ): Agent[] => {
210
+ const arr = agents.map((agent) => {
211
+ const typed = agent as MAgent;
212
+ return typed.__sort ? typed : { ...typed, __sort: buildSortMeta(agent) };
213
+ });
214
+
215
+ arr.sort((a, b) => {
216
+ const sa = a.__sort!;
217
+ const sb = b.__sort!;
218
+ let diff = 0;
219
+
220
+ switch (sortBy) {
221
+ case "recommended":
222
+ diff = sb.recommendedScore - sa.recommendedScore;
223
+ break;
224
+ case "popular":
225
+ diff = sb.useCount - sa.useCount;
226
+ break;
227
+ case "rating":
228
+ diff = sb.rating - sa.rating;
229
+ break;
230
+ case "favorite":
231
+ diff = sb.favoriteCount - sa.favoriteCount;
232
+ break;
233
+ case "outputPriceAsc":
234
+ diff = sa.outputPriceNum - sb.outputPriceNum;
235
+ break;
236
+ case "outputPriceDesc":
237
+ diff = sb.outputPriceNum - sa.outputPriceNum;
238
+ break;
239
+ case "newest":
240
+ default:
241
+ diff = sb.createdAtMs - sa.createdAtMs;
242
+ break;
243
+ }
244
+
245
+ if (diff !== 0) return diff;
246
+ return String(a.id).localeCompare(String(b.id));
247
+ });
248
+
249
+ return arr;
250
+ };
@@ -0,0 +1,146 @@
1
+ export interface ReferencePickerSpaceItem {
2
+ id: string;
3
+ name: string;
4
+ isCurrent: boolean;
5
+ }
6
+
7
+ export interface ReferencePickerContentItem {
8
+ dbKey: string;
9
+ title: string;
10
+ spaceId: string;
11
+ spaceName: string;
12
+ contentType?: string;
13
+ skillSummary?: {
14
+ isSkill: true;
15
+ skillId?: string;
16
+ name?: string;
17
+ description?: string;
18
+ toolNames?: string[];
19
+ triggerMode?: "explicit" | "required" | "recommended";
20
+ } | null;
21
+ }
22
+
23
+ export const ALL_SPACES_ID = "__all_spaces__";
24
+
25
+ export const buildReferencePickerSpaceItems = ({
26
+ currentSpace,
27
+ allMemberSpaces,
28
+ allSpacesLabel,
29
+ }: {
30
+ currentSpace?: { id: string; name: string } | null;
31
+ allMemberSpaces: Array<{ spaceId: string; spaceName: string }>;
32
+ allSpacesLabel: string;
33
+ }) => {
34
+ const items: ReferencePickerSpaceItem[] = [
35
+ {
36
+ id: ALL_SPACES_ID,
37
+ name: allSpacesLabel,
38
+ isCurrent: false,
39
+ },
40
+ ];
41
+ const nameMap = new Map<string, string>();
42
+
43
+ if (currentSpace) {
44
+ items.push({
45
+ id: currentSpace.id,
46
+ name: currentSpace.name,
47
+ isCurrent: true,
48
+ });
49
+ nameMap.set(currentSpace.id, currentSpace.name);
50
+ }
51
+
52
+ const others = allMemberSpaces
53
+ .filter((space) => space.spaceId !== currentSpace?.id)
54
+ .sort((a, b) => a.spaceName.localeCompare(b.spaceName))
55
+ .map((space) => {
56
+ nameMap.set(space.spaceId, space.spaceName);
57
+ return {
58
+ id: space.spaceId,
59
+ name: space.spaceName,
60
+ isCurrent: false,
61
+ };
62
+ });
63
+
64
+ items.push(...others);
65
+
66
+ return { items, nameMap };
67
+ };
68
+
69
+ export const buildReferencePickerContents = ({
70
+ contentsObj,
71
+ spaceId,
72
+ spaceName,
73
+ unnamedLabel,
74
+ }: {
75
+ contentsObj: Record<string, any>;
76
+ spaceId: string;
77
+ spaceName: string;
78
+ unnamedLabel: string;
79
+ }): ReferencePickerContentItem[] =>
80
+ Object.entries(contentsObj)
81
+ .filter(([dbKey, value]) => !dbKey.startsWith("dialog-") && value)
82
+ .map(([dbKey, value]: [string, any]) => ({
83
+ dbKey,
84
+ title: value?.title || unnamedLabel,
85
+ spaceId,
86
+ spaceName,
87
+ contentType: value?.type,
88
+ skillSummary: value?.skillSummary ?? null,
89
+ }));
90
+
91
+ export const filterReferencePickerContents = ({
92
+ spacesData,
93
+ activeSpaceId,
94
+ searchQuery,
95
+ pickerMode,
96
+ skillCandidateMap,
97
+ }: {
98
+ spacesData: Map<string, ReferencePickerContentItem[]>;
99
+ activeSpaceId: string;
100
+ searchQuery: string;
101
+ pickerMode: "skill" | "knowledge" | "instruction";
102
+ skillCandidateMap: Map<string, boolean>;
103
+ }) => {
104
+ const trimmedQuery = searchQuery.trim().toLowerCase();
105
+ const scopedContents =
106
+ activeSpaceId === ALL_SPACES_ID
107
+ ? Array.from(spacesData.values()).flat()
108
+ : spacesData.get(activeSpaceId) || [];
109
+
110
+ const searchedContents = trimmedQuery
111
+ ? scopedContents.filter((item) => {
112
+ const title = item.title.toLowerCase();
113
+ const key = item.dbKey.toLowerCase();
114
+ const space = item.spaceName.toLowerCase();
115
+ return (
116
+ title.includes(trimmedQuery) ||
117
+ key.includes(trimmedQuery) ||
118
+ space.includes(trimmedQuery)
119
+ );
120
+ })
121
+ : scopedContents;
122
+
123
+ if (pickerMode !== "skill") {
124
+ return searchedContents;
125
+ }
126
+
127
+ return searchedContents.filter(
128
+ (item) =>
129
+ item.contentType === "page" &&
130
+ (item.skillSummary?.isSkill || skillCandidateMap.get(item.dbKey))
131
+ );
132
+ };
133
+
134
+ export const collectPendingSkillCandidates = ({
135
+ contents,
136
+ skillCandidateMap,
137
+ }: {
138
+ contents: ReferencePickerContentItem[];
139
+ skillCandidateMap: Map<string, boolean>;
140
+ }) =>
141
+ contents.filter(
142
+ (item) =>
143
+ item.contentType === "page" &&
144
+ !item.skillSummary?.isSkill &&
145
+ !skillCandidateMap.has(item.dbKey)
146
+ );