models - litellm - async

[pgrayy]

Description

Use litellm's acompletion to support async requests.

Related Issues

#83

Type of Change

• Bug fix
• New feature
• Breaking change
• Documentation update
• Other (please describe):

Testing

How have you tested the change? Verify that the changes do not break functionality or introduce warnings in consuming repositories: agents-docs, agents-tools, agents-cli

• I ran hatch run prepare
• Wrote new unit and integ tests.
• Ran the following test script:

prompt = "What is 2+2? Think through the steps."


async def func_a():
    model = LiteLLMModel(model_id="us.anthropic.claude-3-7-sonnet-20250219-v1:0")
    agent = Agent(model=model, callback_handler=None)
    result = await agent.invoke_async(prompt)
    print(f"FUNC_A: {result.message}")


async def func_b():
    model = LiteLLMModel(model_id="us.anthropic.claude-3-7-sonnet-20250219-v1:0")
    agent = Agent(model=model, callback_handler=None)
    result = await agent.invoke_async(prompt)
    print(f"FUNC_B: {result.message}")


async def func_c():
    model = LiteLLMModel(model_id="us.anthropic.claude-3-7-sonnet-20250219-v1:0")
    agent = Agent(model=model, callback_handler=None)
    result = await agent.invoke_async(prompt)
    print(f"FUNC_C: {result.message}")


async def func_d():
    model = LiteLLMModel(model_id="us.anthropic.claude-3-7-sonnet-20250219-v1:0")
    agent = Agent(model=model, callback_handler=None)
    result = await agent.invoke_async(prompt)
    print(f"FUNC_D: {result.message}")


async def main():
    await asyncio.gather(func_a(), func_b(), func_c(), func_d())


asyncio.run(main())

Using the sync Python client, every function ran sequentially and the total run time averaged 12s. Using the acompletion, every function ran concurrently and the total run time averaged 6s.

Checklist

• I have read the CONTRIBUTING document
• I have added any necessary tests that prove my fix is effective or my feature works
• I have updated the documentation accordingly
• I have added an appropriate example to the documentation to outline the feature, or no new docs are needed
• My changes generate no new warnings
• Any dependent changes have been merged and published

---

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.