Unverified Commit 3b55dc08 authored by Yeuoly's avatar Yeuoly

fix: linter

parent ad84b996
import json import json
from collections.abc import Generator
from typing import Union
from flask_restful import Resource, reqparse
from flask import Response from flask import Response
from flask.helpers import stream_with_context from flask.helpers import stream_with_context
from flask_restful import Resource, reqparse
from controllers.console.setup import setup_required from controllers.console.setup import setup_required
from controllers.inner_api import api from controllers.inner_api import api
from controllers.inner_api.wraps import inner_api_only from controllers.inner_api.wraps import inner_api_only
from services.completion_service import CompletionService from services.completion_service import CompletionService
from typing import Generator, Union
class EnterpriseModelInvokeLLMApi(Resource): class EnterpriseModelInvokeLLMApi(Resource):
"""Model invoke API for enterprise edition""" """Model invoke API for enterprise edition"""
......
from core.provider_manager import ProviderManager, ProviderModelBundle from collections.abc import Generator
from typing import Optional, Union, cast
from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk, LLMUsage from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk, LLMUsage
from core.model_runtime.entities.message_entities import PromptMessage, PromptMessageTool from core.model_runtime.entities.message_entities import PromptMessage, PromptMessageTool
from core.model_runtime.model_providers.__base.large_language_model import LargeLanguageModel
from core.model_runtime.entities.model_entities import ModelType from core.model_runtime.entities.model_entities import ModelType
from core.model_runtime.model_providers.__base.large_language_model import LargeLanguageModel
from core.model_runtime.utils.encoders import jsonable_encoder from core.model_runtime.utils.encoders import jsonable_encoder
from core.provider_manager import ProviderModelBundle
from events.inner_event import model_was_invoked from events.inner_event import model_was_invoked
from typing import Generator, Union, cast, Optional
class ModelRunner: class ModelRunner:
""" """
......
...@@ -4,23 +4,23 @@ from typing import Any, Union ...@@ -4,23 +4,23 @@ from typing import Any, Union
from sqlalchemy import and_ from sqlalchemy import and_
from core.app_runner.model_runner import ModelRunner
from core.application_manager import ApplicationManager from core.application_manager import ApplicationManager
from core.provider_manager import ProviderManager
from core.errors.error import ModelCurrentlyNotSupportError, ProviderTokenNotInitError, QuotaExceededError
from core.entities.application_entities import InvokeFrom from core.entities.application_entities import InvokeFrom
from core.entities.model_entities import ModelStatus from core.entities.model_entities import ModelStatus
from core.model_runtime.entities.model_entities import ModelType from core.errors.error import ModelCurrentlyNotSupportError, ProviderTokenNotInitError, QuotaExceededError
from core.file.message_file_parser import MessageFileParser from core.file.message_file_parser import MessageFileParser
from core.model_runtime.entities.message_entities import ( from core.model_runtime.entities.message_entities import (
PromptMessage,
UserPromptMessage,
SystemPromptMessage,
AssistantPromptMessage, AssistantPromptMessage,
ToolPromptMessage, PromptMessage,
PromptMessageRole, PromptMessageRole,
PromptMessageTool PromptMessageTool,
SystemPromptMessage,
ToolPromptMessage,
UserPromptMessage,
) )
from core.app_runner.model_runner import ModelRunner from core.model_runtime.entities.model_entities import ModelType
from core.provider_manager import ProviderManager
from extensions.ext_database import db from extensions.ext_database import db
from models.model import Account, App, AppModelConfig, Conversation, EndUser, Message from models.model import Account, App, AppModelConfig, Conversation, EndUser, Message
from services.app_model_config_service import AppModelConfigService from services.app_model_config_service import AppModelConfigService
...@@ -29,6 +29,7 @@ from services.errors.app_model_config import AppModelConfigBrokenError ...@@ -29,6 +29,7 @@ from services.errors.app_model_config import AppModelConfigBrokenError
from services.errors.conversation import ConversationCompletedError, ConversationNotExistsError from services.errors.conversation import ConversationCompletedError, ConversationNotExistsError
from services.errors.message import MessageNotExistsError from services.errors.message import MessageNotExistsError
class CompletionService: class CompletionService:
@classmethod @classmethod
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment