Update main.py

main.py

@@ -21,6 +21,7 @@ import uvicorn
 
 # LangChain 相關匯入
 from langchain_core.prompts import ChatPromptTemplate
+from langchain.prompts import PromptTemplate
 from langchain_core.tools import tool
 from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain.agents import create_react_agent, AgentExecutor
@@ -173,15 +174,17 @@ tools = [generate_and_upload_image, analyze_image_with_text]
 # 建立 LLM 模型實例
 llm = ChatGoogleGenerativeAI(google_api_key=google_api, model="gemini-2.0-flash", temperature=0.2)
 
-# # 建立提示模板
-# prompt_template = ChatPromptTemplate.from_messages([
-#     ("system", "你是一個強大的助理，可以根據用戶的請求使用提供的工具。"),
-#     ("user", "{input}"),
-#     ("placeholder",  "{agent_scratchpad}"),
-# ])
+# 建立提示模板
+prompt_template = ChatPromptTemplate.from_messages([
+    ("system", "你是一個強大的助理，可以根據用戶的請求使用提供的工具。"),
+    ("user", "{input}"),
+    ("placeholder",  "{agent_scratchpad}"),
+])
+
+prompt = PromptTemplate.from_template(prompt_template)
 
 # 建立代理人，使用 create_react_agent
-react_agent = create_react_agent(llm, tools)
+react_agent = create_react_agent(llm, tools,prompt)
 agent_executor = AgentExecutor(agent=agent, tools=tools, verbose=True)