Maybe this has improved since I last used it, but I was dismayed at not being able to do something like:

    litellm.register("foo", CustomAIProvider)
    litellm.do_whatever("foo/my-cool-model", "what is 2+2")