minhvtt commited on
Commit
b85b8b1
verified
1 Parent(s): 0f17bfd

Upload 12 files

Browse files
Files changed (1) hide show
  1. main.py +14 -5
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 = True
144
  use_compression: bool = True
145
  score_threshold: float = 0.5
146
 
@@ -779,10 +779,19 @@ Example:
779
  )
780
 
781
  # Build messages
782
- messages = [
783
- {"role": "system", "content": system_message},
784
- {"role": "user", "content": request.message}
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 = ""