@lobehub/chat 1.138.1 → 1.138.3

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

@@ -1,4 +1,4 @@
-import { groupChatPrompts, groupSupervisorPrompts } from '@lobechat/prompts';
+import { contextSupervisorMakeDecision } from '@lobechat/prompts';
 import { ChatMessage, GroupMemberWithAgent } from '@lobechat/types';
 
 import { aiChatService } from '@/services/aiChat';
@@ -61,7 +61,7 @@ export class GroupChatSupervisor {
    * Make decision on who should speak next
    */
   async makeDecision(context: SupervisorContext): Promise<SupervisorDecisionResult> {
-    const { messages, availableAgents, userName, systemPrompt, allowDM, todoList } = context;
+    const { availableAgents } = context;
 
     // If no agents available, stop conversation
     if (availableAgents.length === 0) {
@@ -69,22 +69,7 @@ export class GroupChatSupervisor {
     }
 
     try {
-      // Create supervisor prompt with conversation context
-      const conversationHistory = groupSupervisorPrompts(messages);
-
-      const supervisorPrompt = groupChatPrompts.buildSupervisorPrompt({
-        allowDM,
-        availableAgents: availableAgents
-          .filter((agent) => agent.id)
-          .map((agent) => ({ id: agent.id!, title: agent.title })),
-        conversationHistory,
-        scene: context.scene,
-        systemPrompt,
-        todoList,
-        userName,
-      });
-
-      const response = await this.callLLMForDecision(supervisorPrompt, context);
+      const response = await this.callLLMForDecision(context);
       const result = this.parseSupervisorResponse(response, availableAgents, context);
 
       console.log('Supervisor TODO list:', result.todos);
@@ -102,123 +87,25 @@ export class GroupChatSupervisor {
    * Call LLM service to get supervisor decision
    */
   private async callLLMForDecision(
-    prompt: string,
     context: SupervisorContext,
   ): Promise<SupervisorToolCall[] | string> {
-    const supervisorConfig = {
-      model: context.model,
-      provider: context.provider,
-      temperature: 0.3,
-    };
-
-    // Build tools array
-    const tools: any[] = [
-      {
-        description: 'Trigger an agent to speak (group message).',
-        name: '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: 'wait_for_user_input',
-        parameters: {
-          properties: {
-            reason: {
-              description: 'Optional reason for pausing the conversation.',
-              type: 'string',
-            },
-          },
-          required: [],
-          type: 'object',
-        },
-      },
-    ];
-
-    // Add DM tool if allowed
-    if (context.allowDM) {
-      tools.push({
-        description: 'Trigger an agent to DM another agent or user.',
-        name: 'trigger_agent_dm',
-        parameters: {
-          properties: {
-            id: {
-              description: 'The agent id to trigger.',
-              type: 'string',
-            },
-            instruction: {
-              description: 'The instruction or message for the agent.',
-              type: 'string',
-            },
-            target: {
-              description: 'The target agent id. Only used when need DM.',
-              type: 'string',
-            },
-          },
-          required: ['id', 'instruction', 'target'],
-          type: 'object',
-        },
-      });
-    }
-
-    // Add todo tools if in productive scene
-    if (context.scene === 'productive') {
-      tools.push(
-        {
-          description: 'Create a new todo item',
-          name: 'create_todo',
-          parameters: {
-            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',
-          name: 'finish_todo',
-          parameters: {
-            properties: {
-              index: {
-                description: 'The index of the todo to finish.',
-                type: 'number',
-              },
-            },
-            required: ['index'],
-            type: 'object',
-          },
-        },
-      );
-    }
+    const contexts = contextSupervisorMakeDecision({
+      allowDM: context.allowDM,
+      availableAgents: context.availableAgents
+        .filter((agent) => agent.id)
+        .map((agent) => ({ id: agent.id, title: agent.title })),
+      messages: context.messages,
+      scene: context.scene,
+      todoList: context.todoList,
+      userName: context.userName,
+    });
 
     try {
       const response = await aiChatService.generateJSON(
         {
-          messages: [{ content: prompt, role: 'user' }] as any,
-          tools,
-          ...supervisorConfig,
+          ...(contexts as any),
+          model: context.model,
+          provider: context.provider,
         },
         context.abortController || new AbortController(),
       );

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

@@ -1,114 +0,0 @@
-'use client';
-
-import { ActionIcon, Avatar, Block, Grid, Text } from '@lobehub/ui';
-import { Skeleton } from 'antd';
-import { createStyles } from 'antd-style';
-import { RefreshCw } from 'lucide-react';
-import Link from 'next/link';
-import { memo, useState } from 'react';
-import { useTranslation } from 'react-i18next';
-import { Flexbox } from 'react-layout-kit';
-import urlJoin from 'url-join';
-
-import { useDiscoverStore } from '@/store/discover';
-import { DiscoverAssistantItem } from '@/types/discover';
-
-const useStyles = createStyles(({ css, token, responsive }) => ({
-  card: css`
-    position: relative;
-
-    overflow: hidden;
-
-    height: 100%;
-    min-height: 110px;
-    padding: 16px;
-    border-radius: ${token.borderRadiusLG}px;
-
-    background: ${token.colorBgContainer};
-
-    ${responsive.mobile} {
-      min-height: 72px;
-    }
-  `,
-  cardDesc: css`
-    margin-block: 0 !important;
-    color: ${token.colorTextDescription};
-  `,
-  cardTitle: css`
-    margin-block: 0 !important;
-    font-size: 16px;
-    font-weight: bold;
-  `,
-  icon: css`
-    color: ${token.colorTextSecondary};
-  `,
-  title: css`
-    color: ${token.colorTextDescription};
-  `,
-}));
-
-const AgentsSuggest = memo<{ mobile?: boolean }>(({ mobile }) => {
-  const { t } = useTranslation('welcome');
-  const [page, setPage] = useState(1);
-  const useAssistantList = useDiscoverStore((s) => s.useAssistantList);
-
-  const { data: assistantList, isLoading } = useAssistantList({
-    page,
-    pageSize: mobile ? 2 : 4,
-  });
-
-  const { styles } = useStyles();
-
-  const loadingCards = Array.from({ length: mobile ? 2 : 4 }).map((_, index) => (
-    <Block className={styles.card} key={index}>
-      <Skeleton active avatar paragraph={{ rows: 2 }} title={false} />
-    </Block>
-  ));
-
-  const handleRefresh = () => {
-    if (!assistantList) return;
-    setPage(page + 1);
-  };
-
-  // if no assistant data, just hide the component
-  if (!isLoading && !assistantList?.items?.length) return null;
-
-  return (
-    <Flexbox gap={8} width={'100%'}>
-      <Flexbox align={'center'} horizontal justify={'space-between'}>
-        <div className={styles.title}>{t('guide.agents.title')}</div>
-        <ActionIcon
-          icon={RefreshCw}
-          onClick={handleRefresh}
-          size={{ blockSize: 24, size: 14 }}
-          title={t('guide.agents.replaceBtn')}
-        />
-      </Flexbox>
-      <Grid gap={8} rows={2}>
-        {isLoading || !assistantList
-          ? loadingCards
-          : assistantList.items.map((item: DiscoverAssistantItem) => (
-              <Link
-                href={urlJoin('/discover/assistant', item.identifier)}
-                key={item.identifier}
-                prefetch={false}
-              >
-                <Block className={styles.card} clickable gap={12} horizontal variant={'outlined'}>
-                  <Avatar avatar={item.avatar} style={{ flex: 'none' }} />
-                  <Flexbox gap={2} style={{ overflow: 'hidden', width: '100%' }}>
-                    <Text className={styles.cardTitle} ellipsis={{ rows: 1 }}>
-                      {item.title}
-                    </Text>
-                    <Text className={styles.cardDesc} ellipsis={{ rows: mobile ? 1 : 2 }}>
-                      {item.description}
-                    </Text>
-                  </Flexbox>
-                </Block>
-              </Link>
-            ))}
-      </Grid>
-    </Flexbox>
-  );
-});
-
-export default AgentsSuggest;

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

@@ -1,104 +0,0 @@
-'use client';
-
-import { ActionIcon, Block } from '@lobehub/ui';
-import { createStyles } from 'antd-style';
-import { shuffle } from 'lodash-es';
-import { ArrowRight } from 'lucide-react';
-import Link from 'next/link';
-import { memo } from 'react';
-import { useTranslation } from 'react-i18next';
-import { Flexbox } from 'react-layout-kit';
-
-import { BRANDING_NAME } from '@/const/branding';
-import { USAGE_DOCUMENTS } from '@/const/url';
-import { useChatStore } from '@/store/chat';
-
-import { useSend } from '../../../ChatInput/useSend';
-
-const useStyles = createStyles(({ css, token, responsive }) => ({
-  card: css`
-    padding-block: 12px;
-    padding-inline: 24px;
-    border-radius: 48px;
-
-    color: ${token.colorText};
-
-    background: ${token.colorBgContainer};
-
-    ${responsive.mobile} {
-      padding-block: 8px;
-      padding-inline: 16px;
-    }
-  `,
-  icon: css`
-    color: ${token.colorTextSecondary};
-  `,
-  title: css`
-    color: ${token.colorTextDescription};
-  `,
-}));
-
-const qa = shuffle([
-  'q01',
-  'q02',
-  'q03',
-  'q04',
-  'q05',
-  'q06',
-  'q07',
-  'q08',
-  'q09',
-  'q10',
-  'q11',
-  'q12',
-  'q13',
-  'q14',
-  'q15',
-]);
-
-const QuestionSuggest = memo<{ mobile?: boolean }>(({ mobile }) => {
-  const [updateInputMessage] = useChatStore((s) => [s.updateInputMessage]);
-
-  const { t } = useTranslation('welcome');
-  const { styles } = useStyles();
-  const { send: sendMessage } = useSend();
-
-  return (
-    <Flexbox gap={8} width={'100%'}>
-      <Flexbox align={'center'} horizontal justify={'space-between'}>
-        <div className={styles.title}>{t('guide.questions.title')}</div>
-        <Link href={USAGE_DOCUMENTS} target={'_blank'}>
-          <ActionIcon
-            icon={ArrowRight}
-            size={{ blockSize: 24, size: 16 }}
-            title={t('guide.questions.moreBtn')}
-          />
-        </Link>
-      </Flexbox>
-      <Flexbox gap={8} horizontal wrap={'wrap'}>
-        {qa.slice(0, mobile ? 2 : 5).map((item) => {
-          const text = t(`guide.qa.${item}` as any, { appName: BRANDING_NAME });
-          return (
-            <Block
-              align={'center'}
-              className={styles.card}
-              clickable
-              gap={8}
-              horizontal
-              key={item}
-              onClick={() => {
-                updateInputMessage(text);
-                sendMessage({ isWelcomeQuestion: true });
-              }}
-              variant={'outlined'}
-            >
-              {t(text)}
-            </Block>
-          );
-        })}
-      </Flexbox>
-    </Flexbox>
-  );
-});
-
-export default QuestionSuggest;

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

@@ -1,65 +0,0 @@
-import isEqual from 'fast-deep-equal';
-import qs from 'query-string';
-import React, { useMemo } from 'react';
-import { useTranslation } from 'react-i18next';
-import { Flexbox } from 'react-layout-kit';
-
-import { ChatItem } from '@/features/ChatItem';
-import { useAgentStore } from '@/store/agent';
-import { agentChatConfigSelectors, agentSelectors } from '@/store/agent/selectors';
-import { useChatStore } from '@/store/chat';
-import { featureFlagsSelectors, useServerConfigStore } from '@/store/serverConfig';
-import { useSessionStore } from '@/store/session';
-import { sessionMetaSelectors } from '@/store/session/selectors';
-
-import OpeningQuestions from './OpeningQuestions';
-
-const WelcomeMessage = () => {
-  const mobile = useServerConfigStore((s) => s.isMobile);
-  const { t } = useTranslation('chat');
-  const type = useAgentStore(agentChatConfigSelectors.displayMode);
-  const openingMessage = useAgentStore(agentSelectors.openingMessage);
-  const openingQuestions = useAgentStore(agentSelectors.openingQuestions);
-
-  const meta = useSessionStore(sessionMetaSelectors.currentAgentMeta, isEqual);
-  const { isAgentEditable } = useServerConfigStore(featureFlagsSelectors);
-  const activeId = useChatStore((s) => s.activeId);
-
-  const agentSystemRoleMsg = t('agentDefaultMessageWithSystemRole', {
-    name: meta.title || t('defaultAgent'),
-    systemRole: meta.description,
-  });
-
-  const agentMsg = t(isAgentEditable ? 'agentDefaultMessage' : 'agentDefaultMessageWithoutEdit', {
-    name: meta.title || t('defaultAgent'),
-    url: qs.stringifyUrl({
-      query: mobile ? { session: activeId, showMobileWorkspace: mobile } : { session: activeId },
-      url: '/chat/settings',
-    }),
-  });
-
-  const message = useMemo(() => {
-    if (openingMessage) return openingMessage;
-    return !!meta.description ? agentSystemRoleMsg : agentMsg;
-  }, [openingMessage, agentSystemRoleMsg, agentMsg, meta.description]);
-
-  const chatItem = (
-    <ChatItem
-      avatar={meta}
-      editing={false}
-      message={message}
-      placement={'left'}
-      variant={type === 'chat' ? 'bubble' : 'docs'}
-    />
-  );
-
-  return openingQuestions.length > 0 ? (
-    <Flexbox>
-      {chatItem}
-      <OpeningQuestions mobile={mobile} questions={openingQuestions} />
-    </Flexbox>
-  ) : (
-    chatItem
-  );
-};
-export default WelcomeMessage;