Skip to content
Snippets Groups Projects
Unverified Commit 5148c806 authored by Yue Fei's avatar Yue Fei Committed by GitHub
Browse files

Fix incomplete json output in guideline evaluator (#12646)

parent 8795368b
Branches
Tags
No related merge requests found
...@@ -59,6 +59,7 @@ class GuidelineEvaluator(BaseEvaluator): ...@@ -59,6 +59,7 @@ class GuidelineEvaluator(BaseEvaluator):
llm: Optional[LLM] = None, llm: Optional[LLM] = None,
guidelines: Optional[str] = None, guidelines: Optional[str] = None,
eval_template: Optional[Union[str, BasePromptTemplate]] = None, eval_template: Optional[Union[str, BasePromptTemplate]] = None,
output_parser: Optional[PydanticOutputParser] = None,
# deprecated # deprecated
service_context: Optional[ServiceContext] = None, service_context: Optional[ServiceContext] = None,
) -> None: ) -> None:
...@@ -71,7 +72,9 @@ class GuidelineEvaluator(BaseEvaluator): ...@@ -71,7 +72,9 @@ class GuidelineEvaluator(BaseEvaluator):
else: else:
self._eval_template = eval_template or DEFAULT_EVAL_TEMPLATE self._eval_template = eval_template or DEFAULT_EVAL_TEMPLATE
self._output_parser = PydanticOutputParser(output_cls=EvaluationData) self._output_parser = output_parser or PydanticOutputParser(
output_cls=EvaluationData
)
self._eval_template.output_parser = self._output_parser self._eval_template.output_parser = self._output_parser
def _get_prompts(self) -> PromptDictType: def _get_prompts(self) -> PromptDictType:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment