Update app.py

app.py

@@ -30,9 +30,9 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 from smolagents import Tool, CodeAgent, DuckDuckGoSearchTool, FinalAnswerTool, HfApiModel
 
 
-class QwenChatTool(Tool):
-    name = "qwen_chat_tool"
-    description = "Solves reasoning/code questions using Qwen1.5-7B-Chat"
+class CodeLlamaTool(Tool):
+    name = "code_llama_tool"
+    description = "Solves reasoning/code questions using Meta Code Llama 7B Instruct"
 
     inputs = {
         "question": {
@@ -43,7 +43,7 @@ class QwenChatTool(Tool):
     output_type = "string"
 
     def __init__(self):
-        self.model_id = "Qwen/Qwen1.5-7B-Chat"
+        self.model_id = "codellama/CodeLlama-7b-Instruct-hf"
         token = os.getenv("HF_TOKEN")
 
         self.tokenizer = AutoTokenizer.from_pretrained(self.model_id, token=token)
@@ -113,6 +113,14 @@ Answer:"""
             return f"Error executing code:\n{traceback.format_exc()}"
 
 
+from transformers.tools import (
+    DocumentQuestionAnsweringTool,
+    ImageQuestionAnsweringTool,
+    TranslationTool,
+    PythonInterpreterTool
+)
+
+
 #from smolagents import Tool
 #from langchain_community.document_loaders import WikipediaLoader
 
@@ -270,12 +278,13 @@ class VideoTranscriptionTool(Tool):
 class BasicAgent:
     def __init__(self):
         token = os.environ.get("HF_API_TOKEN")
+
         model = HfApiModel(
             temperature=0.1,
             token=token
         )
 
-        # Existing tools
+        # Your other tools
         search_tool = DuckDuckGoSearchTool()
         wiki_search_tool = WikiSearchTool()
         str_reverse_tool = StringReverseTool()
@@ -284,20 +293,21 @@ class BasicAgent:
         visit_webpage_tool = VisitWebpageTool()
         final_answer_tool = FinalAnswerTool()
         video_transcription_tool = VideoTranscriptionTool()
-
-        # ✅ New Llama Tool
-        qwen_chat_tool = QwenChatTool()
-
-        system_prompt = f"""
-You are my general AI assistant. Your task is to answer the question I asked.
-First, provide an explanation of your reasoning, step by step, to arrive at the answer.
-Then, return your final answer in a single line, formatted as follows: "FINAL ANSWER: [YOUR FINAL ANSWER]".
-[YOUR FINAL ANSWER] should be a number, a string, or a comma-separated list of numbers and/or strings, depending on the question.
-If the answer is a number, do not use commas or units (e.g., $, %) unless specified.
-If the answer is a string, do not use articles or abbreviations (e.g., for cities), and write digits in plain text unless specified.
-If the answer is a comma-separated list, apply the above rules for each element based on whether it is a number or a string.
+        code_llama_tool = CodeLlamaTool()
+        arxiv_search_tool = ArxivSearchTool()
+
+        # ✅ Add Hugging Face default tools
+        doc_qa_tool = DocumentQuestionAnsweringTool()
+        image_qa_tool = ImageQuestionAnsweringTool()
+        translation_tool = TranslationTool()
+        python_tool = PythonInterpreterTool()
+
+        system_prompt = """You are my general AI assistant...
+Always return your final result in the format:
+"FINAL ANSWER: [your short answer here]"
 """
 
+
         self.agent = CodeAgent(
             model=model,
             tools=[
@@ -305,7 +315,10 @@ If the answer is a comma-separated list, apply the above rules for each element
                 keywords_extract_tool, speech_to_text_tool,
                 visit_webpage_tool, final_answer_tool,
                 parse_excel_to_json, video_transcription_tool,
-                qwen_chat_tool  # 🔧 Add here
+                arxiv_search_tool,
+                doc_qa_tool, image_qa_tool,
+                translation_tool, python_tool 
+                code_llama_tool  # 🔧 Add here
             ],
             add_base_tools=True
         )