google AI llms

Author: neagualexa

src/agent/agent.py

@@ -29,9 +29,9 @@ class State(TypedDict):
 class BaseAgent:
     def __init__(self):
         # Main chat LLM — change to GoogleAILLMs(), AzureLLMs(), or OllamaLLMs() if preferred
-        self.llm = OpenAILLMs().get_llm()
+        self.llm = GoogleAILLMs().get_llm()
         # Summarisation LLM — can be set to a different/cheaper model than the chat LLM
-        self.summarisation_llm = OpenAILLMs().get_llm()
+        self.summarisation_llm = GoogleAILLMs().get_llm()
 
         self.max_messages_to_summarize = 11
         self.role_prompt = role_prompt