Update app.py

app.py

@@ -18,7 +18,7 @@ class GeminiLLM(LLM):
 
     api_key: str = os.getenv("GEMINI")
     model_name: str = "gemini-2.0-flash"
-    temperature: float = 0.7
+    temperature: float = 0.1
 
     @property
     def _llm_type(self) -> str:
@@ -36,6 +36,10 @@ class GeminiLLM(LLM):
         }
         data = {
             "contents": [
+                {
+                    "role": "system",
+                    "text": "You are an agent. Please respond concisely."
+                },
                 {
                     "parts": [{"text": prompt}]
                 }
@@ -45,6 +49,7 @@ class GeminiLLM(LLM):
             }
         }
 
+
         response = requests.post(url, headers=headers, json=data)
         if response.status_code == 200:
             result = response.json()
@@ -59,9 +64,6 @@ import os
 
 gemini_llm = GeminiLLM()
 
-prompt = PromptTemplate.from_template("Explica qué es la inteligencia artificial en pocas palabras.")
-chain = LLMChain(prompt=prompt, llm=gemini_llm)
-
 import os
 from math import sqrt
 from typing import Dict, List