Spaces:
Running
Running
Upload 12 files
Browse files
main.py
CHANGED
|
@@ -140,7 +140,7 @@ Quy t岷痗 tuy峄噒 膽峄慽:
|
|
| 140 |
# Advanced RAG options
|
| 141 |
use_advanced_rag: bool = True
|
| 142 |
use_query_expansion: bool = True
|
| 143 |
-
use_reranking: bool =
|
| 144 |
use_compression: bool = True
|
| 145 |
score_threshold: float = 0.5
|
| 146 |
|
|
@@ -779,10 +779,19 @@ Example:
|
|
| 779 |
)
|
| 780 |
|
| 781 |
# Build messages
|
| 782 |
-
|
| 783 |
-
|
| 784 |
-
|
| 785 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 786 |
|
| 787 |
# Generate response
|
| 788 |
response = ""
|
|
|
|
| 140 |
# Advanced RAG options
|
| 141 |
use_advanced_rag: bool = True
|
| 142 |
use_query_expansion: bool = True
|
| 143 |
+
use_reranking: bool = False # Disabled - Cross-Encoder not good for Vietnamese
|
| 144 |
use_compression: bool = True
|
| 145 |
score_threshold: float = 0.5
|
| 146 |
|
|
|
|
| 779 |
)
|
| 780 |
|
| 781 |
# Build messages
|
| 782 |
+
if request.use_advanced_rag and context_used:
|
| 783 |
+
# Advanced RAG prompt already contains query + instructions
|
| 784 |
+
# Just send it as system message, user message is empty
|
| 785 |
+
messages = [
|
| 786 |
+
{"role": "system", "content": "You are a helpful assistant."},
|
| 787 |
+
{"role": "user", "content": system_message}
|
| 788 |
+
]
|
| 789 |
+
else:
|
| 790 |
+
# Basic RAG or no RAG - normal message flow
|
| 791 |
+
messages = [
|
| 792 |
+
{"role": "system", "content": system_message},
|
| 793 |
+
{"role": "user", "content": request.message}
|
| 794 |
+
]
|
| 795 |
|
| 796 |
# Generate response
|
| 797 |
response = ""
|