aied-lab commited on
Commit
118ac04
·
verified ·
1 Parent(s): 6c9fd8a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
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字以內。除了文件的相關內容之外,與文件內容不相關的問題,都只能回應:這問題很深奧,需要請示JohnLiao大神...
 
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-4'),
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