feat : 代码梳理 移除所有敏感密钥 通过环境变量方式配置

app/service/chat_robot/script/main.py

@@ -1,15 +1,15 @@
 import json
 import logging
 
-from langchain.agents import Tool
-from langchain.callbacks import FileCallbackHandler
-from langchain.prompts.chat import ChatPromptTemplate, HumanMessagePromptTemplate, MessagesPlaceholder
-from langchain.schema import SystemMessage, AIMessage
-from langchain.utilities import SerpAPIWrapper
+from langchain_community.utilities import SerpAPIWrapper
+from langchain_core.callbacks import FileCallbackHandler
+from langchain_core.messages import SystemMessage, AIMessage
+from langchain_core.prompts import MessagesPlaceholder, HumanMessagePromptTemplate, ChatPromptTemplate
+from langchain_core.tools import Tool
 from langchain_community.chat_models import ChatTongyi
 from loguru import logger
 
-from app.core.config import *
+from app.core.config import settings
 from app.service.chat_robot.script.agents import CustomAgentExecutor, ConversationalFunctionsAgent
 from app.service.chat_robot.script.database import CustomDatabase
 from app.service.chat_robot.script.memory import UserConversationBufferWindowMemory
@@ -30,10 +30,10 @@ log_handler = FileCallbackHandler(logfile)
 #                  # callbacks=[OpenAICallbackHandler()]
 #                  )
 
-llm = ChatTongyi(api_key=QWEN_API_KEY)
+llm = ChatTongyi(api_key=settings.QWEN_API_KEY)
 
 search = SerpAPIWrapper()
-db = CustomDatabase.from_uri(f'mysql+pymysql://{DB_USERNAME}:{DB_PASSWORD}@{DB_HOST}:{DB_PORT}/attribute_retrieval_V3',
+db = CustomDatabase.from_uri(f'mysql+pymysql://{settings.DB_USERNAME}:{settings.DB_PASSWORD}@{settings.DB_HOST}:{settings.DB_PORT}/attribute_retrieval_V3',
                              include_tables=['female_top', 'female_skirt', 'female_pants', 'female_dress',
                                              'female_outwear', 'male_bottom', 'male_top', 'male_outwear'],
                              engine_args={"pool_recycle": 7200})
@@ -43,11 +43,11 @@ tools = [
         description="Can be used to perform Internet searches",
         func=search.run
     ),
-    QuerySQLDataBaseTool(db=db, return_direct=False),
+    QuerySQLDataBaseTool(db=db),
     InfoSQLDatabaseTool(db=db),
     ListSQLDatabaseTool(db=db),
     # QuerySQLCheckerTool(db=db, llm=OpenAI(temperature=0, openai_api_key=OPENAI_API_KEY)),
-    QuerySQLCheckerTool(db=db, llm=ChatTongyi(temperature=0, api_key=QWEN_API_KEY)),
+    QuerySQLCheckerTool(db=db, llm=ChatTongyi(api_key=settings.QWEN_API_KEY)),
     # Tool(
     #     name="tutorial_tool",
     #     description="Utilize this tool to retrieve specific statements related to user guidance tutorials."
@@ -133,5 +133,5 @@ def chat(post_data):
         'completion_tokens': final_outputs['completion_tokens'],
         'response_type': final_outputs["response_type"]
     }
-    logging.info(json.dumps(api_response))
+    logging.info(json.dumps(api_response, indent=4))
     return api_response