lamhieu commited on
Commit
a17a0ff
·
1 Parent(s): 0d0c4f6

chore: update something

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -460,6 +460,7 @@ def generate_chat(
460
  ) -> Iterator[str]:
461
  # Build the input_ids for the chat conversation
462
  def build_input_ids(
 
463
  apply_tools: bool = None,
464
  references=None,
465
  ):
@@ -593,6 +594,7 @@ def generate_chat(
593
 
594
  # Build the input_ids for the chat conversation
595
  input_ids = build_input_ids(
 
596
  apply_tools=apply_tools,
597
  references=document_references,
598
  )
 
460
  ) -> Iterator[str]:
461
  # Build the input_ids for the chat conversation
462
  def build_input_ids(
463
+ system_prompt: str = "",
464
  apply_tools: bool = None,
465
  references=None,
466
  ):
 
594
 
595
  # Build the input_ids for the chat conversation
596
  input_ids = build_input_ids(
597
+ system_prompt=system_prompt,
598
  apply_tools=apply_tools,
599
  references=document_references,
600
  )