참고소스 수정본

참고/instructor-main/tests/docs/_concept_groups.py

@@ -0,0 +1,79 @@
+from __future__ import annotations
+
+import glob
+import os
+from collections.abc import Iterable
+
+from pytest_examples import find_examples
+
+CORE = {
+    "alias.md",
+    "dictionary_operations.md",
+    "distillation.md",
+    "enums.md",
+    "fastapi.md",
+    "fields.md",
+    "index.md",
+    "iterable.md",
+    "lists.md",
+    "logging.md",
+    "maybe.md",
+    "models.md",
+    "parallel.md",
+    "partial.md",
+    "philosophy.md",
+    "prompting.md",
+    "typeadapter.md",
+    "typeddicts.md",
+    "types.md",
+    "union.md",
+    "unions.md",
+    "validation.md",
+}
+
+OPERATIONS = {
+    "caching.md",
+    "prompt_caching.md",
+    "raw_response.md",
+    "retrying.md",
+    "error_handling.md",
+}
+
+PROVIDERS = {
+    "from_provider.md",
+    "migration.md",
+    "mode-migration.md",
+    "patching.md",
+    "usage.md",
+}
+
+ADVANCED = {
+    "batch.md",
+    "hooks.md",
+    "multimodal.md",
+    "reask_validation.md",
+    "semantic_validation.md",
+    "templating.md",
+}
+
+
+def concept_paths(names: Iterable[str]) -> list[str]:
+    return [os.path.join("docs", "concepts", name) for name in names]
+
+
+def all_concept_files() -> list[str]:
+    return sorted(glob.glob("docs/concepts/*.md"))
+
+
+def core_concept_files() -> list[str]:
+    excluded = OPERATIONS | PROVIDERS | ADVANCED
+    return [
+        path for path in all_concept_files() if os.path.basename(path) not in excluded
+    ]
+
+
+def collect_examples(files: Iterable[str]):
+    examples = []
+    for markdown_file in files:
+        examples.extend(find_examples(markdown_file))
+    return examples