| from core.app.apps.base_app_queue_manager import AppQueueManager | from core.app.apps.base_app_queue_manager import AppQueueManager | ||||
| from core.app.entities.app_invoke_entities import InvokeFrom | from core.app.entities.app_invoke_entities import InvokeFrom | ||||
| from core.errors.error import ( | from core.errors.error import ( | ||||
| AppInvokeQuotaExceededError, | |||||
| ModelCurrentlyNotSupportError, | ModelCurrentlyNotSupportError, | ||||
| ProviderTokenNotInitError, | ProviderTokenNotInitError, | ||||
| QuotaExceededError, | QuotaExceededError, | ||||
| raise ProviderModelCurrentlyNotSupportError() | raise ProviderModelCurrentlyNotSupportError() | ||||
| except InvokeError as e: | except InvokeError as e: | ||||
| raise CompletionRequestError(e.description) | raise CompletionRequestError(e.description) | ||||
| except (ValueError, AppInvokeQuotaExceededError) as e: | |||||
| except ValueError as e: | |||||
| raise e | raise e | ||||
| except Exception as e: | except Exception as e: | ||||
| logging.exception("internal server error.") | logging.exception("internal server error.") | ||||
| raise InvokeRateLimitHttpError(ex.description) | raise InvokeRateLimitHttpError(ex.description) | ||||
| except InvokeError as e: | except InvokeError as e: | ||||
| raise CompletionRequestError(e.description) | raise CompletionRequestError(e.description) | ||||
| except (ValueError, AppInvokeQuotaExceededError) as e: | |||||
| except ValueError as e: | |||||
| raise e | raise e | ||||
| except Exception as e: | except Exception as e: | ||||
| logging.exception("internal server error.") | logging.exception("internal server error.") |
| from controllers.console.explore.wraps import InstalledAppResource | from controllers.console.explore.wraps import InstalledAppResource | ||||
| from core.app.apps.base_app_queue_manager import AppQueueManager | from core.app.apps.base_app_queue_manager import AppQueueManager | ||||
| from core.app.entities.app_invoke_entities import InvokeFrom | from core.app.entities.app_invoke_entities import InvokeFrom | ||||
| from core.errors.error import ModelCurrentlyNotSupportError, ProviderTokenNotInitError, QuotaExceededError | |||||
| from core.errors.error import ( | |||||
| ModelCurrentlyNotSupportError, | |||||
| ProviderTokenNotInitError, | |||||
| QuotaExceededError, | |||||
| ) | |||||
| from core.model_runtime.errors.invoke import InvokeError | from core.model_runtime.errors.invoke import InvokeError | ||||
| from extensions.ext_database import db | from extensions.ext_database import db | ||||
| from libs import helper | from libs import helper |
| from controllers.console.explore.wraps import InstalledAppResource | from controllers.console.explore.wraps import InstalledAppResource | ||||
| from core.app.apps.base_app_queue_manager import AppQueueManager | from core.app.apps.base_app_queue_manager import AppQueueManager | ||||
| from core.app.entities.app_invoke_entities import InvokeFrom | from core.app.entities.app_invoke_entities import InvokeFrom | ||||
| from core.errors.error import ModelCurrentlyNotSupportError, ProviderTokenNotInitError, QuotaExceededError | |||||
| from core.errors.error import ( | |||||
| ModelCurrentlyNotSupportError, | |||||
| ProviderTokenNotInitError, | |||||
| QuotaExceededError, | |||||
| ) | |||||
| from core.model_runtime.errors.invoke import InvokeError | from core.model_runtime.errors.invoke import InvokeError | ||||
| from libs import helper | from libs import helper | ||||
| from libs.login import current_user | from libs.login import current_user |
| from core.app.apps.base_app_queue_manager import AppQueueManager | from core.app.apps.base_app_queue_manager import AppQueueManager | ||||
| from core.app.entities.app_invoke_entities import InvokeFrom | from core.app.entities.app_invoke_entities import InvokeFrom | ||||
| from core.errors.error import ( | from core.errors.error import ( | ||||
| AppInvokeQuotaExceededError, | |||||
| ModelCurrentlyNotSupportError, | ModelCurrentlyNotSupportError, | ||||
| ProviderTokenNotInitError, | ProviderTokenNotInitError, | ||||
| QuotaExceededError, | QuotaExceededError, | ||||
| raise ProviderModelCurrentlyNotSupportError() | raise ProviderModelCurrentlyNotSupportError() | ||||
| except InvokeError as e: | except InvokeError as e: | ||||
| raise CompletionRequestError(e.description) | raise CompletionRequestError(e.description) | ||||
| except (ValueError, AppInvokeQuotaExceededError) as e: | |||||
| except ValueError as e: | |||||
| raise e | raise e | ||||
| except Exception as e: | except Exception as e: | ||||
| logging.exception("internal server error.") | logging.exception("internal server error.") | ||||
| raise ProviderModelCurrentlyNotSupportError() | raise ProviderModelCurrentlyNotSupportError() | ||||
| except InvokeError as e: | except InvokeError as e: | ||||
| raise CompletionRequestError(e.description) | raise CompletionRequestError(e.description) | ||||
| except (ValueError, AppInvokeQuotaExceededError) as e: | |||||
| except ValueError as e: | |||||
| raise e | raise e | ||||
| except Exception as e: | except Exception as e: | ||||
| logging.exception("internal server error.") | logging.exception("internal server error.") |
| from core.app.apps.base_app_queue_manager import AppQueueManager | from core.app.apps.base_app_queue_manager import AppQueueManager | ||||
| from core.app.entities.app_invoke_entities import InvokeFrom | from core.app.entities.app_invoke_entities import InvokeFrom | ||||
| from core.errors.error import ( | from core.errors.error import ( | ||||
| AppInvokeQuotaExceededError, | |||||
| ModelCurrentlyNotSupportError, | ModelCurrentlyNotSupportError, | ||||
| ProviderTokenNotInitError, | ProviderTokenNotInitError, | ||||
| QuotaExceededError, | QuotaExceededError, | ||||
| raise ProviderModelCurrentlyNotSupportError() | raise ProviderModelCurrentlyNotSupportError() | ||||
| except InvokeError as e: | except InvokeError as e: | ||||
| raise CompletionRequestError(e.description) | raise CompletionRequestError(e.description) | ||||
| except (ValueError, AppInvokeQuotaExceededError) as e: | |||||
| except ValueError as e: | |||||
| raise e | raise e | ||||
| except Exception as e: | except Exception as e: | ||||
| logging.exception("internal server error.") | logging.exception("internal server error.") |
| from controllers.web.wraps import WebApiResource | from controllers.web.wraps import WebApiResource | ||||
| from core.app.apps.base_app_queue_manager import AppQueueManager | from core.app.apps.base_app_queue_manager import AppQueueManager | ||||
| from core.app.entities.app_invoke_entities import InvokeFrom | from core.app.entities.app_invoke_entities import InvokeFrom | ||||
| from core.errors.error import ModelCurrentlyNotSupportError, ProviderTokenNotInitError, QuotaExceededError | |||||
| from core.errors.error import ( | |||||
| ModelCurrentlyNotSupportError, | |||||
| ProviderTokenNotInitError, | |||||
| QuotaExceededError, | |||||
| ) | |||||
| from core.model_runtime.errors.invoke import InvokeError | from core.model_runtime.errors.invoke import InvokeError | ||||
| from libs import helper | from libs import helper | ||||
| from libs.helper import uuid_value | from libs.helper import uuid_value |
| from controllers.web.wraps import WebApiResource | from controllers.web.wraps import WebApiResource | ||||
| from core.app.apps.base_app_queue_manager import AppQueueManager | from core.app.apps.base_app_queue_manager import AppQueueManager | ||||
| from core.app.entities.app_invoke_entities import InvokeFrom | from core.app.entities.app_invoke_entities import InvokeFrom | ||||
| from core.errors.error import ModelCurrentlyNotSupportError, ProviderTokenNotInitError, QuotaExceededError | |||||
| from core.errors.error import ( | |||||
| ModelCurrentlyNotSupportError, | |||||
| ProviderTokenNotInitError, | |||||
| QuotaExceededError, | |||||
| ) | |||||
| from core.model_runtime.errors.invoke import InvokeError | from core.model_runtime.errors.invoke import InvokeError | ||||
| from libs import helper | from libs import helper | ||||
| from models.model import App, AppMode, EndUser | from models.model import App, AppMode, EndUser |
| from core.app.apps.message_based_app_queue_manager import MessageBasedAppQueueManager | from core.app.apps.message_based_app_queue_manager import MessageBasedAppQueueManager | ||||
| from core.app.entities.app_invoke_entities import AdvancedChatAppGenerateEntity, InvokeFrom | from core.app.entities.app_invoke_entities import AdvancedChatAppGenerateEntity, InvokeFrom | ||||
| from core.app.entities.task_entities import ChatbotAppBlockingResponse, ChatbotAppStreamResponse | from core.app.entities.task_entities import ChatbotAppBlockingResponse, ChatbotAppStreamResponse | ||||
| from core.model_runtime.errors.invoke import InvokeAuthorizationError, InvokeError | |||||
| from core.model_runtime.errors.invoke import InvokeAuthorizationError | |||||
| from core.ops.ops_trace_manager import TraceQueueManager | from core.ops.ops_trace_manager import TraceQueueManager | ||||
| from core.prompt.utils.get_thread_messages_length import get_thread_messages_length | from core.prompt.utils.get_thread_messages_length import get_thread_messages_length | ||||
| from extensions.ext_database import db | from extensions.ext_database import db | ||||
| except ValidationError as e: | except ValidationError as e: | ||||
| logger.exception("Validation Error when generating") | logger.exception("Validation Error when generating") | ||||
| queue_manager.publish_error(e, PublishFrom.APPLICATION_MANAGER) | queue_manager.publish_error(e, PublishFrom.APPLICATION_MANAGER) | ||||
| except (ValueError, InvokeError) as e: | |||||
| except ValueError as e: | |||||
| if dify_config.DEBUG: | if dify_config.DEBUG: | ||||
| logger.exception("Error when generating") | logger.exception("Error when generating") | ||||
| queue_manager.publish_error(e, PublishFrom.APPLICATION_MANAGER) | queue_manager.publish_error(e, PublishFrom.APPLICATION_MANAGER) |
| from core.app.apps.message_based_app_generator import MessageBasedAppGenerator | from core.app.apps.message_based_app_generator import MessageBasedAppGenerator | ||||
| from core.app.apps.message_based_app_queue_manager import MessageBasedAppQueueManager | from core.app.apps.message_based_app_queue_manager import MessageBasedAppQueueManager | ||||
| from core.app.entities.app_invoke_entities import AgentChatAppGenerateEntity, InvokeFrom | from core.app.entities.app_invoke_entities import AgentChatAppGenerateEntity, InvokeFrom | ||||
| from core.model_runtime.errors.invoke import InvokeAuthorizationError, InvokeError | |||||
| from core.model_runtime.errors.invoke import InvokeAuthorizationError | |||||
| from core.ops.ops_trace_manager import TraceQueueManager | from core.ops.ops_trace_manager import TraceQueueManager | ||||
| from extensions.ext_database import db | from extensions.ext_database import db | ||||
| from factories import file_factory | from factories import file_factory | ||||
| except ValidationError as e: | except ValidationError as e: | ||||
| logger.exception("Validation Error when generating") | logger.exception("Validation Error when generating") | ||||
| queue_manager.publish_error(e, PublishFrom.APPLICATION_MANAGER) | queue_manager.publish_error(e, PublishFrom.APPLICATION_MANAGER) | ||||
| except (ValueError, InvokeError) as e: | |||||
| except ValueError as e: | |||||
| if dify_config.DEBUG: | if dify_config.DEBUG: | ||||
| logger.exception("Error when generating") | logger.exception("Error when generating") | ||||
| queue_manager.publish_error(e, PublishFrom.APPLICATION_MANAGER) | queue_manager.publish_error(e, PublishFrom.APPLICATION_MANAGER) |
| from core.app.apps.message_based_app_generator import MessageBasedAppGenerator | from core.app.apps.message_based_app_generator import MessageBasedAppGenerator | ||||
| from core.app.apps.message_based_app_queue_manager import MessageBasedAppQueueManager | from core.app.apps.message_based_app_queue_manager import MessageBasedAppQueueManager | ||||
| from core.app.entities.app_invoke_entities import ChatAppGenerateEntity, InvokeFrom | from core.app.entities.app_invoke_entities import ChatAppGenerateEntity, InvokeFrom | ||||
| from core.model_runtime.errors.invoke import InvokeAuthorizationError, InvokeError | |||||
| from core.model_runtime.errors.invoke import InvokeAuthorizationError | |||||
| from core.ops.ops_trace_manager import TraceQueueManager | from core.ops.ops_trace_manager import TraceQueueManager | ||||
| from extensions.ext_database import db | from extensions.ext_database import db | ||||
| from factories import file_factory | from factories import file_factory | ||||
| except ValidationError as e: | except ValidationError as e: | ||||
| logger.exception("Validation Error when generating") | logger.exception("Validation Error when generating") | ||||
| queue_manager.publish_error(e, PublishFrom.APPLICATION_MANAGER) | queue_manager.publish_error(e, PublishFrom.APPLICATION_MANAGER) | ||||
| except (ValueError, InvokeError) as e: | |||||
| except ValueError as e: | |||||
| if dify_config.DEBUG: | if dify_config.DEBUG: | ||||
| logger.exception("Error when generating") | logger.exception("Error when generating") | ||||
| queue_manager.publish_error(e, PublishFrom.APPLICATION_MANAGER) | queue_manager.publish_error(e, PublishFrom.APPLICATION_MANAGER) |
| from core.app.apps.message_based_app_generator import MessageBasedAppGenerator | from core.app.apps.message_based_app_generator import MessageBasedAppGenerator | ||||
| from core.app.apps.message_based_app_queue_manager import MessageBasedAppQueueManager | from core.app.apps.message_based_app_queue_manager import MessageBasedAppQueueManager | ||||
| from core.app.entities.app_invoke_entities import CompletionAppGenerateEntity, InvokeFrom | from core.app.entities.app_invoke_entities import CompletionAppGenerateEntity, InvokeFrom | ||||
| from core.model_runtime.errors.invoke import InvokeAuthorizationError, InvokeError | |||||
| from core.model_runtime.errors.invoke import InvokeAuthorizationError | |||||
| from core.ops.ops_trace_manager import TraceQueueManager | from core.ops.ops_trace_manager import TraceQueueManager | ||||
| from extensions.ext_database import db | from extensions.ext_database import db | ||||
| from factories import file_factory | from factories import file_factory | ||||
| except ValidationError as e: | except ValidationError as e: | ||||
| logger.exception("Validation Error when generating") | logger.exception("Validation Error when generating") | ||||
| queue_manager.publish_error(e, PublishFrom.APPLICATION_MANAGER) | queue_manager.publish_error(e, PublishFrom.APPLICATION_MANAGER) | ||||
| except (ValueError, InvokeError) as e: | |||||
| except ValueError as e: | |||||
| if dify_config.DEBUG: | if dify_config.DEBUG: | ||||
| logger.exception("Error when generating") | logger.exception("Error when generating") | ||||
| queue_manager.publish_error(e, PublishFrom.APPLICATION_MANAGER) | queue_manager.publish_error(e, PublishFrom.APPLICATION_MANAGER) |
| from core.app.apps.workflow.generate_task_pipeline import WorkflowAppGenerateTaskPipeline | from core.app.apps.workflow.generate_task_pipeline import WorkflowAppGenerateTaskPipeline | ||||
| from core.app.entities.app_invoke_entities import InvokeFrom, WorkflowAppGenerateEntity | from core.app.entities.app_invoke_entities import InvokeFrom, WorkflowAppGenerateEntity | ||||
| from core.app.entities.task_entities import WorkflowAppBlockingResponse, WorkflowAppStreamResponse | from core.app.entities.task_entities import WorkflowAppBlockingResponse, WorkflowAppStreamResponse | ||||
| from core.model_runtime.errors.invoke import InvokeAuthorizationError, InvokeError | |||||
| from core.model_runtime.errors.invoke import InvokeAuthorizationError | |||||
| from core.ops.ops_trace_manager import TraceQueueManager | from core.ops.ops_trace_manager import TraceQueueManager | ||||
| from extensions.ext_database import db | from extensions.ext_database import db | ||||
| from factories import file_factory | from factories import file_factory | ||||
| except ValidationError as e: | except ValidationError as e: | ||||
| logger.exception("Validation Error when generating") | logger.exception("Validation Error when generating") | ||||
| queue_manager.publish_error(e, PublishFrom.APPLICATION_MANAGER) | queue_manager.publish_error(e, PublishFrom.APPLICATION_MANAGER) | ||||
| except (ValueError, InvokeError) as e: | |||||
| except ValueError as e: | |||||
| if dify_config.DEBUG: | if dify_config.DEBUG: | ||||
| logger.exception("Error when generating") | logger.exception("Error when generating") | ||||
| queue_manager.publish_error(e, PublishFrom.APPLICATION_MANAGER) | queue_manager.publish_error(e, PublishFrom.APPLICATION_MANAGER) |