From b4677e0fd6b4b151c2f17d7793c44246b7bc65be Mon Sep 17 00:00:00 2001 From: fit2cloud-chenyw Date: Wed, 28 May 2025 15:31:15 +0800 Subject: [PATCH] perf: Remove unnecessary dependencies --- backend/apps/chat/schemas/chat_base_schema.py | 10 +++++----- backend/pyproject.toml | 1 - 2 files changed, 5 insertions(+), 6 deletions(-) diff --git a/backend/apps/chat/schemas/chat_base_schema.py b/backend/apps/chat/schemas/chat_base_schema.py index 5f0fcd72..691bd010 100644 --- a/backend/apps/chat/schemas/chat_base_schema.py +++ b/backend/apps/chat/schemas/chat_base_schema.py @@ -3,7 +3,7 @@ from typing import Optional, Dict, Any, Type from abc import ABC, abstractmethod from langchain_core.language_models import BaseLLM as LangchainBaseLLM from langchain_openai import ChatOpenAI -from langchain_community.llms import Tongyi, VLLM +# from langchain_community.llms import Tongyi, VLLM class LLMConfig(BaseModel): """Base configuration class for large language models""" @@ -43,7 +43,7 @@ class OpenAILLM(BaseLLM): def generate(self, prompt: str) -> str: return self.llm.invoke(prompt) -class TongyiLLM(BaseLLM): +""" class TongyiLLM(BaseLLM): def _init_llm(self) -> LangchainBaseLLM: return Tongyi( model_name=self.config.model_name, @@ -62,7 +62,7 @@ class VLLMLLM(BaseLLM): ) def generate(self, prompt: str) -> str: - return self.llm.invoke(prompt) + return self.llm.invoke(prompt) """ class LLMFactory: @@ -70,8 +70,8 @@ class LLMFactory: _llm_types: Dict[str, Type[BaseLLM]] = { "openai": OpenAILLM, - "tongyi": TongyiLLM, - "vllm": VLLMLLM + "tongyi": OpenAILLM, + "vllm": OpenAILLM } @classmethod diff --git a/backend/pyproject.toml b/backend/pyproject.toml index 6392348b..192d20b3 100644 --- a/backend/pyproject.toml +++ b/backend/pyproject.toml @@ -24,7 +24,6 @@ dependencies = [ "langchain-openai>=0.3,<0.4", "langchain-community>=0.3,<0.4", "langgraph>=0.3,<0.4", - "vllm>=0.8.5", "dashscope>=1.14.0,<2.0.0", "pymysql (>=1.1.1,<2.0.0)", "cryptography (>=44.0.3,<45.0.0)",