Fix remaining deprecated get_database() calls

aisbf/context.py

@@ -27,7 +27,7 @@ from typing import Dict, List, Optional, Union, Any
 from .utils import count_messages_tokens
 from .config import config
 from .providers import get_provider_handler
-from .database import get_database
+from .database import DatabaseRegistry
 
 
 class ContextManager:

aisbf/providers/base.py

@@ -28,7 +28,7 @@ from typing import Dict, List, Optional, Union, Any
 from ..models import Provider, Model, ErrorTracking
 from ..config import config
 from ..utils import count_messages_tokens
-from ..database import get_database
+from ..database import DatabaseRegistry
 from ..batching import get_request_batcher
 
 # Check if debug mode is enabled

aisbf/providers/claude.py

@@ -124,8 +124,8 @@ class ClaudeProviderHandler(BaseProviderHandler):
         
         # Regular user: ONLY use database credentials, NO file fallback
         try:
-            from ..database import get_database
-            db = get_database()
+            from ..database import DatabaseRegistry
+            db = DatabaseRegistry.get_config_database()
             if db:
                 db_creds = db.get_user_oauth2_credentials(
                     user_id=self.user_id,

aisbf/providers/codex.py

@@ -129,8 +129,8 @@ class CodexProviderHandler(BaseProviderHandler):
         
         # Regular user: ONLY use database credentials, NO file fallback
         try:
-            from ..database import get_database
-            db = get_database()
+            from ..database import DatabaseRegistry
+            db = DatabaseRegistry.get_config_database()
             if db:
                 db_creds = db.get_user_oauth2_credentials(
                     user_id=self.user_id,
@@ -199,8 +199,8 @@ class CodexProviderHandler(BaseProviderHandler):
                 # Save to configuration file or database
                 if self.user_id is not None:
                     # User-specific provider: update in database
-                    from ..database import get_database
-                    db = get_database()
+                    from ..database import DatabaseRegistry
+                    db = DatabaseRegistry.get_config_database()
                     if db:
                         # Update user provider endpoint in database
                         db.update_user_provider_endpoint(

aisbf/providers/kilo.py

@@ -186,8 +186,8 @@ class KiloProviderHandler(BaseProviderHandler):
             return
         
         try:
-            from ..database import get_database
-            db = get_database()
+            from ..database import DatabaseRegistry
+            db = DatabaseRegistry.get_config_database()
             if db:
                 db.save_user_oauth2_credentials(
                     user_id=self.user_id,

aisbf/providers/qwen.py

@@ -120,8 +120,8 @@ class QwenProviderHandler(BaseProviderHandler):
         
         # Regular user: ONLY use database credentials, NO file fallback
         try:
-            from ..database import get_database
-            db = get_database()
+            from ..database import DatabaseRegistry
+            db = DatabaseRegistry.get_config_database()
             if db:
                 db_creds = db.get_user_oauth2_credentials(
                     user_id=self.user_id,
@@ -161,8 +161,8 @@ class QwenProviderHandler(BaseProviderHandler):
             return
         
         try:
-            from ..database import get_database
-            db = get_database()
+            from ..database import DatabaseRegistry
+            db = DatabaseRegistry.get_config_database()
             if db:
                 db.save_user_oauth2_credentials(
                     user_id=self.user_id,