@lobehub/lobehub 2.0.0-next.334 → 2.0.0-next.336

package/src/server/routers/lambda/market/agentGroup.ts

@@ -202,6 +202,66 @@ export const agentGroupRouter = router({
 
   
   /**
+   * Deprecate agent group
+   * POST /market/agent-group/:identifier/deprecate
+   */
+deprecateAgentGroup: agentGroupProcedure
+    .input(z.object({ identifier: z.string() }))
+    .mutation(async ({ input, ctx }) => {
+      log('deprecateAgentGroup input: %O', input);
+
+      try {
+        const deprecateUrl = `${MARKET_BASE_URL}/api/v1/agent-groups/${input.identifier}/deprecate`;
+
+        const headers: Record<string, string> = {
+          'Content-Type': 'application/json',
+        };
+
+        const userInfo = ctx.marketUserInfo as TrustedClientUserInfo | undefined;
+        const accessToken = (ctx as { marketOidcAccessToken?: string }).marketOidcAccessToken;
+
+        if (userInfo) {
+          const trustedClientToken = generateTrustedClientToken(userInfo);
+          if (trustedClientToken) {
+            headers['x-lobe-trust-token'] = trustedClientToken;
+          }
+        }
+
+        if (!headers['x-lobe-trust-token'] && accessToken) {
+          headers['Authorization'] = `Bearer ${accessToken}`;
+        }
+
+        const response = await fetch(deprecateUrl, {
+          headers,
+          method: 'POST',
+        });
+
+        if (!response.ok) {
+          const errorText = await response.text();
+          log(
+            'Deprecate agent group failed: %s %s - %s',
+            response.status,
+            response.statusText,
+            errorText,
+          );
+          throw new Error(`Failed to deprecate agent group: ${response.statusText}`);
+        }
+
+        log('Deprecate agent group success');
+        return { success: true };
+      } catch (error) {
+        log('Error deprecating agent group: %O', error);
+        throw new TRPCError({
+          cause: error,
+          code: 'INTERNAL_SERVER_ERROR',
+          message: error instanceof Error ? error.message : 'Failed to deprecate agent group',
+        });
+      }
+    }),
+
+  
+  
+/**
    * Fork an agent group
    * POST /market/agent-group/:identifier/fork
    */
@@ -340,7 +400,6 @@ getAgentGroupForkSource: agentGroupProcedure
 
   
   
-
 /**
    * Get all forks of an agent group
    * GET /market/agent-group/:identifier/forks
@@ -401,6 +460,66 @@ getAgentGroupForks: agentGroupProcedure
 
   
   
+/**
+   * Publish agent group
+   * POST /market/agent-group/:identifier/publish
+   */
+publishAgentGroup: agentGroupProcedure
+    .input(z.object({ identifier: z.string() }))
+    .mutation(async ({ input, ctx }) => {
+      log('publishAgentGroup input: %O', input);
+
+      try {
+        const publishUrl = `${MARKET_BASE_URL}/api/v1/agent-groups/${input.identifier}/publish`;
+
+        const headers: Record<string, string> = {
+          'Content-Type': 'application/json',
+        };
+
+        const userInfo = ctx.marketUserInfo as TrustedClientUserInfo | undefined;
+        const accessToken = (ctx as { marketOidcAccessToken?: string }).marketOidcAccessToken;
+
+        if (userInfo) {
+          const trustedClientToken = generateTrustedClientToken(userInfo);
+          if (trustedClientToken) {
+            headers['x-lobe-trust-token'] = trustedClientToken;
+          }
+        }
+
+        if (!headers['x-lobe-trust-token'] && accessToken) {
+          headers['Authorization'] = `Bearer ${accessToken}`;
+        }
+
+        const response = await fetch(publishUrl, {
+          headers,
+          method: 'POST',
+        });
+
+        if (!response.ok) {
+          const errorText = await response.text();
+          log(
+            'Publish agent group failed: %s %s - %s',
+            response.status,
+            response.statusText,
+            errorText,
+          );
+          throw new Error(`Failed to publish agent group: ${response.statusText}`);
+        }
+
+        log('Publish agent group success');
+        return { success: true };
+      } catch (error) {
+        log('Error publishing agent group: %O', error);
+        throw new TRPCError({
+          cause: error,
+          code: 'INTERNAL_SERVER_ERROR',
+          message: error instanceof Error ? error.message : 'Failed to publish agent group',
+        });
+      }
+    }),
+
+  
+  
 /**
    * Unified publish or create agent group flow
    * 1. Check if identifier exists and if current user is owner
@@ -494,6 +613,65 @@ publishOrCreate: agentGroupProcedure
         });
       }
     }),
+
+  
+  /**
+   * Unpublish agent group
+   * POST /market/agent-group/:identifier/unpublish
+   */
+unpublishAgentGroup: agentGroupProcedure
+    .input(z.object({ identifier: z.string() }))
+    .mutation(async ({ input, ctx }) => {
+      log('unpublishAgentGroup input: %O', input);
+
+      try {
+        const unpublishUrl = `${MARKET_BASE_URL}/api/v1/agent-groups/${input.identifier}/unpublish`;
+
+        const headers: Record<string, string> = {
+          'Content-Type': 'application/json',
+        };
+
+        const userInfo = ctx.marketUserInfo as TrustedClientUserInfo | undefined;
+        const accessToken = (ctx as { marketOidcAccessToken?: string }).marketOidcAccessToken;
+
+        if (userInfo) {
+          const trustedClientToken = generateTrustedClientToken(userInfo);
+          if (trustedClientToken) {
+            headers['x-lobe-trust-token'] = trustedClientToken;
+          }
+        }
+
+        if (!headers['x-lobe-trust-token'] && accessToken) {
+          headers['Authorization'] = `Bearer ${accessToken}`;
+        }
+
+        const response = await fetch(unpublishUrl, {
+          headers,
+          method: 'POST',
+        });
+
+        if (!response.ok) {
+          const errorText = await response.text();
+          log(
+            'Unpublish agent group failed: %s %s - %s',
+            response.status,
+            response.statusText,
+            errorText,
+          );
+          throw new Error(`Failed to unpublish agent group: ${response.statusText}`);
+        }
+
+        log('Unpublish agent group success');
+        return { success: true };
+      } catch (error) {
+        log('Error unpublishing agent group: %O', error);
+        throw new TRPCError({
+          cause: error,
+          code: 'INTERNAL_SERVER_ERROR',
+          message: error instanceof Error ? error.message : 'Failed to unpublish agent group',
+        });
+      }
+    }),
 });
 
 export type AgentGroupRouter = typeof agentGroupRouter;

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

@@ -1761,6 +1761,7 @@ export class DiscoverService {
         knowledgeCount: agent.knowledgeCount || 0,
         pluginCount: agent.pluginCount || 0,
         schemaVersion: 1,
+        status: agent.status,
         tags: agent.tags || [],
         title: agent.name || agent.identifier,
         tokenUsage: agent.tokenUsage || 0,
@@ -1780,6 +1781,7 @@ export class DiscoverService {
         isOfficial: group.isOfficial || false,
         memberCount: 0, // Will be populated from memberAgents in detail view
         schemaVersion: 1,
+        status: group.status,
         tags: group.tags || [],
         title: group.name || group.identifier,
         updatedAt: group.updatedAt,
@@ -1802,6 +1804,7 @@ export class DiscoverService {
           knowledgeCount: agent.knowledgeCount || 0,
           pluginCount: agent.pluginCount || 0,
           schemaVersion: 1,
+          status: agent.status,
           tags: agent.tags || [],
           title: agent.name || agent.identifier,
           tokenUsage: agent.tokenUsage || 0,
@@ -1824,6 +1827,7 @@ export class DiscoverService {
         isOfficial: group.isOfficial || false,
         memberCount: 0, // Will be populated from memberAgents in detail view
         schemaVersion: 1,
+        status: group.status,
         tags: group.tags || [],
         title: group.name || group.identifier,
         updatedAt: group.updatedAt,

package/src/services/chat/chat.test.ts

@@ -15,6 +15,35 @@ import { aiModelSelectors } from '@/store/aiInfra';
 import { useToolStore } from '@/store/tool';
 
 import { chatService } from './index';
+import type { ResolvedAgentConfig } from './mecha';
+
+/**
+ * Default mock resolvedAgentConfig for tests
+ */
+const createMockResolvedConfig = (overrides?: {
+  agentConfig?: Partial<ResolvedAgentConfig['agentConfig']>;
+  chatConfig?: Partial<ResolvedAgentConfig['chatConfig']>;
+  plugins?: string[];
+  isBuiltinAgent?: boolean;
+}): ResolvedAgentConfig =>
+  ({
+    agentConfig: {
+      model: DEFAULT_AGENT_CONFIG.model,
+      provider: 'openai',
+      systemRole: '',
+      chatConfig: {},
+      params: {},
+      tts: {},
+      ...overrides?.agentConfig,
+    },
+    chatConfig: {
+      searchMode: 'off',
+      autoCreateTopicThreshold: 2,
+      ...overrides?.chatConfig,
+    },
+    isBuiltinAgent: overrides?.isBuiltinAgent ?? false,
+    plugins: overrides?.plugins ?? [],
+  }) as ResolvedAgentConfig;
 
 // Mocking external dependencies
 vi.mock('i18next', () => ({
@@ -109,7 +138,10 @@ describe('ChatService', () => {
           ],
         });
       });
-      await chatService.createAssistantMessage({ messages, plugins: enabledPlugins });
+      await chatService.createAssistantMessage({
+        messages,
+        resolvedAgentConfig: createMockResolvedConfig({ plugins: enabledPlugins }),
+      });
 
       expect(getChatCompletionSpy).toHaveBeenCalledWith(
         expect.objectContaining({
@@ -136,21 +168,14 @@ describe('ChatService', () => {
         vi.spyOn(aiModelSelectors, 'isModelHasExtendParams').mockReturnValue(() => true);
         vi.spyOn(aiModelSelectors, 'modelExtendParams').mockReturnValue(() => ['enableReasoning']);
 
-        // Mock agent chat config with reasoning enabled
-        vi.spyOn(chatConfigByIdSelectors, 'getChatConfigById').mockReturnValue(
-          () =>
-            ({
-              enableReasoning: true,
-              reasoningBudgetToken: 2048,
-              searchMode: 'off',
-            }) as any,
-        );
-
         await chatService.createAssistantMessage({
           messages,
           model: 'deepseek-reasoner',
           provider: 'deepseek',
-          plugins: [],
+          resolvedAgentConfig: createMockResolvedConfig({
+            agentConfig: { model: 'deepseek-reasoner', provider: 'deepseek' },
+            chatConfig: { enableReasoning: true, reasoningBudgetToken: 2048 },
+          }),
         });
 
         expect(getChatCompletionSpy).toHaveBeenCalledWith(
@@ -172,20 +197,14 @@ describe('ChatService', () => {
         vi.spyOn(aiModelSelectors, 'isModelHasExtendParams').mockReturnValue(() => true);
         vi.spyOn(aiModelSelectors, 'modelExtendParams').mockReturnValue(() => ['enableReasoning']);
 
-        // Mock agent chat config with reasoning disabled
-        vi.spyOn(chatConfigByIdSelectors, 'getChatConfigById').mockReturnValue(
-          () =>
-            ({
-              enableReasoning: false,
-              searchMode: 'off',
-            }) as any,
-        );
-
         await chatService.createAssistantMessage({
           messages,
           model: 'deepseek-reasoner',
           provider: 'deepseek',
-          plugins: [],
+          resolvedAgentConfig: createMockResolvedConfig({
+            agentConfig: { model: 'deepseek-reasoner', provider: 'deepseek' },
+            chatConfig: { enableReasoning: false },
+          }),
         });
 
         expect(getChatCompletionSpy).toHaveBeenCalledWith(
@@ -207,21 +226,15 @@ describe('ChatService', () => {
         vi.spyOn(aiModelSelectors, 'isModelHasExtendParams').mockReturnValue(() => true);
         vi.spyOn(aiModelSelectors, 'modelExtendParams').mockReturnValue(() => ['enableReasoning']);
 
-        // Mock agent chat config with reasoning enabled but no custom budget
-        vi.spyOn(chatConfigByIdSelectors, 'getChatConfigById').mockReturnValue(
-          () =>
-            ({
-              enableReasoning: true,
-              // reasoningBudgetToken is undefined
-              searchMode: 'off',
-            }) as any,
-        );
-
         await chatService.createAssistantMessage({
           messages,
           model: 'deepseek-reasoner',
           provider: 'deepseek',
-          plugins: [],
+          resolvedAgentConfig: createMockResolvedConfig({
+            agentConfig: { model: 'deepseek-reasoner', provider: 'deepseek' },
+            // enableReasoning is true, but reasoningBudgetToken is undefined
+            chatConfig: { enableReasoning: true },
+          }),
         });
 
         expect(getChatCompletionSpy).toHaveBeenCalledWith(
@@ -243,20 +256,14 @@ describe('ChatService', () => {
         vi.spyOn(aiModelSelectors, 'isModelHasExtendParams').mockReturnValue(() => true);
         vi.spyOn(aiModelSelectors, 'modelExtendParams').mockReturnValue(() => ['reasoningEffort']);
 
-        // Mock agent chat config with reasoning effort set
-        vi.spyOn(chatConfigByIdSelectors, 'getChatConfigById').mockReturnValue(
-          () =>
-            ({
-              reasoningEffort: 'high',
-              searchMode: 'off',
-            }) as any,
-        );
-
         await chatService.createAssistantMessage({
           messages,
           model: 'test-model',
           provider: 'test-provider',
-          plugins: [],
+          resolvedAgentConfig: createMockResolvedConfig({
+            agentConfig: { model: 'test-model', provider: 'test-provider' },
+            chatConfig: { reasoningEffort: 'high' },
+          }),
         });
 
         expect(getChatCompletionSpy).toHaveBeenCalledWith(
@@ -275,20 +282,14 @@ describe('ChatService', () => {
         vi.spyOn(aiModelSelectors, 'isModelHasExtendParams').mockReturnValue(() => true);
         vi.spyOn(aiModelSelectors, 'modelExtendParams').mockReturnValue(() => ['thinkingBudget']);
 
-        // Mock agent chat config with thinking budget set
-        vi.spyOn(chatConfigByIdSelectors, 'getChatConfigById').mockReturnValue(
-          () =>
-            ({
-              thinkingBudget: 5000,
-              searchMode: 'off',
-            }) as any,
-        );
-
         await chatService.createAssistantMessage({
           messages,
           model: 'test-model',
           provider: 'test-provider',
-          plugins: [],
+          resolvedAgentConfig: createMockResolvedConfig({
+            agentConfig: { model: 'test-model', provider: 'test-provider' },
+            chatConfig: { thinkingBudget: 5000 },
+          }),
         });
 
         expect(getChatCompletionSpy).toHaveBeenCalledWith(
@@ -329,9 +330,11 @@ describe('ChatService', () => {
         const getChatCompletionSpy = vi.spyOn(chatService, 'getChatCompletion');
         await chatService.createAssistantMessage({
           messages,
-          plugins: [],
           model: 'gpt-4-vision-preview',
           provider: 'openai',
+          resolvedAgentConfig: createMockResolvedConfig({
+            agentConfig: { model: 'gpt-4-vision-preview', provider: 'openai' },
+          }),
         });
 
         expect(getChatCompletionSpy).toHaveBeenCalledWith(
@@ -378,7 +381,10 @@ describe('ChatService', () => {
         ] as UIChatMessage[];
 
         const getChatCompletionSpy = vi.spyOn(chatService, 'getChatCompletion');
-        await chatService.createAssistantMessage({ messages, plugins: [] });
+        await chatService.createAssistantMessage({
+          messages,
+          resolvedAgentConfig: createMockResolvedConfig(),
+        });
 
         expect(getChatCompletionSpy).toHaveBeenCalledWith(
           {
@@ -435,8 +441,10 @@ describe('ChatService', () => {
 
         await chatService.createAssistantMessage({
           messages,
-          plugins: [],
           model: 'gpt-4-vision-preview',
+          resolvedAgentConfig: createMockResolvedConfig({
+            agentConfig: { model: 'gpt-4-vision-preview' },
+          }),
         });
 
         // Verify the utility functions were called
@@ -525,8 +533,10 @@ describe('ChatService', () => {
 
         await chatService.createAssistantMessage({
           messages,
-          plugins: [],
           model: 'gpt-4-vision-preview',
+          resolvedAgentConfig: createMockResolvedConfig({
+            agentConfig: { model: 'gpt-4-vision-preview' },
+          }),
         });
 
         // Verify the utility functions were called
@@ -630,8 +640,10 @@ describe('ChatService', () => {
 
         await chatService.createAssistantMessage({
           messages,
-          plugins: [],
           model: 'gpt-4-vision-preview',
+          resolvedAgentConfig: createMockResolvedConfig({
+            agentConfig: { model: 'gpt-4-vision-preview' },
+          }),
         });
 
         // Verify isDesktopLocalStaticServerUrl was called for each image
@@ -730,7 +742,10 @@ describe('ChatService', () => {
           messages,
           model: 'gpt-3.5-turbo-1106',
           top_p: 1,
-          plugins: ['seo'],
+          resolvedAgentConfig: createMockResolvedConfig({
+            agentConfig: { model: 'gpt-3.5-turbo-1106' },
+            plugins: ['seo'],
+          }),
         });
 
         expect(getChatCompletionSpy).toHaveBeenCalledWith(
@@ -832,7 +847,10 @@ describe('ChatService', () => {
           messages,
           model: 'gpt-3.5-turbo-1106',
           top_p: 1,
-          plugins: ['seo'],
+          resolvedAgentConfig: createMockResolvedConfig({
+            agentConfig: { model: 'gpt-3.5-turbo-1106' },
+            plugins: ['seo'],
+          }),
         });
 
         expect(getChatCompletionSpy).toHaveBeenCalledWith(
@@ -888,7 +906,9 @@ describe('ChatService', () => {
           messages,
           model: 'gpt-3.5-turbo-1106',
           top_p: 1,
-          plugins: ['ttt'],
+          resolvedAgentConfig: createMockResolvedConfig({
+            agentConfig: { model: 'gpt-3.5-turbo-1106' },
+          }),
         });
 
         expect(getChatCompletionSpy).toHaveBeenCalledWith(
@@ -949,7 +969,10 @@ describe('ChatService', () => {
           mockToolsEngine as any,
         );
 
-        await chatService.createAssistantMessage({ messages, plugins: [] });
+        await chatService.createAssistantMessage({
+          messages,
+          resolvedAgentConfig: createMockResolvedConfig(),
+        });
 
         // Verify tools were passed to getChatCompletion
         expect(getChatCompletionSpy).toHaveBeenCalledWith(
@@ -1003,7 +1026,10 @@ describe('ChatService', () => {
           mockToolsEngine as any,
         );
 
-        await chatService.createAssistantMessage({ messages, plugins: [] });
+        await chatService.createAssistantMessage({
+          messages,
+          resolvedAgentConfig: createMockResolvedConfig(),
+        });
 
         // Verify enabledSearch was set to true
         expect(getChatCompletionSpy).toHaveBeenCalledWith(
@@ -1051,7 +1077,10 @@ describe('ChatService', () => {
           mockToolsEngine as any,
         );
 
-        await chatService.createAssistantMessage({ messages, plugins: [] });
+        await chatService.createAssistantMessage({
+          messages,
+          resolvedAgentConfig: createMockResolvedConfig(),
+        });
 
         // Verify enabledSearch was not set
         expect(getChatCompletionSpy).toHaveBeenCalledWith(
@@ -1342,20 +1371,14 @@ describe('ChatService private methods', () => {
         'disableContextCaching',
       ]);
 
-      // Mock agent chat config with context caching disabled
-      vi.spyOn(chatConfigByIdSelectors, 'getChatConfigById').mockReturnValue(
-        () =>
-          ({
-            disableContextCaching: true,
-            searchMode: 'off',
-          }) as any,
-      );
-
       await chatService.createAssistantMessage({
         messages,
         model: 'test-model',
         provider: 'test-provider',
-        plugins: [],
+        resolvedAgentConfig: createMockResolvedConfig({
+          agentConfig: { model: 'test-model', provider: 'test-provider' },
+          chatConfig: { disableContextCaching: true },
+        }),
       });
 
       expect(getChatCompletionSpy).toHaveBeenCalledWith(
@@ -1378,20 +1401,14 @@ describe('ChatService private methods', () => {
         'disableContextCaching',
       ]);
 
-      // Mock agent chat config with context caching enabled (default)
-      vi.spyOn(chatConfigByIdSelectors, 'getChatConfigById').mockReturnValue(
-        () =>
-          ({
-            disableContextCaching: false,
-            searchMode: 'off',
-          }) as any,
-      );
-
       await chatService.createAssistantMessage({
         messages,
         model: 'test-model',
         provider: 'test-provider',
-        plugins: [],
+        resolvedAgentConfig: createMockResolvedConfig({
+          agentConfig: { model: 'test-model', provider: 'test-provider' },
+          chatConfig: { disableContextCaching: false },
+        }),
       });
 
       // enabledContextCaching should not be present in the call
@@ -1407,20 +1424,14 @@ describe('ChatService private methods', () => {
       vi.spyOn(aiModelSelectors, 'isModelHasExtendParams').mockReturnValue(() => true);
       vi.spyOn(aiModelSelectors, 'modelExtendParams').mockReturnValue(() => ['reasoningEffort']);
 
-      // Mock agent chat config with reasoning effort set
-      vi.spyOn(chatConfigByIdSelectors, 'getChatConfigById').mockReturnValue(
-        () =>
-          ({
-            reasoningEffort: 'high',
-            searchMode: 'off',
-          }) as any,
-      );
-
       await chatService.createAssistantMessage({
         messages,
         model: 'test-model',
         provider: 'test-provider',
-        plugins: [],
+        resolvedAgentConfig: createMockResolvedConfig({
+          agentConfig: { model: 'test-model', provider: 'test-provider' },
+          chatConfig: { reasoningEffort: 'high' },
+        }),
       });
 
       expect(getChatCompletionSpy).toHaveBeenCalledWith(
@@ -1439,20 +1450,14 @@ describe('ChatService private methods', () => {
       vi.spyOn(aiModelSelectors, 'isModelHasExtendParams').mockReturnValue(() => true);
       vi.spyOn(aiModelSelectors, 'modelExtendParams').mockReturnValue(() => ['thinkingBudget']);
 
-      // Mock agent chat config with thinking budget set
-      vi.spyOn(chatConfigByIdSelectors, 'getChatConfigById').mockReturnValue(
-        () =>
-          ({
-            thinkingBudget: 5000,
-            searchMode: 'off',
-          }) as any,
-      );
-
       await chatService.createAssistantMessage({
         messages,
         model: 'test-model',
         provider: 'test-provider',
-        plugins: [],
+        resolvedAgentConfig: createMockResolvedConfig({
+          agentConfig: { model: 'test-model', provider: 'test-provider' },
+          chatConfig: { thinkingBudget: 5000 },
+        }),
       });
 
       expect(getChatCompletionSpy).toHaveBeenCalledWith(