✨ feat: support DEFAULT_AGENT_CONFIG env

lobehub · Feb 12, 2024 · e512f81 · e512f81
1 parent 57458d4
commit e512f81
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/src/app/api/config/route.ts b/src/app/api/config/route.ts
@@ -1,6 +1,8 @@
 import { getServerConfig } from '@/config/server';
 import { GlobalServerConfig } from '@/types/settings';
 
+import { parseAgentConfig } from './parseDefaultAgent';
+
 export const runtime = 'edge';
 
 /**
@@ -14,10 +16,14 @@ export const GET = async () => {
     ENABLED_AWS_BEDROCK,
     ENABLED_GOOGLE,
     ENABLE_OAUTH_SSO,
+    DEFAULT_AGENT_CONFIG,
   } = getServerConfig();
 
   const config: GlobalServerConfig = {
     customModelName: CUSTOM_MODELS,
+    defaultAgent: {
+      config: parseAgentConfig(DEFAULT_AGENT_CONFIG),
+    },
     enabledOAuthSSO: ENABLE_OAUTH_SSO,
     languageModel: {
       bedrock: { enabled: ENABLED_AWS_BEDROCK },

diff --git a/src/config/server/app.ts b/src/config/server/app.ts
@@ -14,6 +14,8 @@ declare global {
 
       PLUGINS_INDEX_URL?: string;
       PLUGIN_SETTINGS?: string;
+
+      DEFAULT_AGENT_CONFIG?: string;
     }
   }
 }
@@ -32,6 +34,8 @@ export const getAppConfig = () => {
   return {
     ACCESS_CODES,
 
+    DEFAULT_AGENT_CONFIG: process.env.DEFAULT_AGENT_CONFIG || '',
+
     SHOW_ACCESS_CODE_CONFIG: !!ACCESS_CODES.length,
 
     METADATA_BASE_URL: process.env.METADATA_BASE_URL,