feat: add more models released on october 24, 2024 (#1)

* feat: add more models * chore: update example in readme * fix: backward compatibility model naming
sheeiavellie · Oct 24, 2024 · 68b6331 · 68b6331
1 parent d7ed942
commit 68b6331
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/models.go b/models.go
@@ -2,10 +2,17 @@ package yandexgpt
 
 import "fmt"
 
-// TODO: add other models
 var (
-	YandexGPTModel     = yandexGPTModel{modelName: "yandexgpt"}
+	// Yandex GPT Pro 3rd generation
+	YandexGPTModel = yandexGPTModel{modelName: "yandexgpt"}
+	// Yandex GPT Lite 3rd generation
 	YandexGPTModelLite = yandexGPTModel{modelName: "yandexgpt-lite"}
+	// Yandex GPT Pro 4th generation
+	YandexGPT4Model = yandexGPTModel{modelName: "yandexgpt/rc"}
+	// Yandex GPT Lite 4th generation
+	YandexGPT4ModelLite = yandexGPTModel{modelName: "yandexgpt-lite/rc"}
+	// Yandex GPT Pro 32k 4th generation
+	YandexGPT4Model32k = yandexGPTModel{modelName: "yandexgpt-32k/rc"}
 )
 
 type yandexGPTModel struct {