fixing unit test failure

Author: Lucas-Fernandes-Martins

synthetic_data_kit/core/create.py

@@ -37,7 +37,7 @@ def process_file(
     provider: Optional[str] = None,
     chunk_size: Optional[int] = None,
     chunk_overlap: Optional[int] = None,
-    rolling_summary: bool = False,
+    rolling_summary: Optional[bool] = False,
 ) -> str:
     """Process a file to generate content
     

synthetic_data_kit/generators/qa_generator.py

@@ -33,7 +33,7 @@ def __init__(self,
 
     def generate_summary(self, 
                          document_text: str, 
-                         rolling_summary: bool) -> str:
+                         rolling_summary: Optional[bool] = False) -> str:
         """Generate a summary of the document"""
         verbose = os.environ.get('SDK_VERBOSE', 'false').lower() == 'true'
         if verbose:
@@ -322,7 +322,7 @@ def process_documents(self,
                         documents: List[Dict[str, Any]],
                         num_pairs: int = 25,
                         verbose: bool = False,
-                        rolling_summary: bool = False) -> Dict[str, Any]:
+                        rolling_summary: Optional[bool] = False) -> Dict[str, Any]:
         """Process a list of documents to generate QA pairs without rating"""
         # Set the verbose environment variable
         if verbose: