DH-5638/change tests

MohammadrezaPourreza · MohammadrezaPourreza · commit c85b654ece73 · 2024-03-22T16:32:24.000-04:00
diff --git a/dataherald/tests/sql_generator/test_generator.py b/dataherald/tests/sql_generator/test_generator.py
@@ -18,6 +18,7 @@ def generate_response(
         user_prompt: Prompt,
         database_connection: DatabaseConnection,
         context: List[dict] = None,  # noqa: ARG002
+        metadata: dict = None,  # noqa: ARG002
     ) -> SQLGeneration:
         return SQLGeneration(
             question_id="651f2d76275132d5b65175eb",
