nolo-cli 0.1.10 → 0.1.11
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 +0 -32
- package/agentRuntimeCommands.ts +3 -3
- package/commandRegistry.ts +2 -2
- package/machineCommands.ts +31 -6
- package/package.json +6 -22
- package/ai/agent/_executeModel.ts +0 -118
- package/ai/agent/agentSlice.ts +0 -525
- package/ai/agent/appWorkingMemory.ts +0 -126
- package/ai/agent/avatarUtils.ts +0 -24
- package/ai/agent/buildEditingContext.ts +0 -373
- package/ai/agent/buildSystemPrompt.ts +0 -532
- package/ai/agent/cleanAgentMessages.ts +0 -140
- package/ai/agent/cliChatClient.ts +0 -119
- package/ai/agent/cliExecutor.ts +0 -733
- package/ai/agent/cliPrompt.ts +0 -10
- package/ai/agent/contextCompiler.ts +0 -107
- package/ai/agent/contextLayerContract.ts +0 -44
- package/ai/agent/createAgentSchema.ts +0 -234
- package/ai/agent/executeToolCall.ts +0 -58
- package/ai/agent/fetchAgentContexts.ts +0 -42
- package/ai/agent/generatePrompt.ts +0 -3
- package/ai/agent/getFullChatContextKeys.ts +0 -168
- package/ai/agent/hooks/fetchPublicAgents.ts +0 -133
- package/ai/agent/hooks/useAgentConfig.ts +0 -61
- package/ai/agent/hooks/useAgentDialog.ts +0 -35
- package/ai/agent/hooks/useAgentFormValidation.ts +0 -202
- package/ai/agent/hooks/usePublicAgents.ts +0 -473
- package/ai/agent/machineRunPermissions.ts +0 -95
- package/ai/agent/persistMessageWithFixedId.ts +0 -37
- package/ai/agent/planSlice.ts +0 -259
- package/ai/agent/referenceUtils.ts +0 -229
- package/ai/agent/runAgentBackground.ts +0 -238
- package/ai/agent/runAgentClientLoop.ts +0 -138
- package/ai/agent/runtimeGuidance.ts +0 -97
- package/ai/agent/runtimeServerBase.ts +0 -37
- package/ai/agent/server/fetchPublicAgents.ts +0 -128
- package/ai/agent/startParallelAgentStreams.ts +0 -424
- package/ai/agent/startupProtocol.ts +0 -53
- package/ai/agent/streamAgentChatTurn.ts +0 -1278
- package/ai/agent/streamAgentChatTurnUtils.ts +0 -738
- package/ai/agent/types.ts +0 -71
- package/ai/agent/utils/imageOutput.ts +0 -33
- package/ai/agent/utils/sortUtils.ts +0 -250
- package/ai/agent/web/referencePickerUtils.ts +0 -146
- package/ai/ai.locale.ts +0 -1079
- package/ai/chat/accumulateToolCallChunks.ts +0 -95
- package/ai/chat/fetchUtils.native.ts +0 -276
- package/ai/chat/fetchUtils.ts +0 -153
- package/ai/chat/parseApiError.ts +0 -64
- package/ai/chat/parseMultilineSSE.ts +0 -95
- package/ai/chat/sendOpenAICompletionsRequest.native.ts +0 -682
- package/ai/chat/sendOpenAICompletionsRequest.ts +0 -703
- package/ai/chat/sendOpenAIResponseRequest.ts +0 -491
- package/ai/chat/shouldUseServerProxy.ts +0 -18
- package/ai/chat/sseClient.native.ts +0 -91
- package/ai/chat/sseClient.ts +0 -67
- package/ai/chat/streamReader.native.ts +0 -31
- package/ai/chat/streamReader.ts +0 -62
- package/ai/chat/updateTotalUsage.ts +0 -72
- package/ai/context/buildReferenceContext.ts +0 -437
- package/ai/context/calculateContextUsage.ts +0 -133
- package/ai/context/retention.ts +0 -165
- package/ai/context/tokenUtils.ts +0 -78
- package/ai/index.ts +0 -1
- package/ai/llm/calculateGeminiImageTokens.ts +0 -57
- package/ai/llm/deepinfra.ts +0 -28
- package/ai/llm/fireworks.ts +0 -50
- package/ai/llm/generateRequestBody.ts +0 -165
- package/ai/llm/getModelContextWindow.ts +0 -84
- package/ai/llm/getNoloKey.ts +0 -31
- package/ai/llm/getPricing.ts +0 -199
- package/ai/llm/hooks/useModelPricing.ts +0 -75
- package/ai/llm/imagePricing.ts +0 -40
- package/ai/llm/isResponseAPIModel.ts +0 -13
- package/ai/llm/mimo.ts +0 -71
- package/ai/llm/mistral.ts +0 -22
- package/ai/llm/modelAvatar.ts +0 -427
- package/ai/llm/models.ts +0 -45
- package/ai/llm/openrouterModels.ts +0 -269
- package/ai/llm/providers.ts +0 -306
- package/ai/llm/reasoningModels.ts +0 -28
- package/ai/llm/types.ts +0 -59
- package/ai/llm/usageRequestOptions.ts +0 -59
- package/ai/memory/capture.ts +0 -148
- package/ai/memory/consolidate.ts +0 -104
- package/ai/memory/delete.ts +0 -147
- package/ai/memory/overlay.ts +0 -84
- package/ai/memory/query.ts +0 -38
- package/ai/memory/queryShared.ts +0 -160
- package/ai/memory/rank.ts +0 -105
- package/ai/memory/recentRelationshipRecap.ts +0 -249
- package/ai/memory/remember.ts +0 -167
- package/ai/memory/runtime.ts +0 -76
- package/ai/memory/store.ts +0 -20
- package/ai/memory/storeShared.ts +0 -76
- package/ai/memory/types.ts +0 -46
- package/ai/memory/understanding.ts +0 -349
- package/ai/memory/understandingGreeting.ts +0 -264
- package/ai/messages/type.ts +0 -20
- package/ai/policy/personalizationDialog.ts +0 -333
- package/ai/policy/runtimePolicy.ts +0 -440
- package/ai/policy/selfUpdateFields.ts +0 -48
- package/ai/policy/types.ts +0 -64
- package/ai/skills/referenceRuntime.ts +0 -274
- package/ai/skills/skillDiagnostics.ts +0 -251
- package/ai/skills/skillDocBuilder.ts +0 -139
- package/ai/skills/skillDocProtocol.ts +0 -434
- package/ai/skills/skillReferenceSummary.ts +0 -63
- package/ai/skills/skillSummaryMarker.ts +0 -26
- package/ai/token/calculatePrice.ts +0 -544
- package/ai/token/db.ts +0 -98
- package/ai/token/externalToolCost.ts +0 -330
- package/ai/token/hooks/useRecords.ts +0 -65
- package/ai/token/missingUsageEstimate.ts +0 -42
- package/ai/token/modelUsageQuery.ts +0 -252
- package/ai/token/normalizeUsage.ts +0 -84
- package/ai/token/openaiImageGenerationUsage.ts +0 -56
- package/ai/token/prepareTokenUsageData.ts +0 -88
- package/ai/token/query.ts +0 -88
- package/ai/token/queryUserTokens.ts +0 -59
- package/ai/token/resolveBillingTarget.ts +0 -52
- package/ai/token/saveTokenRecord.ts +0 -53
- package/ai/token/serverDialogProjection.ts +0 -78
- package/ai/token/serverTokenWriter.ts +0 -143
- package/ai/token/stats.ts +0 -21
- package/ai/token/tokenThunks.ts +0 -24
- package/ai/token/types.ts +0 -93
- package/ai/tools/agent/agentTools.ts +0 -176
- package/ai/tools/agent/agentUpdateShared.ts +0 -311
- package/ai/tools/agent/callAgentTool.ts +0 -139
- package/ai/tools/agent/createAgentTool.ts +0 -512
- package/ai/tools/agent/createDialogTool.ts +0 -69
- package/ai/tools/agent/createSkillAgentTool.ts +0 -62
- package/ai/tools/agent/parallelBudget.ts +0 -221
- package/ai/tools/agent/presets/appBuilderPreset.ts +0 -145
- package/ai/tools/agent/runLlmTool.ts +0 -96
- package/ai/tools/agent/runStreamingAgentTool.ts +0 -73
- package/ai/tools/agent/skillAgentArgs.ts +0 -106
- package/ai/tools/agent/skillAgentPreset.ts +0 -89
- package/ai/tools/agent/streamParallelAgentsTool.ts +0 -122
- package/ai/tools/agent/updateAgentTool.ts +0 -96
- package/ai/tools/agent/updateSelfTool.ts +0 -113
- package/ai/tools/amazonProductScraperTool.ts +0 -86
- package/ai/tools/apifyActorClient.ts +0 -45
- package/ai/tools/appEditGuard.ts +0 -372
- package/ai/tools/appReadSnapshot.ts +0 -153
- package/ai/tools/appTools.ts +0 -1549
- package/ai/tools/applyEditTool.ts +0 -256
- package/ai/tools/applyLineEditsTool.ts +0 -312
- package/ai/tools/browserTools/click.ts +0 -33
- package/ai/tools/browserTools/closeSession.ts +0 -29
- package/ai/tools/browserTools/common.ts +0 -27
- package/ai/tools/browserTools/openSession.ts +0 -48
- package/ai/tools/browserTools/readContent.ts +0 -38
- package/ai/tools/browserTools/selectOption.ts +0 -46
- package/ai/tools/browserTools/typeText.ts +0 -42
- package/ai/tools/category/createCategoryTool.ts +0 -66
- package/ai/tools/category/queryContentsByCategoryTool.ts +0 -69
- package/ai/tools/category/updateContentCategoryTool.ts +0 -75
- package/ai/tools/cfBrowserTools.ts +0 -319
- package/ai/tools/cfSpeechToTextTool.ts +0 -49
- package/ai/tools/checkEnvTool.ts +0 -65
- package/ai/tools/cloudflareCrawlTool.ts +0 -289
- package/ai/tools/codeSearchTool.ts +0 -111
- package/ai/tools/codeTools.ts +0 -101
- package/ai/tools/createDocTool.ts +0 -132
- package/ai/tools/createPlanTool.ts +0 -999
- package/ai/tools/createSkillDocTool.ts +0 -155
- package/ai/tools/createWorkflowTool.ts +0 -154
- package/ai/tools/deepseekOcrTool.ts +0 -34
- package/ai/tools/delayTool.ts +0 -31
- package/ai/tools/deleteSpacesTool.ts +0 -325
- package/ai/tools/deleteSpacesToolModel.ts +0 -159
- package/ai/tools/devReloadUtils.ts +0 -29
- package/ai/tools/dialogMessageSearch.ts +0 -137
- package/ai/tools/doctorSkillTool.ts +0 -72
- package/ai/tools/ecommerceScraperTool.ts +0 -86
- package/ai/tools/emailTools.ts +0 -549
- package/ai/tools/evalSkillTool.ts +0 -92
- package/ai/tools/exaSearchTool.ts +0 -64
- package/ai/tools/execBashTool.ts +0 -379
- package/ai/tools/executeSqlTool.ts +0 -192
- package/ai/tools/fetchWebpageSupport.ts +0 -309
- package/ai/tools/fetchWebpageTool.ts +0 -84
- package/ai/tools/geminiImagePreviewTool.ts +0 -361
- package/ai/tools/generateDocxTool.ts +0 -215
- package/ai/tools/googleSearchScraperTool.ts +0 -106
- package/ai/tools/importDataTool.ts +0 -133
- package/ai/tools/importSkillTool.ts +0 -162
- package/ai/tools/index.ts +0 -1858
- package/ai/tools/listFilesTool.ts +0 -82
- package/ai/tools/listUserSpacesTool.ts +0 -113
- package/ai/tools/modelUsageTools.ts +0 -142
- package/ai/tools/olmOcrTool.ts +0 -34
- package/ai/tools/openaiImageTool.ts +0 -218
- package/ai/tools/paddleOcrTool.ts +0 -34
- package/ai/tools/prepareTools.ts +0 -23
- package/ai/tools/readDocTool.ts +0 -84
- package/ai/tools/readFileTool.ts +0 -211
- package/ai/tools/readTool.ts +0 -163
- package/ai/tools/readXPostTool.ts +0 -233
- package/ai/tools/rememberMemoryTool.ts +0 -84
- package/ai/tools/remotionVideoTool.ts +0 -151
- package/ai/tools/searchDialogMessagesTool.ts +0 -222
- package/ai/tools/searchRepoTool.ts +0 -115
- package/ai/tools/searchWorkspaceTool.ts +0 -259
- package/ai/tools/skillFollowup.ts +0 -86
- package/ai/tools/surfWeatherTool.ts +0 -169
- package/ai/tools/table/addTableRowTool.ts +0 -217
- package/ai/tools/table/createTableTool.ts +0 -315
- package/ai/tools/table/rowTools.ts +0 -366
- package/ai/tools/table/schemaTools.ts +0 -244
- package/ai/tools/table/shareTableTool.ts +0 -148
- package/ai/tools/table/toolShared.ts +0 -129
- package/ai/tools/toolApiClient.ts +0 -198
- package/ai/tools/toolNameAliases.ts +0 -57
- package/ai/tools/toolResultError.ts +0 -42
- package/ai/tools/toolRunSlice.ts +0 -303
- package/ai/tools/toolSchemaCompatibility.ts +0 -53
- package/ai/tools/toolVisibility.ts +0 -4
- package/ai/tools/types.ts +0 -20
- package/ai/tools/uiAskChoiceTool.ts +0 -104
- package/ai/tools/updateContentTitleTool.ts +0 -84
- package/ai/tools/updateDocTool.ts +0 -105
- package/ai/tools/updateUserPreferenceProfileTool.ts +0 -145
- package/ai/tools/whisperTool.ts +0 -77
- package/ai/tools/writeFileTool.ts +0 -210
- package/ai/tools/youtubeScraperTool.ts +0 -116
- package/ai/tools/ziweiChartTool.ts +0 -678
- package/ai/types.ts +0 -55
- package/ai/workflow/workflowExecutor.ts +0 -323
- package/ai/workflow/workflowSlice.ts +0 -73
- package/ai/workflow/workflowTypes.ts +0 -106
- package/client/compactDialog.ts +0 -222
- package/connector-experimental/capabilities.ts +0 -73
- package/connector-experimental/codexBinary.ts +0 -41
- package/connector-experimental/heartbeatLoop.ts +0 -22
- package/connector-experimental/index.ts +0 -5
- package/connector-experimental/machineInfo.ts +0 -46
- package/connector-experimental/protocol.ts +0 -54
package/ai/context/retention.ts
DELETED
|
@@ -1,165 +0,0 @@
|
|
|
1
|
-
// 文件路径: packages/ai/context/retention.ts
|
|
2
|
-
|
|
3
|
-
/**
|
|
4
|
-
* 将 contextRetention slider (1–100) 映射为实际保留比例 R ∈ (0, 1)。
|
|
5
|
-
*
|
|
6
|
-
* 分三档线性映射:
|
|
7
|
-
* - 低保留: 1–30 → 15% ~ 30%
|
|
8
|
-
* - 中保留: 31–70 → 30% ~ 60%
|
|
9
|
-
* - 高保留: 71–100 → 60% ~ 85%
|
|
10
|
-
*/
|
|
11
|
-
export const mapRetentionSliderToRatio = (slider: number): number => {
|
|
12
|
-
const s = Math.min(100, Math.max(1, slider));
|
|
13
|
-
|
|
14
|
-
if (s <= 30) {
|
|
15
|
-
// Low retention: 15% ~ 30%
|
|
16
|
-
return 0.15 + (0.30 - 0.15) * (s / 30);
|
|
17
|
-
}
|
|
18
|
-
|
|
19
|
-
if (s <= 70) {
|
|
20
|
-
// Medium retention: 30% ~ 60%
|
|
21
|
-
return 0.30 + (0.60 - 0.30) * ((s - 30) / 40);
|
|
22
|
-
}
|
|
23
|
-
|
|
24
|
-
// High retention: 60% ~ 85%
|
|
25
|
-
return 0.60 + (0.85 - 0.60) * ((s - 70) / 30);
|
|
26
|
-
};
|
|
27
|
-
|
|
28
|
-
/**
|
|
29
|
-
* 安全系数:只允许历史对话最多占用 contextWindow 的 SAFE_BUFFER_RATIO 部分,
|
|
30
|
-
* 预留剩余空间给 system prompt / 工具描述等固定开销。
|
|
31
|
-
*/
|
|
32
|
-
export const SAFE_BUFFER_RATIO = 0.9;
|
|
33
|
-
|
|
34
|
-
/**
|
|
35
|
-
* 对话负载分档:
|
|
36
|
-
* - light :短句聊天为主
|
|
37
|
-
* - medium :中等长度混合
|
|
38
|
-
* - heavy :长代码 / 长文档为主
|
|
39
|
-
*/
|
|
40
|
-
export type ConversationLoad = "light" | "medium" | "heavy";
|
|
41
|
-
|
|
42
|
-
export interface ContextPlan {
|
|
43
|
-
/**
|
|
44
|
-
* 历史总预算(summary + 原始消息),单位:token
|
|
45
|
-
* 实际不会超过 contextWindow * SAFE_BUFFER_RATIO
|
|
46
|
-
*/
|
|
47
|
-
historyBudget: number;
|
|
48
|
-
/**
|
|
49
|
-
* 历史中分配给「原始消息」的预算,单位:token
|
|
50
|
-
* 用于 trimMessagesWithSummary 之类的裁剪逻辑。
|
|
51
|
-
*/
|
|
52
|
-
rawMessageBudget: number;
|
|
53
|
-
/**
|
|
54
|
-
* 希望至少保留的「最近尾部消息」的 token 数,用作二次兜底,
|
|
55
|
-
* 确保短句对话能多保几轮,重内容对话能保住最近几条大块内容。
|
|
56
|
-
*/
|
|
57
|
-
minTailTokens: number;
|
|
58
|
-
}
|
|
59
|
-
|
|
60
|
-
const clamp = (v: number, min: number, max: number): number =>
|
|
61
|
-
Math.min(max, Math.max(min, v));
|
|
62
|
-
|
|
63
|
-
/**
|
|
64
|
-
* 统一的上下文使用规划函数:
|
|
65
|
-
* - 输入:模型 window、用户 slider、现有 summary 长度、近期对话负载分档
|
|
66
|
-
* - 输出:历史总预算 + 原始消息预算 + 尾部最少保留 token 数
|
|
67
|
-
*
|
|
68
|
-
* 约束与策略:
|
|
69
|
-
* - 历史最多只能占用 contextWindow * SAFE_BUFFER_RATIO
|
|
70
|
-
* - slider 决定「整体偏好多记还是少记」
|
|
71
|
-
* - recentLoad 决定在该偏好下的微调:
|
|
72
|
-
* - light :适当放宽,给更多历史空间
|
|
73
|
-
* - heavy :适当收紧,为当前大块内容 / 工具描述预留空间
|
|
74
|
-
*/
|
|
75
|
-
export const planContextUsage = (params: {
|
|
76
|
-
contextWindow: number;
|
|
77
|
-
retentionSlider: number;
|
|
78
|
-
summaryTokens: number;
|
|
79
|
-
recentLoad: ConversationLoad;
|
|
80
|
-
}): ContextPlan => {
|
|
81
|
-
const { contextWindow, retentionSlider, summaryTokens, recentLoad } = params;
|
|
82
|
-
|
|
83
|
-
const safeWindowLimit = Math.floor(contextWindow * SAFE_BUFFER_RATIO);
|
|
84
|
-
|
|
85
|
-
const s = Math.min(100, Math.max(1, retentionSlider));
|
|
86
|
-
const iq = s / 100; // 0~1,表示用户偏好的“记忆强度”
|
|
87
|
-
|
|
88
|
-
const isBigWindow = contextWindow >= 128000; // 例如 128k / 1M 等大模型
|
|
89
|
-
const isSmallWindow = contextWindow <= 32000; // 例如 8k / 16k 等小模型
|
|
90
|
-
|
|
91
|
-
// 基础历史比例(只考虑 slider,不考虑对话负载)
|
|
92
|
-
// - 小窗口:历史占 35% ~ 70% 的 safeWindow
|
|
93
|
-
// - 中窗口:历史占 40% ~ 75%
|
|
94
|
-
// - 大窗口:历史占 65% ~ 90%(尽量吃满大模型上下文)
|
|
95
|
-
let baseHistoryRatio: number;
|
|
96
|
-
if (isBigWindow) {
|
|
97
|
-
baseHistoryRatio = 0.65 + 0.25 * iq; // 0.65 ~ 0.90
|
|
98
|
-
} else if (isSmallWindow) {
|
|
99
|
-
baseHistoryRatio = 0.35 + 0.35 * iq; // 0.35 ~ 0.70
|
|
100
|
-
} else {
|
|
101
|
-
baseHistoryRatio = 0.40 + 0.35 * iq; // 0.40 ~ 0.75
|
|
102
|
-
}
|
|
103
|
-
|
|
104
|
-
// 1. 历史比例:在 baseHistoryRatio 基础上,根据对话负载做微调
|
|
105
|
-
let historyRatio = baseHistoryRatio;
|
|
106
|
-
|
|
107
|
-
switch (recentLoad) {
|
|
108
|
-
case "light": {
|
|
109
|
-
// 短句对话:可以多给一点历史空间,但仍保留少量余量给其他上下文
|
|
110
|
-
const maxRatio = isBigWindow ? 0.95 : 0.85;
|
|
111
|
-
historyRatio = clamp(historyRatio * 1.05, 0.3, maxRatio);
|
|
112
|
-
break;
|
|
113
|
-
}
|
|
114
|
-
case "heavy": {
|
|
115
|
-
// 重内容对话:适当收紧历史比例,为当前大块内容 / 工具描述等预留更多窗口
|
|
116
|
-
const maxRatio = isBigWindow ? 0.90 : 0.80;
|
|
117
|
-
historyRatio = clamp(historyRatio * 0.9, 0.3, maxRatio);
|
|
118
|
-
break;
|
|
119
|
-
}
|
|
120
|
-
case "medium":
|
|
121
|
-
default: {
|
|
122
|
-
const maxRatio = isBigWindow ? 0.90 : 0.85;
|
|
123
|
-
historyRatio = clamp(historyRatio, 0.3, maxRatio);
|
|
124
|
-
break;
|
|
125
|
-
}
|
|
126
|
-
}
|
|
127
|
-
|
|
128
|
-
const historyBudget = Math.max(
|
|
129
|
-
0,
|
|
130
|
-
Math.floor(safeWindowLimit * historyRatio),
|
|
131
|
-
);
|
|
132
|
-
|
|
133
|
-
// 2. 在历史预算里分配给「原始消息」的预算
|
|
134
|
-
// - summary 已经占用一部分
|
|
135
|
-
// - 同时保证原始消息至少有一部分空间(heavy 用户更多一点)
|
|
136
|
-
const minRawRatio = recentLoad === "heavy" ? 0.4 : 0.2;
|
|
137
|
-
|
|
138
|
-
const rawMessageBudget = Math.max(
|
|
139
|
-
Math.floor(historyBudget * minRawRatio),
|
|
140
|
-
historyBudget - summaryTokens,
|
|
141
|
-
0,
|
|
142
|
-
);
|
|
143
|
-
|
|
144
|
-
// 3. 希望至少保留的尾部 token 数
|
|
145
|
-
// - light :多保几轮短句(不超过 30% window,且上限 8k)
|
|
146
|
-
// - heavy :至少给几条大块代码空间(>= 16k,或 25% window)
|
|
147
|
-
// - medium:取中间值
|
|
148
|
-
let minTailTokens: number;
|
|
149
|
-
|
|
150
|
-
if (recentLoad === "light") {
|
|
151
|
-
minTailTokens = Math.min(
|
|
152
|
-
Math.floor(safeWindowLimit * 0.3),
|
|
153
|
-
8000,
|
|
154
|
-
);
|
|
155
|
-
} else if (recentLoad === "heavy") {
|
|
156
|
-
minTailTokens = Math.max(
|
|
157
|
-
Math.floor(safeWindowLimit * 0.25),
|
|
158
|
-
16000,
|
|
159
|
-
);
|
|
160
|
-
} else {
|
|
161
|
-
minTailTokens = Math.floor(safeWindowLimit * 0.2);
|
|
162
|
-
}
|
|
163
|
-
|
|
164
|
-
return { historyBudget, rawMessageBudget, minTailTokens };
|
|
165
|
-
};
|
package/ai/context/tokenUtils.ts
DELETED
|
@@ -1,78 +0,0 @@
|
|
|
1
|
-
// ai/context/tokenUtils.ts
|
|
2
|
-
|
|
3
|
-
/**
|
|
4
|
-
* 简单的 Token 估算工具。
|
|
5
|
-
*
|
|
6
|
-
* 规则:
|
|
7
|
-
* - 中文字符:约 1.5 tokens/字
|
|
8
|
-
* - 英文及其他字符:约 0.25 tokens/字符(即 4 字符 ≈ 1 token)
|
|
9
|
-
*
|
|
10
|
-
* 这是一个粗略估算,精确值需要使用 tiktoken 等库。
|
|
11
|
-
*/
|
|
12
|
-
|
|
13
|
-
/**
|
|
14
|
-
* 估算文本的 Token 数量
|
|
15
|
-
*/
|
|
16
|
-
export const estimateTokenCount = (text: string): number => {
|
|
17
|
-
if (!text) return 0;
|
|
18
|
-
|
|
19
|
-
// 匹配中文字符(包括常用汉字范围)
|
|
20
|
-
const chineseChars = (text.match(/[\u4e00-\u9fa5\u3400-\u4dbf]/g) || []).length;
|
|
21
|
-
const otherChars = text.length - chineseChars;
|
|
22
|
-
|
|
23
|
-
// 中文 1.5 token/字,其他 0.25 token/字符
|
|
24
|
-
return Math.ceil(chineseChars * 1.5 + otherChars * 0.25);
|
|
25
|
-
};
|
|
26
|
-
|
|
27
|
-
/**
|
|
28
|
-
* 估算多个文本的 Token 总数
|
|
29
|
-
*/
|
|
30
|
-
export const estimateTotalTokens = (texts: (string | undefined | null)[]): number => {
|
|
31
|
-
return texts.reduce((sum, text) => sum + estimateTokenCount(text || ""), 0);
|
|
32
|
-
};
|
|
33
|
-
|
|
34
|
-
/**
|
|
35
|
-
* 格式化 Token 数量为可读字符串
|
|
36
|
-
* 例如:1500 -> "1.5k", 150000 -> "150k"
|
|
37
|
-
*/
|
|
38
|
-
export const formatTokenCount = (count: number): string => {
|
|
39
|
-
if (count < 1000) return String(count);
|
|
40
|
-
if (count < 10000) return `${(count / 1000).toFixed(1)}k`;
|
|
41
|
-
return `${Math.round(count / 1000)}k`;
|
|
42
|
-
};
|
|
43
|
-
|
|
44
|
-
/**
|
|
45
|
-
* 计算 Token 使用百分比
|
|
46
|
-
*/
|
|
47
|
-
export const calcTokenUsagePercent = (used: number, total: number): number => {
|
|
48
|
-
if (total <= 0) return 0;
|
|
49
|
-
return Math.min(100, Math.round((used / total) * 100));
|
|
50
|
-
};
|
|
51
|
-
|
|
52
|
-
/**
|
|
53
|
-
* Context 分配建议阈值
|
|
54
|
-
* - References/Space Context 不应超过总量的 40%
|
|
55
|
-
* - 为对话历史和用户输入预留 40%
|
|
56
|
-
* - 为系统提示预留 10%
|
|
57
|
-
* - 为输出预留 10%
|
|
58
|
-
*/
|
|
59
|
-
export const CONTEXT_BUDGET = {
|
|
60
|
-
REFERENCES_MAX_PERCENT: 40, // References + Space Context 最大占比
|
|
61
|
-
HISTORY_RESERVE_PERCENT: 40, // 为对话历史预留
|
|
62
|
-
SYSTEM_RESERVE_PERCENT: 10, // 系统提示预留
|
|
63
|
-
OUTPUT_RESERVE_PERCENT: 10, // 输出预留
|
|
64
|
-
};
|
|
65
|
-
|
|
66
|
-
/**
|
|
67
|
-
* 判断 Token 使用是否超过警告阈值
|
|
68
|
-
*/
|
|
69
|
-
export const isTokenUsageWarning = (usedPercent: number): boolean => {
|
|
70
|
-
return usedPercent > CONTEXT_BUDGET.REFERENCES_MAX_PERCENT;
|
|
71
|
-
};
|
|
72
|
-
|
|
73
|
-
/**
|
|
74
|
-
* 判断 Token 使用是否严重超标
|
|
75
|
-
*/
|
|
76
|
-
export const isTokenUsageCritical = (usedPercent: number): boolean => {
|
|
77
|
-
return usedPercent > 60; // 超过 60% 可能影响对话历史
|
|
78
|
-
};
|
package/ai/index.ts
DELETED
|
@@ -1 +0,0 @@
|
|
|
1
|
-
console.log('Hello via Bun!');
|
|
@@ -1,57 +0,0 @@
|
|
|
1
|
-
// src/calcGeminiImageTokens.ts
|
|
2
|
-
|
|
3
|
-
/**
|
|
4
|
-
* Gemini 图片 token 计算(粗略公式版)
|
|
5
|
-
*
|
|
6
|
-
* 规则:
|
|
7
|
-
* - 若宽高都 <= 384,则直接按 1 个图块:258 tokens
|
|
8
|
-
* - 否则:
|
|
9
|
-
* - uRaw = floor( min(width, height) / 1.5 )
|
|
10
|
-
* - u = clamp(uRaw, 256, 768)
|
|
11
|
-
* - tilesX = ceil(width / u)
|
|
12
|
-
* - tilesY = ceil(height / u)
|
|
13
|
-
* - tiles = tilesX * tilesY
|
|
14
|
-
* - tokens = tiles * 258
|
|
15
|
-
*/
|
|
16
|
-
|
|
17
|
-
export const GEMINI_IMAGE_TILE_TOKENS = 258;
|
|
18
|
-
export const GEMINI_SMALL_IMAGE_THRESHOLD = 384;
|
|
19
|
-
export const GEMINI_MIN_TILE_SIZE = 256;
|
|
20
|
-
export const GEMINI_MAX_TILE_SIZE = 768;
|
|
21
|
-
|
|
22
|
-
const clamp = (value: number, min: number, max: number): number =>
|
|
23
|
-
Math.min(Math.max(value, min), max);
|
|
24
|
-
|
|
25
|
-
const ceilDiv = (numerator: number, denominator: number): number =>
|
|
26
|
-
Math.ceil(numerator / denominator);
|
|
27
|
-
|
|
28
|
-
/**
|
|
29
|
-
* 计算单张图片的 token 数(适用于 Gemini 1.5 / 2.0 / 2.5 的粗略估算)
|
|
30
|
-
*/
|
|
31
|
-
export const calculateGeminiImageTokens = (
|
|
32
|
-
width: number,
|
|
33
|
-
height: number,
|
|
34
|
-
): number => {
|
|
35
|
-
if (!Number.isFinite(width) || !Number.isFinite(height)) {
|
|
36
|
-
throw new TypeError('width 和 height 必须是有限数字');
|
|
37
|
-
}
|
|
38
|
-
if (width <= 0 || height <= 0) {
|
|
39
|
-
throw new RangeError('width 和 height 必须为正数');
|
|
40
|
-
}
|
|
41
|
-
|
|
42
|
-
// 情况 1:小图(两个维度都 <= 384)
|
|
43
|
-
if (width <= GEMINI_SMALL_IMAGE_THRESHOLD && height <= GEMINI_SMALL_IMAGE_THRESHOLD) {
|
|
44
|
-
return GEMINI_IMAGE_TILE_TOKENS;
|
|
45
|
-
}
|
|
46
|
-
|
|
47
|
-
// 情况 2:大图,按粗略公式计算图块数量
|
|
48
|
-
const minSide = Math.min(width, height);
|
|
49
|
-
const rawUnit = Math.floor(minSide / 1.5);
|
|
50
|
-
const unit = clamp(rawUnit, GEMINI_MIN_TILE_SIZE, GEMINI_MAX_TILE_SIZE);
|
|
51
|
-
|
|
52
|
-
const tilesX = ceilDiv(width, unit);
|
|
53
|
-
const tilesY = ceilDiv(height, unit);
|
|
54
|
-
const tiles = tilesX * tilesY;
|
|
55
|
-
|
|
56
|
-
return tiles * GEMINI_IMAGE_TILE_TOKENS;
|
|
57
|
-
};
|
package/ai/llm/deepinfra.ts
DELETED
|
@@ -1,28 +0,0 @@
|
|
|
1
|
-
// ai/llm/deepinfra.ts
|
|
2
|
-
export const deepinfraModels = [
|
|
3
|
-
{
|
|
4
|
-
name: "moonshotai/Kimi-K2.5",
|
|
5
|
-
displayName: "MoonshotAI: Kimi K2.5 (DeepInfra)",
|
|
6
|
-
hasVision: true,
|
|
7
|
-
price: {
|
|
8
|
-
input: 0.45 * 8,
|
|
9
|
-
output: 2.25 * 8,
|
|
10
|
-
inputCacheHit: 0.07 * 8,
|
|
11
|
-
},
|
|
12
|
-
maxOutputTokens: 262144,
|
|
13
|
-
contextWindow: 262144,
|
|
14
|
-
supportsTool: true,
|
|
15
|
-
},
|
|
16
|
-
{
|
|
17
|
-
name: "zai-org/GLM-5.1",
|
|
18
|
-
displayName: "Z.AI: GLM 5.1 (DeepInfra)",
|
|
19
|
-
hasVision: false,
|
|
20
|
-
price: {
|
|
21
|
-
input: 1.26 * 8,
|
|
22
|
-
output: 3.96 * 8,
|
|
23
|
-
inputCacheHit: 0.26 * 8,
|
|
24
|
-
},
|
|
25
|
-
contextWindow: 202752,
|
|
26
|
-
supportsTool: true,
|
|
27
|
-
},
|
|
28
|
-
];
|
package/ai/llm/fireworks.ts
DELETED
|
@@ -1,50 +0,0 @@
|
|
|
1
|
-
// ai/llm/fireworks.ts
|
|
2
|
-
export const fireworksModels = [
|
|
3
|
-
// --- MoonshotAI Models (Kimi) ---
|
|
4
|
-
{
|
|
5
|
-
name: "accounts/fireworks/models/kimi-k2p6",
|
|
6
|
-
displayName: "MoonshotAI: Kimi K2.6",
|
|
7
|
-
hasVision: true,
|
|
8
|
-
price: {
|
|
9
|
-
input: 0.95 * 8,
|
|
10
|
-
output: 4.0 * 8,
|
|
11
|
-
cachingRead: 0.16 * 8,
|
|
12
|
-
},
|
|
13
|
-
maxOutputTokens: 262144,
|
|
14
|
-
contextWindow: 262144,
|
|
15
|
-
supportsTool: true,
|
|
16
|
-
},
|
|
17
|
-
{
|
|
18
|
-
name: "accounts/fireworks/models/minimax-m2p7",
|
|
19
|
-
displayName: "MiniMax: MiniMax M2.7",
|
|
20
|
-
hasVision: true,
|
|
21
|
-
price: {
|
|
22
|
-
input: 0.3 * 8,
|
|
23
|
-
output: 1.2 * 8,
|
|
24
|
-
cachingRead: 0.06 * 8,
|
|
25
|
-
},
|
|
26
|
-
supportsTool: true,
|
|
27
|
-
},
|
|
28
|
-
{
|
|
29
|
-
name: "accounts/fireworks/models/qwen3p6-plus",
|
|
30
|
-
displayName: "Qwen: Qwen 3.6 Plus",
|
|
31
|
-
hasVision: true,
|
|
32
|
-
price: {
|
|
33
|
-
input: 0.5 * 8,
|
|
34
|
-
output: 3.0 * 8,
|
|
35
|
-
cachingRead: 0.1 * 8,
|
|
36
|
-
},
|
|
37
|
-
supportsTool: true,
|
|
38
|
-
},
|
|
39
|
-
{
|
|
40
|
-
name: "accounts/fireworks/models/glm-5p1",
|
|
41
|
-
displayName: "Z.AI: GLM 5.1",
|
|
42
|
-
hasVision: true,
|
|
43
|
-
price: {
|
|
44
|
-
input: 1.0 * 8,
|
|
45
|
-
output: 3.2 * 8,
|
|
46
|
-
cachingRead: 0.2 * 8,
|
|
47
|
-
},
|
|
48
|
-
supportsTool: true,
|
|
49
|
-
},
|
|
50
|
-
];
|
|
@@ -1,165 +0,0 @@
|
|
|
1
|
-
// ai/llm/generateRequestBody.ts
|
|
2
|
-
import { Agent, Message } from "app/types";
|
|
3
|
-
import { generateOpenAIRequestBody } from "integrations/openai/generateOpenAIRequestBody";
|
|
4
|
-
import { generateResponseRequestBody } from "integrations/openai/generateResponseRequestBody";
|
|
5
|
-
|
|
6
|
-
import { isResponseAPIModel } from "./isResponseAPIModel";
|
|
7
|
-
import { Contexts } from "../types";
|
|
8
|
-
import { getModelConfig } from "./providers";
|
|
9
|
-
import type { Model } from "./types";
|
|
10
|
-
|
|
11
|
-
export interface GenerateRequestBodyArgs {
|
|
12
|
-
agentConfig: Agent;
|
|
13
|
-
messages: Message[]; // 动态历史消息/本轮新增
|
|
14
|
-
userInput: string; // 本次用户输入
|
|
15
|
-
contexts?: Contexts; // 可选上下文
|
|
16
|
-
stableMessages?: Message[]; // 稳定的、已截断的历史快照(用于缓存)
|
|
17
|
-
prependSystemPrompt?: boolean;
|
|
18
|
-
}
|
|
19
|
-
|
|
20
|
-
const IMAGE_OUTPUT_EXECUTION_PROMPT = [
|
|
21
|
-
"When the conversation already includes an input image and the user asks to change any visual attribute such as hairstyle, glasses, outfit, makeup, background, composition, or style, default to returning the edited image in this turn.",
|
|
22
|
-
"When the user asks you to generate, edit, redraw, or return an image, you must return the image in this turn.",
|
|
23
|
-
"Do not answer with only a textual description.",
|
|
24
|
-
"Do not claim that you uploaded, attached, or sent an image unless the response actually includes image output.",
|
|
25
|
-
"Only return text without an image when the user explicitly asks for analysis, recommendations, or explanation without generating an image.",
|
|
26
|
-
].join(" ");
|
|
27
|
-
|
|
28
|
-
type ImageOutputSettings = {
|
|
29
|
-
shouldEnableImage: boolean;
|
|
30
|
-
modelConfig: Model | null;
|
|
31
|
-
};
|
|
32
|
-
|
|
33
|
-
const getImageOutputSettings = (
|
|
34
|
-
agentConfig: Agent,
|
|
35
|
-
provider: string
|
|
36
|
-
): ImageOutputSettings => {
|
|
37
|
-
const modelName = agentConfig.model;
|
|
38
|
-
if (!modelName) {
|
|
39
|
-
return { shouldEnableImage: false, modelConfig: null };
|
|
40
|
-
}
|
|
41
|
-
|
|
42
|
-
try {
|
|
43
|
-
const modelConfig = getModelConfig(provider as any, modelName);
|
|
44
|
-
const hasImageOutput = !!modelConfig.hasImageOutput;
|
|
45
|
-
const isExplicitlyDisabled = agentConfig.imageConfig?.enabled === false;
|
|
46
|
-
return {
|
|
47
|
-
shouldEnableImage: hasImageOutput && !isExplicitlyDisabled,
|
|
48
|
-
modelConfig,
|
|
49
|
-
};
|
|
50
|
-
} catch {
|
|
51
|
-
return { shouldEnableImage: false, modelConfig: null };
|
|
52
|
-
}
|
|
53
|
-
};
|
|
54
|
-
|
|
55
|
-
const withImageOutputPromptGuard = (
|
|
56
|
-
agentConfig: Agent,
|
|
57
|
-
shouldEnableImage: boolean
|
|
58
|
-
): Agent => {
|
|
59
|
-
if (!shouldEnableImage) return agentConfig;
|
|
60
|
-
|
|
61
|
-
const basePrompt = typeof agentConfig.prompt === "string" ? agentConfig.prompt.trim() : "";
|
|
62
|
-
if (basePrompt.includes(IMAGE_OUTPUT_EXECUTION_PROMPT)) {
|
|
63
|
-
return agentConfig;
|
|
64
|
-
}
|
|
65
|
-
|
|
66
|
-
return {
|
|
67
|
-
...agentConfig,
|
|
68
|
-
prompt: basePrompt
|
|
69
|
-
? `${basePrompt}\n\n${IMAGE_OUTPUT_EXECUTION_PROMPT}`
|
|
70
|
-
: IMAGE_OUTPUT_EXECUTION_PROMPT,
|
|
71
|
-
};
|
|
72
|
-
};
|
|
73
|
-
|
|
74
|
-
/**
|
|
75
|
-
* 根据模型元数据 + Agent 配置,为 chat/completions 请求体注入
|
|
76
|
-
* - modalities: ["image", "text"]
|
|
77
|
-
* - image_config: { aspect_ratio, image_size }
|
|
78
|
-
*/
|
|
79
|
-
const applyImageConfigIfNeeded = (
|
|
80
|
-
body: any,
|
|
81
|
-
agentConfig: Agent,
|
|
82
|
-
provider: string
|
|
83
|
-
): any => {
|
|
84
|
-
const { shouldEnableImage, modelConfig } = getImageOutputSettings(
|
|
85
|
-
agentConfig,
|
|
86
|
-
provider
|
|
87
|
-
);
|
|
88
|
-
const agentImageCfg = agentConfig.imageConfig;
|
|
89
|
-
|
|
90
|
-
if (!shouldEnableImage || !modelConfig) {
|
|
91
|
-
return body;
|
|
92
|
-
}
|
|
93
|
-
|
|
94
|
-
// 1) 处理 modalities
|
|
95
|
-
// 优先级:Agent.forceModalities > Model.defaultModalities > requiresImageModalities ? ["image","text"] : 不强制
|
|
96
|
-
const modalities: Array<"text" | "image"> | undefined =
|
|
97
|
-
agentImageCfg?.forceModalities ??
|
|
98
|
-
modelConfig.defaultModalities ??
|
|
99
|
-
(modelConfig.requiresImageModalities
|
|
100
|
-
? (["image", "text"] as Array<"text" | "image">)
|
|
101
|
-
: undefined);
|
|
102
|
-
|
|
103
|
-
if (modalities && !body.modalities) {
|
|
104
|
-
body.modalities = modalities;
|
|
105
|
-
}
|
|
106
|
-
|
|
107
|
-
// 2) 处理 image_config(仅在模型声明 supportsImageConfig 时生效)
|
|
108
|
-
if (!modelConfig.supportsImageConfig) {
|
|
109
|
-
return body;
|
|
110
|
-
}
|
|
111
|
-
|
|
112
|
-
const aspectRatio = agentImageCfg?.aspectRatio;
|
|
113
|
-
const imageSize = agentImageCfg?.imageSize;
|
|
114
|
-
|
|
115
|
-
if (!aspectRatio && !imageSize) {
|
|
116
|
-
// Agent 未指定任何 image_config 参数,则沿用模型默认
|
|
117
|
-
return body;
|
|
118
|
-
}
|
|
119
|
-
|
|
120
|
-
body.image_config = {
|
|
121
|
-
...(body.image_config ?? {}),
|
|
122
|
-
...(aspectRatio ? { aspect_ratio: aspectRatio } : {}),
|
|
123
|
-
...(imageSize ? { image_size: imageSize } : {}),
|
|
124
|
-
};
|
|
125
|
-
|
|
126
|
-
return body;
|
|
127
|
-
};
|
|
128
|
-
|
|
129
|
-
export const generateRequestBody = ({
|
|
130
|
-
agentConfig,
|
|
131
|
-
messages,
|
|
132
|
-
contexts,
|
|
133
|
-
stableMessages,
|
|
134
|
-
prependSystemPrompt = true,
|
|
135
|
-
}: GenerateRequestBodyArgs) => {
|
|
136
|
-
const provider = (agentConfig.provider || "").toLowerCase();
|
|
137
|
-
const imageSettings = getImageOutputSettings(agentConfig, provider);
|
|
138
|
-
const agentConfigForRequest = withImageOutputPromptGuard(
|
|
139
|
-
agentConfig,
|
|
140
|
-
imageSettings.shouldEnableImage
|
|
141
|
-
);
|
|
142
|
-
|
|
143
|
-
// 1) Response API 走新版 /v1/responses(目前主要是 OpenAI)
|
|
144
|
-
if (isResponseAPIModel(agentConfigForRequest)) {
|
|
145
|
-
return generateResponseRequestBody(
|
|
146
|
-
agentConfigForRequest,
|
|
147
|
-
[...(stableMessages ?? []), ...messages],
|
|
148
|
-
contexts,
|
|
149
|
-
prependSystemPrompt
|
|
150
|
-
);
|
|
151
|
-
}
|
|
152
|
-
|
|
153
|
-
// 2) 其余走老版 chat/completions
|
|
154
|
-
const baseBody = generateOpenAIRequestBody(
|
|
155
|
-
agentConfigForRequest,
|
|
156
|
-
provider,
|
|
157
|
-
messages,
|
|
158
|
-
contexts,
|
|
159
|
-
stableMessages,
|
|
160
|
-
prependSystemPrompt
|
|
161
|
-
);
|
|
162
|
-
|
|
163
|
-
// 在基础请求体上注入图像相关配置(如果模型 + Agent 都支持/需要)
|
|
164
|
-
return applyImageConfigIfNeeded(baseBody, agentConfigForRequest, provider);
|
|
165
|
-
};
|
|
@@ -1,84 +0,0 @@
|
|
|
1
|
-
// ai/llm/getModelContextWindow.ts
|
|
2
|
-
|
|
3
|
-
/**
|
|
4
|
-
* 获取指定模型的 Context Window 大小。
|
|
5
|
-
*
|
|
6
|
-
* 复用 ALL_MODELS 数据源,保持与模型选择器的一致性。
|
|
7
|
-
*/
|
|
8
|
-
|
|
9
|
-
import { ALL_MODELS, type ModelWithProvider } from "./models";
|
|
10
|
-
|
|
11
|
-
// 默认 Context Window(用于未知模型)
|
|
12
|
-
const DEFAULT_CONTEXT_WINDOW = 128000;
|
|
13
|
-
|
|
14
|
-
// 缓存模型映射表
|
|
15
|
-
let modelMap: Map<string, ModelWithProvider> | null = null;
|
|
16
|
-
|
|
17
|
-
const getModelMap = (): Map<string, ModelWithProvider> => {
|
|
18
|
-
if (!modelMap) {
|
|
19
|
-
modelMap = new Map();
|
|
20
|
-
for (const model of ALL_MODELS) {
|
|
21
|
-
modelMap.set(model.name, model);
|
|
22
|
-
// 也按 displayName 索引(如果有)
|
|
23
|
-
if (model.displayName) {
|
|
24
|
-
modelMap.set(model.displayName.toLowerCase(), model);
|
|
25
|
-
}
|
|
26
|
-
}
|
|
27
|
-
}
|
|
28
|
-
return modelMap;
|
|
29
|
-
};
|
|
30
|
-
|
|
31
|
-
/**
|
|
32
|
-
* 根据模型名称获取 Context Window 大小
|
|
33
|
-
* @param modelName 模型名称(如 "gpt-4o", "claude-3-5-sonnet")
|
|
34
|
-
* @returns Context Window 大小(tokens)
|
|
35
|
-
*/
|
|
36
|
-
export const getModelContextWindow = (modelName: string): number => {
|
|
37
|
-
if (!modelName) return DEFAULT_CONTEXT_WINDOW;
|
|
38
|
-
|
|
39
|
-
const map = getModelMap();
|
|
40
|
-
const model =
|
|
41
|
-
map.get(modelName) ||
|
|
42
|
-
map.get(modelName.toLowerCase());
|
|
43
|
-
|
|
44
|
-
if (model?.contextWindow) {
|
|
45
|
-
return typeof model.contextWindow === "number"
|
|
46
|
-
? model.contextWindow
|
|
47
|
-
: DEFAULT_CONTEXT_WINDOW;
|
|
48
|
-
}
|
|
49
|
-
|
|
50
|
-
return DEFAULT_CONTEXT_WINDOW;
|
|
51
|
-
};
|
|
52
|
-
|
|
53
|
-
/**
|
|
54
|
-
* 获取模型的完整信息
|
|
55
|
-
*/
|
|
56
|
-
export const getModelInfo = (modelName: string): ModelWithProvider | null => {
|
|
57
|
-
if (!modelName) return null;
|
|
58
|
-
|
|
59
|
-
const map = getModelMap();
|
|
60
|
-
return (
|
|
61
|
-
map.get(modelName) ||
|
|
62
|
-
map.get(modelName.toLowerCase()) ||
|
|
63
|
-
null
|
|
64
|
-
);
|
|
65
|
-
};
|
|
66
|
-
|
|
67
|
-
/**
|
|
68
|
-
* 获取模型的 Max Output Tokens
|
|
69
|
-
*/
|
|
70
|
-
export const getModelMaxOutputTokens = (modelName: string): number => {
|
|
71
|
-
const model = getModelInfo(modelName);
|
|
72
|
-
if (model?.maxOutputTokens && typeof model.maxOutputTokens === "number") {
|
|
73
|
-
return model.maxOutputTokens;
|
|
74
|
-
}
|
|
75
|
-
return 4096; // 保守默认值
|
|
76
|
-
};
|
|
77
|
-
|
|
78
|
-
/**
|
|
79
|
-
* 获取模型是否支持视觉
|
|
80
|
-
*/
|
|
81
|
-
export const getModelHasVision = (modelName: string): boolean => {
|
|
82
|
-
const model = getModelInfo(modelName);
|
|
83
|
-
return model?.hasVision ?? false;
|
|
84
|
-
};
|
package/ai/llm/getNoloKey.ts
DELETED
|
@@ -1,31 +0,0 @@
|
|
|
1
|
-
import { availableProviderOptions } from "./providers";
|
|
2
|
-
|
|
3
|
-
export const getNoloKey = (
|
|
4
|
-
provider: (typeof availableProviderOptions)[number] | "deepinfra"
|
|
5
|
-
) => {
|
|
6
|
-
switch (provider) {
|
|
7
|
-
case "anthropic":
|
|
8
|
-
return process.env.ANTHROPIC_API_KEY;
|
|
9
|
-
case "google":
|
|
10
|
-
return process.env.GOOGLE_API_KEY;
|
|
11
|
-
case "deepseek":
|
|
12
|
-
return process.env.DEEPSEEK_API_KEY;
|
|
13
|
-
case "openrouter":
|
|
14
|
-
return process.env.OPENROUTER_API_KEY;
|
|
15
|
-
case "fireworks":
|
|
16
|
-
return process.env.FIREWORKS_API_KEY;
|
|
17
|
-
case "openai":
|
|
18
|
-
return process.env.OPENAI_KEY;
|
|
19
|
-
case "mistral":
|
|
20
|
-
return process.env.MISTRAL_KEY;
|
|
21
|
-
case "deepinfra":
|
|
22
|
-
return process.env.DEEPINFRA_API_KEY;
|
|
23
|
-
case "mimo":
|
|
24
|
-
return process.env.MIMO_API_KEY || process.env.OPENCODE_API_KEY;
|
|
25
|
-
|
|
26
|
-
// case "xai":
|
|
27
|
-
// return process.env.XAI_API_KEY;
|
|
28
|
-
default:
|
|
29
|
-
return null;
|
|
30
|
-
}
|
|
31
|
-
};
|