fix: langchain warnings for import from root

chains/llmchain_with_history.py

@@ -1,6 +1,6 @@
 from server.utils import get_ChatOpenAI
 from configs.model_config import LLM_MODEL, TEMPERATURE
-from langchain import LLMChain
+from langchain.chains import LLMChain
 from langchain.prompts.chat import (
     ChatPromptTemplate,
     HumanMessagePromptTemplate,

server/agent/custom_template.py

@@ -45,7 +45,9 @@ Thought:
 
 from langchain.agents import Tool, AgentExecutor, LLMSingleActionAgent, AgentOutputParser
 from langchain.prompts import StringPromptTemplate
-from langchain import OpenAI, SerpAPIWrapper, LLMChain
+from langchain.llms import OpenAI
+from langchain.utilities import SerpAPIWrapper
+from langchain.chains import LLMChain
 from typing import List, Union
 from langchain.schema import AgentAction, AgentFinish, OutputParserException
 from server.agent.tools import tools

server/agent/math.py

@@ -1,4 +1,4 @@
-from langchain import PromptTemplate
+from langchain.prompts import PromptTemplate
 from langchain.chains import LLMMathChain
 from server.utils import wrap_done, get_ChatOpenAI
 from configs.model_config import LLM_MODEL, TEMPERATURE

server/agent/translator.py

@@ -1,4 +1,5 @@
-from langchain import PromptTemplate, LLMChain
+from langchain.prompts import PromptTemplate
+from langchain.chains import LLMChain
 import sys
 import os
 

server/agent/weather.py

@@ -262,7 +262,7 @@ class LLMWeatherChain(Chain):
         return cls(llm_chain=llm_chain, **kwargs)
 
 
-from langchain import PromptTemplate
+from langchain.prompts import PromptTemplate
 
 _PROMPT_TEMPLATE = """用户将会向您咨询天气问题，您不需要自己回答天气问题，而是将用户提问的信息提取出来区，市和时间三个元素后使用我为你编写好的工具进行查询并返回结果，格式为 区+市+时间 每个元素用空格隔开。如果缺少信息，则用 None 代替。
 问题: ${{用户的问题}}

server/chat/agent_chat.py

@@ -7,7 +7,7 @@ from fastapi import Body
 from fastapi.responses import StreamingResponse
 from configs.model_config import LLM_MODEL, TEMPERATURE, HISTORY_LEN
 from server.utils import wrap_done, get_ChatOpenAI
-from langchain import LLMChain
+from langchain.chains import LLMChain
 from typing import AsyncIterable
 import asyncio
 from langchain.prompts.chat import ChatPromptTemplate

server/chat/chat.py

@@ -2,7 +2,7 @@ from fastapi import Body
 from fastapi.responses import StreamingResponse
 from configs import LLM_MODEL, TEMPERATURE
 from server.utils import wrap_done, get_ChatOpenAI
-from langchain import LLMChain
+from langchain.chains import LLMChain
 from langchain.callbacks import AsyncIteratorCallbackHandler
 from typing import AsyncIterable
 import asyncio

server/chat/knowledge_base_chat.py

@@ -3,7 +3,7 @@ from fastapi.responses import StreamingResponse
 from configs import (LLM_MODEL, VECTOR_SEARCH_TOP_K, SCORE_THRESHOLD, TEMPERATURE)
 from server.utils import wrap_done, get_ChatOpenAI
 from server.utils import BaseResponse, get_prompt_template
-from langchain import LLMChain
+from langchain.chains import LLMChain
 from langchain.callbacks import AsyncIteratorCallbackHandler
 from typing import AsyncIterable, List, Optional
 import asyncio

server/chat/search_engine_chat.py

@@ -6,7 +6,7 @@ from fastapi.responses import StreamingResponse
 from fastapi.concurrency import run_in_threadpool
 from server.utils import wrap_done, get_ChatOpenAI
 from server.utils import BaseResponse, get_prompt_template
-from langchain import LLMChain
+from langchain.chains import LLMChain
 from langchain.callbacks import AsyncIteratorCallbackHandler
 from typing import AsyncIterable
 import asyncio

tests/agent/test_agent_function.py

@@ -3,7 +3,7 @@ import os
 sys.path.append(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))))
 from configs import LLM_MODEL, TEMPERATURE
 from server.utils import get_ChatOpenAI
-from langchain import LLMChain
+from langchain.chains import LLMChain
 from langchain.agents import LLMSingleActionAgent, AgentExecutor
 from server.agent.tools import tools, tool_names
 from langchain.memory import ConversationBufferWindowMemory