Small fixes.

tests/test_ai_factory.py

@@ -10,7 +10,7 @@ from chatmastermind.ais.openai import OpenAI
 class TestCreateAI(unittest.TestCase):
     def setUp(self) -> None:
         self.args = MagicMock(spec=argparse.Namespace)
-        self.args.ai = 'default'
+        self.args.AI = 'default'
         self.args.model = None
         self.args.max_tokens = None
         self.args.temperature = None
@@ -18,19 +18,19 @@ class TestCreateAI(unittest.TestCase):
     def test_create_ai_from_args(self) -> None:
         # Create an AI with the default configuration
         config = Config()
-        self.args.ai = 'default'
+        self.args.AI = 'default'
         ai = create_ai(self.args, config)
         self.assertIsInstance(ai, OpenAI)
 
     def test_create_ai_from_default(self) -> None:
-        self.args.ai = None
+        self.args.AI = None
         # Create an AI with the default configuration
         config = Config()
         ai = create_ai(self.args, config)
         self.assertIsInstance(ai, OpenAI)
 
     def test_create_empty_ai_error(self) -> None:
-        self.args.ai = None
+        self.args.AI = None
         # Create Config with empty AIs
         config = Config()
         config.ais = {}
@@ -40,7 +40,7 @@ class TestCreateAI(unittest.TestCase):
 
     def test_create_unsupported_ai_error(self) -> None:
         # Mock argparse.Namespace with ai='invalid_ai'
-        self.args.ai = 'invalid_ai'
+        self.args.AI = 'invalid_ai'
         # Create default Config
         config = Config()
         # Call create_ai function and assert that it raises AIError