Update agent.py

agent.py

@@ -20,7 +20,11 @@ class Agent:
         prompt: Optional[str] = None,
     ):
         logger.info("Initializing Agent")
-        self.model = InferenceClientModel(model_id="deepseek-ai/DeepSeek-R1",provider="together")
+        # Verify correct parameter name for the inference client (e.g. api_provider)
+        self.model = InferenceClientModel(
+            model_id="deepseek-ai/DeepSeek-R1",
+            api_provider="together"  # Changed parameter to api_provider
+        )
         self.tools = tools
         self.imports = [
             "pandas",
@@ -41,35 +45,26 @@ class Agent:
             add_base_tools=True,
             additional_authorized_imports=self.imports,
         )
+        # Improved prompt with clearer context handling instructions
         self.prompt = prompt or (
-            """
-            You are an advanced AI assistant specialized in solving complex, real-world tasks that require multi-step reasoning, factual accuracy, and use of external tools.
-        
-            Follow these principles:
-            - Be precise and concise. The final answer must strictly match the required format with no extra commentary.
-            - Use tools intelligently. If a question involves external information, structured data, images, or audio, call the appropriate tool to retrieve or process it.
-            - Reason step-by-step. Think through the solution logically and plan your actions carefully before answering.
-            - Validate information. Always verify facts when possible instead of guessing.
-            - Use code if needed. For calculations, parsing, or transformations, generate Python code and execute it. But be careful, some questions contains time-consuming tasks, so you should be careful with the code you run. Better analyze the question and think about the best way to solve it.
-            IMPORTANT: When giving the final answer, output only the direct required result without any extra text like "Final Answer:" or explanations. YOU MUST RESPOND IN THE EXACT FORMAT AS THE QUESTION.
+            """You are an advanced AI assistant specialized in solving complex tasks using tools.
+
+            Key Instructions:
+            1. ALWAYS use tools for file paths in the context:
+               - Use read_file for text files
+               - Use extract_text_from_image for images (.png, .jpg)
+            2. Final answer must ONLY contain the requested result in the EXACT format needed.
+            
             QUESTION: {question}
-            CONTEXT: {context}
-            ANSWER:
-            """
+            {context}
+            ANSWER:"""
         )
         logger.info("Agent initialized")
 
     def __call__(self, question: str, file_path: Optional[str] = None) -> str:
-        """
-        Run the agent to answer a question, optionally using a file as context.
-        Args:
-            question (str): The question to answer.
-            file_path (Optional[str]): Path to a file to use as context (if any).
-        Returns:
-            str: The agent's answer as a string.
-        """
+        # Handle context more cleanly
+        context = f"CONTEXT: File available at: {file_path}" if file_path else ""
         answer = self.agent.run(
-            self.prompt.format(question=question, context=file_path)
+            self.prompt.format(question=question, context=context)
         )
-        answer = str(answer).strip("'").strip('"').strip()
-        return answer
+        return str(answer).strip("'").strip('"').strip()