Update agents.py

agents.py

@@ -55,14 +55,15 @@ def get_file_content_as_text(task_id: str) -> str:
 
 def load_hf_model(modelName: str):
     """
-    Lädt ein Hugging Face Modell und gibt den Agenten zurück.
-    :param modelName: Name des Modells
+    Loads a model from the hugging face hub
+    :param modelName: Name of the model
     :return: model
     """
     load_dotenv()  # Lädt automatisch .env im Projektordner
 
-    hf_token = os.getenv("hugging_face")
-    login(token=hf_token)  # Authentifizierung bei Hugging Face
+    # for local usage, we might use a hf token to log in
+    # hf_token = os.getenv("hugging_face")
+    # login(token=hf_token)  # Authentifizierung bei Hugging Face
     # Modell initialisieren
     model = HfApiModel(model_id=modelName)
     return model
@@ -70,9 +71,9 @@ def load_hf_model(modelName: str):
 
 def load_ollama_model(modelName: str):
     """
-    Lädt ein Ollama Modell und gibt den Agenten zurück.
-    :param modelName: Name des Modells
-    :return: model
+    Loads the requested model in ollama
+    :param modelName: Name of the model
+    :return: model (via OpenAI compatible API)
     """
     # Modell initialisieren
     model = OpenAIServerModel(model_id=modelName, api_base="http://localhost:11434/v1")
@@ -80,23 +81,23 @@ def load_ollama_model(modelName: str):
 
 def load_lmStudio_model(modelName: str):
     """
-    Lädt ein LM Studio Modell und gibt den Agenten zurück.
-    :param modelName: Name des Modells
-    :return: model 
+    Loads the requested model into lm studio
+    :param modelName: Name of the model
+    :return: model, accessible through the OpenAI compatible API
     """
     # Modell initialisieren
     #model = LiteLLMModel(model_id=modelName, api_base="http://localhost:1234")
     model = OpenAIServerModel(model_id=modelName, api_base="http://localhost:1234/v1")
     return model
 
-def load_gemini_model():
+def load_gemini_model(model_name: str):
     """
-    Lädt ein Gemini Modell und gibt den Agenten zurück.
+    Loads 
     :return: model
     """
     try:
         print(f"Gemini API Key: {os.getenv('GEMINI_API_KEY')}")
-        model = LiteLLMModel(model_id="gemini/gemini-2.0-flash-exp",
+        model = LiteLLMModel(model_id=f"gemini/{model_name}",
                          api_key=os.getenv("GEMINI_API_KEY"))
         #model = GeminiModel(api_key=os.getenv("GEMINI_API_KEY"))
         return model
@@ -115,7 +116,7 @@ def get_agent(model_name:str, model_type:str) -> Optional[CodeAgent]:
         case "Ollama":
             model = load_ollama_model(model_name)
         case "Gemini":
-            model = load_gemini_model()
+            model = load_gemini_model(model_name)
         case "LMStudio":
             model = load_lmStudio_model(model_name)
         case _: