Bladeren bron

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

-LAN- 9 maanden geleden
bovenliggende
commit
617847e3c0
2 gewijzigde bestanden met toevoegingen van 3 en 2 verwijderingen
  1. 2 1
      api/libs/helper.py
  2. 1 1
      api/services/app_generate_service.py

+ 2 - 1
api/libs/helper.py

@@ -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:

+ 1 - 1
api/services/app_generate_service.py

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