- update документация - промт для ии полу конфигурируемый
This commit is contained in:
@@ -153,11 +153,49 @@ def test_prompt_catalog_loads():
|
||||
from md2gost.help_content import load_prompt_catalog
|
||||
catalog = load_prompt_catalog()
|
||||
names = {name for name, _title, text in catalog}
|
||||
assert "generate-md.md" in names
|
||||
assert "generate-mirea-report.md" in names
|
||||
assert "generate-pis-custom-report.md" in names
|
||||
assert all(text.strip() for _n, _t, text in catalog)
|
||||
|
||||
|
||||
def test_compose_prompt_with_schemes():
|
||||
from md2gost.diagram_schemes import DiagramScheme
|
||||
from md2gost.help_content import compose_prompt, scheme_prompt_block
|
||||
|
||||
scheme = DiagramScheme(
|
||||
id="c4",
|
||||
title="C4 Container",
|
||||
docs='Person(alias, "Label")',
|
||||
ai_prompt="Рисуй C4. Блок ```uml-c4 без @startuml.",
|
||||
)
|
||||
block = scheme_prompt_block(scheme)
|
||||
assert "uml-c4" in block
|
||||
assert "```c4" in block or "```uml-c4" in block
|
||||
assert "Рисуй C4" in block
|
||||
assert "Person(alias" in block
|
||||
|
||||
base = "Базовый промпт про markdown.\n"
|
||||
out = compose_prompt(base, [scheme])
|
||||
assert out.startswith("Базовый промпт")
|
||||
assert "Дополнение: выбранные схемы" in out
|
||||
assert "uml-c4" in out
|
||||
assert compose_prompt(base, []).strip() == base.strip()
|
||||
assert compose_prompt(base, None).strip() == base.strip()
|
||||
|
||||
|
||||
def test_docs_catalog_loads():
|
||||
from md2gost.help_content import load_docs_catalog
|
||||
catalog = load_docs_catalog()
|
||||
assert catalog, "docs/ catalog should not be empty"
|
||||
names = {name for name, _title, _text in catalog}
|
||||
assert "README.md" in names
|
||||
assert "markdown.md" in names
|
||||
assert catalog[0][0].lower() == "readme.md"
|
||||
assert "Документация" in catalog[0][1]
|
||||
assert all(text.strip() for _n, _t, text in catalog)
|
||||
|
||||
|
||||
def test_listing_continuation_mode():
|
||||
import os
|
||||
import docx
|
||||
|
||||
Reference in New Issue
Block a user