Generates structured JSON execution plans for tool/plugin orchestration. Given a user request and available tools, it produces a valid JSON object specifying which tools to call, with what parameters, and in what order.
1from transformers import AutoModelForCausalLM, AutoTokenizer
2import torch
3
4model = AutoModelForCausalLM.from_pretrained("airev-ae/Qwen-0.8B-AgentJSON", torch_dtype=torch.bfloat16)
5tokenizer = AutoTokenizer.from_pretrained("airev-ae/Qwen-0.8B-AgentJSON")
6
7messages = [
8 {"role": "system", "content": "You are an AI agent orchestrator. Generate a JSON execution plan."},
9 {"role": "user", "content": "Search for the latest AI news"}
10]
11
12text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
13inputs = tokenizer(text, return_tensors="pt").to(model.device)
14
15output = model.generate(**inputs, max_new_tokens=512, temperature=0.8, do_sample=True)
16print(tokenizer.decode(output[0][inputs["input_ids"].shape[1]:], skip_special_tokens=True))
1@misc{airev2026agentjson,
2 title={AIREV Qwen-0.8B-AgentJSON: Sub-1B Tool Calling via Progressive Curriculum GRPO},
3 author={AIREV FZ-LLC},
4 year={2026},
5 url={https://huggingface.co/airev-ae/Qwen-0.8B-AgentJSON}
6}