fix: tti model (#2060)

Author: shaohuzhang1

apps/setting/models_provider/impl/aliyun_bai_lian_model_provider/model/llm.py

@@ -1,6 +1,6 @@
 #!/usr/bin/env python
 # -*- coding: UTF-8 -*-
-from typing import List, Dict
+from typing import Dict
 
 from setting.models_provider.base_model_provider import MaxKBBaseModel
 from setting.models_provider.impl.base_chat_open_ai import BaseChatOpenAI

apps/setting/models_provider/impl/aws_bedrock_model_provider/model/llm.py

@@ -1,8 +1,10 @@
-from typing import List, Dict
 import os
 import re
+from typing import Dict
+
 from botocore.config import Config
 from langchain_community.chat_models import BedrockChat
+
 from setting.models_provider.base_model_provider import MaxKBBaseModel
 
 

apps/setting/models_provider/impl/azure_model_provider/model/azure_chat_model.py

@@ -7,13 +7,10 @@
     @desc:
 """
 
-from typing import List, Dict, Optional, Any, Iterator, Type
+from typing import List, Dict
 
-from langchain_core.callbacks import CallbackManagerForLLMRun
-from langchain_core.messages import BaseMessage, get_buffer_string, BaseMessageChunk, AIMessageChunk
-from langchain_core.outputs import ChatGenerationChunk
+from langchain_core.messages import BaseMessage, get_buffer_string
 from langchain_openai import AzureChatOpenAI
-from langchain_openai.chat_models.base import _convert_delta_to_message_chunk
 
 from common.config.tokenizer_manage_config import TokenizerManage
 from setting.models_provider.base_model_provider import MaxKBBaseModel

apps/setting/models_provider/impl/azure_model_provider/model/stt.py

@@ -1,8 +1,7 @@
-import asyncio
 import io
 from typing import Dict
 
-from openai import OpenAI, AzureOpenAI
+from openai import AzureOpenAI
 
 from common.config.tokenizer_manage_config import TokenizerManage
 from setting.models_provider.base_model_provider import MaxKBBaseModel
@@ -61,4 +60,3 @@ def speech_to_text(self, audio_file):
         buffer.name = "file.mp3"  # this is the important line
         res = client.audio.transcriptions.create(model=self.model, language="zh", file=buffer)
         return res.text
-

apps/setting/models_provider/impl/base_chat_open_ai.py

@@ -1,6 +1,7 @@
 # coding=utf-8
 
-from typing import List, Dict, Optional, Any, Iterator, Type, cast
+from typing import List, Dict, Optional, Any, Iterator, cast
+
 from langchain_core.language_models import LanguageModelInput
 from langchain_core.messages import BaseMessage, get_buffer_string
 from langchain_core.outputs import ChatGenerationChunk, ChatGeneration

apps/setting/models_provider/impl/deepseek_model_provider/model/llm.py

@@ -6,7 +6,7 @@
 @Author  ：Brian Yang
 @Date    ：5/12/24 7:44 AM 
 """
-from typing import List, Dict
+from typing import Dict
 
 from setting.models_provider.base_model_provider import MaxKBBaseModel
 from setting.models_provider.impl.base_chat_open_ai import BaseChatOpenAI
@@ -29,4 +29,3 @@ def new_instance(model_type, model_name, model_credential: Dict[str, object], **
             **optional_params
         )
         return deepseek_chat_open_ai
-

apps/setting/models_provider/impl/gemini_model_provider/model/llm.py

@@ -11,14 +11,14 @@
 from google.ai.generativelanguage_v1 import GenerateContentResponse
 from google.generativeai.responder import ToolDict
 from google.generativeai.types import FunctionDeclarationType, SafetySettingDict
+from google.generativeai.types import Tool as GoogleTool
 from langchain_core.callbacks import CallbackManagerForLLMRun
-from langchain_core.messages import BaseMessage, get_buffer_string
+from langchain_core.messages import BaseMessage
 from langchain_core.outputs import ChatGenerationChunk
 from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain_google_genai._function_utils import _ToolConfigDict
 from langchain_google_genai.chat_models import _chat_with_retry, _response_to_result
-from google.generativeai.types import Tool as GoogleTool
-from common.config.tokenizer_manage_config import TokenizerManage
+
 from setting.models_provider.base_model_provider import MaxKBBaseModel
 
 

apps/setting/models_provider/impl/kimi_model_provider/model/llm.py

@@ -6,7 +6,7 @@
     @date：2023/11/10 17:45
     @desc:
 """
-from typing import List, Dict
+from typing import Dict
 
 from setting.models_provider.base_model_provider import MaxKBBaseModel
 from setting.models_provider.impl.base_chat_open_ai import BaseChatOpenAI

apps/setting/models_provider/impl/openai_model_provider/model/llm.py

@@ -6,11 +6,9 @@
     @date：2024/4/18 15:28
     @desc:
 """
-from typing import List, Dict, Optional, Any
+from typing import List, Dict
 
-from langchain_core.language_models import LanguageModelInput
 from langchain_core.messages import BaseMessage, get_buffer_string
-from langchain_core.runnables import RunnableConfig
 from langchain_openai.chat_models import ChatOpenAI
 
 from common.config.tokenizer_manage_config import TokenizerManage

apps/setting/models_provider/impl/qwen_model_provider/model/llm.py

@@ -12,11 +12,10 @@
 from langchain_community.llms.tongyi import generate_with_last_element_mark
 from langchain_core.callbacks import CallbackManagerForLLMRun
 from langchain_core.language_models import LanguageModelInput
-from langchain_core.messages import BaseMessage, get_buffer_string
+from langchain_core.messages import BaseMessage
 from langchain_core.outputs import ChatGenerationChunk, ChatGeneration
 from langchain_core.runnables import RunnableConfig, ensure_config
 
-from common.config.tokenizer_manage_config import TokenizerManage
 from setting.models_provider.base_model_provider import MaxKBBaseModel