oib/aitbc
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: replace SessionDep with explicit Annotated[Session, Depends(get_session)] across all routers

Browse Source 
- Replace SessionDep type alias with explicit Annotated[Session, Depends(get_session)]
- Add missing imports for Session, Annotated, and Depends types
- Update all endpoint function signatures to use explicit dependency annotation
- Remove redundant `= Depends()` default values from session parameters
- Update docstrings and comments to reference new annotation pattern
- Apply changes consistently across all router
This commit is contained in:
oib
2026-03-07 15:45:11 +01:00
parent 93aae0edb3
commit 89e161c906
75 changed files with 371 additions and 372 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
apps/coordinator-api/src/app/services/adaptive_learning.py
View File

@@ -14,7 +14,7 @@ from enum import Enum
import numpy as np
import json
from ..storage import Annotated[Session, Depends(get_session)], get_session
from ..storage import get_session
from ..domain import AIAgentWorkflow, AgentExecution, AgentStatus
logger = get_logger(__name__)
@@ -390,7 +390,7 @@ class ReinforcementLearningAgent:
class AdaptiveLearningService:
"""Service for adaptive learning systems"""
def __init__(self, session: Annotated[Session, Depends(get_session)] = Depends()):
def __init__(self, session: Annotated[Session, Depends(get_session)]):
self.session = session
self.learning_agents = {}
self.environments = {}

10
apps/coordinator-api/src/app/services/adaptive_learning_app.py
View File

@@ -8,7 +8,7 @@ from fastapi import FastAPI, Depends
from fastapi.middleware.cors import CORSMiddleware
from .adaptive_learning import AdaptiveLearningService, LearningAlgorithm, RewardType
from ..storage import Annotated[Session, Depends(get_session)], get_session
from ..storage import get_session
from ..routers.adaptive_learning_health import router as health_router
app = FastAPI(
@@ -36,7 +36,7 @@ async def health():
async def create_learning_environment(
environment_id: str,
config: dict,
session: Annotated[Session, Depends(get_session)] = Depends() = None
session: Annotated[Session, Depends(get_session)] = None
):
"""Create safe learning environment"""
service = AdaptiveLearningService(session)
@@ -51,7 +51,7 @@ async def create_learning_agent(
agent_id: str,
algorithm: str,
config: dict,
session: Annotated[Session, Depends(get_session)] = Depends() = None
session: Annotated[Session, Depends(get_session)] = None
):
"""Create reinforcement learning agent"""
service = AdaptiveLearningService(session)
@@ -67,7 +67,7 @@ async def train_agent(
agent_id: str,
environment_id: str,
training_config: dict,
session: Annotated[Session, Depends(get_session)] = Depends() = None
session: Annotated[Session, Depends(get_session)] = None
):
"""Train agent in environment"""
service = AdaptiveLearningService(session)
@@ -81,7 +81,7 @@ async def train_agent(
@app.get("/agent-performance/{agent_id}")
async def get_agent_performance(
agent_id: str,
session: Annotated[Session, Depends(get_session)] = Depends() = None
session: Annotated[Session, Depends(get_session)] = None
):
"""Get agent performance metrics"""
service = AdaptiveLearningService(session)

2
apps/coordinator-api/src/app/services/advanced_ai_service.py
View File

@@ -23,7 +23,7 @@ from .advanced_reinforcement_learning import AdvancedReinforcementLearningEngine
from .multi_modal_fusion import MultiModalFusionEngine
from .gpu_multimodal import GPUAcceleratedMultiModal
from .advanced_learning import AdvancedLearningService
from ..storage import Annotated[Session, Depends(get_session)], get_session
from ..storage import get_session
logger = get_logger(__name__)

4
apps/coordinator-api/src/app/services/edge_gpu_service.py
View File

@@ -5,11 +5,11 @@ from typing import List, Optional
from sqlmodel import select
from ..domain.gpu_marketplace import ConsumerGPUProfile, GPUArchitecture, EdgeGPUMetrics
from ..data.consumer_gpu_profiles import CONSUMER_GPU_PROFILES
from ..storage import Annotated[Session, Depends(get_session)], get_session
from ..storage import get_session
class EdgeGPUService:
def __init__(self, session: Annotated[Session, Depends(get_session)] = Depends()):
def __init__(self, session: Annotated[Session, Depends(get_session)]):
self.session = session
def list_profiles(

4
apps/coordinator-api/src/app/services/gpu_multimodal.py
View File

@@ -17,7 +17,7 @@ import numpy as np
from datetime import datetime
import time
from ..storage import Annotated[Session, Depends(get_session)], get_session
from ..storage import get_session
from .multimodal_agent import ModalityType, ProcessingMode
logger = get_logger(__name__)
@@ -289,7 +289,7 @@ class GPUAttentionOptimizer:
class GPUAcceleratedMultiModal:
"""GPU-accelerated multi-modal processing with enhanced CUDA optimization"""
def __init__(self, session: Annotated[Session, Depends(get_session)] = Depends()):
def __init__(self, session: Annotated[Session, Depends(get_session)]):
self.session = session
self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
self._cuda_available = self._check_cuda_availability()

4
apps/coordinator-api/src/app/services/gpu_multimodal_app.py
View File

@@ -8,7 +8,7 @@ from fastapi import FastAPI, Depends
from fastapi.middleware.cors import CORSMiddleware
from .gpu_multimodal import GPUAcceleratedMultiModal
from ..storage import Annotated[Session, Depends(get_session)], get_session
from ..storage import get_session
from ..routers.gpu_multimodal_health import router as health_router
app = FastAPI(
@@ -36,7 +36,7 @@ async def health():
async def cross_modal_attention(
modality_features: dict,
attention_config: dict = None,
session: Annotated[Session, Depends(get_session)] = Depends() = None
session: Annotated[Session, Depends(get_session)] = None
):
"""GPU-accelerated cross-modal attention"""
service = GPUAcceleratedMultiModal(session)

2
apps/coordinator-api/src/app/services/memory_manager.py
View File

@@ -15,7 +15,7 @@ from enum import Enum
import json
from .ipfs_storage_service import IPFSStorageService, MemoryMetadata, IPFSUploadResult
from ..storage import Annotated[Session, Depends(get_session)], get_session
from ..storage import get_session
logger = get_logger(__name__)

10
apps/coordinator-api/src/app/services/modality_optimization.py
View File

@@ -13,7 +13,7 @@ from datetime import datetime
from enum import Enum
import numpy as np
from ..storage import Annotated[Session, Depends(get_session)], get_session
from ..storage import get_session
from .multimodal_agent import ModalityType
logger = get_logger(__name__)
@@ -30,7 +30,7 @@ class OptimizationStrategy(str, Enum):
class ModalityOptimizer:
"""Base class for modality-specific optimizers"""
def __init__(self, session: Annotated[Session, Depends(get_session)] = Depends()):
def __init__(self, session: Annotated[Session, Depends(get_session)]):
self.session = session
self._performance_history = {}
@@ -64,7 +64,7 @@ class ModalityOptimizer:
class TextOptimizer(ModalityOptimizer):
"""Text processing optimization strategies"""
def __init__(self, session: Annotated[Session, Depends(get_session)] = Depends()):
def __init__(self, session: Annotated[Session, Depends(get_session)]):
super().__init__(session)
self._token_cache = {}
self._embedding_cache = {}
@@ -318,7 +318,7 @@ class TextOptimizer(ModalityOptimizer):
class ImageOptimizer(ModalityOptimizer):
"""Image processing optimization strategies"""
def __init__(self, session: Annotated[Session, Depends(get_session)] = Depends()):
def __init__(self, session: Annotated[Session, Depends(get_session)]):
super().__init__(session)
self._feature_cache = {}
@@ -862,7 +862,7 @@ class VideoOptimizer(ModalityOptimizer):
class ModalityOptimizationManager:
"""Manager for all modality-specific optimizers"""
def __init__(self, session: Annotated[Session, Depends(get_session)] = Depends()):
def __init__(self, session: Annotated[Session, Depends(get_session)]):
self.session = session
self._optimizers = {
ModalityType.TEXT: TextOptimizer(session),

6
apps/coordinator-api/src/app/services/modality_optimization_app.py
View File

@@ -8,7 +8,7 @@ from fastapi import FastAPI, Depends
from fastapi.middleware.cors import CORSMiddleware
from .modality_optimization import ModalityOptimizationManager, OptimizationStrategy, ModalityType
from ..storage import Annotated[Session, Depends(get_session)], get_session
from ..storage import get_session
from ..routers.modality_optimization_health import router as health_router
app = FastAPI(
@@ -37,7 +37,7 @@ async def optimize_modality(
modality: str,
data: dict,
strategy: str = "balanced",
session: Annotated[Session, Depends(get_session)] = Depends() = None
session: Annotated[Session, Depends(get_session)] = None
):
"""Optimize specific modality"""
manager = ModalityOptimizationManager(session)
@@ -52,7 +52,7 @@ async def optimize_modality(
async def optimize_multimodal(
multimodal_data: dict,
strategy: str = "balanced",
session: Annotated[Session, Depends(get_session)] = Depends() = None
session: Annotated[Session, Depends(get_session)] = None
):
"""Optimize multiple modalities"""
manager = ModalityOptimizationManager(session)

4
apps/coordinator-api/src/app/services/multimodal_agent.py
View File

@@ -13,7 +13,7 @@ from datetime import datetime
from enum import Enum
import json
from ..storage import Annotated[Session, Depends(get_session)], get_session
from ..storage import get_session
from ..domain import AIAgentWorkflow, AgentExecution, AgentStatus
logger = get_logger(__name__)
@@ -40,7 +40,7 @@ class ProcessingMode(str, Enum):
class MultiModalAgentService:
"""Service for advanced multi-modal agent capabilities"""
def __init__(self, session: Annotated[Session, Depends(get_session)] = Depends()):
def __init__(self, session: Annotated[Session, Depends(get_session)]):
self.session = session
self._modality_processors = {
ModalityType.TEXT: self._process_text,

4
apps/coordinator-api/src/app/services/multimodal_app.py
View File

@@ -8,7 +8,7 @@ from fastapi import FastAPI, Depends
from fastapi.middleware.cors import CORSMiddleware
from .multimodal_agent import MultiModalAgentService
from ..storage import Annotated[Session, Depends(get_session)], get_session
from ..storage import get_session
from ..routers.multimodal_health import router as health_router
app = FastAPI(
@@ -37,7 +37,7 @@ async def process_multimodal(
agent_id: str,
inputs: dict,
processing_mode: str = "fusion",
session: Annotated[Session, Depends(get_session)] = Depends() = None
session: Annotated[Session, Depends(get_session)] = None
):
"""Process multi-modal input"""
service = MultiModalAgentService(session)

4
apps/coordinator-api/src/app/services/payments.py
View File

@@ -16,7 +16,7 @@ from ..schemas import (
EscrowRelease,
RefundRequest
)
from ..storage import Annotated[Session, Depends(get_session)], get_session
from ..storage import get_session
logger = get_logger(__name__)
@@ -24,7 +24,7 @@ logger = get_logger(__name__)
class PaymentService:
"""Service for handling job payments"""
def __init__(self, session: Annotated[Session, Depends(get_session)] = Depends()):
def __init__(self, session: Annotated[Session, Depends(get_session)]):
self.session = session
self.wallet_base_url = "http://127.0.0.1:20000" # Wallet daemon URL
self.exchange_base_url = "http://127.0.0.1:23000" # Exchange API URL