Merge pull request #81 from Dogtiti/feautre/endpoint-url

feat: endpoint url
ElricLiu · Apr 27, 2023 · 536eab5 · 536eab5 · vercel · Apr 27, 2023
2 parents 0a40c0e + 30ee1fb
commit 536eab5
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 6 deletions.
diff --git a/next-i18next.config.js b/next-i18next.config.js
@@ -8,7 +8,7 @@ module.exports = {
     typeof window === "undefined"
       ? require("path").resolve("./public/locales")
       : "/locales",
-  debug: process.env.NODE_ENV === "development",
+  // debug: process.env.NODE_ENV === "development",
   reloadOnPrerender: process.env.NODE_ENV === "development",
   ns: ["common", "help", "settings", "chat"],
 };
diff --git a/src/components/AutonomousAgent.ts b/src/components/AutonomousAgent.ts
@@ -149,9 +149,10 @@ class AutonomousAgent {
 
   async getInitialTasks(): Promise<string[]> {
     if (this.shouldRunClientSide()) {
-      if (!env.NEXT_PUBLIC_FF_MOCK_MODE_ENABLED) {
-        await testConnection(this.modelSettings);
-      }
+      //FIXME
+      // if (!env.NEXT_PUBLIC_FF_MOCK_MODE_ENABLED) {
+      //   await testConnection(this.modelSettings);
+      // }
       return await AgentService.startGoalAgent(this.modelSettings, this.goal);
     }
 

diff --git a/src/env/schema.mjs b/src/env/schema.mjs
@@ -46,6 +46,7 @@ export const serverSchema = z.object({
   NEXTAUTH_SECRET: requiredAuthEnabledForProduction(),
   NEXTAUTH_URL: validateNextUrl(),
   OPENAI_API_KEY: z.string(),
+  OPENAI_API_BASE_URL: z.string().url().optional(),
   GOOGLE_CLIENT_ID: requiredAuthEnabledForProduction(),
   GOOGLE_CLIENT_SECRET: requiredAuthEnabledForProduction(),
   GITHUB_CLIENT_ID: requiredAuthEnabledForProduction(),
@@ -73,6 +74,7 @@ export const serverEnv = {
   NEXTAUTH_SECRET: process.env.NEXTAUTH_SECRET,
   NEXTAUTH_URL: process.env.NEXTAUTH_URL ?? "http://localhost:3000",
   OPENAI_API_KEY: process.env.OPENAI_API_KEY,
+  OPENAI_API_BASE_URL: process.env.OPENAI_API_BASE_URL,
   GOOGLE_CLIENT_ID: process.env.GOOGLE_CLIENT_ID,
   GOOGLE_CLIENT_SECRET: process.env.GOOGLE_CLIENT_SECRET,
   GITHUB_CLIENT_ID: process.env.GITHUB_CLIENT_ID,

diff --git a/src/utils/prompts.ts b/src/utils/prompts.ts
@@ -9,12 +9,23 @@ export const createModel = (settings: ModelSettings) => {
     _settings = undefined;
   }
 
-  return new OpenAI({
+  const options = {
     openAIApiKey: _settings?.customApiKey || process.env.OPENAI_API_KEY,
     temperature: _settings?.customTemperature || 0.9,
     modelName: _settings?.customModelName || GPT_35_TURBO,
     maxTokens: _settings?.maxTokens || 400,
-  });
+  };
+
+  const baseOptions = {
+    basePath: process.env.OPENAI_API_BASE_URL,
+  };
+  console.log(
+    "Dogtiti ~ file: prompts.ts:22 ~ createModel ~ options:",
+    options,
+    baseOptions
+  );
+
+  return new OpenAI(options, baseOptions);
 };
 
 export const startGoalPrompt = new PromptTemplate({