google

agent.py

@@ -112,6 +112,24 @@ class BoomBot:
                 max_tokens=8192,
                 temperature=0.7,
             )
+        elif self.provider == "google":
+            meta_model = "google/gemini-2.5-flash"
+            # return OpenAIServerModel(
+            #     model_id=meta_model,
+            #     api_base="https://api.deepinfra.com/v1/openai",
+            #     api_key=os.getenv("DEEPINFRA_API_KEY"),
+            #     flatten_messages_as_text=True,
+            #     max_tokens=8192,
+            #     temperature=0.7,
+            # )
+            return LiteLLMModel(
+                 model_id="deepinfra/"+ meta_model,
+                api_base="https://api.deepinfra.com/v1/openai",
+                api_key=os.getenv("DEEPINFRA_API_KEY"),
+                flatten_messages_as_text=True,
+                max_tokens=8192,
+                temperature=0.7,
+            )
         elif self.provider == "groq":
             # Default to use groq's claude-3-opus or llama-3
             model_id = "claude-3-opus-20240229"
@@ -130,7 +148,7 @@ class BoomBot:
         download_file = DownloadFileFromLinkTool()
         read_file_content = ReadFileContentTool()
         visit_webpage = VisitWebpageTool()
-        transcribe_video = TranscibeVideoFileTool()
+        # transcribe_video = TranscibeVideoFileTool()
         transcribe_audio = TranscribeAudioTool()
         get_wikipedia_info = WikipediaSearchTool()
         web_searcher = DuckDuckGoSearchTool()
@@ -197,7 +215,7 @@ class BoomBot:
         # Create the agent
         agent = CodeAgent(
             tools=agent_tools,
-            max_steps=12,
+            max_steps=15,
             model=self.model,
             add_base_tools=False,
             stream_outputs=True,
@@ -233,7 +251,7 @@ class BoomBot:
             - If the content is long, complex, spans multiple pages, or may be needed later, do NOT rely solely on visit_webpage.  
             - Move quickly to downloading: avoid repeated visits when the content should be archived.
 
-        3. DOWNLOAD (MANDATORY IF CONTENT IS LONG, DENSE, OR CRUCIAL)  
+        3. DOWNLOAD AND ADD TO VECTORSTORE (MANDATORY IF CONTENT IS LONG, DENSE, COMPLEX, MULTIPLE FILES OR LINKS TO VISIT)  
             - Use download_file_from_link on all valuable resources (including html pages or pdfs).
             - Especially when a page is detailed, technical, or multi-part, downloading is preferred.
             - You can (and should) download webpages as HTML. Do this whenever the site might be referenced again later.
@@ -342,7 +360,7 @@ if __name__ == "__main__":
             for row in rows:
                 writer.writerow(row)
 
-    agent = BoomBot(provider="gemma")
+    agent = BoomBot(provider="deepinfra")
     model_name = agent.provider  # e.g. "gemma"
 
     file_online   = load_online_qas(file_path=r"../../Final_Assignment_Template/allqas.jsonl", has_file=True)

app.py

@@ -20,7 +20,7 @@ load_dotenv()
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
-        self.agent = BoomBot(provider="meta")
+        self.agent = BoomBot(provider="google")
 
     def __call__(self, question: str, task_id: str, to_download) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")

tools.py

@@ -289,7 +289,7 @@ class BraveWebSearchTool(Tool):
     output_type = "string"
 
     # api_key = os.getenv("BRAVE_SEARCH_API_KEY")
-    api_key = None
+    api_key = "asdasfd"
     count = 3
     char_limit = 4000  # Adjust based on LLM context window
     tool = BraveSearch.from_api_key(api_key=api_key, search_kwargs={"count": count})