Upload rt_templates.py with huggingface_hub
Browse files- 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"
|