You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
swarms/test_async_litellm.py

17 lines
310 B

from swarms.utils.litellm_wrapper import LiteLLM
llm = LiteLLM(
model_name="gpt-4o-mini",
temperature=0.5,
max_tokens=1000,
stream=True,
)
out = llm.run("What is the capital of France?")
print(out)
for chunk in out:
out = chunk["choices"][0]["delta"]
print(type(out))
print(out)