|
|
|
@@ -15,6 +15,7 @@ from zoneinfo import available_timezones |
|
|
|
from flask import Response, current_app, stream_with_context |
|
|
|
from flask_restful import fields |
|
|
|
|
|
|
|
from core.app.features.rate_limiting.rate_limit import RateLimitGenerator |
|
|
|
from extensions.ext_redis import redis_client |
|
|
|
from models.account import Account |
|
|
|
|
|
|
|
@@ -159,7 +160,7 @@ def generate_text_hash(text: str) -> str: |
|
|
|
return sha256(hash_text.encode()).hexdigest() |
|
|
|
|
|
|
|
|
|
|
|
def compact_generate_response(response: Union[dict, Generator]) -> Response: |
|
|
|
def compact_generate_response(response: Union[dict, RateLimitGenerator]) -> Response: |
|
|
|
if isinstance(response, dict): |
|
|
|
return Response(response=json.dumps(response), status=200, mimetype='application/json') |
|
|
|
else: |