Sure, I'd be happy to take a look. It's currently just a direct call to the OpenRouter models API, though.