From 89d876799be238f1ee259251e328597633322889 Mon Sep 17 00:00:00 2001 From: James Braza <jamesbraza@gmail.com> Date: Fri, 17 Jan 2025 16:38:26 -0800 Subject: [PATCH] Loosening `LLMModel.run_prompt` to allow for `Iterable[Callback]` (#36) --- llmclient/llms.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llmclient/llms.py b/llmclient/llms.py index dcf349d..f497934 100644 --- a/llmclient/llms.py +++ b/llmclient/llms.py @@ -261,7 +261,7 @@ async def run_prompt( self, prompt: str, data: dict, - callbacks: list[Callable] | None = None, + callbacks: Iterable[Callable] | None = None, name: str | None = None, system_prompt: str | None = default_system_prompt, ) -> LLMResult: