fixed tests

Author: Dishant1804

backend/tests/apps/ai/management/commands/ai_update_repository_chunks_test.py

@@ -4,6 +4,7 @@
 
 from apps.ai.common.base.chunk_command import BaseChunkCommand
 from apps.ai.management.commands.ai_update_repository_chunks import Command
+from apps.github.models.repository import Repository
 
 
 @pytest.fixture
@@ -33,8 +34,6 @@ def test_command_help_text(self, command):
 
     def test_model_class_property(self, command):
         """Test the model_class property returns Repository."""
-        from apps.github.models.repository import Repository
-
         assert command.model_class == Repository
 
     def test_entity_name_property(self, command):

backend/tests/apps/ai/management/commands/ai_update_repository_context_test.py

@@ -2,7 +2,9 @@
 
 import pytest
 
+from apps.ai.common.base.context_command import BaseContextCommand
 from apps.ai.management.commands.ai_update_repository_context import Command
+from apps.github.models.repository import Repository
 
 
 @pytest.fixture
@@ -24,8 +26,6 @@ def mock_repository():
 class TestAiUpdateRepositoryContextCommand:
     def test_command_inheritance(self, command):
         """Test that the command inherits from BaseContextCommand."""
-        from apps.ai.common.base.context_command import BaseContextCommand
-
         assert isinstance(command, BaseContextCommand)
 
     def test_command_help_text(self, command):
@@ -34,8 +34,6 @@ def test_command_help_text(self, command):
 
     def test_model_class_property(self, command):
         """Test the model_class property returns Repository."""
-        from apps.github.models.repository import Repository
-
         assert command.model_class == Repository
 
     def test_entity_name_property(self, command):