mirror of
https://github.com/langgenius/dify.git
synced 2026-09-08 11:04:27 +08:00
add interface for review app
This commit is contained in:
parent
e54131e95b
commit
4e5405dce2
@ -116,6 +116,7 @@ class InvokeRateLimitError(BaseHTTPException):
|
|||||||
description = "Rate Limit Error"
|
description = "Rate Limit Error"
|
||||||
code = 429
|
code = 429
|
||||||
|
|
||||||
|
|
||||||
class NeedAddIdsError(BaseHTTPException):
|
class NeedAddIdsError(BaseHTTPException):
|
||||||
error_code = "need_add_ids"
|
error_code = "need_add_ids"
|
||||||
description = "Need to add ids."
|
description = "Need to add ids."
|
||||||
|
|||||||
@ -1,7 +1,8 @@
|
|||||||
import logging
|
import logging
|
||||||
|
from typing import Any, cast
|
||||||
|
|
||||||
from flask import request
|
from flask import request
|
||||||
from flask_restx import Resource, marshal_with, reqparse
|
from flask_restx import Resource, marshal, marshal_with, reqparse
|
||||||
from werkzeug.exceptions import Forbidden, InternalServerError, NotFound
|
from werkzeug.exceptions import Forbidden, InternalServerError, NotFound
|
||||||
|
|
||||||
import services
|
import services
|
||||||
@ -13,12 +14,12 @@ from controllers.console.app.error import (
|
|||||||
AudioTooLargeError,
|
AudioTooLargeError,
|
||||||
CompletionRequestError,
|
CompletionRequestError,
|
||||||
ConversationCompletedError,
|
ConversationCompletedError,
|
||||||
|
NeedAddIdsError,
|
||||||
NoAudioUploadedError,
|
NoAudioUploadedError,
|
||||||
ProviderModelCurrentlyNotSupportError,
|
ProviderModelCurrentlyNotSupportError,
|
||||||
ProviderNotInitializeError,
|
ProviderNotInitializeError,
|
||||||
ProviderNotSupportSpeechToTextError,
|
ProviderNotSupportSpeechToTextError,
|
||||||
ProviderQuotaExceededError,
|
ProviderQuotaExceededError,
|
||||||
NeedAddIdsError,
|
|
||||||
UnsupportedAudioTypeError,
|
UnsupportedAudioTypeError,
|
||||||
)
|
)
|
||||||
from controllers.console.app.wraps import get_app_model_with_trial
|
from controllers.console.app.wraps import get_app_model_with_trial
|
||||||
@ -43,6 +44,7 @@ from core.model_runtime.errors.invoke import InvokeError
|
|||||||
from core.workflow.graph_engine.manager import GraphEngineManager
|
from core.workflow.graph_engine.manager import GraphEngineManager
|
||||||
from extensions.ext_database import db
|
from extensions.ext_database import db
|
||||||
from fields.app_fields import app_detail_fields_with_site
|
from fields.app_fields import app_detail_fields_with_site
|
||||||
|
from fields.dataset_fields import dataset_fields
|
||||||
from fields.workflow_fields import workflow_fields
|
from fields.workflow_fields import workflow_fields
|
||||||
from libs import helper
|
from libs import helper
|
||||||
from libs.helper import uuid_value
|
from libs.helper import uuid_value
|
||||||
@ -50,9 +52,11 @@ from libs.login import current_user
|
|||||||
from models import Account
|
from models import Account
|
||||||
from models.account import TenantStatus
|
from models.account import TenantStatus
|
||||||
from models.model import AppMode, Site
|
from models.model import AppMode, Site
|
||||||
|
from models.workflow import Workflow
|
||||||
from services.app_generate_service import AppGenerateService
|
from services.app_generate_service import AppGenerateService
|
||||||
from services.app_service import AppService
|
from services.app_service import AppService
|
||||||
from services.audio_service import AudioService
|
from services.audio_service import AudioService
|
||||||
|
from services.dataset_service import DatasetService
|
||||||
from services.errors.audio import (
|
from services.errors.audio import (
|
||||||
AudioTooLargeServiceError,
|
AudioTooLargeServiceError,
|
||||||
NoAudioUploadedServiceError,
|
NoAudioUploadedServiceError,
|
||||||
@ -67,11 +71,6 @@ from services.errors.message import (
|
|||||||
)
|
)
|
||||||
from services.message_service import MessageService
|
from services.message_service import MessageService
|
||||||
from services.recommended_app_service import RecommendedAppService
|
from services.recommended_app_service import RecommendedAppService
|
||||||
from models.workflow import Workflow
|
|
||||||
from services.dataset_service import DatasetService
|
|
||||||
from fields.dataset_fields import dataset_fields
|
|
||||||
from flask_restx import marshal
|
|
||||||
from typing import Any, cast
|
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
@ -449,6 +448,7 @@ class AppApi(Resource):
|
|||||||
|
|
||||||
return app_model
|
return app_model
|
||||||
|
|
||||||
|
|
||||||
class AppWorkflowApi(Resource):
|
class AppWorkflowApi(Resource):
|
||||||
@trial_feature_enable
|
@trial_feature_enable
|
||||||
@get_app_model_with_trial
|
@get_app_model_with_trial
|
||||||
@ -467,6 +467,7 @@ class AppWorkflowApi(Resource):
|
|||||||
)
|
)
|
||||||
return workflow
|
return workflow
|
||||||
|
|
||||||
|
|
||||||
class DatasetListApi(Resource):
|
class DatasetListApi(Resource):
|
||||||
def get(self, app_model):
|
def get(self, app_model):
|
||||||
page = request.args.get("page", default=1, type=int)
|
page = request.args.get("page", default=1, type=int)
|
||||||
@ -479,12 +480,12 @@ class DatasetListApi(Resource):
|
|||||||
else:
|
else:
|
||||||
raise NeedAddIdsError()
|
raise NeedAddIdsError()
|
||||||
|
|
||||||
|
|
||||||
data = cast(list[dict[str, Any]], marshal(datasets, dataset_fields))
|
data = cast(list[dict[str, Any]], marshal(datasets, dataset_fields))
|
||||||
|
|
||||||
response = {"data": data, "has_more": len(datasets) == limit, "limit": limit, "total": total, "page": page}
|
response = {"data": data, "has_more": len(datasets) == limit, "limit": limit, "total": total, "page": page}
|
||||||
return response
|
return response
|
||||||
|
|
||||||
|
|
||||||
api.add_resource(TrialChatApi, "/trial-apps/<uuid:app_id>/chat-messages", endpoint="trial_app_chat_completion")
|
api.add_resource(TrialChatApi, "/trial-apps/<uuid:app_id>/chat-messages", endpoint="trial_app_chat_completion")
|
||||||
|
|
||||||
api.add_resource(
|
api.add_resource(
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user