advanced_prompt_template.py 1.0 KB

1234567891011121314151617181920212223242526
  1. from flask_restful import Resource, reqparse
  2. from controllers.console import api
  3. from controllers.console.setup import setup_required
  4. from controllers.console.wraps import account_initialization_required
  5. from libs.login import login_required
  6. from services.advanced_prompt_template_service import AdvancedPromptTemplateService
  7. class AdvancedPromptTemplateList(Resource):
  8. @setup_required
  9. @login_required
  10. @account_initialization_required
  11. def get(self):
  12. parser = reqparse.RequestParser()
  13. parser.add_argument('app_mode', type=str, required=True, location='args')
  14. parser.add_argument('model_mode', type=str, required=True, location='args')
  15. parser.add_argument('has_context', type=str, required=False, default='true', location='args')
  16. parser.add_argument('model_name', type=str, required=True, location='args')
  17. args = parser.parse_args()
  18. return AdvancedPromptTemplateService.get_prompt(args)
  19. api.add_resource(AdvancedPromptTemplateList, '/app/prompt-templates')