13 lines
402 B
Python
13 lines
402 B
Python
from typing import Any
|
|
|
|
from llama_index.callbacks.base_handler import BaseCallbackHandler
|
|
|
|
|
|
def argilla_callback_handler(**kwargs: Any) -> BaseCallbackHandler:
|
|
try:
|
|
# lazy import
|
|
from argilla_llama_index import ArgillaCallbackHandler
|
|
except ImportError:
|
|
raise ImportError("Please install Argilla with `pip install argilla`")
|
|
return ArgillaCallbackHandler(**kwargs)
|