@lobehub/chat 1.138.1 → 1.138.3

package/packages/prompts/src/contexts/supervisor/makeDecision.ts

@@ -0,0 +1,68 @@
+import { ChatCompletionTool, ChatMessage, ChatStreamPayload } from '@lobechat/types';
+
+import { groupChatPrompts, groupSupervisorPrompts } from '../../prompts';
+import { SupervisorToolName, SupervisorTools } from './tools';
+
+interface SupervisorTodoItem {
+  // optional assigned owner (agent id or name)
+  assignee?: string;
+  content: string;
+  finished: boolean;
+}
+
+interface AgentItem {
+  id: string;
+  title?: string | null;
+}
+export interface SupervisorContext {
+  allowDM?: boolean;
+  availableAgents: AgentItem[];
+  messages: ChatMessage[];
+  // Group scene controls which tools are exposed (e.g., todos only in 'productive')
+  scene?: 'casual' | 'productive';
+  systemPrompt?: string;
+  todoList?: SupervisorTodoItem[];
+  userName?: string;
+}
+
+export const contextSupervisorMakeDecision = ({
+  allowDM,
+  scene,
+  systemPrompt,
+  availableAgents,
+  todoList,
+  userName,
+  messages,
+}: SupervisorContext) => {
+  const conversationHistory = groupSupervisorPrompts(messages);
+  const prompt = groupChatPrompts.buildSupervisorPrompt({
+    allowDM,
+    availableAgents: availableAgents.filter((agent) => agent.id),
+    conversationHistory,
+    scene,
+    systemPrompt,
+    todoList,
+    userName,
+  });
+
+  const tools = SupervisorTools.filter((tool) => {
+    if (tool.name === SupervisorToolName.trigger_agent_dm) {
+      return allowDM;
+    }
+
+    if ([SupervisorToolName.finish_todo, SupervisorToolName.create_todo].includes(tool.name)) {
+      return scene === 'productive';
+    }
+
+    return true;
+  }).map<ChatCompletionTool>((tool) => ({
+    function: tool,
+    type: 'function',
+  }));
+
+  return {
+    messages: [{ content: prompt, role: 'user' }],
+    temperature: 0.3,
+    tools,
+  } satisfies Partial<ChatStreamPayload>;
+};

package/packages/prompts/src/contexts/supervisor/tools.ts

@@ -0,0 +1,102 @@
+import { LobeUniformTool } from '@lobechat/types';
+
+export const SupervisorToolName = {
+  create_todo: 'create_todo',
+  finish_todo: 'finish_todo',
+  trigger_agent: 'trigger_agent',
+  trigger_agent_dm: 'trigger_agent_dm',
+  wait_for_user_input: 'wait_for_user_input',
+};
+
+export const SupervisorTools: LobeUniformTool[] = [
+  {
+    description: 'Trigger an agent to speak (group message).',
+    name: SupervisorToolName.trigger_agent,
+    parameters: {
+      properties: {
+        id: {
+          description: 'The agent id to trigger.',
+          type: 'string',
+        },
+        instruction: {
+          description:
+            'The instruction or message for the agent. No longer than 10 words. Always use English.',
+          type: 'string',
+        },
+      },
+      required: ['id', 'instruction'],
+      type: 'object',
+    },
+  },
+  {
+    description:
+      'Wait for user input. Use this when the conversation history looks likes fine for now, or agents are waiting for user input.',
+    name: SupervisorToolName.wait_for_user_input,
+    parameters: {
+      properties: {
+        reason: {
+          description: 'Optional reason for pausing the conversation.',
+          type: 'string',
+        },
+      },
+      required: [],
+      type: 'object',
+    },
+  },
+
+  {
+    description: 'Trigger an agent to DM another agent or user.',
+    name: SupervisorToolName.trigger_agent_dm,
+    parameters: {
+      additionalProperties: false,
+      properties: {
+        id: {
+          description: 'The agent id to trigger.',
+          type: 'string',
+        },
+        instruction: {
+          type: 'string',
+        },
+        target: {
+          description: 'The target agent id. Only used when need DM.',
+          type: 'string',
+        },
+      },
+      required: ['instruction', 'id', 'target'],
+      type: 'object',
+    },
+  },
+  {
+    description: 'Create a new todo item',
+    name: SupervisorToolName.create_todo,
+    parameters: {
+      additionalProperties: false,
+      properties: {
+        assignee: {
+          description: 'Who will do the todo. Can be agent id or empty.',
+          type: 'string',
+        },
+        content: {
+          description: 'The todo content or description.',
+          type: 'string',
+        },
+      },
+      required: ['content', 'assignee'],
+      type: 'object',
+    },
+  },
+  {
+    description: 'Finish a todo by index or all todos',
+    name: SupervisorToolName.finish_todo,
+    parameters: {
+      additionalProperties: false,
+      properties: {
+        index: {
+          type: 'number',
+        },
+      },
+      required: ['index'],
+      type: 'object',
+    },
+  },
+];

package/packages/prompts/src/index.ts

@@ -1,2 +1,3 @@
 export * from './chains';
+export * from './contexts';
 export * from './prompts';

package/packages/types/src/aiChat.ts

@@ -1,6 +1,7 @@
 import { z } from 'zod';
 
 import { ChatMessage } from './message';
+import { OpenAIChatMessage } from './openai/chat';
 import { LobeUniformTool, LobeUniformToolSchema } from './tool';
 import { ChatTopic } from './topic';
 
@@ -75,7 +76,9 @@ export const StructureOutputSchema = z.object({
   provider: z.string(),
   schema: StructureSchema.optional(),
   systemRole: z.string().optional(),
-  tools: z.array(LobeUniformToolSchema).optional(),
+  tools: z
+    .array(z.object({ function: LobeUniformToolSchema, type: z.literal('function') }))
+    .optional(),
 });
 
 interface IStructureSchema {
@@ -92,10 +95,13 @@ interface IStructureSchema {
 
 export interface StructureOutputParams {
   keyVaultsPayload: string;
-  messages: ChatMessage[];
+  messages: OpenAIChatMessage[];
   model: string;
   provider: string;
   schema?: IStructureSchema;
   systemRole?: string;
-  tools?: LobeUniformTool[];
+  tools?: {
+    function: LobeUniformTool;
+    type: 'function';
+  }[];
 }

package/src/app/[variants]/(auth)/next-auth/signin/AuthSignInBox.tsx

@@ -1,7 +1,8 @@
 'use client';
 
+import { BRANDING_NAME, DOCUMENTS_REFER_URL, PRIVACY_URL, TERMS_URL } from '@lobechat/const';
 import { Button, Text } from '@lobehub/ui';
-import { LobeChat } from '@lobehub/ui/brand';
+import { LobeHub } from '@lobehub/ui/brand';
 import { Col, Flex, Row, Skeleton } from 'antd';
 import { createStyles } from 'antd-style';
 import { AuthError } from 'next-auth';
@@ -12,7 +13,6 @@ import { useTranslation } from 'react-i18next';
 
 import BrandWatermark from '@/components/BrandWatermark';
 import AuthIcons from '@/components/NextAuth/AuthIcons';
-import { DOCUMENTS_REFER_URL, PRIVACY_URL, TERMS_URL } from '@/const/url';
 import { useUserStore } from '@/store/user';
 
 const useStyles = createStyles(({ css, token }) => ({
@@ -114,9 +114,9 @@ export default memo(() => {
           <div className={styles.text}>
             <Text as={'h4'} className={styles.title}>
               <div>
-                <LobeChat size={48} />
+                <LobeHub size={48} />
               </div>
-              {t('signIn.start.title', { applicationName: 'LobeChat' })}
+              {t('signIn.start.title', { applicationName: BRANDING_NAME })}
             </Text>
             <Text as={'p'} className={styles.description}>
               {t('signIn.start.subtitle')}

package/src/app/[variants]/(main)/chat/(workspace)/@conversation/features/ChatList/WelcomeChatItem/{OpeningQuestions.tsx → AgentWelcome/OpeningQuestions.tsx}

@@ -8,7 +8,7 @@ import { Flexbox } from 'react-layout-kit';
 
 import { useChatStore } from '@/store/chat';
 
-import { useSend } from '../../ChatInput/useSend';
+import { useSend } from '../../../ChatInput/useSend';
 
 const useStyles = createStyles(({ css, token, responsive }) => ({
   card: css`
@@ -25,7 +25,7 @@ const useStyles = createStyles(({ css, token, responsive }) => ({
 
   container: css`
     padding-block: 0;
-    padding-inline: 64px 16px;
+    padding-inline: 0;
   `,
 
   title: css`

package/src/app/[variants]/(main)/chat/(workspace)/@conversation/features/ChatList/WelcomeChatItem/{InboxWelcome → AgentWelcome}/index.tsx

@@ -1,19 +1,25 @@
 'use client';
 
+import { BRANDING_NAME } from '@lobechat/const';
 import { FluentEmoji, Markdown } from '@lobehub/ui';
 import { createStyles } from 'antd-style';
-import { memo } from 'react';
+import isEqual from 'fast-deep-equal';
+import React, { memo, useMemo } from 'react';
 import { Trans, useTranslation } from 'react-i18next';
 import { Center, Flexbox } from 'react-layout-kit';
 
-import { BRANDING_NAME } from '@/const/branding';
-import { isCustomBranding } from '@/const/version';
 import { useGreeting } from '@/hooks/useGreeting';
+import { useIsMobile } from '@/hooks/useIsMobile';
+import { useAgentStore } from '@/store/agent';
+import { agentSelectors } from '@/store/agent/selectors';
+import { useChatStore } from '@/store/chat';
+import { chatSelectors } from '@/store/chat/selectors';
 import { featureFlagsSelectors, useServerConfigStore } from '@/store/serverConfig';
+import { useSessionStore } from '@/store/session';
+import { sessionMetaSelectors } from '@/store/session/selectors';
 
 import AddButton from './AddButton';
-import AgentsSuggest from './AgentsSuggest';
-import QuestionSuggest from './QuestionSuggest';
+import OpeningQuestions from './OpeningQuestions';
 
 const useStyles = createStyles(({ css, responsive }) => ({
   container: css`
@@ -41,14 +47,30 @@ const useStyles = createStyles(({ css, responsive }) => ({
 }));
 
 const InboxWelcome = memo(() => {
-  const { t } = useTranslation('welcome');
+  const { t } = useTranslation(['welcome', 'chat']);
   const { styles } = useStyles();
-  const mobile = useServerConfigStore((s) => s.isMobile);
+  const mobile = useIsMobile();
   const greeting = useGreeting();
-  const { showWelcomeSuggest, showCreateSession } = useServerConfigStore(featureFlagsSelectors);
+  const { showCreateSession } = useServerConfigStore(featureFlagsSelectors);
+  const openingQuestions = useAgentStore(agentSelectors.openingQuestions);
+
+  const meta = useSessionStore(sessionMetaSelectors.currentAgentMeta, isEqual);
+
+  const agentSystemRoleMsg = t('agentDefaultMessageWithSystemRole', {
+    name: meta.title || t('defaultAgent', { ns: 'chat' }),
+    ns: 'chat',
+  });
+  const openingMessage = useAgentStore(agentSelectors.openingMessage);
+
+  const showInboxWelcome = useChatStore(chatSelectors.showInboxWelcome);
+
+  const message = useMemo(() => {
+    if (openingMessage) return openingMessage;
+    return agentSystemRoleMsg;
+  }, [openingMessage, agentSystemRoleMsg, meta.description]);
 
   return (
-    <Center padding={16} width={'100%'}>
+    <Center gap={12} padding={16} width={'100%'}>
       <Flexbox className={styles.container} gap={16} style={{ maxWidth: 800 }} width={'100%'}>
         <Flexbox align={'center'} gap={8} horizontal>
           <FluentEmoji emoji={'👋'} size={40} type={'anim'} />
@@ -74,15 +96,14 @@ const InboxWelcome = memo(() => {
           }}
           variant={'chat'}
         >
-          {t(showCreateSession ? 'guide.defaultMessage' : 'guide.defaultMessageWithoutCreate', {
-            appName: BRANDING_NAME,
-          })}
+          {showInboxWelcome
+            ? t(showCreateSession ? 'guide.defaultMessage' : 'guide.defaultMessageWithoutCreate', {
+                appName: BRANDING_NAME,
+              })
+            : message}
         </Markdown>
-        {showWelcomeSuggest && (
-          <>
-            <AgentsSuggest mobile={mobile} />
-            {!isCustomBranding && <QuestionSuggest mobile={mobile} />}
-          </>
+        {openingQuestions.length > 0 && (
+          <OpeningQuestions mobile={mobile} questions={openingQuestions} />
         )}
       </Flexbox>
     </Center>

package/src/app/[variants]/(main)/chat/(workspace)/@conversation/features/ChatList/WelcomeChatItem/index.tsx

@@ -3,15 +3,15 @@ import React, { memo } from 'react';
 import { sessionSelectors } from '@/store/session/selectors';
 import { useSessionStore } from '@/store/session/store';
 
+import AgentWelcome from './AgentWelcome';
 import GroupWelcome from './GroupWelcome';
-import WelcomeMessage from './WelcomeMessage';
 
 const WelcomeChatItem = memo(() => {
   const isGroupSession = useSessionStore(sessionSelectors.isCurrentSessionGroupSession);
 
   if (isGroupSession) return <GroupWelcome />;
 
-  return <WelcomeMessage />;
+  return <AgentWelcome />;
 });
 
 export default WelcomeChatItem;

package/src/app/[variants]/(main)/discover/(detail)/assistant/[...slugs]/page.tsx

@@ -48,7 +48,7 @@ export const generateMetadata = async (props: DiscoverPageProps) => {
     authors: [
       { name: author, url: homepage },
       { name: 'LobeHub', url: 'https://github.com/lobehub' },
-      { name: 'LobeChat', url: 'https://github.com/lobehub/lobe-chat' },
+      { name: 'LobeHub Cloud', url: 'https://lobehub.com' },
     ],
     keywords: tags,
     ...metadataModule.generate({

package/src/app/[variants]/(main)/discover/(detail)/mcp/[slug]/page.tsx

@@ -42,7 +42,7 @@ export const generateMetadata = async (props: DiscoverPageProps) => {
     authors: [
       { name: author, url: homepage },
       { name: 'LobeHub', url: 'https://github.com/lobehub' },
-      { name: 'LobeChat', url: 'https://github.com/lobehub/lobe-chat' },
+      { name: 'LobeHub Cloud', url: 'https://lobehub,com' },
     ],
     keywords: tags,
     ...metadataModule.generate({

package/src/config/modelProviders/lobehub.ts

@@ -3,7 +3,7 @@ import { ModelProviderCard } from '@/types/llm';
 const LobeHub: ModelProviderCard = {
   chatModels: [],
   description:
-    'LobeChat Cloud 通过官方部署的 API 来实现 AI 模型的调用，并采用 Credits 计算积分的方式来衡量 AI 模型的用量，对应大模型使用的 Tokens。',
+    'LobeHub Cloud 通过官方部署的 API 来实现 AI 模型的调用，并采用 Credits 计算积分的方式来衡量 AI 模型的用量，对应大模型使用的 Tokens。',
   enabled: true,
   id: 'lobehub',
   modelsUrl: 'https://lobehub.com/zh/docs/usage/subscription/model-pricing',

package/src/locales/default/common.ts

@@ -33,7 +33,7 @@ export default {
   blog: '产品博客',
   branching: '创建子话题',
   branchingDisable:
-    '「子话题」功能在当前模式下不可用，如需该功能，请切换到 Postgres/Pglite DB 模式或使用 LobeChat Cloud',
+    '「子话题」功能在当前模式下不可用，如需该功能，请切换到 Postgres/Pglite DB 模式或使用 LobeHub Cloud',
   branchingRequiresSavedTopic: '当前话题未保存，保存后才可以使用子话题能力',
   cancel: '取消',
   changelog: '更新日志',

package/src/locales/default/plugin.ts

@@ -463,7 +463,7 @@ export default {
     tabs: {
       installed: '已安装',
       mcp: 'MCP 插件',
-      old: 'LobeChat 插件',
+      old: 'LobeHub 插件',
     },
     title: '插件商店',
   },

package/src/locales/default/welcome.ts

@@ -381,7 +381,7 @@ export default {
     },
     questions: {
       moreBtn: '了解更多',
-      title: '大家都在问：',
+      title: '试着问问：',
     },
     welcome: {
       afternoon: '下午好',

package/src/server/ld.ts

@@ -87,7 +87,7 @@ export class Ld {
     return {
       '@id': this.getId(OFFICIAL_URL, '#organization'),
       '@type': 'Organization',
-      'alternateName': 'LobeChat',
+      'alternateName': 'LobeHub',
       'contactPoint': {
         '@type': 'ContactPoint',
         'contactType': 'customer support',

package/src/server/routers/lambda/aiChat.ts

@@ -61,7 +61,7 @@ export const aiChatRouter = router({
       model: input.model,
       schema: input.schema,
       systemRole: input.systemRole,
-      tools: input.tools,
+      tools: input.tools?.map((item) => item.function),
     });
 
     log('generateObject completed, result: %O', result);

package/src/server/services/aiChat/index.test.ts

@@ -32,7 +32,7 @@ describe('AiChatService', () => {
       { includeTopic: true, sessionId: 's1' },
       expect.objectContaining({ postProcessUrl: expect.any(Function) }),
     );
-    expect(mockQueryTopics).toHaveBeenCalledWith({ sessionId: 's1' });
+    expect(mockQueryTopics).toHaveBeenCalledWith({ containerId: 's1' });
     expect(res.messages).toEqual([{ id: 'm1' }]);
     expect(res.topics).toEqual([{ id: 't1' }]);
   });

package/src/server/services/aiChat/index.ts

@@ -29,7 +29,7 @@ export class AiChatService {
       this.messageModel.query(params, {
         postProcessUrl: (path) => this.fileService.getFullFileUrl(path),
       }),
-      params.includeTopic ? this.topicModel.query({ sessionId: params.sessionId }) : undefined,
+      params.includeTopic ? this.topicModel.query({ containerId: params.sessionId }) : undefined,
     ]);
 
     return { messages, topics };

package/src/services/topic/client.ts

@@ -38,7 +38,7 @@ export class ClientService extends BaseClientService implements ITopicService {
   getTopics: ITopicService['getTopics'] = async (params) => {
     const data = await this.topicModel.query({
       ...params,
-      sessionId: this.toDbSessionId(params.containerId),
+      containerId: this.toDbSessionId(params.containerId),
     });
     return data as unknown as Promise<ChatTopic[]>;
   };

package/src/store/chat/slices/message/supervisor.test.ts

@@ -5,10 +5,17 @@ import { aiChatService } from '@/services/aiChat';
 import { GroupChatSupervisor, type SupervisorContext } from './supervisor';
 
 vi.mock('@lobechat/prompts', () => ({
-  groupChatPrompts: {
-    buildSupervisorPrompt: vi.fn(() => 'structured-supervisor-prompt'),
-  },
-  groupSupervisorPrompts: vi.fn(() => 'conversation-history'),
+  contextSupervisorMakeDecision: vi.fn(() => ({
+    messages: [{ content: 'structured-supervisor-prompt', role: 'user' }],
+    temperature: 0.3,
+    tools: [
+      { function: { name: 'trigger_agent' }, type: 'function' },
+      { function: { name: 'wait_for_user_input' }, type: 'function' },
+      { function: { name: 'trigger_agent_dm' }, type: 'function' },
+      { function: { name: 'create_todo' }, type: 'function' },
+      { function: { name: 'finish_todo' }, type: 'function' },
+    ],
+  })),
 }));
 
 vi.mock('@/services/aiChat', () => ({
@@ -76,7 +83,7 @@ describe('GroupChatSupervisor', () => {
       temperature: 0.3,
     });
 
-    const toolNames = (payload.tools ?? []).map((tool: any) => tool.name);
+    const toolNames = (payload.tools ?? []).map((tool: any) => tool.function.name);
     expect(toolNames).toEqual(
       expect.arrayContaining([
         'trigger_agent',