Refactor imports in main_v2.py to include SmartSearchTool, enhance code organization in model_factory.py by removing unnecessary whitespace, and improve formatting in test_questions.py for better readability and consistency.

main_v2.py

@@ -12,8 +12,8 @@ from phoenix.otel import register
 from smolagents import CodeAgent, LiteLLMModel
 from smolagents.monitoring import LogLevel
 
-from tools.smart_search.tool import SmartSearchTool
 from model_factory import ModelFactory
+from tools.smart_search.tool import SmartSearchTool
 
 _disable_debugging()
 

model_factory.py

@@ -1,24 +1,26 @@
 import os
+
 from dotenv import find_dotenv, load_dotenv
 from smolagents import LiteLLMModel, TransformersModel
 
+
 class ModelFactory:
     @staticmethod
     def create_model():
         """
         Creates and returns a LiteLLMModel instance configured with environment variables.
-        
+
         Returns:
             LiteLLMModel: A configured instance of LiteLLMModel
         """
         # Load environment variables
         load_dotenv(find_dotenv())
-        
+
         # Get configuration from environment variables
         api_base = os.getenv("API_BASE")
         api_key = os.getenv("API_KEY")
         model_id = os.getenv("MODEL_ID")
-        
+
         # Create and return the model
         # return LiteLLMModel(
         #     api_base=api_base,

test_questions.py

@@ -1,17 +1,20 @@
 import unittest
+
 import requests
+
 from main_v2 import main
 
+
 class TestQuestions(unittest.TestCase):
     def setUp(self):
         self.api_url = "https://agents-course-unit4-scoring.hf.space"
         self.questions_url = f"{self.api_url}/questions"
-        
+
         # Get questions from the API
         response = requests.get(self.questions_url, timeout=15)
         response.raise_for_status()
         self.questions = response.json()
-        
+
         # Expected answers for each question
         self.expected_answers = {
             "How many studio albums were published by Mercedes Sosa between 2000 and 2009 (included)? You can use the latest 2022 version of english wikipedia.": "3",
@@ -28,8 +31,9 @@ class TestQuestions(unittest.TestCase):
                 self.assertEqual(
                     actual_answer,
                     expected_answer,
-                    f"Question: {question}\nExpected: {expected_answer}\nGot: {actual_answer}"
+                    f"Question: {question}\nExpected: {expected_answer}\nGot: {actual_answer}",
                 )
 
+
 if __name__ == "__main__":
     unittest.main()