| from core.file.file_obj import FileObj | from core.file.file_obj import FileObj | ||||
| from core.model_runtime.entities.message_entities import PromptMessageRole | from core.model_runtime.entities.message_entities import PromptMessageRole | ||||
| from core.model_runtime.entities.model_entities import ModelType | from core.model_runtime.entities.model_entities import ModelType | ||||
| from core.model_runtime.errors.invoke import InvokeAuthorizationError, InvokeError | |||||
| from core.model_runtime.errors.invoke import InvokeAuthorizationError | |||||
| from core.model_runtime.model_providers.__base.large_language_model import LargeLanguageModel | from core.model_runtime.model_providers.__base.large_language_model import LargeLanguageModel | ||||
| from core.prompt.prompt_template import PromptTemplateParser | from core.prompt.prompt_template import PromptTemplateParser | ||||
| from core.provider_manager import ProviderManager | from core.provider_manager import ProviderManager | ||||
| 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: | |||||
| queue_manager.publish_error(e, PublishFrom.APPLICATION_MANAGER) | |||||
| except Exception as e: | except Exception as e: | ||||
| logger.exception("Unknown Error when generating") | logger.exception("Unknown Error when generating") | ||||
| queue_manager.publish_error(e, PublishFrom.APPLICATION_MANAGER) | queue_manager.publish_error(e, PublishFrom.APPLICATION_MANAGER) |