update langchain search wrapper use

Author: lightislost

muagent/chat/search_chat.py

@@ -3,7 +3,7 @@
 
 from langchain.chains.llm import LLMChain
 from langchain.callbacks import AsyncIteratorCallbackHandler
-from langchain.utilities import BingSearchAPIWrapper, DuckDuckGoSearchAPIWrapper
+from langchain_community.utilities import BingSearchAPIWrapper, DuckDuckGoSearchAPIWrapper
 from langchain.prompts.chat import ChatPromptTemplate
 from langchain_community.docstore.document import Document
 

muagent/llm_models/openai_model.py

@@ -113,12 +113,14 @@ def __init__(self, llm_config: LLMConfig, callBack: AsyncIteratorCallbackHandler
 
 
 def getChatModelFromConfig(llm_config: LLMConfig, callBack: AsyncIteratorCallbackHandler = None, ) -> Union[ChatOpenAI, LLM, CustomLLMModel]:
+    # logger.debug(f"{llm_config}")
     if llm_config and llm_config.llm and isinstance(llm_config.llm, LLM):
         return CustomLLMModel(llm=llm_config.llm)
     elif llm_config:
         model_class_dict = {"openai": OpenAILLMModel, "lingyiwanwu": LYWWLLMModel}
         model_class = model_class_dict[llm_config.model_engine]
         model = model_class(llm_config, callBack)
+        # logger.debug(f"{model.llm}")
         return model
     else:
         return OpenAILLMModel(llm_config, callBack)