Merge pull request #641 from xiaodong2008/feat/base-url-option

transitive-bullshit · Jun 7, 2024 · 7bc821d · 7bc821d
2 parents 73fb30b + 9593159
commit 7bc821d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/bin/cli.js b/bin/cli.js
@@ -35,13 +35,15 @@ async function main() {
     .option('-m, --model <model>', 'Model (gpt-3.5-turbo, gpt-4)', {
       default: 'gpt-3.5-turbo'
     })
+    .option('--host <host>', 'API host base')
     .option(
       '-n, --conversationName <conversationName>',
       'Unique name for the conversation'
     )
     .action(async (prompt, options) => {
       const apiOrg = options.apiOrg || process.env.OPENAI_API_ORG
       const apiKey = options.apiKey || process.env.OPENAI_API_KEY
+      const apiBaseUrl = options.host || process.env.OPENAI_API_BASE_URL
       if (!apiKey) {
         console.error('error: either set OPENAI_API_KEY or use --apiKey\n')
         cli.outputHelp()
@@ -74,6 +76,7 @@ async function main() {
       const api = new ChatGPTAPI({
         apiKey,
         apiOrg,
+        apiBaseUrl,
         debug: options.debug,
         completionParams: {
           model