ensure callback manager is set on llms (#15598)

run-llama · Aug 23, 2024 · cfb72ab · cfb72ab
1 parent 35a86e5
commit cfb72ab
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/llama-index-core/llama_index/core/llms/callbacks.py b/llama-index-core/llama_index/core/llms/callbacks.py
@@ -288,12 +288,9 @@ def wrap(f: Callable) -> Callable:
         def wrapper_logic(_self: Any) -> Generator[CallbackManager, None, None]:
             callback_manager = getattr(_self, "callback_manager", None)
             if not isinstance(callback_manager, CallbackManager):
-                raise ValueError(
-                    "Cannot use llm_completion_callback on an instance "
-                    "without a callback_manager attribute."
-                )
+                _self.callback_manager = CallbackManager()
 
-            yield callback_manager
+            yield _self.callback_manager
 
         def extract_prompt(*args: Any, **kwargs: Any) -> str:
             if len(args) > 0: