Update app.py

app.py

@@ -31,19 +31,17 @@ except Exception as e:
 def answer_question(question):
     try:
         question_embedding = embeddings.embed_query(question)
-        return ""
-        # docs_and_scores = db.similarity_search_with_score(question_embedding)
+        question_embedding_np = np.array(question_embedding).reshape(1, -1)  # The fix!
+        docs_and_scores = db.similarity_search_with_score(question_embedding_np)
+        # 正确处理 docs_and_scores 列表
+        context = "\n".join([doc.page_content for doc, _ in docs_and_scores])  # 使用 join() 方法
+        print(context)
 
-        # return ""
-        # # 正确处理 docs_and_scores 列表
-        # context = "\n".join([doc.page_content for doc, _ in docs_and_scores])  # 使用 join() 方法
-        # print(context)
+        prompt = f"请根据以下知识库回答问题：\n{context}\n问题：{question}"
+        print(prompt)
 
-        # prompt = f"请根据以下知识库回答问题：\n{context}\n问题：{question}"
-        # print(prompt)
-
-        # answer = llm(prompt)
-        # return answer
+        answer = llm(prompt)
+        return answer
     except Exception as e:
         st.error(f"问答过程出错：{e}")
         return "An error occurred during the answering process."