format fixes

src/ragas/testsetv3/graph.py

@@ -1,7 +1,7 @@
 import uuid
-
-from graphene import Argument, Enum, Field, JSONString, List, ObjectType, Schema, String
-
+import typing as t
+from graphene import Argument, Field, JSONString, List, ObjectType, Schema, String
+from enum import Enum
 
 class NodeType(Enum):
  DOC = "doc"
@@ -13,7 +13,7 @@ class NodeLevel(Enum):
  LEVEL_1 = 1
  LEVEL_2 = 2
 
- def next_level(self):
+ def next_level(self) -> t.Optional['NodeLevel']:
  level_values = list(NodeLevel)
  current_index = level_values.index(self)
  next_index = current_index + 1

src/ragas/testsetv3/questions/abstract.py

@@ -290,15 +290,19 @@ async def generate_question(
  if critic_verdict:
  source = await self.retrieve_chunks(question, current_nodes, kwargs)
  question = await self.modify_question(question)
- answer = await self.generate_answer(question, source)
- return QAC(
- question=question,
- answer=answer,
- source=source,
- name=self.name,
- style=self.style,
- length=self.length,
- )
+ if source:
+ answer = await self.generate_answer(question, source)
+ return QAC(
+ question=question,
+ answer=answer,
+ source=source,
+ name=self.name,
+ style=self.style,
+ length=self.length,
+ )
+ else:
+ logger.warning("source not found: %s", question)
+ return QAC()
  else:
  logger.warning("Critic rejected the question: %s", question)
  return QAC()

src/ragas/testsetv3/questions/base.py

@@ -79,7 +79,7 @@ def retrieve_chunks(
 
  async def modify_question(self, question: str) -> str:
  assert self.llm is not None, "LLM is not initialized"
- question_modification_prompt_ = self.question_modification_prompt
+ question_modification_prompt_ = self.question_modification_prompt.copy()
  examples = [
  example
  for example in self.question_modification_prompt.examples