diff --git a/ci_scripts/file-check.py b/ci_scripts/file-check.py index 67581e1..ea0e2a2 100644 --- a/ci_scripts/file-check.py +++ b/ci_scripts/file-check.py @@ -7,14 +7,13 @@ folders = BASE_FOLDER.glob("**") def check_files(files): - if len(files) != 2: - raise ValueError(f"Each directory should have two files, got {len(files)}") file_names = [f.name for f in files] if "README.md" not in file_names: raise ValueError(f"Expected to find a README.md file, but found {files}") - other_file = [file for file in files if file.name != "README.md"][0] - if other_file.suffix in (".json", ".yaml"): - load_prompt(other_file) + other_files = [file for file in files if file.name != "README.md"] + for other_file in other_files: + if other_file.suffix in (".json", ".yaml"): + load_prompt(other_file) # TODO: testing for python files diff --git a/prompts/summarize/refine/prompt.yaml b/prompts/summarize/refine/prompt.yaml index d1d1dc2..11b6662 100644 --- a/prompts/summarize/refine/prompt.yaml +++ b/prompts/summarize/refine/prompt.yaml @@ -1,6 +1,6 @@ input_variables: [existing_answer, text] output_parser: null -template: ' +template: " Your job is to produce a final summary\n We have provided an existing summary up to a certain point: {existing_answer}\n We have the opportunity to refine the existing summary @@ -9,5 +9,5 @@ template: ' {text}\n ------------\n Given the new context, refine the original summary - If the context isn't useful, return the original summary.' + If the context isn't useful, return the original summary." template_format: f-string