lamhieu commited on
Commit
54e83f2
·
1 Parent(s): 1d1f741

chore: update something

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -451,6 +451,7 @@ def generate_chat(
451
  ) -> Iterator[str]:
452
  # Build the input_ids for the chat conversation
453
  def build_input_ids(
 
454
  apply_tools: bool = None,
455
  references=None,
456
  ):
@@ -584,6 +585,7 @@ def generate_chat(
584
 
585
  # Build the input_ids for the chat conversation
586
  input_ids = build_input_ids(
 
587
  apply_tools=apply_tools,
588
  references=document_references,
589
  )
 
451
  ) -> Iterator[str]:
452
  # Build the input_ids for the chat conversation
453
  def build_input_ids(
454
+ system_prompt: str = "",
455
  apply_tools: bool = None,
456
  references=None,
457
  ):
 
585
 
586
  # Build the input_ids for the chat conversation
587
  input_ids = build_input_ids(
588
+ system_prompt=system_prompt,
589
  apply_tools=apply_tools,
590
  references=document_references,
591
  )