Oya: Tiny Aya wallet agent (LoRA, v2)

A LoRA adapter that turns Cohere Labs' Tiny Aya Global (3.35B) into a tool-calling wallet agent for English, Nigerian Pidgin, Yoruba and Yoruba-English mix. It buys airtime and data, pays for electricity and sends money through seven tools; it asks when details are missing, refuses to overspend, and ignores instructions planted in tool results. Made for the Lexsi Labs AI Safety & Alignment Hackathon (Track 3) with Lexsi Labs' AgentTune and AuditKIT.

Try it in the demo Space. For a laptop CPU, use the GGUF.

Results

Scored against a mock wallet: what happened to the money, not what the model said it did.

Untrained Tiny Aya Oya v2
Task success, 416 development tasks 39.2% 97.4%
Safety tasks handled correctly 18.8% 100%
Tasks where money moved that should not have 18.3% 2.4%
Task success, 30 held-out requests typed by a native speaker 23.3% 100%
Same 30, with the reply in the right language and asking for the right detail 20.0% 83.3%

Method, comparisons with other models and caveats are in the write-up.

Use

The adapter was trained with the repo's compact system prompt and tool format:

# git clone https://github.com/Timidan/lexsi-hack && cd lexsi-hack
import os
os.environ["OYA_PROMPT"] = "compact"
from naija_agent.hf_policy import HFPolicy
from naija_agent.prompting import system_message

oya = HFPolicy("CohereLabs/tiny-aya-global", adapter="timidan/oya-tiny-aya-sft-v2")
chat = [{"role": "system", "content": system_message()},
        {"role": "user", "content": "Abeg buy ₦500 MTN airtime for 08031234567."}]
print(oya([chat])[0])   # a JSON list of tool calls; naija_agent/runner.py runs them and continues the chat

Training

  • 2,080 reference chats (13 task categories × 4 languages × 40), loss on assistant turns only: the sft config of timidan/oya-wallet-agent-data.
  • LoRA rank 16, alpha 32; 1 epoch; learning rate 2e-4; 75 minutes on two Kaggle T4s.
  • train_log.json and loss.png are in this repo.

Licence

CC-BY-NC 4.0, following Tiny Aya; Cohere Labs' Acceptable Use Policy applies.

Downloads last month
34
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for timidan/oya-tiny-aya-sft-v2

Adapter
(84)
this model

Dataset used to train timidan/oya-tiny-aya-sft-v2

Space using timidan/oya-tiny-aya-sft-v2 1

Collection including timidan/oya-tiny-aya-sft-v2