diff --git a/src/Client.php b/src/Client.php index c4a3544..433d409 100644 --- a/src/Client.php +++ b/src/Client.php @@ -71,6 +71,7 @@ public function geminiPro10(): GenerativeModel { return $this->generativeModel(ModelName::GeminiPro10); } + public function geminiPro10Latest(): GenerativeModel { return $this->generativeModel(ModelName::GeminiPro10Latest); @@ -86,6 +87,10 @@ public function geminiProFlash1_5(): GenerativeModel return $this->generativeModel(ModelName::GeminiPro15Flash); } + public function geminiProFlash1_5_8B(): GenerativeModel + { + return $this->generativeModel(ModelName::GeminiPro15Flash8B); + } public function generativeModel(ModelName $modelName): GenerativeModel { diff --git a/src/Enums/ModelName.php b/src/Enums/ModelName.php index 5103ae2..17b7e50 100644 --- a/src/Enums/ModelName.php +++ b/src/Enums/ModelName.php @@ -12,6 +12,7 @@ enum ModelName: string case GeminiPro10Latest = 'models/gemini-1.0-pro-latest'; case GeminiPro15 = 'models/gemini-1.5-pro'; case GeminiPro15Flash = 'models/gemini-1.5-flash'; + case GeminiPro15Flash8B = 'models/gemini-1.5-flash-8b'; case GeminiProVision = 'models/gemini-pro-vision'; case Embedding = 'models/embedding-001'; case AQA = 'models/aqa';