Update app.py

app.py

@@ -696,47 +696,56 @@ class TextToSQLSystem:
 
     def _build_prompt(self, user_q: str, examples: List[Dict]) -> str:
         """
-        建立結構更清晰的提示詞，明確區分上下文和最終指令，避免模型混淆。
+        建立結構更清晰的提示詞，明確區分上下文和最終指令，避免模型混淆。(已修正 TypeError)
         """
         relevant_tables = self._identify_relevant_tables(user_q)
         
         # 1. 提供背景上下文 (Context)
-        # 這部分告訴模型它的角色和它擁有的參考資料。
         system_context = "You are an expert AI assistant that generates SQLite queries based on a database schema and a user's question."
         schema_str = self._format_relevant_schema(relevant_tables)
 
         # 2. 提供一個清晰的範例 (Few-shot Example)
-        # 這為模型設定了期望的輸入輸出格式。
         ex_str = ""
+        # 檢查 examples 列表是否為空，避免出錯
         if examples:
-            best_example = examples
+            # --- START: 這裡是修正的地方 ---
+            # 從列表中取出第一個元素 (它是一個字典)
+            best_example = examples[0] 
+            # --- END: 修正完成 ---
+            
             # 使用 [QUESTION] 和 [SQL] 標籤來強化結構
             ex_str = f"Here is an example of how to answer:\n[QUESTION]: {best_example['question']}\n[SQL]:\n```sql\n{best_example['sql']}\n```\n\n---\n\n"
 
         # 3. 給出最終的、不可混淆的指令 (Final, Imperative Instruction)
-        # 這部分是提示詞的結尾，告訴模型「現在開始做這個」。
         final_task_instruction = f"""Now, based on the schema and the new user question below, generate a single, valid SQLite query.
-            **CRITICAL RULES TO FOLLOW:**
-            - You **must** only use the tables and columns provided in the schema above.
-            - You **must** use SQLite syntax (e.g., use `strftime('%Y', date_column)` for years).
-            - You **must** output **nothing** else, only the SQL query inside a single ```sql code block.
-            
-            [QUESTION]: {user_q}
-            [SQL]:
-            ```sql
-            """
+
+**CRITICAL RULES TO FOLLOW:**
+- You **must** only use the tables and columns provided in the schema above.
+- You **must** use SQLite syntax (e.g., use `strftime('%Y', date_column)` for years).
+- You **must** output **nothing** else, only the SQL query inside a single ```sql code block.
+
+[QUESTION]: {user_q}
+[SQL]:
+```sql
+"""
 
         # 4. 組合最終的 Prompt
         prompt = f"""{system_context}
-                {schema_str}
-                {ex_str}
-                {final_task_instruction}
-                """
+
+{schema_str}
+{ex_str}
+{final_task_instruction}
+"""
         
-        # 限制總長度 (這個邏輯保持不變)
+        # 5. 限制總長度 (這個邏輯保持不變)
+        # 確保在截斷時，最後的指令部分是完整的
         if len(prompt) > 1500:
-            prompt = prompt[:1500] + "...\n" + final_task_instruction
-
+            # 找到 final_task_instruction 在 prompt 中的起始位置
+            instruction_start_index = prompt.find("Now, based on the schema")
+            # 保留 schema 和一部分範例，然後接上完整的最終指令
+            allowed_context_len = 1500 - len(final_task_instruction)
+            prompt = prompt[:allowed_context_len] + "...\n\n" + final_task_instruction
+            
         return prompt