From 91ed45e978323cb622bc777a29792ab5845f1bc1 Mon Sep 17 00:00:00 2001 From: zhouchengrong Date: Mon, 31 Mar 2025 17:08:52 +0800 Subject: [PATCH] =?UTF-8?q?feat=EF=BC=88=E6=96=B0=E5=8A=9F=E8=83=BD?= =?UTF-8?q?=EF=BC=89:=20=20qwen=20api=20key=20=E4=BF=AE=E6=94=B9=20fix?= =?UTF-8?q?=EF=BC=88=E4=BF=AE=E5=A4=8Dbug=EF=BC=89:=20docs=EF=BC=88?= =?UTF-8?q?=E6=96=87=E6=A1=A3=E5=8F=98=E6=9B=B4=EF=BC=89:=20refactor?= =?UTF-8?q?=EF=BC=88=E9=87=8D=E6=9E=84=EF=BC=89:=20test(=E5=A2=9E=E5=8A=A0?= =?UTF-8?q?=E6=B5=8B=E8=AF=95):?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/service/chat_robot/script/main.py | 18 ++++++++---------- 1 file changed, 8 insertions(+), 10 deletions(-) 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."