@lobehub/chat 1.97.0 → 1.97.2

package/src/components/Analytics/index.tsx

@@ -8,7 +8,6 @@ import Google from './Google';
 import Vercel from './Vercel';
 
 const Plausible = dynamic(() => import('./Plausible'));
-const Posthog = dynamic(() => import('./Posthog'));
 const Umami = dynamic(() => import('./Umami'));
 const Clarity = dynamic(() => import('./Clarity'));
 const ReactScan = dynamic(() => import('./ReactScan'));
@@ -24,13 +23,6 @@ const Analytics = () => {
           scriptBaseUrl={analyticsEnv.PLAUSIBLE_SCRIPT_BASE_URL}
         />
       )}
-      {analyticsEnv.ENABLED_POSTHOG_ANALYTICS && (
-        <Posthog
-          debug={analyticsEnv.DEBUG_POSTHOG_ANALYTICS}
-          host={analyticsEnv.POSTHOG_HOST!}
-          token={analyticsEnv.POSTHOG_KEY}
-        />
-      )}
       {analyticsEnv.ENABLED_UMAMI_ANALYTICS && (
         <Umami
           scriptUrl={analyticsEnv.UMAMI_SCRIPT_URL}

package/src/const/analytics.ts

@@ -0,0 +1,3 @@
+import { isDesktop } from '@/const/version';
+
+export const BUSINESS_LINE = isDesktop ? 'lobe-chat-desktop' : 'lobe-chat';

package/src/database/schemas/relations.ts

@@ -122,6 +122,17 @@ export const agentsToSessionsRelations = relations(agentsToSessions, ({ one }) =
   }),
 }));
 
+export const filesToSessionsRelations = relations(filesToSessions, ({ one }) => ({
+  file: one(files, {
+    fields: [filesToSessions.fileId],
+    references: [files.id],
+  }),
+  session: one(sessions, {
+    fields: [filesToSessions.sessionId],
+    references: [sessions.id],
+  }),
+}));
+
 export const agentsKnowledgeBasesRelations = relations(agentsKnowledgeBases, ({ one }) => ({
   knowledgeBase: one(knowledgeBases, {
     fields: [agentsKnowledgeBases.knowledgeBaseId],
@@ -133,6 +144,39 @@ export const agentsKnowledgeBasesRelations = relations(agentsKnowledgeBases, ({
   }),
 }));
 
+export const agentsFilesRelations = relations(agentsFiles, ({ one }) => ({
+  file: one(files, {
+    fields: [agentsFiles.fileId],
+    references: [files.id],
+  }),
+  agent: one(agents, {
+    fields: [agentsFiles.agentId],
+    references: [agents.id],
+  }),
+}));
+
+export const messagesFilesRelations = relations(messagesFiles, ({ one }) => ({
+  file: one(files, {
+    fields: [messagesFiles.fileId],
+    references: [files.id],
+  }),
+  message: one(messages, {
+    fields: [messagesFiles.messageId],
+    references: [messages.id],
+  }),
+}));
+
+export const fileChunksRelations = relations(fileChunks, ({ one }) => ({
+  file: one(files, {
+    fields: [fileChunks.fileId],
+    references: [files.id],
+  }),
+  chunk: one(chunks, {
+    fields: [fileChunks.chunkId],
+    references: [chunks.id],
+  }),
+}));
+
 export const sessionsRelations = relations(sessions, ({ many, one }) => ({
   filesToSessions: many(filesToSessions),
   agentsToSessions: many(agentsToSessions),

package/src/features/AgentSetting/store/action.ts

@@ -1,3 +1,4 @@
+import { getSingletonAnalyticsOptional } from '@lobehub/analytics';
 import { DeepPartial } from 'utility-types';
 import { StateCreator } from 'zustand/vanilla';
 
@@ -260,7 +261,31 @@ export const store: StateCreator<Store, [['zustand/devtools', never]]> = (set, g
     await get().dispatchConfig({ config, type: 'update' });
   },
   setAgentMeta: async (meta) => {
-    await get().dispatchMeta({ type: 'update', value: meta });
+    const { dispatchMeta, id, meta: currentMeta } = get();
+    const mergedMeta = merge(currentMeta, meta);
+
+    try {
+      const analytics = getSingletonAnalyticsOptional();
+      if (analytics) {
+        analytics.track({
+          name: 'agent_meta_updated',
+          properties: {
+            assistant_avatar: mergedMeta.avatar,
+            assistant_background_color: mergedMeta.backgroundColor,
+            assistant_description: mergedMeta.description,
+            assistant_name: mergedMeta.title,
+            assistant_tags: mergedMeta.tags,
+            is_inbox: id === 'inbox',
+            session_id: id || 'unknown',
+            timestamp: Date.now(),
+            user_id: useUserStore.getState().user?.id || 'anonymous',
+          },
+        });
+      }
+    } catch (error) {
+      console.warn('Failed to track agent meta update:', error);
+    }
+    await dispatchMeta({ type: 'update', value: meta });
   },
 
   setChatConfig: async (config) => {

package/src/features/ChatInput/useSend.ts

@@ -1,8 +1,12 @@
+import { useAnalytics } from '@lobehub/analytics/react';
 import { useCallback, useMemo } from 'react';
 
+import { getAgentStoreState } from '@/store/agent';
+import { agentSelectors } from '@/store/agent/selectors';
 import { useChatStore } from '@/store/chat';
-import { chatSelectors } from '@/store/chat/selectors';
+import { chatSelectors, topicSelectors } from '@/store/chat/selectors';
 import { fileChatSelectors, useFileStore } from '@/store/file';
+import { getUserStoreState } from '@/store/user';
 import { SendMessageParams } from '@/types/message';
 
 export type UseSendMessageParams = Pick<
@@ -15,6 +19,7 @@ export const useSendMessage = () => {
     s.sendMessage,
     s.updateInputMessage,
   ]);
+  const { analytics } = useAnalytics();
 
   const clearChatUploadFileList = useFileStore((s) => s.clearChatUploadFileList);
 
@@ -49,6 +54,29 @@ export const useSendMessage = () => {
     updateInputMessage('');
     clearChatUploadFileList();
 
+    // 获取分析数据
+    const userStore = getUserStoreState();
+    const agentStore = getAgentStoreState();
+
+    // 直接使用现有数据结构判断消息类型
+    const hasImages = fileList.some((file) => file.file?.type?.startsWith('image'));
+    const messageType = fileList.length === 0 ? 'text' : hasImages ? 'image' : 'file';
+
+    analytics?.track({
+      name: 'send_message',
+      properties: {
+        chat_id: store.activeId || 'unknown',
+        current_topic: topicSelectors.currentActiveTopic(store)?.title || null,
+        has_attachments: fileList.length > 0,
+        history_message_count: chatSelectors.activeBaseChats(store).length,
+        message: store.inputMessage,
+        message_length: store.inputMessage.length,
+        message_type: messageType,
+        selected_model: agentSelectors.currentAgentModel(agentStore),
+        session_id: store.activeId || 'inbox', // 当前活跃的会话ID
+        user_id: userStore.user?.id || 'anonymous',
+      },
+    });
     // const hasSystemRole = agentSelectors.hasSystemRole(useAgentStore.getState());
     // const agentSetting = useAgentStore.getState().agentSettingInstance;
 

package/src/features/User/UserLoginOrSignup/Community.tsx

@@ -1,3 +1,4 @@
+import { useAnalytics } from '@lobehub/analytics/react';
 import { Button } from '@lobehub/ui';
 import { memo } from 'react';
 import { useTranslation } from 'react-i18next';
@@ -7,12 +8,24 @@ import UserInfo from '../UserInfo';
 
 const UserLoginOrSignup = memo<{ onClick: () => void }>(({ onClick }) => {
   const { t } = useTranslation('auth');
+  const { analytics } = useAnalytics();
+
+  const handleClick = () => {
+    analytics?.track({
+      name: 'login_or_signup_clicked',
+      properties: {
+        spm: 'homepage.login_or_signup.click',
+      },
+    });
+
+    onClick();
+  };
 
   return (
     <>
       <UserInfo />
       <Flexbox paddingBlock={12} paddingInline={16} width={'100%'}>
-        <Button block onClick={onClick} type={'primary'}>
+        <Button block onClick={handleClick} type={'primary'}>
           {t('loginOrSignup')}
         </Button>
       </Flexbox>

package/src/layout/GlobalProvider/Locale.tsx

@@ -40,7 +40,8 @@ const Locale = memo<LocaleLayoutProps>(({ children, defaultLang, antdLocale }) =
 
   // if run on server side, init i18n instance everytime
   if (isOnServerSide) {
-    i18n.init();
+    // use sync mode to init instantly
+    i18n.init({ initAsync: false });
 
     // load the dayjs locale
     // if (lang) {

package/src/layout/GlobalProvider/index.tsx

@@ -1,5 +1,6 @@
 import { ReactNode, Suspense } from 'react';
 
+import { LobeAnalyticsProviderWrapper } from '@/components/Analytics/LobeAnalyticsProviderWrapper';
 import { getServerFeatureFlagsValue } from '@/config/featureFlags';
 import { appEnv } from '@/envs/app';
 import DevPanel from '@/features/DevPanel';
@@ -54,7 +55,9 @@ const GlobalLayout = async ({
             isMobile={isMobile}
             serverConfig={serverConfig}
           >
-            <QueryProvider>{children}</QueryProvider>
+            <QueryProvider>
+              <LobeAnalyticsProviderWrapper>{children}</LobeAnalyticsProviderWrapper>
+            </QueryProvider>
             <StoreInitialization />
             <Suspense>
               <ImportSettings />

package/src/libs/analytics/index.ts

@@ -0,0 +1,25 @@
+import { createServerAnalytics } from '@lobehub/analytics/server';
+
+import { analyticsEnv } from '@/config/analytics';
+import { BUSINESS_LINE } from '@/const/analytics';
+import { isDev } from '@/utils/env';
+
+export const serverAnalytics = createServerAnalytics({
+  business: BUSINESS_LINE,
+  debug: isDev,
+  providers: {
+    posthogNode: {
+      debug: analyticsEnv.DEBUG_POSTHOG_ANALYTICS,
+      enabled: analyticsEnv.ENABLED_POSTHOG_ANALYTICS,
+      host: analyticsEnv.POSTHOG_HOST,
+      key: analyticsEnv.POSTHOG_KEY ?? '',
+    },
+  },
+});
+
+export const initializeServerAnalytics = async () => {
+  await serverAnalytics.initialize();
+  return serverAnalytics;
+};
+
+export default serverAnalytics;

package/src/locales/create.ts

@@ -31,8 +31,10 @@ export const createI18nNext = (lang?: string) => {
     }
   });
   return {
-    init: () =>
-      instance.init({
+    init: (params: { initAsync?: boolean } = {}) => {
+      const { initAsync = true } = params;
+
+      return instance.init({
         debug: debugMode,
         defaultNS: ['error', 'common', 'chat'],
         // detection: {
@@ -50,11 +52,14 @@ export const createI18nNext = (lang?: string) => {
         //   lookupCookie: LOBE_LOCALE_COOKIE,
         // },
         fallbackLng: DEFAULT_LANG,
+
+        initAsync,
         interpolation: {
           escapeValue: false,
         },
         lng: lang,
-      }),
+      });
+    },
     instance,
   };
 };

package/src/server/routers/async/caller.ts

@@ -1,4 +1,4 @@
-import { createTRPCClient, httpBatchLink } from '@trpc/client';
+import { createTRPCClient, httpLink } from '@trpc/client';
 import superjson from 'superjson';
 import urlJoin from 'url-join';
 
@@ -26,7 +26,7 @@ export const createAsyncServerClient = async (userId: string, payload: JWTPayloa
 
   return createTRPCClient<AsyncRouter>({
     links: [
-      httpBatchLink({
+      httpLink({
         headers,
         transformer: superjson,
         url: urlJoin(appEnv.APP_URL!, '/trpc/async'),

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

@@ -8,6 +8,14 @@ import { AgentService } from '@/server/services/agent';
 
 import { UserService } from './index';
 
+// Mock @/libs/analytics to avoid server-side environment variable access in client test environment
+vi.mock('@/libs/analytics', () => ({
+  initializeServerAnalytics: vi.fn().mockResolvedValue({
+    identify: vi.fn(),
+    track: vi.fn(),
+  }),
+}));
+
 // Mock dependencies
 vi.mock('@/database/models/user', () => {
   const MockUserModel = vi.fn();

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

@@ -2,6 +2,7 @@ import { UserJSON } from '@clerk/backend';
 
 import { UserModel } from '@/database/models/user';
 import { serverDB } from '@/database/server';
+import { initializeServerAnalytics } from '@/libs/analytics';
 import { pino } from '@/libs/logger';
 import { KeyVaultsGateKeeper } from '@/server/modules/KeyVaultsEncrypt';
 import { S3 } from '@/server/modules/S3';
@@ -51,6 +52,23 @@ export class UserService {
 
     /* ↑ cloud slot ↑ */
 
+    //analytics
+    const analytics = await initializeServerAnalytics();
+    analytics?.identify(id, {
+      email: email?.email_address,
+      firstName: params.first_name,
+      lastName: params.last_name,
+      phone: phone?.phone_number,
+      username: params.username,
+    });
+    analytics?.track({
+      name: 'user_register_completed',
+      properties: {
+        spm: 'user_service.create_user.user_created',
+      },
+      userId: id,
+    });
+
     return { message: 'user created', success: true };
   };
 

package/src/store/session/slices/session/action.ts

@@ -1,3 +1,4 @@
+import { getSingletonAnalyticsOptional } from '@lobehub/analytics';
 import isEqual from 'fast-deep-equal';
 import { t } from 'i18next';
 import useSWR, { SWRResponse, mutate } from 'swr';
@@ -10,8 +11,8 @@ import { DEFAULT_AGENT_LOBE_SESSION, INBOX_SESSION_ID } from '@/const/session';
 import { useClientDataSWR } from '@/libs/swr';
 import { sessionService } from '@/services/session';
 import { SessionStore } from '@/store/session';
-import { useUserStore } from '@/store/user';
-import { settingsSelectors } from '@/store/user/selectors';
+import { getUserStoreState, useUserStore } from '@/store/user';
+import { settingsSelectors, userProfileSelectors } from '@/store/user/selectors';
 import { MetaData } from '@/types/meta';
 import {
   ChatSessionList,
@@ -24,6 +25,7 @@ import {
 import { merge } from '@/utils/merge';
 import { setNamespace } from '@/utils/storeDebug';
 
+import { sessionGroupSelectors } from '../sessionGroup/selectors';
 import { SessionDispatch, sessionsReducer } from './reducers';
 import { sessionSelectors } from './selectors';
 import { sessionMetaSelectors } from './selectors/meta';
@@ -114,6 +116,30 @@ export const createSessionSlice: StateCreator<
     const id = await sessionService.createSession(LobeSessionType.Agent, newSession);
     await refreshSessions();
 
+    // Track new agent creation analytics
+    const analytics = getSingletonAnalyticsOptional();
+    if (analytics) {
+      const userStore = getUserStoreState();
+      const userId = userProfileSelectors.userId(userStore);
+
+      // Get group information
+      const groupId = newSession.group || 'default';
+      const group = sessionGroupSelectors.getGroupById(groupId)(get());
+      const groupName = group?.name || (groupId === 'default' ? 'Default' : 'Unknown');
+
+      analytics.track({
+        name: 'new_agent_created',
+        properties: {
+          assistant_name: newSession.meta?.title || 'Untitled Agent',
+          assistant_tags: newSession.meta?.tags || [],
+          group_id: groupId,
+          group_name: groupName,
+          session_id: id,
+          user_id: userId || 'anonymous',
+        },
+      });
+    }
+
     // Whether to goto  to the new session after creation, the default is to switch to
     if (isSwitchSession) switchSession(id);
 

package/src/store/user/slices/common/action.ts

@@ -1,3 +1,4 @@
+import { getSingletonAnalyticsOptional } from '@lobehub/analytics';
 import useSWR, { SWRResponse, mutate } from 'swr';
 import { DeepPartial } from 'utility-types';
 import type { StateCreator } from 'zustand/vanilla';
@@ -22,7 +23,6 @@ const GET_USER_STATE_KEY = 'initUserState';
  */
 export interface CommonAction {
   refreshUserState: () => Promise<void>;
-
   updateAvatar: (avatar: string) => Promise<void>;
   useCheckTrace: (shouldFetch: boolean) => SWRResponse;
   useInitUserState: (
@@ -118,7 +118,14 @@ export const createCommonSlice: StateCreator<
               false,
               n('initUserState'),
             );
-
+            //analytics
+            const analytics = getSingletonAnalyticsOptional();
+            analytics?.identify(data.userId || '', {
+              email: data.email,
+              firstName: data.firstName,
+              lastName: data.lastName,
+              username: data.username,
+            });
             get().refreshDefaultModelProviderList({ trigger: 'fetchUserState' });
           }
         },

package/src/utils/locale.ts

@@ -1,7 +1,8 @@
 import { resolveAcceptLanguage } from 'resolve-accept-language';
 
 import { DEFAULT_LANG } from '@/const/locale';
-import { locales, normalizeLocale } from '@/locales/resources';
+import { Locales, locales, normalizeLocale } from '@/locales/resources';
+import { RouteVariants } from '@/utils/server/routeVariants';
 
 export const getAntdLocale = async (lang?: string) => {
   let normalLang: any = normalizeLocale(lang);
@@ -44,3 +45,17 @@ export const parseBrowserLanguage = (headers: Headers, defaultLang: string = DEF
 
   return browserLang;
 };
+
+/**
+ * Parse the page locale from the URL and search
+ * used in cloud
+ */
+export const parsePageLocale = async (props: {
+  params: Promise<{ variants: string }>;
+  searchParams: Promise<any>;
+}) => {
+  const searchParams = await props.searchParams;
+
+  const browserLocale = await RouteVariants.getLocale(props);
+  return normalizeLocale(searchParams?.hl || browserLocale) as Locales;
+};

package/tests/setup.ts

@@ -5,6 +5,17 @@ import { theme } from 'antd';
 // refs: https://github.com/dumbmatter/fakeIndexedDB#dexie-and-other-indexeddb-api-wrappers
 import 'fake-indexeddb/auto';
 import React from 'react';
+import { vi } from 'vitest';
+
+// Global mock for @lobehub/analytics/react to avoid AnalyticsProvider dependency
+// This prevents tests from failing when components use useAnalytics hook
+vi.mock('@lobehub/analytics/react', () => ({
+  useAnalytics: () => ({
+    analytics: {
+      track: vi.fn(),
+    },
+  }),
+}));
 
 // only inject in the dom environment
 if (