Loosening LLMModel.run_prompt to allow for Iterable[Callback] (#36)

Future-House · Jan 18, 2025 · 89d8767 · 89d8767
1 parent 1eae8a2
commit 89d8767
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: