faiss_rag_enterprise/llama_index/callbacks/deepeval_callback.py

12 lines
414 B
Python

from typing import Any
from llama_index.callbacks.base_handler import BaseCallbackHandler
def deepeval_callback_handler(**kwargs: Any) -> BaseCallbackHandler:
try:
from deepeval.tracing.integrations.llama_index import LlamaIndexCallbackHandler
except ImportError:
raise ImportError("Please install DeepEval with `pip install -U deepeval`")
return LlamaIndexCallbackHandler(**kwargs)