nolo-cli 0.1.40 → 0.1.42

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 (482) hide show
  1. package/README.md +27 -12
  2. package/agent-runtime/agentThread.ts +222 -0
  3. package/agent-runtime/agentThreadAdmission.ts +74 -0
  4. package/agent-runtime/dialogWritePlan.ts +42 -0
  5. package/agent-runtime/hostAdapter.ts +1 -0
  6. package/agent-runtime/hybridRecordStore.ts +5 -3
  7. package/agent-runtime/index.ts +43 -0
  8. package/agent-runtime/localLoop.ts +31 -2
  9. package/agent-runtime/localToolPolicy.ts +1 -0
  10. package/agent-runtime/localWorkspaceTools.ts +1258 -78
  11. package/agent-runtime/noloWorkspaceTools.ts +20 -1
  12. package/agent-runtime/runtimeToolPolicy.ts +1 -5
  13. package/agent-runtime/runtimeToolSurface.ts +1 -0
  14. package/agentAliases.ts +29 -9
  15. package/agentInternalCommandEntries.ts +11 -3
  16. package/agentMachineCommands.ts +1 -0
  17. package/agentRecordCommands.ts +2 -2
  18. package/agentRecordHelpers.ts +59 -6
  19. package/agentRunCommand.ts +84 -19
  20. package/ai/agent/_executeModel.ts +146 -0
  21. package/ai/agent/agentSlice.ts +556 -1
  22. package/ai/agent/appWorkingMemory.ts +126 -0
  23. package/ai/agent/buildEditingContext.ts +378 -0
  24. package/ai/agent/buildSystemPrompt.ts +599 -0
  25. package/ai/agent/canvasEditingContext.ts +49 -0
  26. package/ai/agent/cleanAgentMessages.ts +140 -0
  27. package/ai/agent/cliChatClient.ts +153 -0
  28. package/ai/agent/cliExecutor.ts +264 -46
  29. package/ai/agent/contextCompiler.ts +107 -0
  30. package/ai/agent/contextLayerContract.ts +44 -0
  31. package/ai/agent/createAgentSchema.ts +272 -0
  32. package/ai/agent/currentTimeContext.ts +65 -0
  33. package/ai/agent/executeToolCall.ts +58 -0
  34. package/ai/agent/fetchAgentContexts.ts +42 -0
  35. package/ai/agent/generatePrompt.ts +3 -0
  36. package/ai/agent/getFullChatContextKeys.ts +168 -0
  37. package/ai/agent/guidedCreation/capabilities.ts +84 -0
  38. package/ai/agent/guidedCreation/types.ts +38 -0
  39. package/ai/agent/isLiveAudioOnlyAgent.ts +26 -0
  40. package/ai/agent/liveAudioModel.ts +2 -0
  41. package/ai/agent/pageBuilderHandoffRules.ts +75 -0
  42. package/ai/agent/persistMessageWithFixedId.ts +37 -0
  43. package/ai/agent/planSlice.ts +259 -0
  44. package/ai/agent/referenceUtils.ts +229 -0
  45. package/ai/agent/runAgentBackground.ts +310 -0
  46. package/ai/agent/runAgentClientLoop.ts +164 -0
  47. package/ai/agent/runtimeGuidance.ts +116 -0
  48. package/ai/agent/serverOrigin.ts +7 -0
  49. package/ai/agent/startParallelAgentStreams.ts +424 -0
  50. package/ai/agent/startupProtocol.ts +53 -0
  51. package/ai/agent/streamAgentChatTurn.ts +2135 -0
  52. package/ai/agent/streamAgentChatTurnUtils.ts +923 -0
  53. package/ai/agent/toolFailureGuard.ts +99 -0
  54. package/ai/agent/types.ts +73 -0
  55. package/ai/agent/utils/imageOutput.ts +37 -0
  56. package/ai/agent/utils/publicImageAgentMode.ts +26 -0
  57. package/ai/ai.locale.ts +1165 -0
  58. package/ai/chat/accumulateToolCallChunks.ts +126 -0
  59. package/ai/chat/fetchUtils.ts +143 -0
  60. package/ai/chat/inlineImageUrlsForCustomProvider.ts +122 -0
  61. package/ai/chat/parseApiError.ts +67 -0
  62. package/ai/chat/parseMultilineSSE.ts +95 -0
  63. package/ai/chat/sendOpenAICompletionsRequest.ts +921 -0
  64. package/ai/chat/sendOpenAIResponseRequest.ts +720 -0
  65. package/ai/chat/serverProxyRetry.ts +102 -0
  66. package/ai/chat/shouldUseServerProxy.ts +18 -0
  67. package/ai/chat/streamRetry.ts +46 -0
  68. package/ai/chat/updateTotalUsage.ts +89 -0
  69. package/ai/context/buildReferenceContext.ts +437 -0
  70. package/ai/context/retention.ts +165 -0
  71. package/ai/context/tokenUtils.ts +78 -0
  72. package/ai/llm/agentCapabilities.ts +74 -0
  73. package/ai/llm/deepinfra.ts +79 -0
  74. package/ai/llm/fireworks.ts +68 -0
  75. package/ai/llm/generateRequestBody.ts +165 -0
  76. package/ai/llm/getModelContextWindow.ts +93 -0
  77. package/ai/llm/getPricing.ts +252 -0
  78. package/ai/llm/imagePricing.ts +66 -0
  79. package/ai/llm/isResponseAPIModel.ts +13 -0
  80. package/ai/llm/kimi.ts +22 -0
  81. package/ai/llm/mimo.ts +99 -0
  82. package/ai/llm/mistral.ts +21 -0
  83. package/ai/llm/models.ts +45 -0
  84. package/ai/llm/openrouterModels.ts +31 -0
  85. package/ai/llm/providers.ts +313 -0
  86. package/ai/llm/reasoningModels.ts +27 -0
  87. package/ai/llm/types.ts +82 -0
  88. package/ai/llm/usageRequestOptions.ts +59 -0
  89. package/ai/memory/policy.ts +49 -0
  90. package/ai/memory/queryShared.ts +167 -0
  91. package/ai/memory/recentRelationshipRecap.ts +251 -0
  92. package/ai/memory/remember.ts +111 -0
  93. package/ai/memory/scope.ts +74 -0
  94. package/ai/memory/store.ts +20 -0
  95. package/ai/memory/storeShared.ts +76 -0
  96. package/ai/memory/types.ts +52 -0
  97. package/ai/memory/understanding.ts +341 -0
  98. package/ai/memory/understandingGreeting.ts +264 -0
  99. package/ai/policy/personalizationDialog.ts +333 -0
  100. package/ai/policy/runtimePolicy.ts +526 -0
  101. package/ai/policy/selfUpdateFields.ts +48 -0
  102. package/ai/policy/types.ts +64 -0
  103. package/ai/skills/referenceRuntime.ts +274 -0
  104. package/ai/skills/skillDiagnostics.ts +251 -0
  105. package/ai/skills/skillDocBuilder.ts +139 -0
  106. package/ai/skills/skillSummaryMarker.ts +26 -0
  107. package/ai/token/calculatePrice.ts +576 -0
  108. package/ai/token/missingUsageEstimate.ts +42 -0
  109. package/ai/token/normalizeUsage.ts +124 -0
  110. package/ai/token/openaiImageGenerationUsage.ts +56 -0
  111. package/ai/token/prepareTokenUsageData.ts +93 -0
  112. package/ai/token/resolveBillingTarget.ts +52 -0
  113. package/ai/token/saveTokenRecord.ts +53 -0
  114. package/ai/token/types.ts +103 -0
  115. package/ai/tools/agent/agentTools.ts +214 -0
  116. package/ai/tools/agent/agentUpdateShared.ts +311 -0
  117. package/ai/tools/agent/callAgentTool.ts +139 -0
  118. package/ai/tools/agent/createAgentTool.ts +512 -0
  119. package/ai/tools/agent/createDialogTool.ts +69 -0
  120. package/ai/tools/agent/createSkillAgentTool.ts +62 -0
  121. package/ai/tools/agent/parallelBudget.ts +221 -0
  122. package/ai/tools/agent/prepareAgentDraftTool.ts +159 -0
  123. package/ai/tools/agent/presets/appBuilderPreset.ts +154 -0
  124. package/ai/tools/agent/runLlmTool.ts +96 -0
  125. package/ai/tools/agent/runStreamingAgentTool.ts +73 -0
  126. package/ai/tools/agent/skillAgentArgs.ts +106 -0
  127. package/ai/tools/agent/startAgentDialogTool.ts +54 -0
  128. package/ai/tools/agent/streamParallelAgentsTool.ts +123 -0
  129. package/ai/tools/agent/updateAgentTool.ts +96 -0
  130. package/ai/tools/agent/updateSelfTool.ts +113 -0
  131. package/ai/tools/amazonProductScraperTool.ts +86 -0
  132. package/ai/tools/apifyActorClient.ts +45 -0
  133. package/ai/tools/appEditGuard.ts +372 -0
  134. package/ai/tools/appReadSnapshot.ts +153 -0
  135. package/ai/tools/appTools.ts +1549 -0
  136. package/ai/tools/applyEditTool.ts +259 -0
  137. package/ai/tools/applyLineEditsTool.ts +315 -0
  138. package/ai/tools/browserTools/click.ts +33 -0
  139. package/ai/tools/browserTools/closeSession.ts +29 -0
  140. package/ai/tools/browserTools/common.ts +27 -0
  141. package/ai/tools/browserTools/openSession.ts +48 -0
  142. package/ai/tools/browserTools/readContent.ts +38 -0
  143. package/ai/tools/browserTools/selectOption.ts +46 -0
  144. package/ai/tools/browserTools/typeText.ts +42 -0
  145. package/ai/tools/category/createCategoryTool.ts +66 -0
  146. package/ai/tools/category/queryContentsByCategoryTool.ts +69 -0
  147. package/ai/tools/category/updateContentCategoryTool.ts +75 -0
  148. package/ai/tools/cfBrowserTools.ts +319 -0
  149. package/ai/tools/cfSpeechToTextTool.ts +49 -0
  150. package/ai/tools/checkEnvTool.ts +102 -0
  151. package/ai/tools/cloudflareCrawlTool.ts +289 -0
  152. package/ai/tools/codeSearchTool.ts +123 -0
  153. package/ai/tools/codeTools.ts +101 -0
  154. package/ai/tools/convertMarxistsBookTool.ts +103 -0
  155. package/ai/tools/createDocTool.ts +132 -0
  156. package/ai/tools/createSkillDocTool.ts +155 -0
  157. package/ai/tools/createWorkflowTool.ts +154 -0
  158. package/ai/tools/delayTool.ts +31 -0
  159. package/ai/tools/deleteSpacesTool.ts +325 -0
  160. package/ai/tools/deleteSpacesToolModel.ts +159 -0
  161. package/ai/tools/devReloadUtils.ts +29 -0
  162. package/ai/tools/dialogMessageSearch.ts +137 -0
  163. package/ai/tools/doctorSkillTool.ts +72 -0
  164. package/ai/tools/ecommerceScraperTool.ts +86 -0
  165. package/ai/tools/emailTools.ts +549 -0
  166. package/ai/tools/evalSkillTool.ts +92 -0
  167. package/ai/tools/exaSearchTool.ts +63 -0
  168. package/ai/tools/execShellTool.ts +266 -0
  169. package/ai/tools/executeSqlTool.ts +192 -0
  170. package/ai/tools/fetchWebpageSupport.ts +333 -0
  171. package/ai/tools/fetchWebpageTool.ts +97 -0
  172. package/ai/tools/geminiImagePreviewTool.ts +386 -0
  173. package/ai/tools/generateDocxTool.ts +215 -0
  174. package/ai/tools/googleSearchScraperTool.ts +106 -0
  175. package/ai/tools/importDataTool.ts +133 -0
  176. package/ai/tools/importSkillTool.ts +162 -0
  177. package/ai/tools/index.ts +2290 -0
  178. package/ai/tools/jdProductScraperTool.ts +821 -0
  179. package/ai/tools/listUserSpacesTool.ts +113 -0
  180. package/ai/tools/localFilesTool.ts +276 -0
  181. package/ai/tools/marxistsOfflineBook.ts +253 -0
  182. package/ai/tools/modelUsageTools.ts +206 -0
  183. package/ai/tools/noloWorkspaceReadTools.ts +467 -0
  184. package/ai/tools/olmOcrTool.ts +33 -0
  185. package/ai/tools/openaiImageTool.ts +267 -0
  186. package/ai/tools/prepareTools.ts +193 -0
  187. package/ai/tools/readDocTool.ts +88 -0
  188. package/ai/tools/readFileTool.ts +214 -0
  189. package/ai/tools/readTool.ts +194 -0
  190. package/ai/tools/readXPostTool.ts +229 -0
  191. package/ai/tools/readXhsProfileTool.ts +638 -0
  192. package/ai/tools/rememberMemoryTool.ts +84 -0
  193. package/ai/tools/remotionVideoTool.ts +151 -0
  194. package/ai/tools/searchDialogMessagesTool.ts +222 -0
  195. package/ai/tools/searchWorkspaceTool.ts +259 -0
  196. package/ai/tools/skillFollowup.ts +86 -0
  197. package/ai/tools/surfWeatherTool.ts +169 -0
  198. package/ai/tools/table/addTableRowTool.ts +217 -0
  199. package/ai/tools/table/createTableTool.ts +326 -0
  200. package/ai/tools/table/rowTools.ts +386 -0
  201. package/ai/tools/table/schemaTools.ts +244 -0
  202. package/ai/tools/table/shareTableTool.ts +148 -0
  203. package/ai/tools/table/toolShared.ts +129 -0
  204. package/ai/tools/taobaoTmallProductScraperTool.ts +49 -0
  205. package/ai/tools/toolApiClient.ts +279 -0
  206. package/ai/tools/toolPacks.ts +19 -0
  207. package/ai/tools/toolResultError.ts +42 -0
  208. package/ai/tools/toolRunSlice.ts +303 -0
  209. package/ai/tools/toolSchemaCompatibility.ts +53 -0
  210. package/ai/tools/uiAskChoiceTool.ts +104 -0
  211. package/ai/tools/updateContentTitleTool.ts +84 -0
  212. package/ai/tools/updateDocTool.ts +105 -0
  213. package/ai/tools/updateUserPreferenceProfileTool.ts +153 -0
  214. package/ai/tools/wereadGatewayTool.ts +152 -0
  215. package/ai/tools/whisperTool.ts +76 -0
  216. package/ai/tools/writeFileTool.ts +213 -0
  217. package/ai/tools/youtubeScraperTool.ts +116 -0
  218. package/ai/tools/ziweiChartTool.ts +678 -0
  219. package/ai/types.ts +58 -0
  220. package/ai/workflow/workflowExecutor.ts +323 -0
  221. package/ai/workflow/workflowSlice.ts +73 -0
  222. package/ai/workflow/workflowTypes.ts +106 -0
  223. package/app/actions/syncAppRecord.ts +73 -0
  224. package/app/constants/animationSets.ts +21 -0
  225. package/app/constants/appEditor.ts +99 -0
  226. package/app/favorite/deletedFavoriteProjection.ts +31 -0
  227. package/app/favorite/favoriteSlice.ts +690 -0
  228. package/app/favorite/type.ts +43 -0
  229. package/app/hooks/deleteDbKey.ts +254 -0
  230. package/app/i18n/client.ts +13 -0
  231. package/app/i18n/i18n.base.ts +14 -0
  232. package/app/i18n/i18n.config.ts +77 -0
  233. package/app/i18n/index.ts +15 -0
  234. package/app/i18n/mapLanguage.ts +11 -0
  235. package/app/i18n/translations/appBuilder.locale.ts +453 -0
  236. package/app/i18n/translations/error.locale.ts +246 -0
  237. package/app/i18n/translations/interface.locale.ts +2206 -0
  238. package/app/i18n/translations/pricing.locale.ts +397 -0
  239. package/app/i18n/translations/privacy.locale.ts +168 -0
  240. package/app/i18n/translations/seo.locale.ts +129 -0
  241. package/app/i18n/translations/terms.locale.ts +124 -0
  242. package/app/i18n/types.ts +6 -0
  243. package/app/notifications/model.ts +77 -0
  244. package/app/notifications/notificationSlice.ts +141 -0
  245. package/app/reducer.ts +36 -0
  246. package/app/settings/dbActionThunks.ts +34 -0
  247. package/app/settings/defaultSpacePreference.ts +87 -0
  248. package/app/settings/settingSlice.tsx +1212 -0
  249. package/app/settings/settingsRecord.ts +8 -0
  250. package/app/stateViews/runtime.ts +90 -0
  251. package/app/store.ts +55 -0
  252. package/app/theme/colors.ts +280 -0
  253. package/app/theme/fontPreference.ts +77 -0
  254. package/app/theme/theme.config.ts +120 -0
  255. package/app/theme/themeModeBootstrap.ts +123 -0
  256. package/app/types/appSummary.ts +118 -0
  257. package/app/types.ts +358 -0
  258. package/app/utils/appKeys.ts +55 -0
  259. package/app/utils/async.ts +28 -0
  260. package/app/utils/desktopAgentRuntimeTurnClient.ts +75 -0
  261. package/app/utils/env.ts +17 -0
  262. package/app/utils/fileUtils.ts +229 -0
  263. package/app/utils/imageUtils.ts +274 -0
  264. package/app/utils/myContentItems.ts +289 -0
  265. package/app/utils/retryAfter.ts +36 -0
  266. package/app/utils/retryFetch.ts +61 -0
  267. package/app/utils/toast.ts +3 -0
  268. package/auth/action/signUpAction.ts +227 -0
  269. package/auth/adminPermissions.ts +78 -0
  270. package/auth/authSlice.ts +531 -0
  271. package/auth/client/loginRequest.ts +14 -0
  272. package/auth/resetAuthScopedClientState.ts +27 -0
  273. package/auth/routes.ts +183 -0
  274. package/auth/token.ts +63 -0
  275. package/auth/types.ts +40 -0
  276. package/chat/chat.locale.ts +817 -0
  277. package/chat/dialog/actions/addCybotAction.ts +52 -0
  278. package/chat/dialog/actions/addReferenceKeysAction.ts +62 -0
  279. package/chat/dialog/actions/builtinDialogLlm.ts +35 -0
  280. package/chat/dialog/actions/cleanupCliSession.ts +40 -0
  281. package/chat/dialog/actions/createDialogAction.ts +379 -0
  282. package/chat/dialog/actions/createScheduledTaskAction.ts +78 -0
  283. package/chat/dialog/actions/handleSendMessageAction.ts +202 -0
  284. package/chat/dialog/actions/handleSendMessageResolver.ts +27 -0
  285. package/chat/dialog/actions/removeCybotAction.ts +6 -0
  286. package/chat/dialog/actions/removeDialogAgentAction.ts +44 -0
  287. package/chat/dialog/actions/replacePrimaryCybotAction.ts +6 -0
  288. package/chat/dialog/actions/setPrimaryDialogAgentAction.ts +52 -0
  289. package/chat/dialog/actions/updateDialogSummaryAction.ts +430 -0
  290. package/chat/dialog/actions/updateDialogTitleAction.ts +151 -0
  291. package/chat/dialog/actions/updateDialogTitlePolicy.ts +30 -0
  292. package/chat/dialog/actions/updateTokensAction.ts +233 -0
  293. package/chat/dialog/dialogAgents.ts +72 -0
  294. package/chat/dialog/dialogAttachmentCleanup.ts +135 -0
  295. package/chat/dialog/dialogSlice.ts +985 -0
  296. package/chat/dialog/dialogTitle.ts +81 -0
  297. package/chat/dialog/dialogTokenStats.ts +12 -0
  298. package/chat/dialog/dialogUrl.ts +6 -0
  299. package/chat/dialog/goal.ts +108 -0
  300. package/chat/dialog/objectAssistantRegistry.ts +286 -0
  301. package/chat/dialog/schedule.ts +17 -0
  302. package/chat/messages/activityCompletion.ts +97 -0
  303. package/chat/messages/fetchAndCacheMessages.ts +119 -0
  304. package/chat/messages/fetchConvMsgs.ts +51 -0
  305. package/chat/messages/fileUrl.ts +13 -0
  306. package/chat/messages/imagePayloadPersistence.ts +22 -0
  307. package/chat/messages/messageAgent.ts +8 -0
  308. package/chat/messages/messageContent.ts +203 -0
  309. package/chat/messages/messageContract.ts +88 -0
  310. package/chat/messages/messageSlice.ts +1386 -0
  311. package/chat/messages/toolThunks.ts +646 -0
  312. package/chat/messages/types.ts +8 -1
  313. package/chat/messages/web/toolDisplayName.ts +686 -0
  314. package/cli/agentAliases.ts +29 -9
  315. package/cli/agentInternalCommandEntries.ts +11 -3
  316. package/cli/agentMachineCommands.ts +1 -0
  317. package/cli/agentRecordCommands.ts +2 -2
  318. package/cli/agentRecordHelpers.ts +59 -6
  319. package/cli/agentRunCommand.ts +84 -19
  320. package/cli/client/agentRun.ts +92 -261
  321. package/cli/client/localRuntimeAdapter.ts +292 -50
  322. package/cli/commandRegistry.ts +7 -4
  323. package/cli/dialogAttachmentCleanup.ts +151 -0
  324. package/cli/dialogCommands.ts +163 -61
  325. package/cli/dialogInternalCommandEntries.ts +1 -1
  326. package/cli/docCreateCommands.ts +345 -0
  327. package/cli/docInternalCommandEntries.ts +11 -0
  328. package/cli/docPageHelpers.ts +80 -0
  329. package/cli/docSkillHelpers.ts +67 -0
  330. package/cli/docSpaceHelpers.ts +46 -0
  331. package/cli/internalCommandEntries.ts +4 -0
  332. package/cli/machineCommands.ts +18 -34
  333. package/cli/memoryCommands.ts +253 -0
  334. package/cli/memoryInternalCommandEntries.ts +9 -0
  335. package/cli/qoderUsageProbe.ts +252 -0
  336. package/cli/scriptCommandEntries.ts +0 -3
  337. package/cli/secretScan.ts +31 -0
  338. package/cli/tableCommands.ts +248 -13
  339. package/cli/workflowInternalCommandEntries.ts +2 -3
  340. package/client/agentRun.test.ts +119 -83
  341. package/client/agentRun.ts +92 -261
  342. package/client/localRuntimeAdapter.test.ts +894 -32
  343. package/client/localRuntimeAdapter.ts +292 -50
  344. package/commandRegistry.ts +7 -4
  345. package/connector-experimental/capabilities.ts +10 -9
  346. package/core/builtinAgents.ts +209 -0
  347. package/core/config.ts +15 -0
  348. package/core/crypto.ts +51 -0
  349. package/core/generateKeyPairFromSeedV1.ts +15 -0
  350. package/core/generateMainKey.ts +22 -0
  351. package/core/init.ts +41 -0
  352. package/core/localOrigins.ts +8 -0
  353. package/core/password.ts +13 -0
  354. package/core/prefix.ts +23 -0
  355. package/create/editor/transforms/fromMarkdown/index.ts +260 -0
  356. package/create/editor/transforms/fromMarkdown/inline.ts +285 -0
  357. package/create/editor/transforms/fromMarkdown/table.ts +130 -0
  358. package/create/editor/transforms/markdownToSlate.ts +24 -0
  359. package/create/editor/transforms/slateToRenderMarkdown.ts +218 -0
  360. package/create/editor/transforms/slateToSimplifiedMarkdown.ts +224 -0
  361. package/create/editor/transforms/slateToText.ts +33 -0
  362. package/create/editor/utils/slateUtils.ts +233 -0
  363. package/create/space/addSpaceAction.ts +193 -0
  364. package/create/space/category/categoryActions.ts +427 -0
  365. package/create/space/changeSpaceAction.ts +36 -0
  366. package/create/space/constants.ts +8 -0
  367. package/create/space/content/addContentAction.ts +116 -0
  368. package/create/space/content/contentThunks.ts +90 -0
  369. package/create/space/content/deleteContentFromSpaceAction.ts +187 -0
  370. package/create/space/content/deleteMultipleContentAction.ts +112 -0
  371. package/create/space/content/moveContentAction.ts +285 -0
  372. package/create/space/content/updateContentCategoryAction.ts +154 -0
  373. package/create/space/content/updateContentTitleAction.ts +151 -0
  374. package/create/space/content/uploadAndAddFileToSpaceAction.ts +98 -0
  375. package/create/space/contentKeyUtils.ts +117 -0
  376. package/create/space/deleteSpaceAction.ts +276 -0
  377. package/create/space/fetchSpaceAction.ts +70 -0
  378. package/create/space/fetchSpaceSidebarStateAction.ts +26 -0
  379. package/create/space/loadDefaultSpaceAction.ts +57 -0
  380. package/create/space/member/addMemberAction.ts +79 -0
  381. package/create/space/member/fetchUserSpaceMembershipsAction.ts +386 -0
  382. package/create/space/member/memberThunks.ts +47 -0
  383. package/create/space/member/removeMemberAction.ts +67 -0
  384. package/create/space/resolvePreferredSpaceId.ts +113 -0
  385. package/create/space/sidebarVisibleTypes.ts +260 -0
  386. package/create/space/space.locale.ts +1087 -0
  387. package/create/space/spaceAccess.ts +182 -0
  388. package/create/space/spaceCollapsedState.ts +54 -0
  389. package/create/space/spaceKeys.ts +65 -0
  390. package/create/space/spaceSlice.ts +328 -0
  391. package/create/space/spaceThunks.ts +410 -0
  392. package/create/space/types.ts +27 -0
  393. package/create/space/updateSpaceAction.ts +107 -0
  394. package/create/space/utils/permissions.ts +47 -0
  395. package/create/types.ts +21 -0
  396. package/database/actions/agentReadResolution.ts +33 -0
  397. package/database/actions/common.ts +243 -0
  398. package/database/actions/deleteFile.ts +40 -0
  399. package/database/actions/fileContent.ts +125 -0
  400. package/database/actions/patch.ts +155 -0
  401. package/database/actions/read.ts +358 -0
  402. package/database/actions/readAndWait.ts +260 -0
  403. package/database/actions/readRequestManager.ts +120 -0
  404. package/database/actions/readResolution.ts +125 -0
  405. package/database/actions/remove.ts +94 -0
  406. package/database/actions/replication.ts +477 -0
  407. package/database/actions/upload.ts +186 -0
  408. package/database/actions/upsert.ts +56 -0
  409. package/database/actions/write.ts +146 -0
  410. package/database/authority/ownerKey.ts +80 -0
  411. package/database/authority/recordAuthority.ts +67 -0
  412. package/database/authority/userAuthorityRegistry.ts +54 -0
  413. package/database/client/fetchUserData.ts +61 -0
  414. package/database/dbActionThunks.ts +1 -0
  415. package/database/dbSlice.ts +149 -0
  416. package/database/fileRing.ts +51 -0
  417. package/database/fileStorage.ts +232 -0
  418. package/database/fileUrl.ts +34 -0
  419. package/database/keys.ts +767 -0
  420. package/database/queryPrefixes.ts +14 -0
  421. package/database/requests.ts +443 -0
  422. package/database/runtimeServerContext.ts +54 -0
  423. package/database/tenantPlacement.ts +57 -0
  424. package/database/userPreferenceRegister.ts +131 -0
  425. package/database/utils/ulid.ts +1 -0
  426. package/dialogAttachmentCleanup.ts +151 -0
  427. package/dialogCommands.ts +163 -61
  428. package/dialogInternalCommandEntries.ts +1 -1
  429. package/docCreateCommands.ts +345 -0
  430. package/docInternalCommandEntries.ts +11 -0
  431. package/docPageHelpers.ts +80 -0
  432. package/docSkillHelpers.ts +67 -0
  433. package/docSpaceHelpers.ts +46 -0
  434. package/integrations/anthropic/anthropicModels.ts +47 -0
  435. package/integrations/deepseek/models.ts +37 -0
  436. package/integrations/google/models.ts +341 -0
  437. package/integrations/openai/chatCompletionStreamMode.ts +10 -0
  438. package/integrations/openai/filterAndCleanMessages.ts +284 -0
  439. package/integrations/openai/flexTier.ts +102 -0
  440. package/integrations/openai/generateOpenAIRequestBody.ts +271 -0
  441. package/integrations/openai/generateResponseRequestBody.ts +77 -0
  442. package/integrations/openai/models.ts +271 -0
  443. package/integrations/openai/providerBodyCompatibility.ts +23 -0
  444. package/integrations/openai/responsesHelpers.ts +297 -0
  445. package/integrations/x-reader/types.ts +109 -0
  446. package/integrations/xai/models.ts +13 -0
  447. package/integrations/xhs-reader/redaction.ts +68 -0
  448. package/integrations/xhs-reader/types.ts +277 -0
  449. package/integrations/xhs-reader/url.ts +80 -0
  450. package/internalCommandEntries.ts +4 -0
  451. package/machineCommands.ts +18 -34
  452. package/memoryCommands.ts +253 -0
  453. package/memoryInternalCommandEntries.ts +9 -0
  454. package/package.json +454 -9
  455. package/qoderUsageProbe.ts +252 -0
  456. package/render/contentIcon/types.ts +167 -0
  457. package/render/page/createPageAction.ts +183 -0
  458. package/render/page/docSlice.ts +495 -0
  459. package/render/page/types.ts +19 -0
  460. package/render/table/createTableAction.ts +274 -0
  461. package/render/table/deleteTableAction.ts +193 -0
  462. package/render/table/fetchAndCacheTableRows.ts +174 -0
  463. package/render/table/tableSlice.ts +1167 -0
  464. package/render/table/toolValueUtils.ts +363 -0
  465. package/render/table/types.ts +262 -0
  466. package/render/table/utils/tableSerialization.ts +50 -0
  467. package/scriptCommandEntries.ts +0 -3
  468. package/secretScan.ts +31 -0
  469. package/server/handlers/agentRun/types.ts +105 -0
  470. package/share/action.ts +225 -0
  471. package/share/helpers.ts +302 -0
  472. package/share/keys.ts +243 -0
  473. package/share/publicSharePolicy.ts +27 -0
  474. package/share/shareSlice.ts +53 -0
  475. package/share/types.ts +100 -0
  476. package/tableCommands.ts +248 -13
  477. package/workflowInternalCommandEntries.ts +2 -3
  478. package/ai/tools/agent/taskRunTool.ts +0 -112
  479. package/cli/client/taskRunPrompt.ts +0 -27
  480. package/cli/taskRunCommand.ts +0 -265
  481. package/client/taskRunPrompt.ts +0 -27
  482. package/taskRunCommand.ts +0 -265
@@ -0,0 +1,921 @@
1
+ // 文件路径: chat/sendOpenAICompletionsRequest.ts
2
+
3
+ import {
4
+ addActiveController,
5
+ removeActiveController,
6
+ tokenUsageLiveUpdate,
7
+ } from "../../chat/dialog/dialogSlice";
8
+ import {
9
+ messageStreamEnd,
10
+ messageStreaming,
11
+ } from "../../chat/messages/messageSlice";
12
+ import { handleToolCalls } from "../../chat/messages/toolThunks";
13
+ import {
14
+ MessageContentPart,
15
+ ImageGenerationState,
16
+ Message,
17
+ OpenAITextContent,
18
+ } from "../../chat/messages/types";
19
+ import { selectCurrentServer } from "../../app/settings/settingSlice";
20
+ import { selectCurrentSpaceId } from "../../create/space/spaceSlice";
21
+ import { getApiEndpoint } from "../llm/providers";
22
+ import { createDialogMessageKeyAndId } from "../../database/keys";
23
+ import { selectCurrentToken } from "../../auth/authSlice";
24
+ import { extractCustomId } from "../../core/prefix";
25
+
26
+ import { performFetchRequest } from "./fetchUtils";
27
+ import { createSSEParser } from "./parseMultilineSSE";
28
+ import { parseApiError } from "./parseApiError";
29
+ import { updateTotalUsage } from "./updateTotalUsage";
30
+ import { accumulateToolCallChunks } from "./accumulateToolCallChunks";
31
+ import { prepareTools } from "../tools/prepareTools";
32
+ import {
33
+ inlineImageUrlsForCustomProvider,
34
+ shouldInlineImageUrlsForAgent,
35
+ } from "./inlineImageUrlsForCustomProvider";
36
+
37
+ import { getModelInfo } from "../llm/getModelContextWindow";
38
+ import { supportsImageGeneration } from "../agent/utils/imageOutput";
39
+
40
+ import type { RootState } from "../../app/store";
41
+
42
+ /**
43
+ * 追加文本 chunk 到 contentBuffer(不可变更新)
44
+ */
45
+ function appendTextChunk(
46
+ currentContentBuffer: MessageContentPart[],
47
+ textChunk: string
48
+ ): MessageContentPart[] {
49
+ if (!textChunk) return currentContentBuffer;
50
+
51
+ const updatedContentBuffer = [...currentContentBuffer];
52
+ const lastIndex = updatedContentBuffer.length - 1;
53
+
54
+ if (lastIndex >= 0 && updatedContentBuffer[lastIndex].type === "text") {
55
+ const last = updatedContentBuffer[lastIndex] as OpenAITextContent;
56
+ updatedContentBuffer[lastIndex] = {
57
+ ...last,
58
+ text: (last.text || "") + textChunk,
59
+ };
60
+ } else {
61
+ updatedContentBuffer.push({ type: "text", text: textChunk });
62
+ }
63
+
64
+ return updatedContentBuffer;
65
+ }
66
+
67
+ /** OpenAI 标准 tool_call 结构 */
68
+ type AssistantToolCall = {
69
+ id: string;
70
+ type: "function";
71
+ function: { name: string; arguments: string };
72
+ };
73
+
74
+ const EXPLICIT_IMAGE_TOOL_NAMES = new Set([
75
+ "openAIGptImage",
76
+ "openAIGptImageGenerate",
77
+ "openAIGptImageEdit",
78
+ "geminiProImagePreview",
79
+ ]);
80
+
81
+ function getStreamErrorMessage(data: any): string {
82
+ const message =
83
+ typeof data?.error?.message === "string" && data.error.message.trim()
84
+ ? data.error.message.trim()
85
+ : typeof data?.error?.msg === "string" && data.error.msg.trim()
86
+ ? data.error.msg.trim()
87
+ : typeof data?.message === "string" && data.message.trim()
88
+ ? data.message.trim()
89
+ : null;
90
+ if (message) return message;
91
+
92
+ const code =
93
+ typeof data?.error?.code === "string" && data.error.code.trim()
94
+ ? data.error.code.trim()
95
+ : typeof data?.code === "string" && data.code.trim()
96
+ ? data.code.trim()
97
+ : null;
98
+ if (code) return code;
99
+
100
+ const type =
101
+ typeof data?.error?.type === "string" && data.error.type.trim()
102
+ ? data.error.type.trim()
103
+ : null;
104
+ if (type) return type;
105
+
106
+ return "Unknown error";
107
+ }
108
+
109
+ function formatStreamErrorMessage(data: any): string {
110
+ const rawMessage = getStreamErrorMessage(data);
111
+ if (
112
+ /prohibited|violation|terms\s+of\s+service|content\s+policy|safety/i.test(
113
+ rawMessage
114
+ )
115
+ ) {
116
+ return "当前模型服务商拒绝了这次请求。你可以稍后重试,或切换到其他模型继续。";
117
+ }
118
+ return rawMessage;
119
+ }
120
+
121
+ function getChoiceFinishErrorMessage(data: any, choice: any): string | null {
122
+ if (choice?.error || choice?.message || choice?.code) {
123
+ const message = formatStreamErrorMessage(choice);
124
+ if (message && message !== "Unknown error") return message;
125
+ }
126
+ if (data?.error || data?.message || data?.code) {
127
+ const message = formatStreamErrorMessage(data);
128
+ if (message && message !== "Unknown error") return message;
129
+ }
130
+ const messageContent = choice?.message?.content;
131
+ if (typeof messageContent === "string" && messageContent.trim()) {
132
+ return messageContent.trim();
133
+ }
134
+ return null;
135
+ }
136
+
137
+ /** 单次流式请求过程中的全部中间状态(显式 state) */
138
+ type StreamState = {
139
+ contentBuffer: MessageContentPart[];
140
+ totalUsage: any | null;
141
+ accumulatedToolCalls: any[];
142
+ reasoningBuffer: string;
143
+ assistantToolCalls?: AssistantToolCall[];
144
+ hasHandedOff: boolean;
145
+ hasProcessedToolCalls: boolean;
146
+ alreadyFinalized: boolean;
147
+ finishReason: string | null;
148
+ };
149
+
150
+ type FinalizeContext = {
151
+ dispatch: any;
152
+ msgKey: string;
153
+ dialogId: string;
154
+ dialogKey: string;
155
+ messageId: string;
156
+ agentConfig: any;
157
+ spaceId?: string;
158
+ messageMetadata?: Partial<Message>;
159
+ };
160
+
161
+ type ToolCallsContext = {
162
+ dispatch: any;
163
+ agentConfig: any;
164
+ dialogId: string;
165
+ dialogKey: string;
166
+ messageId: string;
167
+ messageMetadata?: Partial<Message>;
168
+ };
169
+
170
+ type StreamCompletionContext = {
171
+ dispatch: any;
172
+ dialogId: string;
173
+ dialogKey: string;
174
+ messageId: string;
175
+ agentConfig: any;
176
+ };
177
+
178
+ const withImageGenerationStage = (
179
+ messageMetadata: Partial<Message> | undefined,
180
+ stage: ImageGenerationState["stage"]
181
+ ): Partial<Message> | undefined => {
182
+ const previousState = messageMetadata?.imageGenerationState;
183
+ if (previousState?.kind !== "image_generation") {
184
+ return messageMetadata;
185
+ }
186
+
187
+ return {
188
+ ...(messageMetadata ?? {}),
189
+ imageGenerationState: {
190
+ ...previousState,
191
+ stage,
192
+ },
193
+ };
194
+ };
195
+
196
+ const shouldShowImageSavingState = (contentBuffer: MessageContentPart[]) =>
197
+ contentBuffer.some((part) => part?.type === "image_url") &&
198
+ !contentBuffer.some(
199
+ (part) =>
200
+ part?.type === "text" &&
201
+ typeof part.text === "string" &&
202
+ part.text.trim().length > 0
203
+ );
204
+
205
+ /** 单轮调用后返回给 Agent Loop 的元信息 */
206
+ export type CompletionMeta = {
207
+ hasToolCalls: boolean;
208
+ hasPendingInteraction: boolean;
209
+ hasHandedOff: boolean;
210
+ finishReason: string | null;
211
+ messageId: string;
212
+ usage?: any;
213
+ };
214
+
215
+ const STREAM_READ_TIMEOUT_MS = 45_000;
216
+
217
+ const logQuickChatPerfStage = (
218
+ startedAt: number | undefined,
219
+ stage: string,
220
+ details?: Record<string, unknown>
221
+ ) => {
222
+ if (!startedAt) return;
223
+ const now = typeof performance !== "undefined" ? performance.now() : Date.now();
224
+ console.info("[QuickChatPerf]", {
225
+ stage,
226
+ elapsedMs: now - startedAt,
227
+ ...(typeof performance !== "undefined" ? { atMs: now } : {}),
228
+ ...(details ?? {}),
229
+ });
230
+ };
231
+
232
+ /**
233
+ * 初始化流式状态
234
+ */
235
+ function createInitialStreamState(): StreamState {
236
+ return {
237
+ contentBuffer: [],
238
+ totalUsage: null,
239
+ accumulatedToolCalls: [],
240
+ reasoningBuffer: "",
241
+ assistantToolCalls: undefined,
242
+ hasHandedOff: false,
243
+ hasProcessedToolCalls: false,
244
+ alreadyFinalized: false,
245
+ finishReason: null,
246
+ };
247
+ }
248
+
249
+ /**
250
+ * 根据 tools 配置生成本次请求体(不改动外部传入的 bodyData)
251
+ */
252
+ function buildRequestBodyWithTools(
253
+ bodyData: any,
254
+ agentConfig: any,
255
+ disableToolsForThisRequest: boolean
256
+ ): any {
257
+ if (disableToolsForThisRequest) return bodyData;
258
+
259
+ // 如果模型拥有图像输出能力,则不组装任何 tools
260
+ const modelInfo = getModelInfo(agentConfig.model);
261
+ if (modelInfo?.hasImageOutput) {
262
+ return bodyData;
263
+ }
264
+
265
+ const rawTools = agentConfig.tools;
266
+ if (!Array.isArray(rawTools) || rawTools.length === 0) return bodyData;
267
+ const hasExplicitImageTool = rawTools.some(
268
+ (toolName: unknown) =>
269
+ typeof toolName === "string" && EXPLICIT_IMAGE_TOOL_NAMES.has(toolName)
270
+ );
271
+ if (supportsImageGeneration(agentConfig) && !hasExplicitImageTool) {
272
+ return bodyData;
273
+ }
274
+
275
+ const tools = prepareTools(rawTools, { provider: agentConfig.provider });
276
+ if (!tools.length) return bodyData;
277
+
278
+ return {
279
+ ...bodyData,
280
+ tools,
281
+ tool_choice: bodyData.tool_choice ?? "auto",
282
+ };
283
+ }
284
+
285
+ /**
286
+ * 结束当前流式消息,派发 messageStreamEnd(带幂等保护)
287
+ */
288
+ async function finalizeStream(
289
+ state: StreamState,
290
+ ctx: FinalizeContext
291
+ ): Promise<StreamState> {
292
+ if (state.hasHandedOff || state.alreadyFinalized) return state;
293
+
294
+ const savingMetadata = withImageGenerationStage(
295
+ ctx.messageMetadata,
296
+ "saving"
297
+ );
298
+ if (savingMetadata && shouldShowImageSavingState(state.contentBuffer)) {
299
+ ctx.dispatch(
300
+ messageStreaming({
301
+ id: ctx.messageId,
302
+ dialogId: ctx.dialogId,
303
+ dbKey: ctx.msgKey,
304
+ content: "",
305
+ thinkContent: state.reasoningBuffer,
306
+ role: "assistant",
307
+ agentKey: ctx.agentConfig.dbKey,
308
+ cybotKey: ctx.agentConfig.dbKey,
309
+ ...(typeof ctx.agentConfig?.name === "string" && ctx.agentConfig.name.trim()
310
+ ? { agentName: ctx.agentConfig.name.trim() }
311
+ : {}),
312
+ ...(savingMetadata ?? {}),
313
+ })
314
+ );
315
+ ctx.messageMetadata = savingMetadata;
316
+ }
317
+
318
+ await ctx.dispatch(
319
+ messageStreamEnd({
320
+ finalContentBuffer: state.contentBuffer,
321
+ totalUsage: state.totalUsage,
322
+ msgKey: ctx.msgKey,
323
+ agentConfig: ctx.agentConfig,
324
+ dialogId: ctx.dialogId,
325
+ dialogKey: ctx.dialogKey,
326
+ messageId: ctx.messageId,
327
+ reasoningBuffer: state.reasoningBuffer,
328
+ messageMetadata: ctx.messageMetadata,
329
+ toolCalls: state.assistantToolCalls,
330
+ spaceId: ctx.spaceId,
331
+ })
332
+ );
333
+
334
+ return {
335
+ ...state,
336
+ alreadyFinalized: true,
337
+ };
338
+ }
339
+
340
+ /**
341
+ * 累积 usage
342
+ */
343
+ function applyUsage(state: StreamState, data: any): StreamState {
344
+ if (!data.usage) return state;
345
+
346
+ return {
347
+ ...state,
348
+ totalUsage: updateTotalUsage(state.totalUsage, data.usage),
349
+ };
350
+ }
351
+
352
+ /**
353
+ * 处理单个 delta(文本 / 推理 / tool_calls / 图片)
354
+ */
355
+ function applyDelta(
356
+ state: StreamState,
357
+ delta: any
358
+ ): { state: StreamState; hasNewVisibleContent: boolean } {
359
+ let hasNewVisibleContent = false;
360
+ let next: StreamState = { ...state };
361
+
362
+ // reasoning_content 增量(DeepSeek)/ reasoning 增量(Ollama/Qwen3)
363
+ const reasoningChunk: string = delta.reasoning_content ?? delta.reasoning ?? "";
364
+ if (reasoningChunk) {
365
+ next.reasoningBuffer = (next.reasoningBuffer || "") + reasoningChunk;
366
+ }
367
+
368
+ // tool_calls 累积 + 映射为 OpenAI 标准格式
369
+ if (Array.isArray(delta.tool_calls) && delta.tool_calls.length > 0) {
370
+ const accumulated = accumulateToolCallChunks(
371
+ next.accumulatedToolCalls,
372
+ delta.tool_calls
373
+ );
374
+
375
+ next = {
376
+ ...next,
377
+ accumulatedToolCalls: accumulated,
378
+ assistantToolCalls: accumulated.map((call: any) => ({
379
+ id: call.id,
380
+ type: "function",
381
+ function: {
382
+ name: call.function?.name,
383
+ arguments:
384
+ typeof call.function?.arguments === "string"
385
+ ? call.function.arguments
386
+ : JSON.stringify(call.function?.arguments ?? {}),
387
+ },
388
+ })),
389
+ };
390
+ }
391
+
392
+ // 图片增量
393
+ const deltaAny = delta as any;
394
+ if (Array.isArray(deltaAny.images) && deltaAny.images.length > 0) {
395
+ next = {
396
+ ...next,
397
+ contentBuffer: [...next.contentBuffer, ...deltaAny.images],
398
+ };
399
+ hasNewVisibleContent = true;
400
+ }
401
+
402
+ // 文本增量
403
+ const contentChunk = delta.content || "";
404
+ if (contentChunk) {
405
+ next = {
406
+ ...next,
407
+ contentBuffer: appendTextChunk(next.contentBuffer, contentChunk),
408
+ };
409
+ hasNewVisibleContent = true;
410
+ }
411
+
412
+ return { state: next, hasNewVisibleContent };
413
+ }
414
+
415
+ /**
416
+ * 如果有新可见内容,则派发一次 messageStreaming 更新前端
417
+ */
418
+ function emitStreamingUpdate(
419
+ hasNewVisibleContent: boolean,
420
+ state: StreamState,
421
+ ctx: {
422
+ dispatch: any;
423
+ agentConfig: any;
424
+ messageId: string;
425
+ msgKey: string;
426
+ dialogId: string;
427
+ messageMetadata?: Partial<Message>;
428
+ }
429
+ ) {
430
+ if (!hasNewVisibleContent) return;
431
+
432
+ ctx.dispatch(
433
+ messageStreaming({
434
+ id: ctx.messageId,
435
+ dialogId: ctx.dialogId,
436
+ dbKey: ctx.msgKey,
437
+ content: state.contentBuffer,
438
+ thinkContent: state.reasoningBuffer,
439
+ role: "assistant",
440
+ agentKey: ctx.agentConfig.dbKey,
441
+ cybotKey: ctx.agentConfig.dbKey,
442
+ ...(typeof ctx.agentConfig?.name === "string" && ctx.agentConfig.name.trim()
443
+ ? { agentName: ctx.agentConfig.name.trim() }
444
+ : {}),
445
+ ...(ctx.messageMetadata ?? {}),
446
+ })
447
+ );
448
+ }
449
+
450
+ /**
451
+ * 处理已经累积好的 tool_calls
452
+ */
453
+ async function processAccumulatedToolCalls(
454
+ state: StreamState,
455
+ ctx: ToolCallsContext
456
+ ): Promise<{
457
+ state: StreamState;
458
+ hasHandedOff: boolean;
459
+ hasPendingInteraction: boolean;
460
+ }> {
461
+ if (!state.accumulatedToolCalls.length) {
462
+ return {
463
+ state,
464
+ hasHandedOff: false,
465
+ hasPendingInteraction: false,
466
+ };
467
+ }
468
+
469
+ const result = await ctx
470
+ .dispatch(
471
+ handleToolCalls({
472
+ accumulatedCalls: state.accumulatedToolCalls,
473
+ currentContentBuffer: state.contentBuffer,
474
+ agentConfig: ctx.agentConfig,
475
+ messageId: ctx.messageId,
476
+ dialogId: ctx.dialogId,
477
+ dialogKey: ctx.dialogKey,
478
+ parallelSessionId: ctx.messageMetadata?.parallelSessionId,
479
+ parallelBranchId: ctx.messageMetadata?.parallelBranchId,
480
+ parallelLabel: ctx.messageMetadata?.parallelLabel,
481
+ parallelIndex: ctx.messageMetadata?.parallelIndex,
482
+ })
483
+ )
484
+ .unwrap();
485
+
486
+ const nextState: StreamState = {
487
+ ...state,
488
+ contentBuffer: result.finalContentBuffer,
489
+ accumulatedToolCalls: [],
490
+ hasProcessedToolCalls: true,
491
+ hasHandedOff: result.hasHandedOff,
492
+ };
493
+
494
+ return {
495
+ state: nextState,
496
+ hasHandedOff: result.hasHandedOff,
497
+ hasPendingInteraction: result.hasPendingInteraction,
498
+ };
499
+ }
500
+
501
+ async function readStreamChunkWithTimeout(
502
+ reader: ReadableStreamDefaultReader<Uint8Array>
503
+ ) {
504
+ let timeoutId: ReturnType<typeof setTimeout> | undefined;
505
+ try {
506
+ return await Promise.race([
507
+ reader.read(),
508
+ new Promise<ReadableStreamReadResult<Uint8Array>>((_, reject) => {
509
+ timeoutId = setTimeout(() => {
510
+ reject(
511
+ new Error(
512
+ `模型响应流 ${Math.round(STREAM_READ_TIMEOUT_MS / 1000)} 秒内没有返回新内容`
513
+ )
514
+ );
515
+ }, STREAM_READ_TIMEOUT_MS);
516
+ }),
517
+ ]);
518
+ } finally {
519
+ if (timeoutId) clearTimeout(timeoutId);
520
+ }
521
+ }
522
+
523
+ /**
524
+ * 处理「流结束」场景(done === true)
525
+ */
526
+ async function handleStreamCompletion(
527
+ state: StreamState,
528
+ ctx: StreamCompletionContext,
529
+ finalizeCtx: FinalizeContext
530
+ ): Promise<{
531
+ state: StreamState;
532
+ hasHandedOff: boolean;
533
+ hasPendingInteraction: boolean;
534
+ }> {
535
+ let hasHandedOff = false;
536
+ let hasPendingInteraction = false;
537
+
538
+ if (!state.hasProcessedToolCalls && state.accumulatedToolCalls.length > 0) {
539
+ // Tool 开始前先更新 TopBar token 显示
540
+ if (state.totalUsage) {
541
+ ctx.dispatch(tokenUsageLiveUpdate({
542
+ input_tokens: state.totalUsage.prompt_tokens ?? state.totalUsage.input_tokens,
543
+ output_tokens: state.totalUsage.completion_tokens ?? state.totalUsage.output_tokens,
544
+ cost: state.totalUsage.cost,
545
+ dialogKey: ctx.dialogKey,
546
+ }));
547
+ }
548
+ const toolResult = await processAccumulatedToolCalls(state, {
549
+ dispatch: ctx.dispatch,
550
+ agentConfig: ctx.agentConfig,
551
+ dialogId: ctx.dialogId,
552
+ dialogKey: ctx.dialogKey,
553
+ messageId: ctx.messageId,
554
+ messageMetadata: ctx.messageMetadata,
555
+ });
556
+
557
+ let next = toolResult.state;
558
+ hasHandedOff ||= toolResult.hasHandedOff;
559
+ hasPendingInteraction ||= toolResult.hasPendingInteraction;
560
+
561
+ next = await finalizeStream(next, finalizeCtx);
562
+ return { state: next, hasHandedOff, hasPendingInteraction };
563
+ }
564
+
565
+ const finalized = await finalizeStream(state, finalizeCtx);
566
+ return { state: finalized, hasHandedOff, hasPendingInteraction };
567
+ }
568
+
569
+ /**
570
+ * 主流程:发送请求 + 处理流式 SSE + 工具调用
571
+ */
572
+ export const sendOpenAICompletionsRequest = async ({
573
+ bodyData,
574
+ agentConfig,
575
+ thunkApi,
576
+ dialogKey,
577
+ parentMessageId,
578
+ messageMetadata,
579
+ disableToolsForThisRequest = false,
580
+ quickChatPerfStartedAt,
581
+ }: {
582
+ bodyData: any;
583
+ agentConfig: any;
584
+ thunkApi: any;
585
+ dialogKey: string;
586
+ parentMessageId?: string;
587
+ messageMetadata?: Partial<Message>;
588
+ disableToolsForThisRequest?: boolean;
589
+ quickChatPerfStartedAt?: number;
590
+ }): Promise<CompletionMeta> => {
591
+ const { dispatch, getState, signal: thunkSignal } = thunkApi;
592
+
593
+ const dialogId = extractCustomId(dialogKey);
594
+ const controller = new AbortController();
595
+ thunkSignal.addEventListener("abort", () => controller.abort());
596
+ const signal = controller.signal;
597
+ const streamSpaceId = selectCurrentSpaceId(getState() as RootState) || undefined;
598
+
599
+ let messageId: string;
600
+ let msgKey: string;
601
+
602
+ if (parentMessageId) {
603
+ messageId = parentMessageId;
604
+ msgKey = `msg:${dialogId}:${messageId}`;
605
+ } else {
606
+ const newIds = createDialogMessageKeyAndId(dialogId);
607
+ messageId = newIds.messageId;
608
+ msgKey = newIds.key;
609
+ }
610
+
611
+ dispatch(addActiveController({ messageId, controller, dialogKey }));
612
+
613
+ const requestBody = await inlineImageUrlsForCustomProvider(
614
+ buildRequestBodyWithTools(
615
+ bodyData,
616
+ agentConfig,
617
+ disableToolsForThisRequest
618
+ ),
619
+ {
620
+ shouldInline: shouldInlineImageUrlsForAgent(agentConfig),
621
+ },
622
+ );
623
+
624
+ let streamState: StreamState = createInitialStreamState();
625
+ const parseSSE = createSSEParser();
626
+ let reader: ReadableStreamDefaultReader<Uint8Array> | undefined;
627
+
628
+ const finalizeCtx: FinalizeContext = {
629
+ dispatch,
630
+ msgKey,
631
+ dialogId,
632
+ dialogKey,
633
+ messageId,
634
+ agentConfig,
635
+ spaceId: streamSpaceId,
636
+ messageMetadata,
637
+ };
638
+
639
+ let hasHandedOffOverall = false;
640
+ let hasPendingInteractionOverall = false;
641
+ let lastFinishReason: string | null = null;
642
+ let activeMessageMetadata = messageMetadata;
643
+
644
+ const buildMeta = (): CompletionMeta => ({
645
+ hasToolCalls:
646
+ Array.isArray(streamState.assistantToolCalls) &&
647
+ streamState.assistantToolCalls.length > 0,
648
+ hasPendingInteraction: hasPendingInteractionOverall,
649
+ hasHandedOff: hasHandedOffOverall,
650
+ finishReason: lastFinishReason,
651
+ messageId,
652
+ usage: streamState.totalUsage ?? undefined,
653
+ });
654
+
655
+ try {
656
+ if (!parentMessageId) {
657
+ dispatch(
658
+ messageStreaming({
659
+ id: messageId,
660
+ dialogId,
661
+ dbKey: msgKey,
662
+ content: "",
663
+ role: "assistant",
664
+ agentKey: agentConfig.dbKey,
665
+ cybotKey: agentConfig.dbKey,
666
+ ...(typeof agentConfig?.name === "string" && agentConfig.name.trim()
667
+ ? { agentName: agentConfig.name.trim() }
668
+ : {}),
669
+ ...(activeMessageMetadata ?? {}),
670
+ isStreaming: true,
671
+ })
672
+ );
673
+ }
674
+
675
+ const api = getApiEndpoint(agentConfig);
676
+ const token = selectCurrentToken(getState() as RootState);
677
+ logQuickChatPerfStage(quickChatPerfStartedAt, "openai-completions-fetch-starting", {
678
+ api,
679
+ dialogKey,
680
+ });
681
+ const response = await performFetchRequest({
682
+ agentConfig,
683
+ api,
684
+ bodyData: requestBody,
685
+ currentServer: selectCurrentServer(getState() as RootState),
686
+ signal,
687
+ token,
688
+ });
689
+ logQuickChatPerfStage(quickChatPerfStartedAt, "openai-completions-fetch-response", {
690
+ ok: response.ok,
691
+ status: response.status,
692
+ dialogKey,
693
+ });
694
+ activeMessageMetadata = withImageGenerationStage(
695
+ activeMessageMetadata,
696
+ "generating"
697
+ );
698
+ finalizeCtx.messageMetadata = activeMessageMetadata;
699
+ if (!parentMessageId && activeMessageMetadata?.imageGenerationState) {
700
+ dispatch(
701
+ messageStreaming({
702
+ id: messageId,
703
+ dialogId,
704
+ dbKey: msgKey,
705
+ content: "",
706
+ role: "assistant",
707
+ agentKey: agentConfig.dbKey,
708
+ cybotKey: agentConfig.dbKey,
709
+ ...(typeof agentConfig?.name === "string" && agentConfig.name.trim()
710
+ ? { agentName: agentConfig.name.trim() }
711
+ : {}),
712
+ ...(activeMessageMetadata ?? {}),
713
+ isStreaming: true,
714
+ })
715
+ );
716
+ }
717
+
718
+ if (!response.ok) {
719
+ const errorMessage = await parseApiError(response);
720
+ streamState = {
721
+ ...streamState,
722
+ contentBuffer: appendTextChunk(
723
+ streamState.contentBuffer,
724
+ `[错误: ${errorMessage}]`
725
+ ),
726
+ };
727
+ streamState = await finalizeStream(streamState, finalizeCtx);
728
+ return buildMeta();
729
+ }
730
+
731
+ reader = response.body?.getReader();
732
+ if (!reader) {
733
+ streamState = await finalizeStream(streamState, finalizeCtx);
734
+ return buildMeta();
735
+ }
736
+
737
+ const decoder = new TextDecoder();
738
+ let loggedFirstStreamChunk = false;
739
+ let loggedFirstParsedEvent = false;
740
+ let loggedFirstVisibleDelta = false;
741
+
742
+ while (true) {
743
+ const { done, value } = await readStreamChunkWithTimeout(reader);
744
+
745
+ if (done) {
746
+ const completion = await handleStreamCompletion(
747
+ streamState,
748
+ {
749
+ dispatch,
750
+ dialogId,
751
+ dialogKey: finalizeCtx.dialogKey,
752
+ messageId,
753
+ agentConfig,
754
+ },
755
+ finalizeCtx
756
+ );
757
+
758
+ streamState = completion.state;
759
+ hasHandedOffOverall ||= completion.hasHandedOff;
760
+ hasPendingInteractionOverall ||= completion.hasPendingInteraction;
761
+
762
+ break;
763
+ }
764
+
765
+ if (!loggedFirstStreamChunk) {
766
+ loggedFirstStreamChunk = true;
767
+ logQuickChatPerfStage(
768
+ quickChatPerfStartedAt,
769
+ "openai-completions-first-stream-chunk",
770
+ { dialogKey, byteLength: value.byteLength }
771
+ );
772
+ }
773
+
774
+ const chunk = decoder.decode(value, { stream: true });
775
+ const parsedResults = parseSSE(chunk);
776
+ if (parsedResults.length > 0 && !loggedFirstParsedEvent) {
777
+ loggedFirstParsedEvent = true;
778
+ logQuickChatPerfStage(
779
+ quickChatPerfStartedAt,
780
+ "openai-completions-first-sse-event",
781
+ { dialogKey, eventCount: parsedResults.length }
782
+ );
783
+ }
784
+
785
+ for (const parsedData of parsedResults) {
786
+ const dataList = Array.isArray(parsedData) ? parsedData : [parsedData];
787
+
788
+ for (const data of dataList) {
789
+ streamState = applyUsage(streamState, data);
790
+
791
+ if (data === "[DONE]") {
792
+ // Web 端通常依赖 reader.read() 的 done 状态来结束,
793
+ // 但如果流中显式包含 [DONE],也可以直接 break。
794
+ // 这里我们选择忽略它,让 reader 自然结束。
795
+ continue;
796
+ }
797
+
798
+ if (data.error) {
799
+ const errorMsg = `Error: ${formatStreamErrorMessage(data)}`;
800
+ streamState = {
801
+ ...streamState,
802
+ contentBuffer: appendTextChunk(
803
+ streamState.contentBuffer,
804
+ `\n[API Error] ${errorMsg}`
805
+ ),
806
+ };
807
+ await reader.cancel();
808
+ break;
809
+ }
810
+
811
+ const choice = data.choices?.[0];
812
+ if (!choice) continue;
813
+
814
+ const delta = choice.delta || {};
815
+
816
+ const { state: updatedState, hasNewVisibleContent } = applyDelta(
817
+ streamState,
818
+ delta
819
+ );
820
+ streamState = updatedState;
821
+ if (hasNewVisibleContent && !loggedFirstVisibleDelta) {
822
+ loggedFirstVisibleDelta = true;
823
+ logQuickChatPerfStage(
824
+ quickChatPerfStartedAt,
825
+ "openai-completions-first-visible-delta",
826
+ { dialogKey }
827
+ );
828
+ }
829
+
830
+ emitStreamingUpdate(hasNewVisibleContent, streamState, {
831
+ dispatch,
832
+ agentConfig,
833
+ messageId,
834
+ msgKey,
835
+ dialogId,
836
+ messageMetadata: activeMessageMetadata,
837
+ });
838
+
839
+ const finishReason = choice.finish_reason;
840
+ if (finishReason) {
841
+ lastFinishReason = finishReason;
842
+ streamState.finishReason = finishReason;
843
+
844
+ if (finishReason === "tool_calls") {
845
+ // Tool 开始前先更新 TopBar token 显示
846
+ if (streamState.totalUsage) {
847
+ dispatch(tokenUsageLiveUpdate({
848
+ input_tokens: streamState.totalUsage.prompt_tokens ?? streamState.totalUsage.input_tokens,
849
+ output_tokens: streamState.totalUsage.completion_tokens ?? streamState.totalUsage.output_tokens,
850
+ cost: streamState.totalUsage.cost,
851
+ dialogKey,
852
+ }));
853
+ }
854
+ const toolResult = await processAccumulatedToolCalls(
855
+ streamState,
856
+ {
857
+ dispatch,
858
+ agentConfig,
859
+ dialogId,
860
+ dialogKey,
861
+ messageId,
862
+ messageMetadata,
863
+ }
864
+ );
865
+
866
+ streamState = toolResult.state;
867
+ hasHandedOffOverall ||= toolResult.hasHandedOff;
868
+ hasPendingInteractionOverall ||= toolResult.hasPendingInteraction;
869
+
870
+ // 无论是否 handoff,都 finalize 当前 assistant 消息(stub)
871
+ streamState = await finalizeStream(streamState, finalizeCtx);
872
+ } else if (finishReason !== "stop") {
873
+ const finishErrorMessage =
874
+ finishReason === "error"
875
+ ? getChoiceFinishErrorMessage(data, choice)
876
+ : null;
877
+ streamState = {
878
+ ...streamState,
879
+ contentBuffer: appendTextChunk(
880
+ streamState.contentBuffer,
881
+ finishErrorMessage
882
+ ? `\n[API Error] Error: ${finishErrorMessage}`
883
+ : finishReason === "error"
884
+ ? "\n[API Error] Error: 模型响应以 error 结束,但上游未返回具体错误。请重试,或切换到其他支持图片输入的模型。"
885
+ : `\n[流结束原因: ${finishReason}]`
886
+ ),
887
+ };
888
+ }
889
+ }
890
+ }
891
+ }
892
+ }
893
+ } catch (error: any) {
894
+ let errorText: string;
895
+ if (error?.name === "AbortError") {
896
+ errorText = "\n[用户中断]";
897
+ } else {
898
+ errorText = `\n[错误: ${error?.message || String(error)}]`;
899
+ }
900
+
901
+ console.error("[SSE] sendOpenAICompletionsRequest error:", error);
902
+
903
+ streamState = {
904
+ ...streamState,
905
+ contentBuffer: appendTextChunk(streamState.contentBuffer, errorText),
906
+ };
907
+ streamState = await finalizeStream(streamState, finalizeCtx);
908
+ } finally {
909
+ logQuickChatPerfStage(quickChatPerfStartedAt, "openai-completions-stream-finished", {
910
+ dialogKey,
911
+ });
912
+ dispatch(removeActiveController({ messageId, dialogKey }));
913
+ try {
914
+ await reader?.cancel();
915
+ } catch (_e) {
916
+ // ignore
917
+ }
918
+ }
919
+
920
+ return buildMeta();
921
+ };