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.
- package/README.md +9 -2
- package/agent-runtime/hostAdapter.ts +53 -0
- package/agent-runtime/index.ts +28 -0
- package/agent-runtime/localLoop.ts +62 -0
- package/agent-runtime/runtimeDecision.ts +70 -0
- package/agent-runtime/types.ts +87 -0
- package/agentRuntimeCommands.ts +139 -22
- package/agentRuntimeLocal.ts +7 -0
- package/ai/agent/_executeModel.ts +118 -0
- package/ai/agent/agentSlice.ts +544 -1
- package/ai/agent/appWorkingMemory.ts +126 -0
- package/ai/agent/avatarUtils.ts +24 -0
- package/ai/agent/buildEditingContext.ts +373 -0
- package/ai/agent/buildSystemPrompt.ts +532 -0
- package/ai/agent/cleanAgentMessages.ts +140 -0
- package/ai/agent/cliChatClient.ts +119 -0
- package/ai/agent/contextCompiler.ts +107 -0
- package/ai/agent/contextLayerContract.ts +44 -0
- package/ai/agent/createAgentSchema.ts +234 -0
- package/ai/agent/executeToolCall.ts +58 -0
- package/ai/agent/fetchAgentContexts.ts +42 -0
- package/ai/agent/generatePrompt.ts +3 -0
- package/ai/agent/getFullChatContextKeys.ts +168 -0
- package/ai/agent/hooks/fetchPublicAgents.ts +133 -0
- package/ai/agent/hooks/useAgentConfig.ts +61 -0
- package/ai/agent/hooks/useAgentDialog.ts +35 -0
- package/ai/agent/hooks/useAgentFormValidation.ts +202 -0
- package/ai/agent/hooks/usePublicAgents.ts +473 -0
- package/ai/agent/persistMessageWithFixedId.ts +37 -0
- package/ai/agent/planSlice.ts +259 -0
- package/ai/agent/referenceUtils.ts +229 -0
- package/ai/agent/runAgentBackground.ts +238 -0
- package/ai/agent/runAgentClientLoop.ts +138 -0
- package/ai/agent/runtimeGuidance.ts +97 -0
- package/ai/agent/runtimeServerBase.ts +37 -0
- package/ai/agent/server/fetchPublicAgents.ts +128 -0
- package/ai/agent/startParallelAgentStreams.ts +424 -0
- package/ai/agent/startupProtocol.ts +53 -0
- package/ai/agent/streamAgentChatTurn.ts +1299 -0
- package/ai/agent/streamAgentChatTurnUtils.ts +738 -0
- package/ai/agent/types.ts +71 -0
- package/ai/agent/utils/imageOutput.ts +39 -0
- package/ai/agent/utils/publicImageAgentMode.ts +26 -0
- package/ai/agent/utils/sortUtils.ts +250 -0
- package/ai/agent/web/referencePickerUtils.ts +146 -0
- package/ai/ai.locale.ts +1083 -0
- package/ai/chat/accumulateToolCallChunks.ts +95 -0
- package/ai/chat/fetchUtils.native.ts +276 -0
- package/ai/chat/fetchUtils.ts +153 -0
- package/ai/chat/inlineImageUrlsForCustomProvider.ts +117 -0
- package/ai/chat/parseApiError.ts +64 -0
- package/ai/chat/parseMultilineSSE.ts +95 -0
- package/ai/chat/sendOpenAICompletionsRequest.native.ts +682 -0
- package/ai/chat/sendOpenAICompletionsRequest.ts +712 -0
- package/ai/chat/sendOpenAIResponseRequest.ts +512 -0
- package/ai/chat/shouldUseServerProxy.ts +18 -0
- package/ai/chat/sseClient.native.ts +91 -0
- package/ai/chat/sseClient.ts +67 -0
- package/ai/chat/streamReader.native.ts +31 -0
- package/ai/chat/streamReader.ts +62 -0
- package/ai/chat/updateTotalUsage.ts +72 -0
- package/ai/context/buildReferenceContext.ts +437 -0
- package/ai/context/calculateContextUsage.ts +133 -0
- package/ai/context/retention.ts +165 -0
- package/ai/context/tokenUtils.ts +78 -0
- package/ai/index.ts +1 -1
- package/ai/llm/agentCapabilities.ts +74 -0
- package/ai/llm/calculateGeminiImageTokens.ts +57 -0
- package/ai/llm/deepinfra.ts +28 -0
- package/ai/llm/fireworks.ts +68 -0
- package/ai/llm/generateRequestBody.ts +165 -0
- package/ai/llm/getModelContextWindow.ts +84 -0
- package/ai/llm/getNoloKey.ts +37 -0
- package/ai/llm/getPricing.ts +232 -0
- package/ai/llm/hooks/useModelPricing.ts +75 -0
- package/ai/llm/imagePricing.ts +66 -0
- package/ai/llm/isResponseAPIModel.ts +13 -0
- package/ai/llm/kimi.ts +18 -0
- package/ai/llm/mimo.ts +71 -0
- package/ai/llm/mistral.ts +22 -0
- package/ai/llm/modelAvatar.ts +427 -0
- package/ai/llm/models.ts +45 -0
- package/ai/llm/openrouterModels.ts +141 -0
- package/ai/llm/providers.ts +307 -0
- package/ai/llm/reasoningModels.ts +28 -0
- package/ai/llm/types.ts +59 -0
- package/ai/llm/usageRequestOptions.ts +59 -0
- package/ai/memory/capture.ts +148 -0
- package/ai/memory/consolidate.ts +104 -0
- package/ai/memory/delete.ts +147 -0
- package/ai/memory/overlay.ts +84 -0
- package/ai/memory/query.ts +38 -0
- package/ai/memory/queryShared.ts +160 -0
- package/ai/memory/rank.ts +105 -0
- package/ai/memory/recentRelationshipRecap.ts +247 -0
- package/ai/memory/remember.ts +167 -0
- package/ai/memory/runtime.ts +76 -0
- package/ai/memory/store.ts +20 -0
- package/ai/memory/storeShared.ts +76 -0
- package/ai/memory/types.ts +46 -0
- package/ai/memory/understanding.ts +349 -0
- package/ai/memory/understandingGreeting.ts +264 -0
- package/ai/messages/type.ts +20 -0
- package/ai/policy/personalizationDialog.ts +333 -0
- package/ai/policy/runtimePolicy.ts +440 -0
- package/ai/policy/selfUpdateFields.ts +48 -0
- package/ai/policy/types.ts +64 -0
- package/ai/skills/referenceRuntime.ts +274 -0
- package/ai/skills/skillDiagnostics.ts +251 -0
- package/ai/skills/skillDocBuilder.ts +139 -0
- package/ai/skills/skillDocProtocol.ts +434 -0
- package/ai/skills/skillReferenceSummary.ts +63 -0
- package/ai/skills/skillSummaryMarker.ts +26 -0
- package/ai/token/calculatePrice.ts +546 -0
- package/ai/token/db.ts +98 -0
- package/ai/token/externalToolCost.ts +321 -0
- package/ai/token/hooks/useRecords.ts +65 -0
- package/ai/token/missingUsageEstimate.ts +42 -0
- package/ai/token/modelUsageQuery.ts +252 -0
- package/ai/token/normalizeUsage.ts +84 -0
- package/ai/token/openaiImageGenerationUsage.ts +56 -0
- package/ai/token/prepareTokenUsageData.ts +88 -0
- package/ai/token/query.ts +88 -0
- package/ai/token/queryUserTokens.ts +59 -0
- package/ai/token/resolveBillingTarget.ts +52 -0
- package/ai/token/saveTokenRecord.ts +53 -0
- package/ai/token/serverDialogProjection.ts +78 -0
- package/ai/token/serverTokenWriter.ts +143 -0
- package/ai/token/stats.ts +21 -0
- package/ai/token/tokenThunks.ts +24 -0
- package/ai/token/types.ts +93 -0
- package/ai/tools/agent/agentTools.ts +176 -0
- package/ai/tools/agent/agentUpdateShared.ts +311 -0
- package/ai/tools/agent/callAgentTool.ts +139 -0
- package/ai/tools/agent/createAgentTool.ts +512 -0
- package/ai/tools/agent/createDialogTool.ts +69 -0
- package/ai/tools/agent/createSkillAgentTool.ts +62 -0
- package/ai/tools/agent/parallelBudget.ts +221 -0
- package/ai/tools/agent/presets/appBuilderPreset.ts +147 -0
- package/ai/tools/agent/runLlmTool.ts +96 -0
- package/ai/tools/agent/runStreamingAgentTool.ts +73 -0
- package/ai/tools/agent/skillAgentArgs.ts +106 -0
- package/ai/tools/agent/skillAgentPreset.ts +89 -0
- package/ai/tools/agent/streamParallelAgentsTool.ts +122 -0
- package/ai/tools/agent/updateAgentTool.ts +96 -0
- package/ai/tools/agent/updateSelfTool.ts +113 -0
- package/ai/tools/amazonProductScraperTool.ts +86 -0
- package/ai/tools/apifyActorClient.ts +45 -0
- package/ai/tools/appEditGuard.ts +372 -0
- package/ai/tools/appReadSnapshot.ts +153 -0
- package/ai/tools/appTools.ts +1549 -0
- package/ai/tools/applyEditTool.ts +256 -0
- package/ai/tools/applyLineEditsTool.ts +312 -0
- package/ai/tools/browserTools/click.ts +33 -0
- package/ai/tools/browserTools/closeSession.ts +29 -0
- package/ai/tools/browserTools/common.ts +27 -0
- package/ai/tools/browserTools/openSession.ts +48 -0
- package/ai/tools/browserTools/readContent.ts +38 -0
- package/ai/tools/browserTools/selectOption.ts +46 -0
- package/ai/tools/browserTools/typeText.ts +42 -0
- package/ai/tools/category/createCategoryTool.ts +66 -0
- package/ai/tools/category/queryContentsByCategoryTool.ts +69 -0
- package/ai/tools/category/updateContentCategoryTool.ts +75 -0
- package/ai/tools/cfBrowserTools.ts +319 -0
- package/ai/tools/cfSpeechToTextTool.ts +49 -0
- package/ai/tools/checkEnvTool.ts +65 -0
- package/ai/tools/cloudflareCrawlTool.ts +289 -0
- package/ai/tools/codeSearchTool.ts +111 -0
- package/ai/tools/codeTools.ts +101 -0
- package/ai/tools/createDocTool.ts +132 -0
- package/ai/tools/createPlanTool.ts +999 -0
- package/ai/tools/createSkillDocTool.ts +155 -0
- package/ai/tools/createWorkflowTool.ts +154 -0
- package/ai/tools/deepseekOcrTool.ts +34 -0
- package/ai/tools/delayTool.ts +31 -0
- package/ai/tools/deleteSpacesTool.ts +325 -0
- package/ai/tools/deleteSpacesToolModel.ts +159 -0
- package/ai/tools/devReloadUtils.ts +29 -0
- package/ai/tools/dialogMessageSearch.ts +137 -0
- package/ai/tools/doctorSkillTool.ts +72 -0
- package/ai/tools/ecommerceScraperTool.ts +86 -0
- package/ai/tools/emailTools.ts +549 -0
- package/ai/tools/evalSkillTool.ts +92 -0
- package/ai/tools/exaSearchTool.ts +64 -0
- package/ai/tools/execBashTool.ts +379 -0
- package/ai/tools/executeSqlTool.ts +192 -0
- package/ai/tools/fetchWebpageSupport.ts +309 -0
- package/ai/tools/fetchWebpageTool.ts +84 -0
- package/ai/tools/geminiImagePreviewTool.ts +361 -0
- package/ai/tools/generateDocxTool.ts +215 -0
- package/ai/tools/googleSearchScraperTool.ts +106 -0
- package/ai/tools/importDataTool.ts +133 -0
- package/ai/tools/importSkillTool.ts +162 -0
- package/ai/tools/index.ts +1927 -0
- package/ai/tools/listFilesTool.ts +82 -0
- package/ai/tools/listUserSpacesTool.ts +113 -0
- package/ai/tools/modelUsageTools.ts +199 -0
- package/ai/tools/olmOcrTool.ts +34 -0
- package/ai/tools/openaiImageTool.ts +267 -0
- package/ai/tools/prepareTools.ts +23 -0
- package/ai/tools/readDocTool.ts +84 -0
- package/ai/tools/readFileTool.ts +211 -0
- package/ai/tools/readTool.ts +163 -0
- package/ai/tools/readXPostTool.ts +233 -0
- package/ai/tools/rememberMemoryTool.ts +84 -0
- package/ai/tools/remotionVideoTool.ts +151 -0
- package/ai/tools/searchDialogMessagesTool.ts +222 -0
- package/ai/tools/searchRepoTool.ts +115 -0
- package/ai/tools/searchWorkspaceTool.ts +259 -0
- package/ai/tools/skillFollowup.ts +86 -0
- package/ai/tools/surfWeatherTool.ts +169 -0
- package/ai/tools/table/addTableRowTool.ts +217 -0
- package/ai/tools/table/createTableTool.ts +315 -0
- package/ai/tools/table/rowTools.ts +366 -0
- package/ai/tools/table/schemaTools.ts +244 -0
- package/ai/tools/table/shareTableTool.ts +148 -0
- package/ai/tools/table/toolShared.ts +129 -0
- package/ai/tools/toolApiClient.ts +198 -0
- package/ai/tools/toolNameAliases.ts +57 -0
- package/ai/tools/toolResultError.ts +42 -0
- package/ai/tools/toolRunSlice.ts +303 -0
- package/ai/tools/toolSchemaCompatibility.ts +53 -0
- package/ai/tools/toolVisibility.ts +4 -0
- package/ai/tools/types.ts +20 -0
- package/ai/tools/uiAskChoiceTool.ts +104 -0
- package/ai/tools/updateContentTitleTool.ts +84 -0
- package/ai/tools/updateDocTool.ts +105 -0
- package/ai/tools/updateUserPreferenceProfileTool.ts +145 -0
- package/ai/tools/whisperTool.ts +77 -0
- package/ai/tools/writeFileTool.ts +210 -0
- package/ai/tools/youtubeScraperTool.ts +116 -0
- package/ai/tools/ziweiChartTool.ts +678 -0
- package/ai/types.ts +55 -0
- package/ai/workflow/workflowExecutor.ts +323 -0
- package/ai/workflow/workflowSlice.ts +73 -0
- package/ai/workflow/workflowTypes.ts +106 -0
- package/client/agentRun.test.ts +240 -0
- package/client/agentRun.ts +182 -19
- package/client/compactDialog.test.ts +238 -0
- package/client/localRuntimeAdapter.test.ts +135 -0
- package/client/localRuntimeAdapter.ts +244 -0
- package/client/profileConfig.test.ts +40 -0
- package/client/streamingOutput.test.ts +22 -0
- package/client/streamingOutput.ts +38 -0
- package/commandRegistry.ts +9 -2
- package/connector-experimental/index.ts +5 -0
- package/database/actions/cacheMergedUserData.ts +64 -0
- package/database/actions/common.ts +242 -0
- package/database/actions/deleteFile.ts +40 -0
- package/database/actions/fetchUserData.ts +16 -0
- package/database/actions/fileContent.ts +125 -0
- package/database/actions/patch.ts +155 -0
- package/database/actions/read.ts +337 -0
- package/database/actions/readAndWait.ts +224 -0
- package/database/actions/readRequestManager.ts +120 -0
- package/database/actions/remove.ts +94 -0
- package/database/actions/replication.ts +366 -0
- package/database/actions/upload.ts +174 -0
- package/database/actions/upsert.ts +56 -0
- package/database/actions/write.ts +126 -0
- package/database/client/db.native.ts +73 -0
- package/database/client/db.ts +51 -0
- package/database/client/fetchUserData.ts +61 -0
- package/database/client/handleError.ts +19 -0
- package/database/client/queryRequest.ts +21 -0
- package/database/config.ts +21 -0
- package/database/dbActionThunks.ts +1 -0
- package/database/dbSlice.ts +149 -0
- package/database/email.ts +42 -0
- package/database/fileRing.ts +51 -0
- package/database/fileSharding.ts +70 -0
- package/database/fileStorage.native.ts +92 -0
- package/database/fileStorage.ts +232 -0
- package/database/fileUrl.ts +34 -0
- package/database/hooks/useUserData.ts +489 -0
- package/database/index.ts +1 -0
- package/database/keys.ts +765 -0
- package/database/queryPrefixes.ts +14 -0
- package/database/requests.ts +443 -0
- package/database/runtimeServerContext.ts +35 -0
- package/database/server/MemoryDB.ts +76 -0
- package/database/server/actorAccess.ts +76 -0
- package/database/server/agentDelegation.ts +124 -0
- package/database/server/coreDataOwnership.ts +13 -0
- package/database/server/coreDataProxy.ts +76 -0
- package/database/server/cybotReadonly.ts +18 -0
- package/database/server/dataHandlers.ts +111 -0
- package/database/server/db.ts +118 -0
- package/database/server/dbPath.ts +20 -0
- package/database/server/delete.ts +499 -0
- package/database/server/emailRepository.ts +1480 -0
- package/database/server/ensureDbOpen.ts +12 -0
- package/database/server/fileRead.ts +337 -0
- package/database/server/fileService.ts +436 -0
- package/database/server/handleTransaction.ts +86 -0
- package/database/server/patch.ts +282 -0
- package/database/server/query.ts +138 -0
- package/database/server/read.ts +325 -0
- package/database/server/resourceAccess.ts +211 -0
- package/database/server/routes.ts +110 -0
- package/database/server/spaceMemberAuthority.ts +67 -0
- package/database/server/upload.ts +159 -0
- package/database/server/write.ts +494 -0
- package/database/server/writeAuthority.ts +133 -0
- package/database/sqliteDb.ts +46 -0
- package/database/table/deleteTable.ts +120 -0
- package/database/tenantPlacement.ts +57 -0
- package/database/tombstones.ts +52 -0
- package/database/userDataLoadDecision.ts +17 -0
- package/database/userDataMerge.ts +95 -0
- package/database/userPreferenceRegister.ts +108 -0
- package/database/utils/dbPath.ts +47 -0
- package/database/utils/ulid.native.ts +6 -0
- package/database/utils/ulid.ts +1 -0
- package/index.ts +25 -15
- package/localRuntimeDb.ts +28 -0
- package/package.json +16 -4
- package/runtimeModeArgs.ts +33 -0
- package/tui/readlineWorkspace.ts +1 -0
- 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
|
+
);
|