diff --git a/app/service/chat_robot/script/main.py b/app/service/chat_robot/script/main.py
index 3342a5c..ab1cf8d 100644
--- a/app/service/chat_robot/script/main.py
+++ b/app/service/chat_robot/script/main.py
@@ -1,23 +1,21 @@
 import json
 import logging
-from langchain_community.chat_models import ChatTongyi
-from loguru import logger
+
 from langchain.agents import Tool
 from langchain.callbacks import FileCallbackHandler
-from langchain.utilities import SerpAPIWrapper
 from langchain.prompts.chat import ChatPromptTemplate, HumanMessagePromptTemplate, MessagesPlaceholder
 from langchain.schema import SystemMessage, AIMessage
-from langchain.chat_models import ChatOpenAI
-from langchain.llms.openai import OpenAI
-from langchain.callbacks import FileCallbackHandler
+from langchain.utilities import SerpAPIWrapper
+from langchain_community.chat_models import ChatTongyi
+from loguru import logger
+
+from app.core.config import *
 from app.service.chat_robot.script.agents import CustomAgentExecutor, ConversationalFunctionsAgent
-from app.service.chat_robot.script.callbacks import OpenAITokenRecordCallbackHandler
 from app.service.chat_robot.script.database import CustomDatabase
 from app.service.chat_robot.script.memory import UserConversationBufferWindowMemory
 from app.service.chat_robot.script.prompt import FASHION_CHAT_BOT_PREFIX, TOOLS_FUNCTIONS_SUFFIX
 from app.service.chat_robot.script.service import CallQWen
 from app.service.chat_robot.script.tools import (QuerySQLDataBaseTool, InfoSQLDatabaseTool, QuerySQLCheckerTool, ListSQLDatabaseTool)
-from app.core.config import *
 
 # os.environ["http_proxy"] = "http://127.0.0.1:7890"
 # os.environ["https_proxy"] = "http://127.0.0.1:7890"
@@ -32,7 +30,7 @@ log_handler = FileCallbackHandler(logfile)
 #                  # callbacks=[OpenAICallbackHandler()]
 #                  )
 
-llm = ChatTongyi(api_key="sk-7658298c6b99443c98184a5e634fe6ab")
+llm = ChatTongyi(api_key=QWEN_API_KEY)
 
 search = SerpAPIWrapper()
 db = CustomDatabase.from_uri(f'mysql+pymysql://{DB_USERNAME}:{DB_PASSWORD}@{DB_HOST}:{DB_PORT}/attribute_retrieval_V3',
@@ -49,7 +47,7 @@ tools = [
     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="sk-7658298c6b99443c98184a5e634fe6ab")),
+    QuerySQLCheckerTool(db=db, llm=ChatTongyi(temperature=0, api_key=QWEN_API_KEY)),
     # Tool(
     #     name="tutorial_tool",
     #     description="Utilize this tool to retrieve specific statements related to user guidance tutorials."