Merge pull request #115 from awslabs/fix/package-import-paths

fix: remove hardcoded src prefix from package imports

Author: fgogolli

src/infrastructure/di/components/dependency_resolver.py

@@ -353,9 +353,9 @@ def _resolve_string_annotation(self, annotation: str, context_class: type) -> ty
                 # Last resort: try to import from common locations
                 for module_name in [
                     "typing",
-                    "src.domain",
-                    "src.application",
-                    "src.infrastructure",
+                    "domain",
+                    "application",
+                    "infrastructure",
                 ]:
                     try:
                         module_obj = __import__(module_name, fromlist=[annotation])

src/infrastructure/di/handler_discovery.py

@@ -93,7 +93,7 @@ def _resolve_cache_path_fallback(self) -> str:
         os.makedirs(cache_dir, exist_ok=True)
         return os.path.join(cache_dir, "handler_discovery.json")
 
-    def discover_and_register_handlers(self, base_package: str = "src.application") -> None:
+    def discover_and_register_handlers(self, base_package: str = "application") -> None:
         """
         Discover all handlers and register them with the DI container.
         Uses caching to improve performance on subsequent runs.
@@ -336,7 +336,7 @@ def _register_handlers_from_cache(self, cached_handlers: dict[str, Any]) -> None
     def _fallback_to_full_discovery(self) -> None:
         """Fall back to full discovery if cache loading fails."""
         logger.info("Falling back to full handler discovery")
-        self._discover_handlers("src.application")
+        self._discover_handlers("application")
         self._register_handlers()
 
     def _get_source_file_mtimes(self, base_package: str) -> dict[str, float]: