Update OpenAIService.kt (#12)

bump model increase maxRokens
Marc-R2 · Jun 8, 2024 · f5ed130 · f5ed130
1 parent b8e4e97
commit f5ed130
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/main/kotlin/com/github/blarc/ai/commits/intellij/plugin/OpenAIService.kt b/src/main/kotlin/com/github/blarc/ai/commits/intellij/plugin/OpenAIService.kt
@@ -17,7 +17,7 @@ import com.knuddels.jtokkit.api.EncodingType
 class OpenAIService {
 
     companion object {
-        const val model = "gpt-3.5-turbo"
+        const val model = "gpt-4o"
         val instance: OpenAIService
             get() = ApplicationManager.getApplication().getService(OpenAIService::class.java)
     }
@@ -49,7 +49,7 @@ class OpenAIService {
             topP = 1.0,
             frequencyPenalty = 0.0,
             presencePenalty = 0.0,
-            maxTokens = 200,
+            maxTokens = 2048,
             n = completions
         )
 
@@ -66,4 +66,4 @@ class OpenAIService {
         val encoding = registry.getEncoding(EncodingType.CL100K_BASE)
         return encoding.countTokens(prompt) > 4000
     }
-}
+}