@lobehub/lobehub 2.0.0-next.3 → 2.0.0-next.31
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/.env.desktop +1 -2
- package/.env.example +0 -3
- package/.env.example.development +0 -2
- package/.github/workflows/claude-auto-testing.yml +73 -0
- package/.github/workflows/claude-translate-comments.yml +67 -0
- package/.github/workflows/desktop-pr-build.yml +18 -16
- package/.github/workflows/docker.yml +25 -20
- package/.github/workflows/e2e.yml +17 -3
- package/.github/workflows/release-desktop-beta.yml +12 -12
- package/.github/workflows/release.yml +3 -5
- package/.github/workflows/test.yml +47 -12
- package/.nvmrc +1 -1
- package/CHANGELOG.md +684 -0
- package/Dockerfile +1 -3
- package/README.md +2 -45
- package/README.zh-CN.md +2 -45
- package/apps/desktop/src/main/controllers/AuthCtr.ts +53 -39
- package/apps/desktop/src/main/controllers/MenuCtr.ts +5 -5
- package/apps/desktop/src/main/controllers/NotificationCtr.ts +29 -29
- package/apps/desktop/src/main/controllers/RemoteServerConfigCtr.ts +16 -16
- package/apps/desktop/src/main/controllers/ShortcutCtr.ts +2 -2
- package/apps/desktop/src/main/controllers/TrayMenuCtr.ts +18 -18
- package/apps/desktop/src/main/controllers/UpdaterCtr.ts +4 -4
- package/apps/desktop/src/main/controllers/__tests__/AuthCtr.test.ts +706 -0
- package/apps/desktop/src/main/controllers/__tests__/TrayMenuCtr.test.ts +5 -5
- package/apps/desktop/src/main/controllers/index.ts +4 -4
- package/apps/desktop/src/main/utils/next-electron-rsc.ts +7 -5
- package/apps/desktop/tsconfig.json +0 -1
- package/changelog/v1.json +232 -0
- package/docs/development/database-schema.dbml +12 -0
- package/docs/self-hosting/advanced/auth/next-auth/auth0.mdx +2 -2
- package/docs/self-hosting/advanced/auth/next-auth/auth0.zh-CN.mdx +2 -2
- package/docs/self-hosting/advanced/auth/next-auth/authelia.mdx +2 -2
- package/docs/self-hosting/advanced/auth/next-auth/authelia.zh-CN.mdx +2 -2
- package/docs/self-hosting/advanced/auth/next-auth/authentik.mdx +2 -2
- package/docs/self-hosting/advanced/auth/next-auth/authentik.zh-CN.mdx +2 -2
- package/docs/self-hosting/advanced/auth/next-auth/casdoor.mdx +2 -2
- package/docs/self-hosting/advanced/auth/next-auth/casdoor.zh-CN.mdx +2 -2
- package/docs/self-hosting/advanced/auth/next-auth/cloudflare-zero-trust.mdx +2 -2
- package/docs/self-hosting/advanced/auth/next-auth/cloudflare-zero-trust.zh-CN.mdx +2 -2
- package/docs/self-hosting/advanced/auth/next-auth/github.mdx +2 -2
- package/docs/self-hosting/advanced/auth/next-auth/github.zh-CN.mdx +2 -2
- package/docs/self-hosting/advanced/auth/next-auth/google.mdx +32 -29
- package/docs/self-hosting/advanced/auth/next-auth/keycloak.mdx +2 -2
- package/docs/self-hosting/advanced/auth/next-auth/keycloak.zh-CN.mdx +2 -2
- package/docs/self-hosting/advanced/auth/next-auth/logto.mdx +5 -3
- package/docs/self-hosting/advanced/auth/next-auth/logto.zh-CN.mdx +5 -3
- package/docs/self-hosting/advanced/auth/next-auth/microsoft-entra-id.mdx +2 -2
- package/docs/self-hosting/advanced/auth/next-auth/microsoft-entra-id.zh-CN.mdx +2 -2
- package/docs/self-hosting/advanced/auth/next-auth/okta.mdx +2 -2
- package/docs/self-hosting/advanced/auth/next-auth/okta.zh-CN.mdx +2 -2
- package/docs/self-hosting/advanced/auth/next-auth/wechat.mdx +2 -2
- package/docs/self-hosting/advanced/auth/next-auth/wechat.zh-CN.mdx +2 -2
- package/docs/self-hosting/advanced/auth/next-auth/zitadel.mdx +2 -2
- package/docs/self-hosting/advanced/auth/next-auth/zitadel.zh-CN.mdx +2 -2
- package/docs/self-hosting/advanced/auth.mdx +32 -21
- package/docs/self-hosting/advanced/auth.zh-CN.mdx +30 -19
- package/docs/self-hosting/advanced/feature-flags.mdx +0 -1
- package/docs/self-hosting/advanced/feature-flags.zh-CN.mdx +0 -1
- package/docs/self-hosting/advanced/online-search.mdx +30 -25
- package/docs/self-hosting/advanced/online-search.zh-CN.mdx +25 -23
- package/e2e/src/features/discover/detail-pages.feature +95 -0
- package/e2e/src/features/discover/interactions.feature +113 -0
- package/e2e/src/features/discover/smoke.feature +34 -1
- package/e2e/src/steps/discover/detail-pages.steps.ts +295 -0
- package/e2e/src/steps/discover/interactions.steps.ts +451 -0
- package/e2e/src/steps/discover/smoke.steps.ts +116 -4
- package/e2e/tsconfig.json +0 -1
- package/locales/ar/labs.json +4 -0
- package/locales/ar/models.json +15 -6
- package/locales/ar/oauth.json +1 -0
- package/locales/bg-BG/labs.json +4 -0
- package/locales/bg-BG/models.json +15 -6
- package/locales/bg-BG/oauth.json +1 -0
- package/locales/de-DE/labs.json +4 -0
- package/locales/de-DE/models.json +15 -6
- package/locales/de-DE/oauth.json +1 -0
- package/locales/en-US/labs.json +4 -0
- package/locales/en-US/models.json +15 -6
- package/locales/en-US/oauth.json +1 -0
- package/locales/es-ES/labs.json +4 -0
- package/locales/es-ES/models.json +15 -6
- package/locales/es-ES/oauth.json +1 -0
- package/locales/fa-IR/labs.json +4 -0
- package/locales/fa-IR/models.json +15 -6
- package/locales/fa-IR/oauth.json +1 -0
- package/locales/fr-FR/labs.json +4 -0
- package/locales/fr-FR/models.json +15 -6
- package/locales/fr-FR/oauth.json +1 -0
- package/locales/it-IT/labs.json +4 -0
- package/locales/it-IT/models.json +15 -6
- package/locales/it-IT/oauth.json +1 -0
- package/locales/ja-JP/labs.json +4 -0
- package/locales/ja-JP/models.json +15 -6
- package/locales/ja-JP/oauth.json +1 -0
- package/locales/ko-KR/labs.json +4 -0
- package/locales/ko-KR/models.json +21 -12
- package/locales/ko-KR/oauth.json +1 -0
- package/locales/nl-NL/labs.json +4 -0
- package/locales/nl-NL/models.json +15 -6
- package/locales/nl-NL/oauth.json +1 -0
- package/locales/pl-PL/labs.json +4 -0
- package/locales/pl-PL/models.json +15 -6
- package/locales/pl-PL/oauth.json +1 -0
- package/locales/pt-BR/labs.json +4 -0
- package/locales/pt-BR/models.json +15 -6
- package/locales/pt-BR/oauth.json +1 -0
- package/locales/ru-RU/labs.json +4 -0
- package/locales/ru-RU/models.json +15 -6
- package/locales/ru-RU/oauth.json +1 -0
- package/locales/tr-TR/labs.json +4 -0
- package/locales/tr-TR/models.json +15 -6
- package/locales/tr-TR/oauth.json +1 -0
- package/locales/vi-VN/labs.json +4 -0
- package/locales/vi-VN/models.json +15 -6
- package/locales/vi-VN/oauth.json +1 -0
- package/locales/zh-CN/labs.json +4 -0
- package/locales/zh-CN/models.json +15 -6
- package/locales/zh-CN/oauth.json +1 -0
- package/locales/zh-TW/labs.json +4 -0
- package/locales/zh-TW/models.json +15 -6
- package/locales/zh-TW/oauth.json +1 -0
- package/next.config.ts +2 -3
- package/package.json +76 -82
- package/packages/const/src/index.ts +0 -1
- package/packages/const/src/url.ts +1 -4
- package/packages/const/src/user.ts +5 -2
- package/packages/const/src/version.ts +3 -3
- package/packages/context-engine/src/index.ts +1 -6
- package/packages/context-engine/src/processors/GroupMessageFlatten.ts +12 -2
- package/packages/context-engine/src/processors/__tests__/GroupMessageFlatten.test.ts +73 -9
- package/packages/context-engine/src/providers/index.ts +0 -2
- package/packages/database/migrations/0041_improve_index.sql +10 -0
- package/packages/database/migrations/0042_improve_agent_index.sql +1 -0
- package/packages/database/migrations/0043_add_ai_model_settings.sql +1 -0
- package/packages/database/migrations/meta/0041_snapshot.json +7784 -0
- package/packages/database/migrations/meta/0042_snapshot.json +7800 -0
- package/packages/database/migrations/meta/0043_snapshot.json +8419 -0
- package/packages/database/migrations/meta/_journal.json +21 -0
- package/packages/database/package.json +2 -2
- package/packages/database/src/core/migrations.json +33 -0
- package/packages/database/src/core/web-server.ts +2 -1
- package/packages/database/src/models/__tests__/message.grouping.test.ts +812 -0
- package/packages/database/src/models/__tests__/message.test.ts +256 -233
- package/packages/database/src/models/agent.ts +16 -13
- package/packages/database/src/models/message.ts +90 -74
- package/packages/database/src/models/session.ts +80 -28
- package/packages/database/src/models/user.ts +2 -1
- package/packages/database/src/repositories/aiInfra/index.test.ts +198 -0
- package/packages/database/src/repositories/aiInfra/index.ts +2 -1
- package/packages/database/src/repositories/dataImporter/deprecated/__tests__/index.test.ts +2 -1
- package/packages/database/src/repositories/dataImporter/deprecated/index.ts +7 -1
- package/packages/database/src/schemas/agent.ts +13 -11
- package/packages/database/src/schemas/aiInfra.ts +2 -0
- package/packages/database/src/schemas/message.ts +5 -1
- package/packages/database/src/schemas/relations.ts +6 -4
- package/packages/database/src/schemas/session.ts +2 -0
- package/packages/database/src/schemas/topic.ts +6 -1
- package/packages/database/src/utils/__tests__/groupMessages.test.ts +145 -2
- package/packages/database/src/utils/groupMessages.ts +7 -5
- package/packages/electron-client-ipc/package.json +4 -1
- package/packages/file-loaders/package.json +1 -0
- package/packages/memory-extract/package.json +2 -2
- package/packages/model-bank/src/aiModels/anthropic.ts +0 -63
- package/packages/model-bank/src/aiModels/higress.ts +0 -55
- package/packages/model-bank/src/aiModels/infiniai.ts +21 -0
- package/packages/model-bank/src/aiModels/ollamacloud.ts +13 -0
- package/packages/model-bank/src/aiModels/siliconcloud.ts +19 -0
- package/packages/model-runtime/src/core/streams/openai/__snapshots__/responsesStream.test.ts.snap +0 -38
- package/packages/model-runtime/src/providers/minimax/index.ts +5 -5
- package/packages/model-runtime/src/providers/search1api/index.test.ts +2 -2
- package/packages/model-runtime/src/utils/googleErrorParser.test.ts +125 -0
- package/packages/model-runtime/src/utils/googleErrorParser.ts +103 -77
- package/packages/obervability-otel/package.json +1 -3
- package/packages/obervability-otel/src/node.ts +3 -7
- package/packages/types/src/discover/mcp.ts +6 -0
- package/packages/types/src/index.ts +0 -1
- package/packages/types/src/message/common/base.ts +13 -0
- package/packages/types/src/message/common/image.ts +8 -0
- package/packages/types/src/message/common/metadata.ts +39 -0
- package/packages/types/src/message/common/tools.ts +10 -0
- package/packages/types/src/message/db/params.ts +47 -1
- package/packages/types/src/message/ui/chat.ts +4 -1
- package/packages/types/src/message/ui/params.ts +49 -4
- package/packages/types/src/plugins/mcp.ts +4 -1
- package/packages/types/src/search.ts +16 -0
- package/packages/types/src/serverConfig.ts +2 -6
- package/packages/types/src/topic/topic.ts +14 -0
- package/packages/types/src/user/index.ts +2 -76
- package/packages/types/src/user/preference.ts +105 -0
- package/packages/types/src/user/settings/index.ts +22 -0
- package/packages/utils/src/apiKey.test.ts +139 -0
- package/packages/utils/src/client/clipboard.ts +2 -2
- package/packages/utils/src/client/exportFile.ts +10 -10
- package/packages/utils/src/client/parserPlaceholder.ts +18 -18
- package/packages/utils/src/client/topic.ts +10 -10
- package/packages/utils/src/client/xor-obfuscation.ts +11 -11
- package/packages/utils/src/server/auth.ts +6 -6
- package/packages/utils/src/server/geo.ts +9 -9
- package/packages/utils/src/server/xor.ts +7 -7
- package/packages/web-crawler/src/crawImpl/firecrawl.ts +39 -12
- package/packages/web-crawler/tsconfig.json +0 -1
- package/renovate.json +23 -1
- package/scripts/migrateServerDB/errorHint.js +1 -7
- package/scripts/migrateServerDB/index.ts +2 -1
- package/src/app/(backend)/webapi/revalidate/route.ts +1 -1
- package/src/app/[variants]/(auth)/signup/[[...signup]]/page.tsx +1 -8
- package/src/app/[variants]/(main)/(mobile)/me/(home)/__tests__/useCategory.test.tsx +9 -0
- package/src/app/[variants]/(main)/(mobile)/me/(home)/features/UserBanner.tsx +3 -6
- package/src/app/[variants]/(main)/(mobile)/me/(home)/layout.tsx +0 -2
- package/src/app/[variants]/(main)/_layout/Desktop/SideBar/TopActions.tsx +1 -4
- package/src/app/[variants]/(main)/chat/(workspace)/@conversation/features/ChatInput/V1Mobile/index.tsx +2 -2
- package/src/app/[variants]/(main)/chat/(workspace)/@conversation/features/ChatInput/V1Mobile/useSend.ts +6 -4
- package/src/app/[variants]/(main)/chat/(workspace)/@conversation/features/ChatInput/useSend.ts +15 -10
- package/src/app/[variants]/(main)/chat/@session/features/SessionListContent/List/Item/Actions.tsx +3 -28
- package/src/app/[variants]/(main)/chat/@session/features/SessionListContent/List/Item/index.tsx +4 -2
- package/src/app/[variants]/(main)/chat/_layout/Desktop/index.tsx +0 -2
- package/src/app/[variants]/(main)/chat/_layout/Mobile.tsx +1 -5
- package/src/app/[variants]/(main)/chat/settings/features/HeaderContent.tsx +2 -62
- package/src/app/[variants]/(main)/discover/(list)/features/Pagination.tsx +1 -0
- package/src/app/[variants]/(main)/discover/(list)/features/SortButton/index.tsx +1 -1
- package/src/app/[variants]/(main)/discover/(list)/mcp/features/List/Item.tsx +1 -0
- package/src/app/[variants]/(main)/discover/(list)/model/features/List/Item.tsx +1 -0
- package/src/app/[variants]/(main)/discover/(list)/provider/features/List/Item.tsx +1 -0
- package/src/app/[variants]/(main)/discover/components/CategoryMenu.tsx +9 -1
- package/src/app/[variants]/(main)/image/@topic/features/Topics/TopicList.tsx +1 -0
- package/src/app/[variants]/(main)/image/features/PromptInput/index.tsx +1 -1
- package/src/app/[variants]/(main)/image/page.tsx +0 -2
- package/src/app/[variants]/(main)/labs/components/LabCard.tsx +8 -6
- package/src/app/[variants]/(main)/labs/page.tsx +19 -22
- package/src/app/[variants]/(main)/profile/_layout/Desktop/index.tsx +23 -24
- package/src/app/[variants]/(main)/profile/_layout/Mobile/index.tsx +5 -9
- package/src/app/[variants]/(main)/settings/_layout/Desktop/index.tsx +0 -2
- package/src/app/[variants]/(main)/settings/_layout/Mobile/index.tsx +0 -2
- package/src/app/[variants]/(main)/settings/provider/(list)/ProviderGrid/Card.tsx +1 -1
- package/src/app/[variants]/(main)/settings/provider/detail/azure/index.tsx +6 -8
- package/src/app/[variants]/(main)/settings/provider/detail/azureai/index.tsx +1 -1
- package/src/app/[variants]/(main)/settings/provider/detail/bedrock/index.tsx +1 -1
- package/src/app/[variants]/(main)/settings/provider/detail/cloudflare/index.tsx +1 -1
- package/src/app/[variants]/(main)/settings/provider/detail/comfyui/index.tsx +1 -1
- package/src/app/[variants]/(main)/settings/provider/detail/github/index.tsx +1 -1
- package/src/app/[variants]/(main)/settings/provider/detail/vertexai/index.tsx +1 -1
- package/src/app/[variants]/(main)/settings/provider/features/ProviderConfig/index.tsx +2 -4
- package/src/app/[variants]/layout.tsx +1 -0
- package/src/app/[variants]/loading/index.tsx +1 -10
- package/src/app/[variants]/oauth/ResultLayout.tsx +47 -0
- package/src/app/[variants]/oauth/callback/error/page.tsx +20 -33
- package/src/app/[variants]/oauth/callback/layout.tsx +1 -0
- package/src/app/[variants]/oauth/callback/success/page.tsx +8 -22
- package/src/app/[variants]/oauth/consent/[uid]/Consent/BuiltinConsent.tsx +47 -0
- package/src/app/[variants]/oauth/consent/[uid]/{Consent.tsx → Consent/index.tsx} +12 -1
- package/src/app/[variants]/oauth/consent/[uid]/Login.tsx +19 -2
- package/src/app/sitemap.tsx +7 -1
- package/src/components/InvalidAPIKey/APIKeyForm/Bedrock.tsx +8 -13
- package/src/components/Link.tsx +12 -0
- package/src/components/Skeleton/SkeletonSwitch.tsx +13 -0
- package/src/components/Skeleton/index.ts +2 -0
- package/src/components/Thinking/index.tsx +4 -3
- package/src/config/db.ts +0 -5
- package/src/config/featureFlags/schema.test.ts +0 -2
- package/src/config/featureFlags/schema.ts +0 -6
- package/src/config/modelProviders/ai21.ts +1 -16
- package/src/config/modelProviders/ai302.ts +1 -128
- package/src/config/modelProviders/ai360.ts +1 -32
- package/src/config/modelProviders/anthropic.ts +1 -94
- package/src/config/modelProviders/azure.ts +1 -51
- package/src/config/modelProviders/baichuan.ts +1 -57
- package/src/config/modelProviders/bedrock.ts +1 -276
- package/src/config/modelProviders/cloudflare.ts +1 -64
- package/src/config/modelProviders/deepseek.ts +1 -19
- package/src/config/modelProviders/fireworksai.ts +1 -174
- package/src/config/modelProviders/giteeai.ts +1 -135
- package/src/config/modelProviders/github.ts +1 -254
- package/src/config/modelProviders/google.ts +1 -130
- package/src/config/modelProviders/groq.ts +1 -119
- package/src/config/modelProviders/higress.ts +1 -1736
- package/src/config/modelProviders/huggingface.ts +1 -54
- package/src/config/modelProviders/hunyuan.ts +1 -83
- package/src/config/modelProviders/infiniai.ts +1 -74
- package/src/config/modelProviders/internlm.ts +1 -20
- package/src/config/modelProviders/minimax.ts +1 -1
- package/src/config/modelProviders/mistral.ts +1 -95
- package/src/config/modelProviders/modelscope.ts +1 -27
- package/src/config/modelProviders/moonshot.ts +1 -29
- package/src/config/modelProviders/novita.ts +1 -105
- package/src/config/modelProviders/ollama.ts +1 -325
- package/src/config/modelProviders/openai.ts +1 -242
- package/src/config/modelProviders/openrouter.ts +1 -240
- package/src/config/modelProviders/perplexity.ts +1 -45
- package/src/config/modelProviders/ppio.ts +1 -152
- package/src/config/modelProviders/qiniu.ts +2 -19
- package/src/config/modelProviders/qwen.ts +1 -245
- package/src/config/modelProviders/search1api.ts +1 -34
- package/src/config/modelProviders/sensenova.ts +1 -69
- package/src/config/modelProviders/siliconcloud.ts +1 -417
- package/src/config/modelProviders/spark.ts +1 -59
- package/src/config/modelProviders/stepfun.ts +1 -98
- package/src/config/modelProviders/taichu.ts +1 -18
- package/src/config/modelProviders/togetherai.ts +1 -274
- package/src/config/modelProviders/upstage.ts +1 -28
- package/src/config/modelProviders/wenxin.ts +1 -140
- package/src/config/modelProviders/xai.ts +1 -38
- package/src/config/modelProviders/zeroone.ts +1 -81
- package/src/config/modelProviders/zhipu.ts +1 -108
- package/src/envs/app.ts +5 -8
- package/src/envs/auth.ts +0 -179
- package/src/features/AgentSetting/AgentPlugin/index.tsx +2 -2
- package/src/features/ChatInput/ActionBar/STT/browser.tsx +2 -2
- package/src/features/ChatInput/ActionBar/STT/openai.tsx +2 -2
- package/src/features/ChatInput/ActionBar/Tools/useControls.tsx +1 -3
- package/src/features/ChatInput/ActionBar/index.tsx +2 -2
- package/src/features/ChatInput/InputEditor/index.tsx +2 -2
- package/src/features/Conversation/Error/ErrorJsonViewer.tsx +4 -3
- package/src/features/Conversation/Error/OllamaBizError/index.tsx +7 -2
- package/src/features/Conversation/Error/index.tsx +15 -5
- package/src/features/Conversation/MarkdownElements/LobeArtifact/Render/index.tsx +2 -2
- package/src/features/Conversation/Messages/Assistant/Extra/index.tsx +2 -2
- package/src/features/Conversation/Messages/Assistant/MessageContent.tsx +5 -3
- package/src/features/Conversation/Messages/Assistant/Tool/Inspector/BuiltinPluginTitle.tsx +2 -2
- package/src/features/Conversation/Messages/Assistant/Tool/Inspector/ToolTitle.tsx +4 -2
- package/src/features/Conversation/Messages/Assistant/Tool/Render/CustomRender.tsx +2 -2
- package/src/features/Conversation/Messages/Assistant/Tool/Render/index.tsx +2 -2
- package/src/features/Conversation/Messages/Assistant/Tool/index.tsx +2 -2
- package/src/features/Conversation/Messages/Assistant/index.tsx +4 -4
- package/src/features/Conversation/Messages/Default.tsx +2 -2
- package/src/features/Conversation/Messages/Group/Actions/WithContentId.tsx +152 -0
- package/src/features/Conversation/Messages/Group/Actions/WithoutContentId.tsx +70 -0
- package/src/features/Conversation/Messages/Group/Actions/index.tsx +21 -0
- package/src/features/Conversation/Messages/Group/ContentBlock.tsx +91 -0
- package/src/features/Conversation/Messages/Group/EditState.tsx +51 -0
- package/src/features/Conversation/Messages/Group/Error/index.tsx +53 -0
- package/src/features/Conversation/Messages/Group/GroupChildren.tsx +73 -0
- package/src/features/Conversation/Messages/Group/MessageContent.tsx +39 -0
- package/src/features/Conversation/Messages/Group/Tool/Inspector/BuiltinPluginTitle.tsx +49 -0
- package/src/features/Conversation/Messages/Group/Tool/Inspector/Debug.tsx +70 -0
- package/src/features/Conversation/Messages/Group/Tool/Inspector/PluginResult.tsx +34 -0
- package/src/features/Conversation/Messages/Group/Tool/Inspector/PluginState.tsx +18 -0
- package/src/features/Conversation/Messages/Group/Tool/Inspector/Settings.tsx +40 -0
- package/src/features/Conversation/Messages/Group/Tool/Inspector/ToolTitle.tsx +92 -0
- package/src/features/Conversation/Messages/Group/Tool/Inspector/index.tsx +176 -0
- package/src/features/Conversation/Messages/Group/Tool/Render/Arguments/ObjectEntity.tsx +81 -0
- package/src/features/Conversation/Messages/Group/Tool/Render/Arguments/ValueCell.tsx +43 -0
- package/src/features/Conversation/Messages/Group/Tool/Render/Arguments/index.tsx +134 -0
- package/src/features/Conversation/Messages/Group/Tool/Render/CustomRender.tsx +88 -0
- package/src/features/Conversation/Messages/Group/Tool/Render/ErrorResponse.tsx +35 -0
- package/src/features/Conversation/Messages/Group/Tool/Render/LoadingPlaceholder/index.tsx +29 -0
- package/src/features/Conversation/Messages/Group/Tool/Render/PluginSettings.tsx +66 -0
- package/src/features/Conversation/Messages/Group/Tool/Render/index.tsx +105 -0
- package/src/features/Conversation/Messages/Group/Tool/index.tsx +75 -0
- package/src/features/Conversation/Messages/Group/Tools.tsx +46 -0
- package/src/features/Conversation/Messages/Group/index.tsx +140 -0
- package/src/features/Conversation/Messages/User/Extra.tsx +2 -2
- package/src/features/Conversation/Messages/User/index.tsx +4 -4
- package/src/features/Conversation/Messages/index.tsx +15 -3
- package/src/features/Conversation/components/AutoScroll.tsx +2 -2
- package/src/features/Conversation/components/Extras/Usage/UsageDetail/index.tsx +9 -6
- package/src/features/Conversation/components/ShareMessageModal/ShareImage/Preview.tsx +2 -2
- package/src/features/DataImporter/index.tsx +15 -60
- package/src/features/DevPanel/PostgresViewer/usePgTable.ts +3 -2
- package/src/features/MCP/utils.test.ts +91 -0
- package/src/features/MCP/utils.ts +20 -2
- package/src/features/PluginStore/Content.tsx +2 -3
- package/src/features/PluginStore/McpList/index.tsx +6 -2
- package/src/features/PluginTag/index.tsx +1 -3
- package/src/features/PluginsUI/Render/BuiltinType/index.test.tsx +37 -28
- package/src/features/Portal/Artifacts/Body/index.tsx +2 -2
- package/src/helpers/isCanUseFC.ts +0 -8
- package/src/hooks/useEnabledChatModels.ts +0 -8
- package/src/hooks/useInterceptingRoutes.test.ts +21 -3
- package/src/hooks/useModelContextWindowTokens.ts +0 -8
- package/src/hooks/useModelHasContextWindowToken.ts +1 -10
- package/src/hooks/useModelSupportFiles.ts +1 -11
- package/src/hooks/useModelSupportReasoning.ts +1 -11
- package/src/hooks/useModelSupportToolUse.ts +1 -11
- package/src/hooks/useModelSupportVision.ts +1 -11
- package/src/layout/AuthProvider/Clerk/index.tsx +2 -16
- package/src/libs/next-auth/auth.config.ts +3 -6
- package/src/libs/next-auth/sso-providers/auth0.ts +0 -7
- package/src/libs/next-auth/sso-providers/authelia.ts +3 -5
- package/src/libs/next-auth/sso-providers/authentik.ts +0 -7
- package/src/libs/next-auth/sso-providers/cloudflare-zero-trust.ts +3 -6
- package/src/libs/next-auth/sso-providers/cognito.ts +1 -5
- package/src/libs/next-auth/sso-providers/generic-oidc.ts +3 -5
- package/src/libs/next-auth/sso-providers/github.ts +0 -6
- package/src/libs/next-auth/sso-providers/google.ts +0 -2
- package/src/libs/next-auth/sso-providers/index.ts +0 -2
- package/src/libs/next-auth/sso-providers/keycloak.ts +0 -3
- package/src/libs/next-auth/sso-providers/logto.ts +3 -5
- package/src/libs/next-auth/sso-providers/okta.ts +0 -4
- package/src/libs/next-auth/sso-providers/zitadel.ts +0 -7
- package/src/libs/oidc-provider/provider.ts +1 -1
- package/src/libs/trpc/client/index.ts +0 -1
- package/src/libs/trpc/client/lambda.ts +8 -5
- package/src/libs/trpc/lambda/context.ts +4 -1
- package/src/locales/default/labs.ts +4 -0
- package/src/locales/default/oauth.ts +1 -0
- package/src/server/globalConfig/index.ts +0 -23
- package/src/server/modules/AssistantStore/index.ts +1 -1
- package/src/server/modules/ModelRuntime/trace.ts +11 -4
- package/src/server/routers/desktop/mcp.ts +1 -3
- package/src/server/routers/lambda/__tests__/integration/message.integration.test.ts +810 -70
- package/src/server/routers/lambda/config/__snapshots__/index.test.ts.snap +175 -12
- package/src/server/routers/lambda/config/index.test.ts +38 -30
- package/src/server/routers/lambda/market/index.ts +4 -2
- package/src/server/routers/lambda/message.ts +60 -52
- package/src/server/routers/lambda/session.ts +8 -5
- package/src/server/routers/lambda/topic.ts +7 -1
- package/src/server/routers/lambda/user.ts +32 -31
- package/src/server/routers/tools/mcp.ts +2 -3
- package/src/server/routers/tools/search.test.ts +1 -7
- package/src/server/routers/tools/search.ts +1 -4
- package/src/server/services/mcp/deps/MCPSystemDepsCheckService.test.ts +541 -0
- package/src/server/services/message/__tests__/index.test.ts +348 -0
- package/src/server/services/message/index.ts +159 -0
- package/src/server/services/search/impls/firecrawl/index.ts +51 -11
- package/src/server/services/search/impls/firecrawl/type.ts +60 -9
- package/src/services/__tests__/tool.test.ts +0 -3
- package/src/services/aiModel/index.test.ts +3 -6
- package/src/services/aiModel/index.ts +55 -7
- package/src/services/aiProvider/index.test.ts +2 -5
- package/src/services/aiProvider/index.ts +47 -7
- package/src/services/chat/chat.test.ts +13 -40
- package/src/services/chat/contextEngineering.test.ts +0 -30
- package/src/services/chat/contextEngineering.ts +7 -17
- package/src/services/chat/helper.ts +7 -31
- package/src/services/chat/index.ts +5 -10
- package/src/services/chat/types.ts +1 -1
- package/src/services/chatGroup/index.ts +64 -9
- package/src/services/config.ts +1 -65
- package/src/services/export/index.ts +9 -4
- package/src/services/file/index.ts +59 -10
- package/src/services/import/index.ts +132 -7
- package/src/services/mcp.test.ts +777 -0
- package/src/services/mcp.ts +40 -6
- package/src/services/message/index.ts +203 -10
- package/src/services/message/{__tests__/server.test.ts → server.test.ts} +3 -3
- package/src/services/models.ts +2 -11
- package/src/services/plugin/index.test.ts +8 -0
- package/src/services/plugin/index.ts +51 -10
- package/src/services/session/index.test.ts +8 -0
- package/src/services/session/index.ts +143 -10
- package/src/services/tableViewer/client.ts +12 -15
- package/src/services/thread/index.test.ts +8 -0
- package/src/services/thread/index.ts +37 -7
- package/src/services/topic/index.test.ts +8 -0
- package/src/services/topic/index.ts +74 -10
- package/src/services/user/index.test.ts +8 -0
- package/src/services/user/index.ts +50 -11
- package/src/store/aiInfra/slices/aiModel/action.test.ts +17 -9
- package/src/store/chat/helpers.test.ts +99 -0
- package/src/store/chat/helpers.ts +21 -2
- package/src/store/chat/selectors.ts +1 -1
- package/src/store/chat/slices/aiChat/actions/__tests__/generateAIChat.test.ts +17 -251
- package/src/store/chat/slices/aiChat/actions/__tests__/generateAIChatV2.test.ts +339 -7
- package/src/store/chat/slices/aiChat/actions/__tests__/helpers.ts +9 -5
- package/src/store/chat/slices/aiChat/actions/generateAIChat.ts +49 -178
- package/src/store/chat/slices/aiChat/actions/generateAIChatV2.ts +272 -14
- package/src/store/chat/slices/aiChat/actions/generateAIGroupChat.ts +7 -2
- package/src/store/chat/slices/builtinTool/actions/index.ts +1 -4
- package/src/store/chat/slices/builtinTool/actions/search.ts +3 -3
- package/src/store/chat/slices/message/action.test.ts +157 -16
- package/src/store/chat/slices/message/action.ts +153 -77
- package/src/store/chat/slices/message/reducer.test.ts +363 -5
- package/src/store/chat/slices/message/reducer.ts +87 -3
- package/src/store/chat/slices/message/{selectors.test.ts → selectors/chat.test.ts} +266 -30
- package/src/store/chat/slices/message/{selectors.ts → selectors/chat.ts} +29 -79
- package/src/store/chat/slices/message/selectors/index.ts +2 -0
- package/src/store/chat/slices/message/selectors/messageState.test.ts +36 -0
- package/src/store/chat/slices/message/selectors/messageState.ts +80 -0
- package/src/store/chat/slices/plugin/action.test.ts +118 -157
- package/src/store/chat/slices/plugin/action.ts +53 -68
- package/src/store/chat/slices/thread/action.test.ts +13 -4
- package/src/store/chat/slices/thread/action.ts +3 -1
- package/src/store/chat/slices/topic/action.test.ts +1 -1
- package/src/store/chat/slices/topic/action.ts +1 -2
- package/src/store/chat/slices/topic/reducer.ts +1 -2
- package/src/store/file/slices/chat/action.ts +1 -4
- package/src/store/file/slices/fileManager/action.ts +2 -3
- package/src/store/global/store.ts +1 -7
- package/src/store/session/slices/sessionGroup/action.test.ts +5 -5
- package/src/store/tool/selectors/tool.test.ts +1 -1
- package/src/store/tool/selectors/tool.ts +6 -8
- package/src/store/tool/slices/builtin/action.test.ts +83 -35
- package/src/store/tool/slices/builtin/action.ts +0 -9
- package/src/store/tool/slices/builtin/selectors.test.ts +4 -30
- package/src/store/tool/slices/builtin/selectors.ts +15 -21
- package/src/store/tool/slices/mcpStore/action.test.ts +95 -3
- package/src/store/tool/slices/mcpStore/action.ts +177 -53
- package/src/store/tool/slices/oldStore/initialState.ts +1 -2
- package/src/store/user/initialState.ts +1 -7
- package/src/store/user/selectors.ts +1 -5
- package/src/store/user/slices/common/action.test.ts +1 -4
- package/src/store/user/slices/common/action.ts +5 -4
- package/src/store/user/slices/preference/action.ts +8 -1
- package/src/store/user/slices/preference/selectors/index.ts +2 -0
- package/src/store/user/slices/preference/selectors/labPrefer.ts +13 -0
- package/src/store/user/slices/preference/{selectors.ts → selectors/preference.ts} +0 -2
- package/src/store/user/slices/settings/selectors/index.ts +1 -0
- package/src/store/user/slices/settings/selectors/keyVaults.ts +21 -0
- package/src/store/user/store.ts +0 -3
- package/src/tools/index.ts +0 -6
- package/src/tools/renders.ts +0 -3
- package/src/tools/web-browsing/Portal/Search/Footer.tsx +2 -2
- package/src/tools/web-browsing/Render/Search/ConfigForm/Form.tsx +1 -1
- package/tsconfig.json +9 -3
- package/packages/const/src/guide.ts +0 -89
- package/packages/context-engine/src/providers/InboxGuide.ts +0 -102
- package/packages/context-engine/src/providers/__tests__/InboxGuideProvider.test.ts +0 -121
- package/packages/utils/src/_deprecated/__snapshots__/parseModels.test.ts.snap +0 -104
- package/packages/utils/src/_deprecated/parseModels.test.ts +0 -287
- package/packages/utils/src/_deprecated/parseModels.ts +0 -165
- package/src/app/(backend)/trpc/edge/[trpc]/route.ts +0 -26
- package/src/app/[variants]/(main)/(mobile)/me/data/features/Category.tsx +0 -48
- package/src/app/[variants]/(main)/(mobile)/me/data/features/Header.tsx +0 -33
- package/src/app/[variants]/(main)/(mobile)/me/data/layout.tsx +0 -13
- package/src/app/[variants]/(main)/(mobile)/me/data/loading.tsx +0 -5
- package/src/app/[variants]/(main)/(mobile)/me/data/page.tsx +0 -29
- package/src/app/[variants]/(main)/chat/features/Migration/DBReader.ts +0 -290
- package/src/app/[variants]/(main)/chat/features/Migration/ExportConfigButton.tsx +0 -35
- package/src/app/[variants]/(main)/chat/features/Migration/Failed.tsx +0 -120
- package/src/app/[variants]/(main)/chat/features/Migration/Modal.tsx +0 -81
- package/src/app/[variants]/(main)/chat/features/Migration/Start.tsx +0 -108
- package/src/app/[variants]/(main)/chat/features/Migration/UpgradeButton.tsx +0 -71
- package/src/app/[variants]/(main)/chat/features/Migration/const.ts +0 -15
- package/src/app/[variants]/(main)/chat/features/Migration/index.tsx +0 -50
- package/src/app/[variants]/loading/Client/Content.tsx +0 -48
- package/src/app/[variants]/loading/Client/Error.tsx +0 -27
- package/src/app/[variants]/loading/Client/Redirect.tsx +0 -47
- package/src/app/[variants]/loading/Client/index.tsx +0 -22
- package/src/components/InnerLink.tsx +0 -20
- package/src/database/_deprecated/core/__tests__/db-upgrade.test.ts +0 -42
- package/src/database/_deprecated/core/__tests__/db.test.ts +0 -79
- package/src/database/_deprecated/core/__tests__/model.test.ts +0 -55
- package/src/database/_deprecated/core/db.ts +0 -246
- package/src/database/_deprecated/core/index.ts +0 -2
- package/src/database/_deprecated/core/migrations/migrateSettingsToUser/fixtures/input.json +0 -55
- package/src/database/_deprecated/core/migrations/migrateSettingsToUser/fixtures/output.json +0 -60
- package/src/database/_deprecated/core/migrations/migrateSettingsToUser/index.test.ts +0 -14
- package/src/database/_deprecated/core/migrations/migrateSettingsToUser/index.ts +0 -22
- package/src/database/_deprecated/core/migrations/migrateSettingsToUser/type.ts +0 -105
- package/src/database/_deprecated/core/model.ts +0 -218
- package/src/database/_deprecated/core/schemas.ts +0 -88
- package/src/database/_deprecated/core/types/db.ts +0 -15
- package/src/database/_deprecated/models/__DEBUG.ts +0 -124
- package/src/database/_deprecated/models/__tests__/file.test.ts +0 -83
- package/src/database/_deprecated/models/__tests__/message.test.ts +0 -426
- package/src/database/_deprecated/models/__tests__/plugin.test.ts +0 -81
- package/src/database/_deprecated/models/__tests__/session.test.ts +0 -253
- package/src/database/_deprecated/models/__tests__/sessionGroup.test.ts +0 -220
- package/src/database/_deprecated/models/__tests__/topic.test.ts +0 -523
- package/src/database/_deprecated/models/__tests__/user.test.ts +0 -82
- package/src/database/_deprecated/models/file.ts +0 -51
- package/src/database/_deprecated/models/message.ts +0 -277
- package/src/database/_deprecated/models/plugin.ts +0 -62
- package/src/database/_deprecated/models/session.ts +0 -271
- package/src/database/_deprecated/models/sessionGroup.ts +0 -93
- package/src/database/_deprecated/models/topic.ts +0 -250
- package/src/database/_deprecated/models/user.ts +0 -69
- package/src/database/_deprecated/schemas/files.ts +0 -39
- package/src/database/_deprecated/schemas/message.ts +0 -50
- package/src/database/_deprecated/schemas/plugin.ts +0 -12
- package/src/database/_deprecated/schemas/session.ts +0 -54
- package/src/database/_deprecated/schemas/sessionGroup.ts +0 -8
- package/src/database/_deprecated/schemas/topic.ts +0 -12
- package/src/database/_deprecated/schemas/user.ts +0 -40
- package/src/envs/__tests__/auth.test.ts +0 -200
- package/src/features/DataImporter/_deprecated.ts +0 -43
- package/src/features/InitClientDB/EnableModal.tsx +0 -118
- package/src/features/InitClientDB/ErrorResult.tsx +0 -143
- package/src/features/InitClientDB/InitIndicator.tsx +0 -124
- package/src/features/InitClientDB/PGliteIcon.tsx +0 -28
- package/src/features/InitClientDB/features/DatabaseRepair/Backup.tsx +0 -75
- package/src/features/InitClientDB/features/DatabaseRepair/Diagnosis.tsx +0 -98
- package/src/features/InitClientDB/features/DatabaseRepair/Repair.tsx +0 -218
- package/src/features/InitClientDB/features/DatabaseRepair/index.tsx +0 -91
- package/src/features/InitClientDB/index.tsx +0 -37
- package/src/hooks/_header.ts +0 -23
- package/src/libs/next-auth/sso-providers/azure-ad.ts +0 -33
- package/src/libs/trpc/client/edge.ts +0 -26
- package/src/libs/trpc/edge/context.ts +0 -71
- package/src/libs/trpc/edge/index.ts +0 -45
- package/src/libs/trpc/edge/init.ts +0 -26
- package/src/libs/trpc/edge/middleware/jwtPayload.test.ts +0 -75
- package/src/libs/trpc/edge/middleware/jwtPayload.ts +0 -14
- package/src/migrations/FromV0ToV1.ts +0 -10
- package/src/migrations/FromV1ToV2/fixtures/input-v1-session.json +0 -191
- package/src/migrations/FromV1ToV2/fixtures/output-v2.json +0 -202
- package/src/migrations/FromV1ToV2/index.ts +0 -82
- package/src/migrations/FromV1ToV2/migrations.test.ts +0 -224
- package/src/migrations/FromV1ToV2/types/v1.ts +0 -78
- package/src/migrations/FromV1ToV2/types/v2.ts +0 -52
- package/src/migrations/FromV2ToV3/fixtures/input-v2-session.json +0 -72
- package/src/migrations/FromV2ToV3/fixtures/output-v3-from-v1.json +0 -203
- package/src/migrations/FromV2ToV3/fixtures/output-v3.json +0 -74
- package/src/migrations/FromV2ToV3/index.ts +0 -30
- package/src/migrations/FromV2ToV3/migrations.test.ts +0 -42
- package/src/migrations/FromV2ToV3/types/v3.ts +0 -27
- package/src/migrations/FromV3ToV4/fixtures/azure-input-v3.json +0 -79
- package/src/migrations/FromV3ToV4/fixtures/azure-output-v4.json +0 -75
- package/src/migrations/FromV3ToV4/fixtures/ollama-input-v3.json +0 -85
- package/src/migrations/FromV3ToV4/fixtures/ollama-output-v4.json +0 -86
- package/src/migrations/FromV3ToV4/fixtures/openai-input-v3.json +0 -77
- package/src/migrations/FromV3ToV4/fixtures/openai-output-v4.json +0 -77
- package/src/migrations/FromV3ToV4/fixtures/openrouter-input-v3.json +0 -82
- package/src/migrations/FromV3ToV4/fixtures/openrouter-output-v4.json +0 -85
- package/src/migrations/FromV3ToV4/fixtures/output-v4-from-v1.json +0 -203
- package/src/migrations/FromV3ToV4/index.ts +0 -102
- package/src/migrations/FromV3ToV4/migrations.test.ts +0 -195
- package/src/migrations/FromV3ToV4/types/v3.ts +0 -52
- package/src/migrations/FromV3ToV4/types/v4.ts +0 -37
- package/src/migrations/FromV4ToV5/fixtures/from-v1-to-v5-output.json +0 -245
- package/src/migrations/FromV4ToV5/fixtures/function-input-v4.json +0 -96
- package/src/migrations/FromV4ToV5/fixtures/function-output-v5.json +0 -120
- package/src/migrations/FromV4ToV5/index.ts +0 -58
- package/src/migrations/FromV4ToV5/migrations.test.ts +0 -49
- package/src/migrations/FromV4ToV5/types/v4.ts +0 -21
- package/src/migrations/FromV4ToV5/types/v5.ts +0 -27
- package/src/migrations/FromV5ToV6/fixtures/from-v1-to-v6-output.json +0 -247
- package/src/migrations/FromV5ToV6/fixtures/session-input-v5.json +0 -81
- package/src/migrations/FromV5ToV6/fixtures/session-output-v6.json +0 -85
- package/src/migrations/FromV5ToV6/index.ts +0 -61
- package/src/migrations/FromV5ToV6/migrations.test.ts +0 -50
- package/src/migrations/FromV5ToV6/types/v5.ts +0 -48
- package/src/migrations/FromV5ToV6/types/v6.ts +0 -63
- package/src/migrations/FromV6ToV7/fixtures/output-v7-from-v1.json +0 -203
- package/src/migrations/FromV6ToV7/fixtures/provider-input-v6.json +0 -103
- package/src/migrations/FromV6ToV7/fixtures/provider-output-v7.json +0 -118
- package/src/migrations/FromV6ToV7/index.ts +0 -101
- package/src/migrations/FromV6ToV7/migrations.test.ts +0 -64
- package/src/migrations/FromV6ToV7/types/v6.ts +0 -61
- package/src/migrations/FromV6ToV7/types/v7.ts +0 -69
- package/src/migrations/VersionController.test.ts +0 -88
- package/src/migrations/VersionController.ts +0 -67
- package/src/migrations/index.ts +0 -61
- package/src/server/globalConfig/_deprecated.test.ts +0 -92
- package/src/server/globalConfig/_deprecated.ts +0 -41
- package/src/server/routers/edge/appStatus.ts +0 -3
- package/src/server/routers/edge/index.ts +0 -14
- package/src/server/routers/edge/upload.ts +0 -16
- package/src/services/aiModel/client.ts +0 -70
- package/src/services/aiModel/server.test.ts +0 -122
- package/src/services/aiModel/server.ts +0 -51
- package/src/services/aiModel/type.ts +0 -32
- package/src/services/aiProvider/client.ts +0 -58
- package/src/services/aiProvider/server.ts +0 -43
- package/src/services/aiProvider/type.ts +0 -27
- package/src/services/baseClientService/index.ts +0 -9
- package/src/services/chat/__snapshots__/chat.test.ts.snap +0 -110
- package/src/services/chatGroup/client.ts +0 -63
- package/src/services/chatGroup/server.ts +0 -67
- package/src/services/chatGroup/type.ts +0 -22
- package/src/services/export/_deprecated.ts +0 -155
- package/src/services/export/client.ts +0 -15
- package/src/services/export/server.ts +0 -9
- package/src/services/export/type.ts +0 -5
- package/src/services/file/_deprecated.test.ts +0 -119
- package/src/services/file/_deprecated.ts +0 -80
- package/src/services/file/client.test.ts +0 -199
- package/src/services/file/client.ts +0 -85
- package/src/services/file/server.ts +0 -53
- package/src/services/file/type.ts +0 -13
- package/src/services/import/_deprecated.ts +0 -115
- package/src/services/import/client.test.ts +0 -1015
- package/src/services/import/client.ts +0 -64
- package/src/services/import/server.ts +0 -133
- package/src/services/import/type.ts +0 -17
- package/src/services/message/_deprecated.test.ts +0 -398
- package/src/services/message/_deprecated.ts +0 -168
- package/src/services/message/client.test.ts +0 -410
- package/src/services/message/client.ts +0 -186
- package/src/services/message/server.ts +0 -150
- package/src/services/message/type.ts +0 -54
- package/src/services/plugin/_deprecated.test.ts +0 -162
- package/src/services/plugin/_deprecated.ts +0 -42
- package/src/services/plugin/client.test.ts +0 -177
- package/src/services/plugin/client.ts +0 -46
- package/src/services/plugin/server.ts +0 -42
- package/src/services/plugin/type.ts +0 -23
- package/src/services/session/_deprecated.test.ts +0 -440
- package/src/services/session/_deprecated.ts +0 -190
- package/src/services/session/client.test.ts +0 -413
- package/src/services/session/client.ts +0 -193
- package/src/services/session/server.test.ts +0 -260
- package/src/services/session/server.ts +0 -125
- package/src/services/session/type.ts +0 -82
- package/src/services/thread/client.ts +0 -51
- package/src/services/thread/server.ts +0 -32
- package/src/services/thread/type.ts +0 -21
- package/src/services/topic/_deprecated.test.ts +0 -245
- package/src/services/topic/_deprecated.ts +0 -75
- package/src/services/topic/client.ts +0 -89
- package/src/services/topic/pglite.test.ts +0 -212
- package/src/services/topic/server.ts +0 -57
- package/src/services/topic/type.ts +0 -40
- package/src/services/user/_deprecated.test.ts +0 -101
- package/src/services/user/_deprecated.ts +0 -70
- package/src/services/user/client.test.ts +0 -108
- package/src/services/user/client.ts +0 -104
- package/src/services/user/server.test.ts +0 -149
- package/src/services/user/server.ts +0 -47
- package/src/services/user/type.ts +0 -21
- package/src/store/chat/slices/builtinTool/actions/__tests__/dalle.test.ts +0 -121
- package/src/store/chat/slices/builtinTool/actions/dalle.ts +0 -124
- package/src/store/global/actions/clientDb.ts +0 -67
- package/src/store/user/slices/modelList/__snapshots__/action.test.ts.snap +0 -12
- package/src/store/user/slices/modelList/action.test.ts +0 -359
- package/src/store/user/slices/modelList/action.ts +0 -223
- package/src/store/user/slices/modelList/initialState.ts +0 -15
- package/src/store/user/slices/modelList/reducers/customModelCard.test.ts +0 -204
- package/src/store/user/slices/modelList/reducers/customModelCard.ts +0 -64
- package/src/store/user/slices/modelList/selectors/index.ts +0 -3
- package/src/store/user/slices/modelList/selectors/keyVaults.test.ts +0 -201
- package/src/store/user/slices/modelList/selectors/keyVaults.ts +0 -50
- package/src/store/user/slices/modelList/selectors/modelConfig.test.ts +0 -219
- package/src/store/user/slices/modelList/selectors/modelConfig.ts +0 -95
- package/src/store/user/slices/modelList/selectors/modelProvider.test.ts +0 -138
- package/src/store/user/slices/modelList/selectors/modelProvider.ts +0 -170
- package/src/tools/dalle/Render/GalleyGrid.tsx +0 -60
- package/src/tools/dalle/Render/Item/EditMode.tsx +0 -66
- package/src/tools/dalle/Render/Item/Error.tsx +0 -49
- package/src/tools/dalle/Render/Item/Image.tsx +0 -44
- package/src/tools/dalle/Render/Item/ImageFileItem.tsx +0 -57
- package/src/tools/dalle/Render/Item/index.tsx +0 -88
- package/src/tools/dalle/Render/ToolBar.tsx +0 -56
- package/src/tools/dalle/Render/index.tsx +0 -52
- package/src/tools/dalle/index.ts +0 -92
- /package/src/{app/[variants]/(main)/settings/provider/features/ProviderConfig → components/Skeleton}/SkeletonInput.tsx +0 -0
- /package/src/{middleware.ts → proxy.ts} +0 -0
|
@@ -28,19 +28,21 @@ export class AgentModel {
|
|
|
28
28
|
};
|
|
29
29
|
|
|
30
30
|
getAgentAssignedKnowledge = async (id: string) => {
|
|
31
|
-
|
|
32
|
-
|
|
33
|
-
.
|
|
34
|
-
|
|
35
|
-
|
|
36
|
-
|
|
37
|
-
|
|
38
|
-
|
|
39
|
-
.
|
|
40
|
-
|
|
41
|
-
|
|
42
|
-
|
|
43
|
-
|
|
31
|
+
// Run both queries in parallel for better performance
|
|
32
|
+
const [knowledgeBaseResult, fileResult] = await Promise.all([
|
|
33
|
+
this.db
|
|
34
|
+
.select({ enabled: agentsKnowledgeBases.enabled, knowledgeBases })
|
|
35
|
+
.from(agentsKnowledgeBases)
|
|
36
|
+
.where(eq(agentsKnowledgeBases.agentId, id))
|
|
37
|
+
.orderBy(desc(agentsKnowledgeBases.createdAt))
|
|
38
|
+
.leftJoin(knowledgeBases, eq(knowledgeBases.id, agentsKnowledgeBases.knowledgeBaseId)),
|
|
39
|
+
this.db
|
|
40
|
+
.select({ enabled: agentsFiles.enabled, files })
|
|
41
|
+
.from(agentsFiles)
|
|
42
|
+
.where(eq(agentsFiles.agentId, id))
|
|
43
|
+
.orderBy(desc(agentsFiles.createdAt))
|
|
44
|
+
.leftJoin(files, eq(files.id, agentsFiles.fileId)),
|
|
45
|
+
]);
|
|
44
46
|
|
|
45
47
|
return {
|
|
46
48
|
files: fileResult.map((item) => ({
|
|
@@ -61,6 +63,7 @@ export class AgentModel {
|
|
|
61
63
|
const item = await this.db.query.agentsToSessions.findFirst({
|
|
62
64
|
where: eq(agentsToSessions.sessionId, sessionId),
|
|
63
65
|
});
|
|
66
|
+
|
|
64
67
|
if (!item) return;
|
|
65
68
|
|
|
66
69
|
const agentId = item.agentId;
|
|
@@ -1,3 +1,4 @@
|
|
|
1
|
+
import { INBOX_SESSION_ID } from '@lobechat/const';
|
|
1
2
|
import {
|
|
2
3
|
ChatFileItem,
|
|
3
4
|
ChatImageItem,
|
|
@@ -6,7 +7,6 @@ import {
|
|
|
6
7
|
ChatTranslate,
|
|
7
8
|
ChatVideoItem,
|
|
8
9
|
CreateMessageParams,
|
|
9
|
-
CreateMessageResult,
|
|
10
10
|
DBMessageItem,
|
|
11
11
|
ModelRankItem,
|
|
12
12
|
NewMessageQueryParams,
|
|
@@ -14,6 +14,7 @@ import {
|
|
|
14
14
|
UIChatMessage,
|
|
15
15
|
UpdateMessageParams,
|
|
16
16
|
UpdateMessageRAGParams,
|
|
17
|
+
UpdateMessageResult,
|
|
17
18
|
} from '@lobechat/types';
|
|
18
19
|
import type { HeatmapsProps } from '@lobehub/charts';
|
|
19
20
|
import dayjs from 'dayjs';
|
|
@@ -39,6 +40,7 @@ import {
|
|
|
39
40
|
} from '../schemas';
|
|
40
41
|
import { LobeChatDatabase } from '../type';
|
|
41
42
|
import { genEndDateWhere, genRangeWhere, genStartDateWhere, genWhere } from '../utils/genWhere';
|
|
43
|
+
import { groupAssistantMessages } from '../utils/groupMessages';
|
|
42
44
|
import { idGenerator } from '../utils/idGenerator';
|
|
43
45
|
|
|
44
46
|
export class MessageModel {
|
|
@@ -54,6 +56,7 @@ export class MessageModel {
|
|
|
54
56
|
query = async (
|
|
55
57
|
{ current = 0, pageSize = 1000, sessionId, topicId, groupId }: QueryMessageParams = {},
|
|
56
58
|
options: {
|
|
59
|
+
groupAssistantMessages?: boolean;
|
|
57
60
|
postProcessUrl?: (path: string | null, file: { fileType: string }) => Promise<string>;
|
|
58
61
|
} = {},
|
|
59
62
|
) => {
|
|
@@ -211,7 +214,7 @@ export class MessageModel {
|
|
|
211
214
|
.from(messageQueries)
|
|
212
215
|
.where(inArray(messageQueries.messageId, messageIds));
|
|
213
216
|
|
|
214
|
-
|
|
217
|
+
const mappedMessages = result.map(
|
|
215
218
|
({ model, provider, translate, ttsId, ttsFile, ttsContentMd5, ttsVoice, ...item }) => {
|
|
216
219
|
const messageQuery = messageQueriesList.find((relation) => relation.messageId === item.id);
|
|
217
220
|
return {
|
|
@@ -246,13 +249,15 @@ export class MessageModel {
|
|
|
246
249
|
size: size!,
|
|
247
250
|
url,
|
|
248
251
|
})),
|
|
249
|
-
|
|
250
252
|
imageList: imageList
|
|
251
253
|
.filter((relation) => relation.messageId === item.id)
|
|
252
254
|
// eslint-disable-next-line @typescript-eslint/no-unused-vars
|
|
253
255
|
.map<ChatImageItem>(({ id, url, name }) => ({ alt: name!, id, url })),
|
|
254
|
-
|
|
255
256
|
meta: {},
|
|
257
|
+
|
|
258
|
+
model,
|
|
259
|
+
|
|
260
|
+
provider,
|
|
256
261
|
ragQuery: messageQuery?.rewriteQuery,
|
|
257
262
|
ragQueryId: messageQuery?.id,
|
|
258
263
|
ragRawQuery: messageQuery?.userQuery,
|
|
@@ -263,6 +268,10 @@ export class MessageModel {
|
|
|
263
268
|
} as unknown as UIChatMessage;
|
|
264
269
|
},
|
|
265
270
|
);
|
|
271
|
+
|
|
272
|
+
// Group assistant messages with their tool results
|
|
273
|
+
const { groupAssistantMessages: useGroup = false } = options;
|
|
274
|
+
return useGroup ? groupAssistantMessages(mappedMessages) : mappedMessages;
|
|
266
275
|
};
|
|
267
276
|
|
|
268
277
|
findById = async (id: string) => {
|
|
@@ -522,54 +531,6 @@ export class MessageModel {
|
|
|
522
531
|
});
|
|
523
532
|
};
|
|
524
533
|
|
|
525
|
-
/**
|
|
526
|
-
* Create a new message and return the complete message list
|
|
527
|
-
*
|
|
528
|
-
* This method combines message creation and querying into a single operation,
|
|
529
|
-
* reducing the need for separate refresh calls and improving performance.
|
|
530
|
-
*
|
|
531
|
-
* @param params - Message creation parameters
|
|
532
|
-
* @param options - Query options for post-processing
|
|
533
|
-
* @returns Object containing the created message ID and full message list
|
|
534
|
-
*
|
|
535
|
-
* @example
|
|
536
|
-
* const { id, messages } = await messageModel.createNewMessage({
|
|
537
|
-
* role: 'assistant',
|
|
538
|
-
* content: 'Hello',
|
|
539
|
-
* tools: [...],
|
|
540
|
-
* sessionId: 'session-1',
|
|
541
|
-
* });
|
|
542
|
-
* // messages already contains grouped structure, no need to refresh
|
|
543
|
-
*/
|
|
544
|
-
createNewMessage = async (
|
|
545
|
-
params: CreateMessageParams,
|
|
546
|
-
options: {
|
|
547
|
-
postProcessUrl?: (path: string | null, file: { fileType: string }) => Promise<string>;
|
|
548
|
-
} = {},
|
|
549
|
-
): Promise<CreateMessageResult> => {
|
|
550
|
-
// 1. Create the message (reuse existing create method)
|
|
551
|
-
const item = await this.create(params);
|
|
552
|
-
|
|
553
|
-
// 2. Query all messages for this session/topic
|
|
554
|
-
// query() method internally applies groupAssistantMessages transformation
|
|
555
|
-
const messages = await this.query(
|
|
556
|
-
{
|
|
557
|
-
current: 0,
|
|
558
|
-
groupId: params.groupId,
|
|
559
|
-
pageSize: 9999,
|
|
560
|
-
sessionId: params.sessionId,
|
|
561
|
-
topicId: params.topicId, // Get all messages
|
|
562
|
-
},
|
|
563
|
-
options,
|
|
564
|
-
);
|
|
565
|
-
|
|
566
|
-
// 3. Return the result
|
|
567
|
-
return {
|
|
568
|
-
id: item.id,
|
|
569
|
-
messages,
|
|
570
|
-
};
|
|
571
|
-
};
|
|
572
|
-
|
|
573
534
|
batchCreate = async (newMessages: DBMessageItem[]) => {
|
|
574
535
|
const messagesToInsert = newMessages.map((m) => {
|
|
575
536
|
// TODO: need a better way to handle this
|
|
@@ -589,27 +550,54 @@ export class MessageModel {
|
|
|
589
550
|
};
|
|
590
551
|
// **************** Update *************** //
|
|
591
552
|
|
|
592
|
-
update = async (
|
|
593
|
-
|
|
594
|
-
|
|
595
|
-
|
|
553
|
+
update = async (
|
|
554
|
+
id: string,
|
|
555
|
+
{ imageList, ...message }: Partial<UpdateMessageParams>,
|
|
556
|
+
options?: {
|
|
557
|
+
groupAssistantMessages?: boolean;
|
|
558
|
+
postProcessUrl?: (path: string | null, file: { fileType: string }) => Promise<string>;
|
|
559
|
+
sessionId?: string | null;
|
|
560
|
+
topicId?: string | null;
|
|
561
|
+
},
|
|
562
|
+
): Promise<UpdateMessageResult> => {
|
|
563
|
+
try {
|
|
564
|
+
await this.db.transaction(async (trx) => {
|
|
565
|
+
// 1. insert message files
|
|
566
|
+
if (imageList && imageList.length > 0) {
|
|
567
|
+
await trx
|
|
568
|
+
.insert(messagesFiles)
|
|
569
|
+
.values(
|
|
570
|
+
imageList.map((file) => ({ fileId: file.id, messageId: id, userId: this.userId })),
|
|
571
|
+
);
|
|
572
|
+
}
|
|
573
|
+
|
|
596
574
|
await trx
|
|
597
|
-
.
|
|
598
|
-
.
|
|
599
|
-
|
|
600
|
-
|
|
575
|
+
.update(messages)
|
|
576
|
+
.set({ ...message })
|
|
577
|
+
.where(and(eq(messages.id, id), eq(messages.userId, this.userId)));
|
|
578
|
+
});
|
|
579
|
+
|
|
580
|
+
// if sessionId or topicId provided, return the updated message list
|
|
581
|
+
if (options?.sessionId !== undefined || options?.topicId !== undefined) {
|
|
582
|
+
const messageList = await this.query(
|
|
583
|
+
{
|
|
584
|
+
sessionId: options.sessionId,
|
|
585
|
+
topicId: options.topicId,
|
|
586
|
+
},
|
|
587
|
+
{
|
|
588
|
+
groupAssistantMessages: options.groupAssistantMessages ?? false,
|
|
589
|
+
postProcessUrl: options.postProcessUrl,
|
|
590
|
+
},
|
|
591
|
+
);
|
|
592
|
+
|
|
593
|
+
return { messages: messageList, success: true };
|
|
601
594
|
}
|
|
602
595
|
|
|
603
|
-
return
|
|
604
|
-
|
|
605
|
-
|
|
606
|
-
|
|
607
|
-
|
|
608
|
-
// TODO: but I forget why 🤡
|
|
609
|
-
role: message.role as any,
|
|
610
|
-
})
|
|
611
|
-
.where(and(eq(messages.id, id), eq(messages.userId, this.userId)));
|
|
612
|
-
});
|
|
596
|
+
return { success: true };
|
|
597
|
+
} catch (error) {
|
|
598
|
+
console.error('Update message error:', error);
|
|
599
|
+
return { success: false };
|
|
600
|
+
}
|
|
613
601
|
};
|
|
614
602
|
|
|
615
603
|
updateMetadata = async (id: string, metadata: Record<string, any>) => {
|
|
@@ -625,16 +613,41 @@ export class MessageModel {
|
|
|
625
613
|
.where(and(eq(messages.userId, this.userId), eq(messages.id, id)));
|
|
626
614
|
};
|
|
627
615
|
|
|
628
|
-
updatePluginState = async (
|
|
616
|
+
updatePluginState = async (
|
|
617
|
+
id: string,
|
|
618
|
+
state: Record<string, any>,
|
|
619
|
+
options?: {
|
|
620
|
+
groupAssistantMessages?: boolean;
|
|
621
|
+
postProcessUrl?: (path: string | null, file: { fileType: string }) => Promise<string>;
|
|
622
|
+
sessionId?: string | null;
|
|
623
|
+
topicId?: string | null;
|
|
624
|
+
},
|
|
625
|
+
): Promise<UpdateMessageResult> => {
|
|
629
626
|
const item = await this.db.query.messagePlugins.findFirst({
|
|
630
627
|
where: eq(messagePlugins.id, id),
|
|
631
628
|
});
|
|
632
629
|
if (!item) throw new Error('Plugin not found');
|
|
633
630
|
|
|
634
|
-
|
|
631
|
+
await this.db
|
|
635
632
|
.update(messagePlugins)
|
|
636
633
|
.set({ state: merge(item.state || {}, state) })
|
|
637
634
|
.where(eq(messagePlugins.id, id));
|
|
635
|
+
|
|
636
|
+
// Return updated messages if sessionId or topicId is provided
|
|
637
|
+
if (options?.sessionId !== undefined || options?.topicId !== undefined) {
|
|
638
|
+
const messageList = await this.query(
|
|
639
|
+
{
|
|
640
|
+
sessionId: options.sessionId,
|
|
641
|
+
topicId: options.topicId,
|
|
642
|
+
},
|
|
643
|
+
{
|
|
644
|
+
groupAssistantMessages: options.groupAssistantMessages ?? false,
|
|
645
|
+
postProcessUrl: options.postProcessUrl,
|
|
646
|
+
},
|
|
647
|
+
);
|
|
648
|
+
return { messages: messageList, success: true };
|
|
649
|
+
}
|
|
650
|
+
return { success: true };
|
|
638
651
|
};
|
|
639
652
|
|
|
640
653
|
updateMessagePlugin = async (id: string, value: Partial<MessagePluginItem>) => {
|
|
@@ -778,8 +791,11 @@ export class MessageModel {
|
|
|
778
791
|
|
|
779
792
|
private genId = () => idGenerator('messages', 14);
|
|
780
793
|
|
|
781
|
-
private matchSession = (sessionId?: string | null) =>
|
|
782
|
-
|
|
794
|
+
private matchSession = (sessionId?: string | null) => {
|
|
795
|
+
if (sessionId === INBOX_SESSION_ID) return isNull(messages.sessionId);
|
|
796
|
+
|
|
797
|
+
return sessionId ? eq(messages.sessionId, sessionId) : isNull(messages.sessionId);
|
|
798
|
+
};
|
|
783
799
|
|
|
784
800
|
private matchTopic = (topicId?: string | null) =>
|
|
785
801
|
topicId ? eq(messages.topicId, topicId) : isNull(messages.topicId);
|
|
@@ -53,13 +53,44 @@ export class SessionModel {
|
|
|
53
53
|
query = async ({ current = 0, pageSize = 9999 } = {}) => {
|
|
54
54
|
const offset = current * pageSize;
|
|
55
55
|
|
|
56
|
-
|
|
57
|
-
|
|
58
|
-
|
|
59
|
-
|
|
60
|
-
|
|
61
|
-
|
|
62
|
-
|
|
56
|
+
// Use leftJoin instead of nested with for better performance
|
|
57
|
+
const result = await this.db
|
|
58
|
+
.select({
|
|
59
|
+
// Agent fields (from agentsToSessions join)
|
|
60
|
+
agent: agents,
|
|
61
|
+
// Group fields
|
|
62
|
+
group: sessionGroups,
|
|
63
|
+
// Session fields
|
|
64
|
+
session: sessions,
|
|
65
|
+
})
|
|
66
|
+
.from(sessions)
|
|
67
|
+
.leftJoin(agentsToSessions, eq(sessions.id, agentsToSessions.sessionId))
|
|
68
|
+
.leftJoin(agents, eq(agentsToSessions.agentId, agents.id))
|
|
69
|
+
.leftJoin(sessionGroups, eq(sessions.groupId, sessionGroups.id))
|
|
70
|
+
.where(and(eq(sessions.userId, this.userId), not(eq(sessions.slug, INBOX_SESSION_ID))))
|
|
71
|
+
.orderBy(desc(sessions.updatedAt))
|
|
72
|
+
.limit(pageSize)
|
|
73
|
+
.offset(offset);
|
|
74
|
+
|
|
75
|
+
// Group results by session (since leftJoin can create multiple rows per session)
|
|
76
|
+
// Use Map to preserve order
|
|
77
|
+
const groupedResults = new Map<string, any>();
|
|
78
|
+
|
|
79
|
+
for (const row of result) {
|
|
80
|
+
const sessionId = row.session.id;
|
|
81
|
+
if (!groupedResults.has(sessionId)) {
|
|
82
|
+
groupedResults.set(sessionId, {
|
|
83
|
+
...row.session,
|
|
84
|
+
agentsToSessions: [],
|
|
85
|
+
group: row.group,
|
|
86
|
+
});
|
|
87
|
+
}
|
|
88
|
+
if (row.agent) {
|
|
89
|
+
groupedResults.get(sessionId)!.agentsToSessions.push({ agent: row.agent });
|
|
90
|
+
}
|
|
91
|
+
}
|
|
92
|
+
|
|
93
|
+
return Array.from(groupedResults.values());
|
|
63
94
|
};
|
|
64
95
|
|
|
65
96
|
queryWithGroups = async (): Promise<ChatSessionList> => {
|
|
@@ -71,9 +102,11 @@ export class SessionModel {
|
|
|
71
102
|
where: eq(sessions.userId, this.userId),
|
|
72
103
|
});
|
|
73
104
|
|
|
105
|
+
const mappedSessions = result.map((item) => this.mapSessionItem(item as any));
|
|
106
|
+
|
|
74
107
|
return {
|
|
75
108
|
sessionGroups: groups as unknown as ChatSessionList['sessionGroups'],
|
|
76
|
-
sessions:
|
|
109
|
+
sessions: mappedSessions,
|
|
77
110
|
};
|
|
78
111
|
};
|
|
79
112
|
|
|
@@ -90,17 +123,28 @@ export class SessionModel {
|
|
|
90
123
|
findByIdOrSlug = async (
|
|
91
124
|
idOrSlug: string,
|
|
92
125
|
): Promise<(SessionItem & { agent: AgentItem }) | undefined> => {
|
|
93
|
-
|
|
94
|
-
|
|
95
|
-
|
|
96
|
-
|
|
97
|
-
|
|
98
|
-
|
|
99
|
-
|
|
126
|
+
// Use leftJoin instead of nested 'with' for better performance
|
|
127
|
+
const result = await this.db
|
|
128
|
+
.select({
|
|
129
|
+
agent: agents,
|
|
130
|
+
group: sessionGroups,
|
|
131
|
+
session: sessions,
|
|
132
|
+
})
|
|
133
|
+
.from(sessions)
|
|
134
|
+
.where(
|
|
135
|
+
and(
|
|
136
|
+
or(eq(sessions.id, idOrSlug), eq(sessions.slug, idOrSlug)),
|
|
137
|
+
eq(sessions.userId, this.userId),
|
|
138
|
+
),
|
|
139
|
+
)
|
|
140
|
+
.leftJoin(agentsToSessions, eq(sessions.id, agentsToSessions.sessionId))
|
|
141
|
+
.leftJoin(agents, eq(agentsToSessions.agentId, agents.id))
|
|
142
|
+
.leftJoin(sessionGroups, eq(sessions.groupId, sessionGroups.id))
|
|
143
|
+
.limit(1);
|
|
100
144
|
|
|
101
|
-
if (!result) return;
|
|
145
|
+
if (!result || !result[0]) return;
|
|
102
146
|
|
|
103
|
-
return { ...result, agent:
|
|
147
|
+
return { ...result[0].session, agent: result[0].agent, group: result[0].group } as any;
|
|
104
148
|
};
|
|
105
149
|
|
|
106
150
|
count = async (params?: {
|
|
@@ -332,7 +376,7 @@ export class SessionModel {
|
|
|
332
376
|
.delete(agentsToSessions)
|
|
333
377
|
.where(and(eq(agentsToSessions.sessionId, id), eq(agentsToSessions.userId, this.userId)));
|
|
334
378
|
|
|
335
|
-
// Delete the session
|
|
379
|
+
// Delete the session (this will cascade delete messages, topics, etc.)
|
|
336
380
|
const result = await trx
|
|
337
381
|
.delete(sessions)
|
|
338
382
|
.where(and(eq(sessions.id, id), eq(sessions.userId, this.userId)));
|
|
@@ -397,17 +441,25 @@ export class SessionModel {
|
|
|
397
441
|
};
|
|
398
442
|
|
|
399
443
|
clearOrphanAgent = async (agentIds: string[], trx: any) => {
|
|
400
|
-
|
|
401
|
-
for (const agentId of agentIds) {
|
|
402
|
-
const remaining = await trx
|
|
403
|
-
.select()
|
|
404
|
-
.from(agentsToSessions)
|
|
405
|
-
.where(eq(agentsToSessions.agentId, agentId))
|
|
406
|
-
.limit(1);
|
|
444
|
+
if (agentIds.length === 0) return;
|
|
407
445
|
|
|
408
|
-
|
|
409
|
-
|
|
410
|
-
}
|
|
446
|
+
// Batch query to find which agents still have sessions
|
|
447
|
+
const remainingLinks = (await trx
|
|
448
|
+
.select({ agentId: agentsToSessions.agentId })
|
|
449
|
+
.from(agentsToSessions)
|
|
450
|
+
.where(inArray(agentsToSessions.agentId, agentIds))) as { agentId: string }[];
|
|
451
|
+
|
|
452
|
+
const linkedAgentIds = new Set(remainingLinks.map((link) => link.agentId));
|
|
453
|
+
|
|
454
|
+
// Find orphaned agents (those not in the linked set)
|
|
455
|
+
const orphanedAgentIds = agentIds.filter((id) => !linkedAgentIds.has(id));
|
|
456
|
+
|
|
457
|
+
// Batch delete orphaned agents (this will cascade to agentsFiles, agentsKnowledgeBases, etc.)
|
|
458
|
+
// and SET NULL on messages.agentId
|
|
459
|
+
if (orphanedAgentIds.length > 0) {
|
|
460
|
+
await trx
|
|
461
|
+
.delete(agents)
|
|
462
|
+
.where(and(inArray(agents.id, orphanedAgentIds), eq(agents.userId, this.userId)));
|
|
411
463
|
}
|
|
412
464
|
};
|
|
413
465
|
|
|
@@ -82,7 +82,8 @@ export class UserModel {
|
|
|
82
82
|
})
|
|
83
83
|
.from(users)
|
|
84
84
|
.where(eq(users.id, this.userId))
|
|
85
|
-
.leftJoin(userSettings, eq(users.id, userSettings.id))
|
|
85
|
+
.leftJoin(userSettings, eq(users.id, userSettings.id))
|
|
86
|
+
.limit(1);
|
|
86
87
|
|
|
87
88
|
if (!result || !result[0]) {
|
|
88
89
|
throw new UserNotFoundError();
|
|
@@ -731,6 +731,105 @@ describe('AiInfraRepos', () => {
|
|
|
731
731
|
// 无 settings
|
|
732
732
|
expect(merged?.settings).toBeUndefined();
|
|
733
733
|
});
|
|
734
|
+
|
|
735
|
+
it('should prefer user settings over builtin settings', async () => {
|
|
736
|
+
const mockProviders = [
|
|
737
|
+
{ enabled: true, id: 'openai', name: 'OpenAI', source: 'builtin' as const },
|
|
738
|
+
];
|
|
739
|
+
|
|
740
|
+
const userModel: EnabledAiModel = {
|
|
741
|
+
id: 'gpt-4',
|
|
742
|
+
providerId: 'openai',
|
|
743
|
+
enabled: true,
|
|
744
|
+
type: 'chat',
|
|
745
|
+
abilities: {},
|
|
746
|
+
settings: { searchImpl: 'params', searchProvider: 'user-provider' },
|
|
747
|
+
};
|
|
748
|
+
|
|
749
|
+
const builtinModel = {
|
|
750
|
+
id: 'gpt-4',
|
|
751
|
+
enabled: true,
|
|
752
|
+
type: 'chat' as const,
|
|
753
|
+
settings: { searchImpl: 'tool', searchProvider: 'builtin-provider' },
|
|
754
|
+
};
|
|
755
|
+
|
|
756
|
+
vi.spyOn(repo, 'getAiProviderList').mockResolvedValue(mockProviders);
|
|
757
|
+
vi.spyOn(repo.aiModelModel, 'getAllModels').mockResolvedValue([userModel]);
|
|
758
|
+
vi.spyOn(repo as any, 'fetchBuiltinModels').mockResolvedValue([builtinModel]);
|
|
759
|
+
|
|
760
|
+
const result = await repo.getEnabledModels();
|
|
761
|
+
|
|
762
|
+
const merged = result.find((m) => m.id === 'gpt-4');
|
|
763
|
+
expect(merged).toBeDefined();
|
|
764
|
+
// 应该使用用户的 settings,不是内置的
|
|
765
|
+
expect(merged?.settings).toEqual({ searchImpl: 'params', searchProvider: 'user-provider' });
|
|
766
|
+
});
|
|
767
|
+
|
|
768
|
+
it('should use builtin settings when user has no settings', async () => {
|
|
769
|
+
const mockProviders = [
|
|
770
|
+
{ enabled: true, id: 'openai', name: 'OpenAI', source: 'builtin' as const },
|
|
771
|
+
];
|
|
772
|
+
|
|
773
|
+
const userModel: EnabledAiModel = {
|
|
774
|
+
id: 'gpt-4',
|
|
775
|
+
providerId: 'openai',
|
|
776
|
+
enabled: true,
|
|
777
|
+
type: 'chat',
|
|
778
|
+
abilities: { vision: true },
|
|
779
|
+
// 用户未设置 settings
|
|
780
|
+
};
|
|
781
|
+
|
|
782
|
+
const builtinModel = {
|
|
783
|
+
id: 'gpt-4',
|
|
784
|
+
enabled: true,
|
|
785
|
+
type: 'chat' as const,
|
|
786
|
+
settings: { searchImpl: 'tool', searchProvider: 'google' },
|
|
787
|
+
};
|
|
788
|
+
|
|
789
|
+
vi.spyOn(repo, 'getAiProviderList').mockResolvedValue(mockProviders);
|
|
790
|
+
vi.spyOn(repo.aiModelModel, 'getAllModels').mockResolvedValue([userModel]);
|
|
791
|
+
vi.spyOn(repo as any, 'fetchBuiltinModels').mockResolvedValue([builtinModel]);
|
|
792
|
+
|
|
793
|
+
const result = await repo.getEnabledModels();
|
|
794
|
+
|
|
795
|
+
const merged = result.find((m) => m.id === 'gpt-4');
|
|
796
|
+
expect(merged).toBeDefined();
|
|
797
|
+
// 应该使用内置的 settings
|
|
798
|
+
expect(merged?.settings).toEqual({ searchImpl: 'tool', searchProvider: 'google' });
|
|
799
|
+
});
|
|
800
|
+
|
|
801
|
+
it('should have no settings when both user and builtin have no settings', async () => {
|
|
802
|
+
const mockProviders = [
|
|
803
|
+
{ enabled: true, id: 'openai', name: 'OpenAI', source: 'builtin' as const },
|
|
804
|
+
];
|
|
805
|
+
|
|
806
|
+
const userModel: EnabledAiModel = {
|
|
807
|
+
id: 'gpt-4',
|
|
808
|
+
providerId: 'openai',
|
|
809
|
+
enabled: true,
|
|
810
|
+
type: 'chat',
|
|
811
|
+
abilities: { vision: true },
|
|
812
|
+
// 用户未设置 settings
|
|
813
|
+
};
|
|
814
|
+
|
|
815
|
+
const builtinModel = {
|
|
816
|
+
id: 'gpt-4',
|
|
817
|
+
enabled: true,
|
|
818
|
+
type: 'chat' as const,
|
|
819
|
+
// 内置也无 settings
|
|
820
|
+
};
|
|
821
|
+
|
|
822
|
+
vi.spyOn(repo, 'getAiProviderList').mockResolvedValue(mockProviders);
|
|
823
|
+
vi.spyOn(repo.aiModelModel, 'getAllModels').mockResolvedValue([userModel]);
|
|
824
|
+
vi.spyOn(repo as any, 'fetchBuiltinModels').mockResolvedValue([builtinModel]);
|
|
825
|
+
|
|
826
|
+
const result = await repo.getEnabledModels();
|
|
827
|
+
|
|
828
|
+
const merged = result.find((m) => m.id === 'gpt-4');
|
|
829
|
+
expect(merged).toBeDefined();
|
|
830
|
+
// 无 settings
|
|
831
|
+
expect(merged?.settings).toBeUndefined();
|
|
832
|
+
});
|
|
734
833
|
});
|
|
735
834
|
|
|
736
835
|
describe('getAiProviderModelList', () => {
|
|
@@ -1270,6 +1369,105 @@ describe('AiInfraRepos', () => {
|
|
|
1270
1369
|
// 无 settings
|
|
1271
1370
|
expect(merged?.settings).toBeUndefined();
|
|
1272
1371
|
});
|
|
1372
|
+
|
|
1373
|
+
it('should prefer user settings over builtin settings in getAiProviderModelList', async () => {
|
|
1374
|
+
const providerId = 'openai';
|
|
1375
|
+
|
|
1376
|
+
const userModels: AiProviderModelListItem[] = [
|
|
1377
|
+
{
|
|
1378
|
+
id: 'gpt-4',
|
|
1379
|
+
type: 'chat',
|
|
1380
|
+
enabled: true,
|
|
1381
|
+
abilities: {},
|
|
1382
|
+
settings: { searchImpl: 'params', searchProvider: 'user-provider' },
|
|
1383
|
+
} as any,
|
|
1384
|
+
];
|
|
1385
|
+
|
|
1386
|
+
const builtinModels: AiProviderModelListItem[] = [
|
|
1387
|
+
{
|
|
1388
|
+
id: 'gpt-4',
|
|
1389
|
+
type: 'chat',
|
|
1390
|
+
enabled: true,
|
|
1391
|
+
settings: { searchImpl: 'tool', searchProvider: 'builtin-provider' },
|
|
1392
|
+
} as any,
|
|
1393
|
+
];
|
|
1394
|
+
|
|
1395
|
+
vi.spyOn(repo.aiModelModel, 'getModelListByProviderId').mockResolvedValue(userModels);
|
|
1396
|
+
vi.spyOn(repo as any, 'fetchBuiltinModels').mockResolvedValue(builtinModels);
|
|
1397
|
+
|
|
1398
|
+
const result = await repo.getAiProviderModelList(providerId);
|
|
1399
|
+
|
|
1400
|
+
const merged = result.find((m) => m.id === 'gpt-4');
|
|
1401
|
+
expect(merged).toBeDefined();
|
|
1402
|
+
// 应该使用用户的 settings
|
|
1403
|
+
expect(merged?.settings).toEqual({ searchImpl: 'params', searchProvider: 'user-provider' });
|
|
1404
|
+
});
|
|
1405
|
+
|
|
1406
|
+
it('should use builtin settings when user has no settings in getAiProviderModelList', async () => {
|
|
1407
|
+
const providerId = 'openai';
|
|
1408
|
+
|
|
1409
|
+
const userModels: AiProviderModelListItem[] = [
|
|
1410
|
+
{
|
|
1411
|
+
id: 'gpt-4',
|
|
1412
|
+
type: 'chat',
|
|
1413
|
+
enabled: true,
|
|
1414
|
+
abilities: { vision: true },
|
|
1415
|
+
// 用户未设置 settings
|
|
1416
|
+
},
|
|
1417
|
+
];
|
|
1418
|
+
|
|
1419
|
+
const builtinModels: AiProviderModelListItem[] = [
|
|
1420
|
+
{
|
|
1421
|
+
id: 'gpt-4',
|
|
1422
|
+
type: 'chat',
|
|
1423
|
+
enabled: true,
|
|
1424
|
+
settings: { searchImpl: 'tool', searchProvider: 'google' },
|
|
1425
|
+
} as any,
|
|
1426
|
+
];
|
|
1427
|
+
|
|
1428
|
+
vi.spyOn(repo.aiModelModel, 'getModelListByProviderId').mockResolvedValue(userModels);
|
|
1429
|
+
vi.spyOn(repo as any, 'fetchBuiltinModels').mockResolvedValue(builtinModels);
|
|
1430
|
+
|
|
1431
|
+
const result = await repo.getAiProviderModelList(providerId);
|
|
1432
|
+
|
|
1433
|
+
const merged = result.find((m) => m.id === 'gpt-4');
|
|
1434
|
+
expect(merged).toBeDefined();
|
|
1435
|
+
// 应该使用内置的 settings
|
|
1436
|
+
expect(merged?.settings).toEqual({ searchImpl: 'tool', searchProvider: 'google' });
|
|
1437
|
+
});
|
|
1438
|
+
|
|
1439
|
+
it('should have no settings when both user and builtin have no settings in getAiProviderModelList', async () => {
|
|
1440
|
+
const providerId = 'openai';
|
|
1441
|
+
|
|
1442
|
+
const userModels: AiProviderModelListItem[] = [
|
|
1443
|
+
{
|
|
1444
|
+
id: 'gpt-4',
|
|
1445
|
+
type: 'chat',
|
|
1446
|
+
enabled: true,
|
|
1447
|
+
abilities: { vision: true },
|
|
1448
|
+
// 用户未设置 settings
|
|
1449
|
+
},
|
|
1450
|
+
];
|
|
1451
|
+
|
|
1452
|
+
const builtinModels: AiProviderModelListItem[] = [
|
|
1453
|
+
{
|
|
1454
|
+
id: 'gpt-4',
|
|
1455
|
+
type: 'chat',
|
|
1456
|
+
enabled: true,
|
|
1457
|
+
// 内置也无 settings
|
|
1458
|
+
},
|
|
1459
|
+
];
|
|
1460
|
+
|
|
1461
|
+
vi.spyOn(repo.aiModelModel, 'getModelListByProviderId').mockResolvedValue(userModels);
|
|
1462
|
+
vi.spyOn(repo as any, 'fetchBuiltinModels').mockResolvedValue(builtinModels);
|
|
1463
|
+
|
|
1464
|
+
const result = await repo.getAiProviderModelList(providerId);
|
|
1465
|
+
|
|
1466
|
+
const merged = result.find((m) => m.id === 'gpt-4');
|
|
1467
|
+
expect(merged).toBeDefined();
|
|
1468
|
+
// 无 settings
|
|
1469
|
+
expect(merged?.settings).toBeUndefined();
|
|
1470
|
+
});
|
|
1273
1471
|
});
|
|
1274
1472
|
|
|
1275
1473
|
describe('getAiProviderRuntimeState', () => {
|
|
@@ -91,6 +91,7 @@ const injectSearchSettings = (providerId: string, item: any) => {
|
|
|
91
91
|
if (item?.settings?.searchImpl || item?.settings?.searchProvider) {
|
|
92
92
|
const next = { ...item } as any;
|
|
93
93
|
if (next.settings) {
|
|
94
|
+
// eslint-disable-next-line unused-imports/no-unused-vars, @typescript-eslint/no-unused-vars
|
|
94
95
|
const { searchImpl, searchProvider, ...restSettings } = next.settings;
|
|
95
96
|
next.settings = Object.keys(restSettings).length > 0 ? restSettings : undefined;
|
|
96
97
|
}
|
|
@@ -224,7 +225,7 @@ export class AiInfraRepos {
|
|
|
224
225
|
enabled: typeof user.enabled === 'boolean' ? user.enabled : item.enabled,
|
|
225
226
|
id: item.id,
|
|
226
227
|
providerId: provider.id,
|
|
227
|
-
settings: item.settings,
|
|
228
|
+
settings: user.settings || item.settings,
|
|
228
229
|
sort: user.sort || undefined,
|
|
229
230
|
type: user.type || item.type,
|
|
230
231
|
};
|