Support for more generic Agents

[kitrakrev]

I am trying to use litellm , and found the agents are not as customizable as I thought. I could potentially make it as generic as possible. Raising this issue to check if the fix already exists or can I make the changes and raise the PR ?