mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-06 00:57:33 +08:00
[Feature] add custom chat template (#3251)
* add custom chat_template * add custom chat_template * add unittest * fix * add docs * fix comment * add offline chat * fix unit test * fix unit test * fix * fix pre commit * fix unit test * add unit test * add unit test * add unit test * fix pre_commit * fix enable_thinking * fix pre commit * fix pre commit * fix unit test * add requirements
This commit is contained in:
@@ -72,6 +72,7 @@ class Request:
|
||||
guided_json_object: Optional[bool] = None,
|
||||
enable_thinking: Optional[bool] = True,
|
||||
trace_carrier: dict = dict(),
|
||||
chat_template: Optional[str] = None,
|
||||
) -> None:
|
||||
self.request_id = request_id
|
||||
self.prompt = prompt
|
||||
@@ -111,6 +112,8 @@ class Request:
|
||||
self.enable_thinking = enable_thinking
|
||||
self.trace_carrier = trace_carrier
|
||||
|
||||
self.chat_template = chat_template
|
||||
|
||||
# token num
|
||||
self.block_tables = []
|
||||
self.output_token_ids = []
|
||||
@@ -152,6 +155,7 @@ class Request:
|
||||
guided_json_object=d.get("guided_json_object", None),
|
||||
enable_thinking=d.get("enable_thinking", True),
|
||||
trace_carrier=d.get("trace_carrier", {}),
|
||||
chat_template=d.get("chat_template", None),
|
||||
)
|
||||
|
||||
@property
|
||||
@@ -191,6 +195,7 @@ class Request:
|
||||
"draft_token_ids": self.draft_token_ids,
|
||||
"enable_thinking": self.enable_thinking,
|
||||
"trace_carrier": self.trace_carrier,
|
||||
"chat_template": self.chat_template,
|
||||
}
|
||||
add_params = [
|
||||
"guided_json",
|
||||
|
Reference in New Issue
Block a user