faiss_rag_enterprise/llama_index/callbacks/honeyhive_callback.py

12 lines
406 B
Python

from typing import Any
from llama_index.callbacks.base_handler import BaseCallbackHandler
def honeyhive_callback_handler(**kwargs: Any) -> BaseCallbackHandler:
try:
from honeyhive.utils.llamaindex_tracer import HoneyHiveLlamaIndexTracer
except ImportError:
raise ImportError("Please install HoneyHive with `pip install honeyhive`")
return HoneyHiveLlamaIndexTracer(**kwargs)