fix(api/services/app_generate_service.py): Remove wrong type hints. (#6535)

This commit is contained in:
-LAN- 2024-07-22 22:58:07 +08:00 committed by GitHub
parent 71a7211411
commit 617847e3c0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3 additions and 2 deletions

View File

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

View File

@ -21,7 +21,7 @@ class AppGenerateService:
args: Any, args: Any,
invoke_from: InvokeFrom, invoke_from: InvokeFrom,
streaming: bool = True, streaming: bool = True,
) -> Union[dict, Generator[dict, None, None]]: ):
""" """
App Content Generate App Content Generate
:param app_model: app model :param app_model: app model