This commit is contained in:
xds
2026-02-18 17:01:06 +03:00
parent ecc88aca62
commit 7488665d04
2 changed files with 11 additions and 13 deletions

View File

@@ -178,12 +178,11 @@ async def import_external_generation(
Import a generation from an external source. Import a generation from an external source.
Requires server-to-server authentication via HMAC signature. Requires server-to-server authentication via HMAC signature.
""" """
import os import os
from utils.external_auth import verify_signature from utils.external_auth import verify_signature
from api.models.ExternalGenerationDTO import ExternalGenerationRequest from api.models import ExternalGenerationRequest
logger.info("import_external_generation called")
logger.info("import_external_generation called")
# Get raw request body for signature verification # Get raw request body for signature verification
body = await request.body() body = await request.body()

View File

@@ -1,25 +1,25 @@
import asyncio import asyncio
import base64
import logging import logging
import random import random
import base64
from datetime import datetime, UTC from datetime import datetime, UTC
from typing import List, Optional, Tuple, Any, Dict from typing import List, Optional, Tuple, Any, Dict
from io import BytesIO
from uuid import uuid4 from uuid import uuid4
import httpx
import httpx
from aiogram import Bot from aiogram import Bot
from aiogram.types import BufferedInputFile from aiogram.types import BufferedInputFile
from adapters.Exception import GoogleGenerationException from adapters.Exception import GoogleGenerationException
from adapters.google_adapter import GoogleAdapter from adapters.google_adapter import GoogleAdapter
from api.models import GenerationRequest, GenerationResponse, GenerationsResponse, GenerationGroupResponse from adapters.s3_adapter import S3Adapter
from api.models import FinancialReport, UsageStats, UsageByEntity from api.models import FinancialReport, UsageStats, UsageByEntity
from api.models import GenerationRequest, GenerationResponse, GenerationsResponse, GenerationGroupResponse
# Импортируйте ваши модели DAO, Asset, Generation корректно # Импортируйте ваши модели DAO, Asset, Generation корректно
from models.Asset import Asset, AssetType, AssetContentType from models.Asset import Asset, AssetType, AssetContentType
from models.Generation import Generation, GenerationStatus from models.Generation import Generation, GenerationStatus
from models.enums import AspectRatios, Quality, GenType from models.enums import AspectRatios, Quality
from repos.dao import DAO from repos.dao import DAO
from adapters.s3_adapter import S3Adapter
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@@ -378,8 +378,7 @@ class GenerationService:
Returns: Returns:
Created Generation object Created Generation object
""" """
from api.models.ExternalGenerationDTO import ExternalGenerationRequest
# Validate image source # Validate image source
external_gen.validate_image_source() external_gen.validate_image_source()