Update app.py
Browse files
app.py
CHANGED
|
@@ -83,7 +83,7 @@ def get_conversation_chain(vectorstore, tokenH):
|
|
| 83 |
def generate_response(messages):
|
| 84 |
try:
|
| 85 |
completion = client.chat.completions.create(
|
| 86 |
-
model="
|
| 87 |
messages=messages,
|
| 88 |
max_tokens=500
|
| 89 |
)
|
|
@@ -97,7 +97,7 @@ def get_conversation_chain(vectorstore, tokenH):
|
|
| 97 |
retriever = vectorstore.as_retriever(search_type="similarity", search_kwargs={"k": 5})
|
| 98 |
documents = retriever.get_relevant_documents(user_input)
|
| 99 |
documents_text = "\n".join(doc.page_content for doc in documents)
|
| 100 |
-
messages = [{"role": "user", "content": user_input}, {"role": "
|
| 101 |
return generate_response(messages)
|
| 102 |
|
| 103 |
return conversation_chain
|
|
|
|
| 83 |
def generate_response(messages):
|
| 84 |
try:
|
| 85 |
completion = client.chat.completions.create(
|
| 86 |
+
model="Qwen/Qwen2.5-72B-Instruct",
|
| 87 |
messages=messages,
|
| 88 |
max_tokens=500
|
| 89 |
)
|
|
|
|
| 97 |
retriever = vectorstore.as_retriever(search_type="similarity", search_kwargs={"k": 5})
|
| 98 |
documents = retriever.get_relevant_documents(user_input)
|
| 99 |
documents_text = "\n".join(doc.page_content for doc in documents)
|
| 100 |
+
messages = [{"role": "user", "content": user_input}, {"role": "system", "content": documents_text}]
|
| 101 |
return generate_response(messages)
|
| 102 |
|
| 103 |
return conversation_chain
|