Improve Error Message for failed callback (#6247)

Include the handler class name in the warning
This commit is contained in:
Zander Chase 2023-06-15 19:18:37 -07:00 committed by GitHub
parent 0013256e81
commit bc9b8c8239
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 14 additions and 7 deletions

View File

@ -200,11 +200,15 @@ def _handle_event(
**kwargs, **kwargs,
) )
else: else:
logger.warning(f"Error in {event_name} callback: {e}") logger.warning(
f"Error in {handler.__class__.__name__}.{event_name} callback: {e}"
)
except Exception as e: except Exception as e:
logger.warning(
f"Error in {handler.__class__.__name__}.{event_name} callback: {e}"
)
if handler.raise_error: if handler.raise_error:
raise e raise e
logger.warning(f"Error in {event_name} callback: {e}")
async def _ahandle_event_for_handler( async def _ahandle_event_for_handler(
@ -236,11 +240,15 @@ async def _ahandle_event_for_handler(
**kwargs, **kwargs,
) )
else: else:
logger.warning(f"Error in {event_name} callback: {e}") logger.warning(
f"Error in {handler.__class__.__name__}.{event_name} callback: {e}"
)
except Exception as e: except Exception as e:
logger.warning(
f"Error in {handler.__class__.__name__}.{event_name} callback: {e}"
)
if handler.raise_error: if handler.raise_error:
raise e raise e
logger.warning(f"Error in {event_name} callback: {e}")
async def _ahandle_event( async def _ahandle_event(

View File

@ -11,7 +11,6 @@ from langchain.callbacks.manager import (
CallbackManagerForChainRun, CallbackManagerForChainRun,
) )
from langchain.chains.base import Chain from langchain.chains.base import Chain
from langchain.chains.llm import LLMChain
from langchain.schema import RUN_KEY, BaseOutputParser from langchain.schema import RUN_KEY, BaseOutputParser
@ -39,7 +38,7 @@ class RunEvaluatorChain(Chain, RunEvaluator):
input_mapper: RunEvaluatorInputMapper input_mapper: RunEvaluatorInputMapper
"""Maps the Run and Optional example to a dictionary for the eval chain.""" """Maps the Run and Optional example to a dictionary for the eval chain."""
eval_chain: LLMChain eval_chain: Chain
"""The evaluation chain.""" """The evaluation chain."""
output_parser: RunEvaluatorOutputParser output_parser: RunEvaluatorOutputParser
"""Parse the output of the eval chain into feedback.""" """Parse the output of the eval chain into feedback."""

View File

@ -1,6 +1,6 @@
from typing import Any, Dict, Mapping, Optional, Sequence, Union from typing import Any, Dict, Mapping, Optional, Sequence, Union
from langchainplus_sdk.evaluation.evaluator import EvaluationResult from langchainplus_sdk.evaluation import EvaluationResult
from langchainplus_sdk.schemas import Example, Run from langchainplus_sdk.schemas import Example, Run
from pydantic import BaseModel from pydantic import BaseModel