Merge pull request #4 from kromiii/model-as-env

Specify GPT model as an environment variable
pyama86 · Jun 9, 2024 · bad4a3e · bad4a3e
2 parents c71a0e8 + fe3ae06
commit bad4a3e
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/manifests/deployment.yml b/manifests/deployment.yml
@@ -45,6 +45,8 @@ spec:
               secretKeyRef:
                 name: slack-gpt
                 key: openai-secret
+          - name: GPT_MODEL
+            value: gpt-4o
           - name: LANG
             value: C.UTF-8
           - name: BOT_USER_ID

diff --git a/src/mention.ts b/src/mention.ts
@@ -18,7 +18,7 @@ export const appMention: any = async ({ event, client, say }) => {
     }
 
     const nonNullable = <T>(value: T): value is NonNullable<T> => value != null
-    let model = 'gpt-4o'
+    let model = process.env.GPT_MODEL || 'gpt-4-turbo'
     let max_tokens = null
     const threadMessages = await Promise.all(
       replies.messages.map(async (message) => {
@@ -38,6 +38,7 @@ export const appMention: any = async ({ event, client, say }) => {
               }
 
               if (encodedImage) {
+                model = 'gpt-4-vision-preview'
                 max_tokens = 4096
                 contents.push(
                   {

diff --git a/src/utils.ts b/src/utils.ts
@@ -10,7 +10,7 @@ if (!apiKey) {
 }
 const openai = new OpenAI({ apiKey })
 
-const GPT_MODEL = 'gpt-4-1106-preview'
+const GPT_MODEL = process.env.GPT_MODEL || 'gpt-4-turbo'
 
 const GPT_MAX_TOKEN = 128000