Fixing local mistake

Fixing return structure

Updating with jais's new handler

Trying max length of 2k

Improving variable names

Fixing handler

Adding handler for inference endpoints