Unverified Commit 3b55dc08 authored by Yeuoly's avatar Yeuoly

fix: linter

parent ad84b996
import json
from collections.abc import Generator
from typing import Union
from flask_restful import Resource, reqparse
from flask import Response
from flask.helpers import stream_with_context
from flask_restful import Resource, reqparse
from controllers.console.setup import setup_required
from controllers.inner_api import api
from controllers.inner_api.wraps import inner_api_only
from services.completion_service import CompletionService
from typing import Generator, Union
class EnterpriseModelInvokeLLMApi(Resource):
"""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.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.model_providers.__base.large_language_model import LargeLanguageModel
from core.model_runtime.utils.encoders import jsonable_encoder
from core.provider_manager import ProviderModelBundle
from events.inner_event import model_was_invoked
from typing import Generator, Union, cast, Optional
class ModelRunner:
"""
......
......@@ -4,23 +4,23 @@ from typing import Any, Union
from sqlalchemy import and_
from core.app_runner.model_runner import ModelRunner
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.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.model_runtime.entities.message_entities import (
PromptMessage,
UserPromptMessage,
SystemPromptMessage,
AssistantPromptMessage,
ToolPromptMessage,
PromptMessage,
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 models.model import Account, App, AppModelConfig, Conversation, EndUser, Message
from services.app_model_config_service import AppModelConfigService
......@@ -29,6 +29,7 @@ from services.errors.app_model_config import AppModelConfigBrokenError
from services.errors.conversation import ConversationCompletedError, ConversationNotExistsError
from services.errors.message import MessageNotExistsError
class CompletionService:
@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