reverse commit for build_prompt and bib

README.md

@@ -203,26 +203,10 @@ def convert_to_xlam_tool(tools):
 
 # Helper function to build the input prompt for our model
 def build_prompt(task_instruction: str, format_instruction: str, tools: list, query: str):
-    prompt = f"[BEGIN OF TASK INSTRUCTION]
-{task_instruction}
-[END OF TASK INSTRUCTION]
-
-"
-    prompt += f"[BEGIN OF AVAILABLE TOOLS]
-{json.dumps(xlam_format_tools)}
-[END OF AVAILABLE TOOLS]
-
-"
-    prompt += f"[BEGIN OF FORMAT INSTRUCTION]
-{format_instruction}
-[END OF FORMAT INSTRUCTION]
-
-"
-    prompt += f"[BEGIN OF QUERY]
-{query}
-[END OF QUERY]
-
-"
+    prompt = f"[BEGIN OF TASK INSTRUCTION]\n{task_instruction}\n[END OF TASK INSTRUCTION]\n\n"
+    prompt += f"[BEGIN OF AVAILABLE TOOLS]\n{json.dumps(xlam_format_tools)}\n[END OF AVAILABLE TOOLS]\n\n"
+    prompt += f"[BEGIN OF FORMAT INSTRUCTION]\n{format_instruction}\n[END OF FORMAT INSTRUCTION]\n\n"
+    prompt += f"[BEGIN OF QUERY]\n{query}\n[END OF QUERY]\n\n"
     return prompt
     
 # Build the input and start the inference
@@ -320,13 +304,16 @@ This release is for research purposes only in support of an academic paper. Our
 If you find this repo helpful, please cite our paper:
 
 ```bibtex
-@misc{tan2025actionstudio,
-      title={ActionStudio: A Lightweight Framework for Data and Training of Action Models}, 
-      author={Zhenxiong Tan and Jianguo Zhang and Tian Lan and Thai Hoang and Linh Tran and Ming Zhu and Shervin Raissi and Caiming Xiong},
-      year={2025},\
-      eprint={2503.22673},\
-      archivePrefix={arXiv},\
-      primaryClass={cs.LG},\
-      url={https://arxiv.org/abs/2503.22673}\
+@article{liu2024apigen,
+  title={APIGen: Automated Pipeline for Generating Verifiable and Diverse Function-Calling Datasets},
+  author={Liu, Zuxin and Hoang, Thai and Zhang, Jianguo and Zhu, Ming and Lan, Tian and Kokane, Shirley and Tan, Juntao and Yao, Weiran and Liu, Zhiwei and Feng, Yihao and others},
+  journal={arXiv preprint arXiv:2406.18518},
+  year={2024}
+
+@article{zhang2025actionstudio,
+  title={ActionStudio: A Lightweight Framework for Data and Training of Action Models},
+  author={Zhang, Jianguo and Hoang, Thai and Zhu, Ming and Liu, Zuxin and Wang, Shiyu and Awalgaonkar, Tulika and Prabhakar, Akshara and Chen, Haolin and Yao, Weiran and Liu, Zhiwei and others},
+  journal={arXiv preprint arXiv:2503.22673},
+  year={2025}
 }
 ```