From 78be193fe4e5aa0289d4374a373c48699b1ab6f4 Mon Sep 17 00:00:00 2001
From: Nicola Di Cicco <93935338+nicoladicicco@users.noreply.github.com>
Date: Thu, 19 Sep 2024 14:10:51 +0900
Subject: [PATCH] Add default models to llm constructor

---
 src/llm.jl | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/llm.jl b/src/llm.jl
index 2ce08da..7ad7d2e 100644
--- a/src/llm.jl
+++ b/src/llm.jl
@@ -14,7 +14,7 @@ struct GroqLLM <: AbstractLLM
     api_key::String
     model_id::String
 
-    function GroqLLM(model_id::String)
+    function GroqLLM(model_id::String = "llama-3.1-8b-instant")
         api_key = get(ENV, "GROQ_API_KEY", "")
         if isempty(api_key)
             error("Environment variable GROQ_API_KEY is not set")
@@ -33,7 +33,7 @@ struct GoogleLLM <: AbstractLLM
     api_key::String
     model_id::String
 
-    function GoogleLLM(model_id::String)
+    function GoogleLLM(model_id::String = "gemini-1.5-pro")
         api_key = get(ENV, "GOOGLE_API_KEY", "")
         if isempty(api_key)
             error("Environment variable GOOGLE_API_KEY is not set")