Fix testing crashed tests

Author: Einswilli

forklet/core/orchestrator.py

@@ -24,6 +24,7 @@
 from .progress_tracker import ProgressTracker
 from .state_controller import StateController
 
+from forklet.infrastructure import RateLimitInfo
 from forklet.infrastructure.logger import logger
 
 

forklet/infrastructure/__init__.py

@@ -1,15 +1,16 @@
 from .error_handler import (
-    DownloadError, RateLimitError, RateLimitError,
+    DownloadError, RateLimitError,
     AuthenticationError, RepositoryNotFoundError,
     handle_api_error, retry_on_error
 )
 
-from .rate_limiter import RateLimiter
+from .rate_limiter import RateLimiter, RateLimitInfo
 from .retry_manager import RetryManager
+from .cache_manager import CacheManager, CacheEntry
 
 __all__ = [
-    DownloadError, RateLimitError, RateLimitError,
+    DownloadError, RateLimitError,
     AuthenticationError, RepositoryNotFoundError,
     handle_api_error, retry_on_error, RateLimiter,
-    RetryManager
+    RetryManager, RateLimitInfo, CacheManager, CacheEntry
 ]

forklet/models/download.py

@@ -11,7 +11,7 @@
 from datetime import datetime
 from enum import Enum
 from pathlib import Path
-from typing import Dict, List, Optional, Set
+from typing import Dict, List, Optional, Set, Any
 
 from .github import RepositoryInfo, GitReference
 

forklet/services/github_api.py

@@ -2,22 +2,26 @@
 Service for interacting with GitHub API with rate limiting and error handling.
 """
 
-from typing import List, Optional, Dict, Any, AsyncIterator
-
+from typing import List, Optional, Dict, Any, AsyncIterator, Callable
+import datetime
 import asyncio
 import httpx
 from github import Github, GithubException
 # from github.Repository import Repository as GithubRepository
 
-from ..infrastructure.rate_limiter import RateLimiter
-from ..infrastructure.retry_manager import RetryManager
-from ..infrastructure.error_handler import (
+# from ..infrastructure.rate_limiter import RateLimiter
+# from ..infrastructure.retry_manager import RetryManager
+from ..infrastructure import (
     handle_api_error,
     RateLimitError,
     RepositoryNotFoundError,
     DownloadError,
+    RateLimiter,
+    RetryManager,
+    CacheManager,
+    RateLimitInfo
 )
-from ..infrastructure.cache_manager import CacheManager
+# from ..infrastructure.cache_manager import CacheManager
 from ..models import RepositoryInfo, GitReference, RepositoryType, GitHubFile
 from ..models.constants import USER_AGENT
 
@@ -75,19 +79,19 @@ def _on_rate_limit_update(self, rate_limit_info: RateLimitInfo) -> None:
         # Configure HTTP client
         headers = {"Accept": "application/vnd.github.v3+json", "User-Agent": USER_AGENT}
 
-        if auth_token:
-            headers["Authorization"] = f"token {auth_token}"
+        if self.auth_token:
+            headers["Authorization"] = f"token {self.auth_token}"
 
         self.http_client = httpx.AsyncClient(
-            headers=headers, timeout=httpx.Timeout(timeout)
+            headers=headers, timeout=httpx.Timeout(self.timeout)
         )
 
         # Sync client for PyGithub (used only for metadata)
         self.github_client = (
             Github(
-                auth_token, retry=self.retry_manager.max_retries, user_agent=USER_AGENT
+                self.auth_token, retry=self.retry_manager.max_retries, user_agent=USER_AGENT
             )
-            if auth_token
+            if self.auth_token
             else Github(retry=self.retry_manager.max_retries, user_agent=USER_AGENT)
         )
 
@@ -129,19 +133,19 @@ async def update_rate_limit_info(self, headers: Dict[str, str]) -> None:
         # Configure HTTP client
         headers = {"Accept": "application/vnd.github.v3+json", "User-Agent": USER_AGENT}
 
-        if auth_token:
-            headers["Authorization"] = f"token {auth_token}"
+        if self.auth_token:
+            headers["Authorization"] = f"token {self.auth_token}"
 
         self.http_client = httpx.AsyncClient(
-            headers=headers, timeout=httpx.Timeout(timeout)
+            headers=headers, timeout=httpx.Timeout(self.timeout)
         )
 
         # Sync client for PyGithub (used only for metadata)
         self.github_client = (
             Github(
-                auth_token, retry=self.retry_manager.max_retries, user_agent=USER_AGENT
+                self.auth_token, retry=self.retry_manager.max_retries, user_agent=USER_AGENT
             )
-            if auth_token
+            if self.auth_token
             else Github(retry=self.retry_manager.max_retries, user_agent=USER_AGENT)
         )