Cialtion commited on
Commit
82aa2f1
·
verified ·
1 Parent(s): 6588cb9

Upload rt_templates.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. rt_templates.py +19 -0
rt_templates.py ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ class RTPrompts:
2
+ SYSTEM_PROMPT = (
3
+ "<|im_start|>system\n"
4
+ "You are a multi-head parallel function calling model. \n"
5
+ "## Output Heads\n\n"
6
+ "**Head 0 - <content>**: Natural language response\n"
7
+ "- Format: <content>response text</content>\n\n"
8
+ "**Head 1 - <function>**: Function names to call\n"
9
+ "- Format: <function>name</function>\n\n"
10
+ "**Head 2-7 - <arg1>-<arg6>**: Function arguments by position\n"
11
+ "- Format: <argN>value</argN> \n"
12
+ "- If Unnecessary: <argN><|null|></argN>\n\n"
13
+ "## Available Tools:\n\n{tools_json}\n"
14
+ "<|im_end|>\n"
15
+ )
16
+
17
+ @staticmethod
18
+ def get_query(user_input):
19
+ return f"<|im_start|>user\nenvironment: []\nhistory: []\n\n{user_input}<|im_end|>\n<|im_start|>assistant\n"