diff --git a/operations/nlp/dffml_operations_nlp/tools/dffml_docs.py b/operations/nlp/dffml_operations_nlp/tools/dffml_docs.py index c6d5910995..e7cf5564d8 100644 --- a/operations/nlp/dffml_operations_nlp/tools/dffml_docs.py +++ b/operations/nlp/dffml_operations_nlp/tools/dffml_docs.py @@ -99,7 +99,7 @@ class WritePaperSectionContent(BaseModel): ) text: str = Field( json_schema_extra={ - "description": "The content for this section. This should be at least 500 words", + "description": "The content for this section. This should be at least 1000 words. Please include mermaid diagrams to illustrate points when you think necessary", } ) comments: str = Field( @@ -187,8 +187,6 @@ def generate_markdown(write_paper_model: WritePaper): rich_console.print(rich.markdown.Markdown(markdown_content)) -sys.exit(0) - # with open("WritePaper.schema.json", "wt") as fileobj: # print(json.dumps(WritePaperSection.model_json_schema(), indent=4, @@ -874,6 +872,21 @@ def do_llm_call(prompt): data_path.write_text(write_paper.model_dump_json()) +write_paper_instance = write_paper + +# Generate Markdown content +markdown_content = generate_markdown(write_paper_instance) + +md_path.write_text(markdown_content) + +# Print or save the markdown content +import rich.console +import rich.markdown + +rich_console = rich.console.Console(width=80) + +rich_console.print(rich.markdown.Markdown(markdown_content)) + print() sys.exit(0)