Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -54,7 +54,7 @@ def load_and_process_documents(folder_path):
|
|
| 54 |
vectordb = Chroma.from_documents(
|
| 55 |
documents,
|
| 56 |
embedding=OpenAIEmbeddings(),
|
| 57 |
-
persist_directory="./"
|
| 58 |
)
|
| 59 |
return vectordb
|
| 60 |
|
|
@@ -70,7 +70,8 @@ def handle_query(user_message, temperature, chat_history):
|
|
| 70 |
|
| 71 |
# 使用 LangChain 的 ConversationalRetrievalChain 處理查詢
|
| 72 |
preface = """
|
| 73 |
-
指令: 全部以繁體中文呈現,200
|
|
|
|
| 74 |
"""
|
| 75 |
query = f"{preface} 查詢內容:{user_message}"
|
| 76 |
|
|
@@ -78,7 +79,7 @@ def handle_query(user_message, temperature, chat_history):
|
|
| 78 |
previous_answers = transform_history_for_langchain(chat_history)
|
| 79 |
|
| 80 |
pdf_qa = ConversationalRetrievalChain.from_llm(
|
| 81 |
-
ChatOpenAI(temperature=temperature, model_name='gpt-
|
| 82 |
retriever=vectordb.as_retriever(search_kwargs={'k': 6}),
|
| 83 |
return_source_documents=True,
|
| 84 |
verbose=False
|
|
|
|
| 54 |
vectordb = Chroma.from_documents(
|
| 55 |
documents,
|
| 56 |
embedding=OpenAIEmbeddings(),
|
| 57 |
+
persist_directory="./tmp"
|
| 58 |
)
|
| 59 |
return vectordb
|
| 60 |
|
|
|
|
| 70 |
|
| 71 |
# 使用 LangChain 的 ConversationalRetrievalChain 處理查詢
|
| 72 |
preface = """
|
| 73 |
+
指令: 全部以繁體中文呈現,200字以內。
|
| 74 |
+
除了與文件相關內容之外,與文件內容不相關的問題都必須回應:這問題很深奧,需要請示JohnLiao大神...
|
| 75 |
"""
|
| 76 |
query = f"{preface} 查詢內容:{user_message}"
|
| 77 |
|
|
|
|
| 79 |
previous_answers = transform_history_for_langchain(chat_history)
|
| 80 |
|
| 81 |
pdf_qa = ConversationalRetrievalChain.from_llm(
|
| 82 |
+
ChatOpenAI(temperature=temperature, model_name='gpt-4o-mini'),
|
| 83 |
retriever=vectordb.as_retriever(search_kwargs={'k': 6}),
|
| 84 |
return_source_documents=True,
|
| 85 |
verbose=False
|