Skip to content

Commit bcf870d

Browse files
lovets18valovtsov
andauthored
fix: enum values instead of names in testset synthesizers (#2137)
**Change** Using Enum.value instead of Enum.name It seems that value (description) for QueryStyle is more logical than using names. The current name and value are almost identical, but this might change in the future Co-authored-by: valovtsov <[email protected]>
1 parent 4be53fa commit bcf870d

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

ragas/src/ragas/testset/synthesizers/multi_hop/base.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -167,8 +167,8 @@ async def _generate_sample(
167167
persona=scenario.persona,
168168
themes=scenario.combinations,
169169
context=reference_context,
170-
query_length=scenario.length.name,
171-
query_style=scenario.style.name,
170+
query_length=scenario.length.value,
171+
query_style=scenario.style.value,
172172
)
173173
response = await self.generate_query_reference_prompt.generate(
174174
data=prompt_input, llm=self.llm, callbacks=callbacks

ragas/src/ragas/testset/synthesizers/single_hop/base.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -128,8 +128,8 @@ async def _generate_sample(
128128
persona=scenario.persona,
129129
term=scenario.term,
130130
context=reference_context,
131-
query_length=scenario.length.name,
132-
query_style=scenario.style.name,
131+
query_length=scenario.length.value,
132+
query_style=scenario.style.value,
133133
)
134134
response = await self.generate_query_reference_prompt.generate(
135135
data=prompt_input, llm=self.llm, callbacks=callbacks

0 commit comments

Comments
 (0)