# 準備few-shot樣例
  examples =[
{
"inn":"我想知道東方阿爾法優(yōu)勢產(chǎn)業(yè)混合C基金，在2021年年度報告中，前10大重倉股中，有多少只股票在報告期內(nèi)取得正收益。",
"out":"rag_question***我想知道東方阿爾法優(yōu)勢產(chǎn)業(yè)混合C基金，在2021年年度報告中，前10大重倉股中，有多少只股票在報告期內(nèi)取得正收益。"
},
{
"inn":"森赫電梯股份有限公司產(chǎn)品生產(chǎn)材料是什么？",
"out":"rag_question***森赫電梯股份有限公司產(chǎn)品生產(chǎn)材料是什么？"
},
{
"inn":"20210930日，一級行業(yè)為機械的股票的成交金額合計是多少？取整。",
"out":"agent_question***20210930日，一級行業(yè)為機械的股票的成交金額合計是多少？取整。"
},
...
  ]

if intent =="rag_question":
# 如果是RAG相關的問題
            result = self.rag.get_result(question=question)

return result

elif intent =="agent_question":
# 如果是Agent相關的問題
            result, result_list = self.agent.get_result(input=question)

return result
else:
# 其他類問題
            result = self.chat.invoke(input=question).content
return result

import logging
import datetime
from langgraph.prebuilt import create_react_agent
from langchain.tools.retriever import create_retriever_tool
from langchain_community.utilities importSQLDatabase
from langchain_community.agent_toolkits importSQLDatabaseToolkit
from rag.rag importRagManager
import settings


classFinanceBotEx:
def__init__(self, llm=settings.llm, chat=settings.chat, embed=settings.embed):
        self.llm = llm
        self.chat = chat
        self.embed = embed
        self.tools =[]

        self.rag =RagManager(llm=llm, embed=embed)

def init_sql_tool(self, path):
        # 連接數(shù)據(jù)庫
        db = SQLDatabase.from_uri(f"sqlite:///{path}")
        toolkit = SQLDatabaseToolkit(db=db, llm=self.llm)
        sql_tools = toolkit.get_tools()  # 工具

        return sql_tools

def init_rag_tools(self):
        # 給大模型 RAG 檢索器工具
        retriever = self.rag.get_retriever()
        retriever_tool = create_retriever_tool(
            retriever=retriever,
            name="rag_search",
            description="按照用戶的問題搜索相關的資料，對于招股書類的問題，you must use this tool!",
        )
        return retriever_tool

def create_prompt():
        system_prompt = """你是一位金融助手，可以幫助用戶查詢數(shù)據(jù)庫中的信息。
            你要盡可能的回答用戶提出的問題，為了更好的回答問題，你可以使用工具進行多輪的嘗試。

            # 關于retriever_tool工具的使用：
            1、你需要結合對檢索出來的上下文進行回答問題。
            2、如果你不知道答案，就說你不知道。請使用不超過三句話的簡潔回答。

            # 關于sql類工具的使用： 
            ## 工具使用規(guī)則                                     
            1、你需要根據(jù)用戶的問題，創(chuàng)建一個語法正確的SQLite查詢來運行，然后查看查詢的結果并返回答案。
            2、除非用戶指定了他們希望獲得的特定數(shù)量的示例，否則總是將查詢限制為最多5個結果。
            3、您可以按相關列對結果進行排序，以返回數(shù)據(jù)庫中最有趣的示例。
            4、永遠不要查詢指定表的所有列以避免查詢性能問題，你只查詢給定問題的相關列即可。
            5、你必須在執(zhí)行查詢之前仔細檢查查詢。如果執(zhí)行查詢時出現(xiàn)錯誤，請重新編寫查詢并重試。
            6、請勿對數(shù)據(jù)庫進行任何DML語句（INSERT，UPDATE，DELETE，DROP等）。

            ## 工具使用過程
            1、首先，你應該始終查看數(shù)據(jù)庫中的表，看看可以查詢什么，這一步驟很重要，注意不要跳過。
            2、然后，你應該查詢最相關表的schema。

            ## 工具使用注意事項：
            1、如果生成的SQL語句中，字段帶有英文括號()，請使用雙引號包裹起來，例如：收盤價(元) 雙引號包裹為 "收盤價(元)"。
            2、如果查詢過程中SQL語句有語法錯誤，減少查詢量,總體查詢次數(shù)應控制在15次以內(nèi)。      

            # 關于你的思考和行動過程，請按照如下格式：
            問題：你必須回答的輸入問題
            思考：你應該總是考慮該怎么做
            行動：你應該采取的行動，應該是以下工具之一：{tool_names}
            行動輸入：行動的輸入
            觀察：行動的結果
            ... (這個思考/行動/行動輸入/觀察可以重復N次)
            思考: 我現(xiàn)在知道最終答案了
            最終答案：原始輸入問題的最終答案


            Begin!

            """
        return system_prompt

def init_agent(self):
# 初始化 RAG 工具
        retriever_tool = self.init_rag_tools()

# 初始化 SQL 工具
        sql_tools = self.init_sql_tool(settings.SQLDATABASE_URI)

# 創(chuàng)建系統(tǒng)Prompt提示語
        system_prompt = self.create_prompt()

# 創(chuàng)建Agent
        agent_executor = create_react_agent(
            self.chat,
            tools=[retriever_tool]+ sql_tools,
            state_modifier=system_prompt
)
return agent_executor

def handle_query(self, example_query):
# 流式處理事件
        events = self.agent_executor.stream(
{"messages":[("user", example_query)]},
            stream_mode="values",
)

# 打印流式事件的消息
for event in events:
            event["messages"][-1].pretty_print()

return event["messages"][-1].content

# 測試 FinanceBotEx 主流程
deftest_financebot_ex():
from finance_bot_ex importFinanceBotEx
    llm, chat, embed = get_qwen_models()
    financebot =FinanceBotEx(llm=llm, chat=chat, embed=embed)

    example_query ="20210304日，一級行業(yè)為非銀金融的股票的成交量合計是多少？取整。"

    financebot.handle_query(example_query)

if __name__ =="__main__":
# test_rag()
# test_import()
    test_financebot_ex()

example_query = "根據(jù)聯(lián)化科技股份有限公司招股意見書，精細化工產(chǎn)品的通常利潤率是多少？"