refactor apps

This commit is contained in:
takatost
2024-03-02 02:40:18 +08:00
parent 5e38996222
commit 5c7ea08bdf
111 changed files with 1979 additions and 1819 deletions

View File

@ -37,7 +37,7 @@ class ChatMessageAudioApi(Resource):
@setup_required
@login_required
@account_initialization_required
@get_app_model(mode=AppMode.CHAT)
@get_app_model(mode=[AppMode.CHAT, AppMode.AGENT_CHAT])
def post(self, app_model):
file = request.files['file']

View File

@ -22,7 +22,7 @@ from controllers.console.app.wraps import get_app_model
from controllers.console.setup import setup_required
from controllers.console.wraps import account_initialization_required
from core.app.app_queue_manager import AppQueueManager
from core.entities.application_entities import InvokeFrom
from core.app.entities.app_invoke_entities import InvokeFrom
from core.errors.error import ModelCurrentlyNotSupportError, ProviderTokenNotInitError, QuotaExceededError
from core.model_runtime.errors.invoke import InvokeError
from libs.helper import uuid_value
@ -103,7 +103,7 @@ class ChatMessageApi(Resource):
@setup_required
@login_required
@account_initialization_required
@get_app_model(mode=AppMode.CHAT)
@get_app_model(mode=[AppMode.CHAT, AppMode.AGENT_CHAT])
def post(self, app_model):
parser = reqparse.RequestParser()
parser.add_argument('inputs', type=dict, required=True, location='json')
@ -168,7 +168,7 @@ class ChatMessageStopApi(Resource):
@setup_required
@login_required
@account_initialization_required
@get_app_model(mode=AppMode.CHAT)
@get_app_model(mode=[AppMode.CHAT, AppMode.AGENT_CHAT])
def post(self, app_model, task_id):
account = flask_login.current_user

View File

@ -112,7 +112,7 @@ class CompletionConversationDetailApi(Resource):
@setup_required
@login_required
@account_initialization_required
@get_app_model(mode=AppMode.CHAT)
@get_app_model(mode=[AppMode.CHAT, AppMode.AGENT_CHAT])
def delete(self, app_model, conversation_id):
conversation_id = str(conversation_id)
@ -133,7 +133,7 @@ class ChatConversationApi(Resource):
@setup_required
@login_required
@account_initialization_required
@get_app_model(mode=AppMode.CHAT)
@get_app_model(mode=[AppMode.CHAT, AppMode.AGENT_CHAT])
@marshal_with(conversation_with_summary_pagination_fields)
def get(self, app_model):
parser = reqparse.RequestParser()
@ -218,7 +218,7 @@ class ChatConversationDetailApi(Resource):
@setup_required
@login_required
@account_initialization_required
@get_app_model(mode=AppMode.CHAT)
@get_app_model(mode=[AppMode.CHAT, AppMode.AGENT_CHAT])
@marshal_with(conversation_detail_fields)
def get(self, app_model, conversation_id):
conversation_id = str(conversation_id)
@ -227,7 +227,7 @@ class ChatConversationDetailApi(Resource):
@setup_required
@login_required
@get_app_model(mode=AppMode.CHAT)
@get_app_model(mode=[AppMode.CHAT, AppMode.AGENT_CHAT])
@account_initialization_required
def delete(self, app_model, conversation_id):
conversation_id = str(conversation_id)

View File

@ -42,7 +42,7 @@ class ChatMessageListApi(Resource):
@setup_required
@login_required
@get_app_model(mode=AppMode.CHAT)
@get_app_model(mode=[AppMode.CHAT, AppMode.AGENT_CHAT])
@account_initialization_required
@marshal_with(message_infinite_scroll_pagination_fields)
def get(self, app_model):
@ -194,7 +194,7 @@ class MessageSuggestedQuestionApi(Resource):
@setup_required
@login_required
@account_initialization_required
@get_app_model(mode=AppMode.CHAT)
@get_app_model(mode=[AppMode.CHAT, AppMode.AGENT_CHAT])
def get(self, app_model, message_id):
message_id = str(message_id)

View File

@ -203,7 +203,7 @@ class AverageSessionInteractionStatistic(Resource):
@setup_required
@login_required
@account_initialization_required
@get_app_model(mode=AppMode.CHAT)
@get_app_model(mode=[AppMode.CHAT, AppMode.AGENT_CHAT])
def get(self, app_model):
account = current_user

View File

@ -22,7 +22,7 @@ from controllers.console.app.error import (
from controllers.console.explore.error import NotChatAppError, NotCompletionAppError
from controllers.console.explore.wraps import InstalledAppResource
from core.app.app_queue_manager import AppQueueManager
from core.entities.application_entities import InvokeFrom
from core.app.entities.app_invoke_entities import InvokeFrom
from core.errors.error import ModelCurrentlyNotSupportError, ProviderTokenNotInitError, QuotaExceededError
from core.model_runtime.errors.invoke import InvokeError
from extensions.ext_database import db

View File

@ -24,7 +24,7 @@ from controllers.console.explore.error import (
NotCompletionAppError,
)
from controllers.console.explore.wraps import InstalledAppResource
from core.entities.application_entities import InvokeFrom
from core.app.entities.app_invoke_entities import InvokeFrom
from core.errors.error import ModelCurrentlyNotSupportError, ProviderTokenNotInitError, QuotaExceededError
from core.model_runtime.errors.invoke import InvokeError
from fields.message_fields import message_infinite_scroll_pagination_fields