Update app.py

app.py

@@ -111,7 +111,7 @@ class TextToSQLSystem:
             corpus = [item['messages'][0]['content'] for item in dataset]
             self._log(f"正在編碼 {len(corpus)} 個問題...")
             all_embeddings = torch.cat([self._encode_texts(corpus[i:i+32]) for i in range(0, len(corpus), 32)], dim=0).numpy()
-            index = faiss.IndexFlatIP(all_embeddings.shape)
+            index = faiss.IndexFlatIP(all_embeddings.shape[1])
             index.add(all_embeddings.astype('float32'))
             self._log("✅ 向量索引建立完成")
             return dataset, index