Update main.py

main.py

@@ -23,6 +23,8 @@ from langchain_core.prompts import ChatPromptTemplate
 from langchain_core.tools import tool
 from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain.agents import AgentExecutor, create_tool_calling_agent
+from pydantic import BaseModel, Field
+from typing import List
 
 # ==========================#  環境設定與工具函式# ==========================#
 # 設置日誌記錄，級別為 INFO，格式包含時間、級別和訊息
@@ -124,20 +126,26 @@ def get_image_url_from_line(message_id):
         return None
 
 # ==========================#  LangChain 工具定義# ==========================#
-@tool
+# 定義工具的輸入模型，明確指定參數的型別和描述
+class OutfitInput(BaseModel):
+    """用於生成穿搭圖片的輸入參數。"""
+    upper_body_urls: List[str] = Field(..., description="上衣圖片的 URL 列表。")
+    lower_body_urls: List[str] = Field(..., description="褲子/裙子圖片的 URL 列表。")
+
+@tool(args_schema=OutfitInput)
 def generate_outfit_from_clothes(upper_body_urls: list, lower_body_urls: list) -> str:
     """
     這個工具可以根據提供的上衣和褲子/裙子圖片 URLs，生成一套全新的穿搭圖片。
-
+    
     Args:
         upper_body_urls: 一組上衣圖片的 URL 列表。
         lower_body_urls: 一組褲子/裙子圖片的 URL 列表。
-
+        
     Returns:
         回傳生成圖片的 URL。
     """
     logging.info("Attempting to generate a new outfit image.")
-    prompt = "使用提供的上衣和褲子/裙子圖片，生成一套完整且時尚的穿搭圖片。請將衣服呈現在一個有模特兒穿著或是在平面上呈現的完整畫面中。風格應與提供的衣物相符。都使用繁體中文回覆。"
+    prompt = "使用提供的上衣和褲子/裙子圖片，生成一套完整且時尚的穿搭圖片。請將衣服呈現在一個有模特兒穿著或是在平面上呈現的完整畫面中。風格應與提供的衣物相符。"
     
     all_images_base64 = (
         [get_base64_from_url(url) for url in upper_body_urls] +
@@ -224,7 +232,7 @@ tools = [generate_outfit_from_clothes, analyze_image_with_text]
 llm = ChatGoogleGenerativeAI(google_api_key=google_api, model="gemini-2.5-flash", temperature=0.2)
 # 建立提示模板
 prompt_template = ChatPromptTemplate([
-    ("system", "你是一個強大的圖像生成與虛擬穿搭問答助理，可以根據用戶的請求使用提供的工具。當你執行 generate_and_upload_image 工具\    成功後會獲得一個 URL，然後你回答的 output 要包含有這個 URL 的完整資訊。如果工具有產生錯誤訊息請解讀並回應。都使用繁體中文回覆。"),
+    ("system", "你是一個強大的圖像生成與問答助理，可以根據用戶的請求使用提供的工具。當你執行 generate_and_upload_image 工具成功後會獲得一個 URL，然後你回答的 output 要包含有這個 URL 的完整資訊。如果工具有產生錯誤訊息請解讀並回應。"),
     ("user", "{input}"),
     ("placeholder", "{agent_scratchpad}"),
 ])