Fix create config CLI command

src/canopy/utils/config.py

@@ -1,6 +1,18 @@
+import os
+import yaml
 from typing import Dict, Any
 
 
+def load_config_template(name: str) -> Dict[str, Any]:
+    if not name.endswith(".yaml"):
+        name += ".yaml"
+
+    config_template_path = os.path.join(os.path.dirname(os.path.dirname(__file__)),
+                                        "config_templates", name)
+    with open(config_template_path, "r") as f:
+        return yaml.safe_load(f)
+
+
 class ConfigurableMixin:
     _DEFAULT_COMPONENTS: Dict[str, type] = {}
 

src/canopy_cli/cli.py

@@ -1,8 +1,8 @@
 import os
+import shutil
 from typing import Dict, Any, Optional, List, Iterable
 
 import click
-import importlib.resources as pkg_resources
 from prompt_toolkit import prompt
 import time
 
@@ -211,38 +211,28 @@ def health(url):
     return
 
 
-@cli.command(help="Writes a config template YAML file to the specified path.")
-@click.argument("path", type=click.Path(), required=True)
-@click.option("--template", default="default", help="The name of the template to use.")
-def create_config(path, template):
+@cli.command(help="Writes the config templates to a directory.")
+@click.argument("out_path", type=click.Path(), required=True)
+def create_config(out_path):
 
-    if not template.endswith('.yaml'):
-        template += '.yaml'
+    if os.path.isfile(out_path):
+        raise CLIError(f"Path expected to be a directory,"
+                       f"but found a file at {out_path}")
 
-    if os.path.exists(path):
-        click.confirm(click.style(f"File {path} already exists. Overwrite?", fg="red"),
+    if os.path.exists(out_path) and os.path.isdir(out_path) and os.listdir(out_path):
+        click.confirm(click.style(f"Path {out_path} is not empty. Overwrite?",
+                                  fg="red"),
                       abort=True)
 
+    project_root = os.path.dirname(os.path.dirname(__file__))
+    config_templates_path = os.path.join(project_root, "canopy", "config_templates")
+
     try:
-        with pkg_resources.open_text('canopy.config', template) as f:
-            config = yaml.safe_load(f)
-    except FileNotFoundError:
-        files = pkg_resources.files('canopy.config')
-        template_names = [f.name for f in files.iterdir()]
-        click.echo(f"Template '{template}' not found."
-                   f"Available templates: {template_names}",
-                   err=True)
-        return
+        shutil.copytree(config_templates_path, out_path, dirs_exist_ok=True)
     except Exception as e:
-        click.echo(f"Failed to load template '{template}'. Reason: {e}"
-                   f"Make sure you pip installed `canopy-sdk`.",
-                   err=True)
-        return
-
-    with open(path, 'w') as dst_file:
-        yaml.dump(config, dst_file)
+        raise CLIError(f"Failed to write config template to {out_path}. Reason:\n{e}")
 
-    click.echo(f"Config template '{template}' written to {path}")
+    click.echo(click.style(f"Config templates written to {out_path}", fg="green"))
 
 
 @cli.command(

tests/system/utils/test_config.py

@@ -1,13 +1,11 @@
 import os
 
 import pytest
-import yaml
 
 from canopy.chat_engine import ChatEngine
 from canopy.context_engine import ContextEngine
 from canopy.knowledge_base import KnowledgeBase
-
-DEFAULT_CONFIG_PATH = 'src/canopy/config/default.yaml'
+from canopy.utils.config import load_config_template
 
 
 @pytest.fixture(scope='module')
@@ -24,8 +22,7 @@ def temp_index_name():
 
 
 def test_default_config_matches_code_defaults(temp_index_name):
-    with open(DEFAULT_CONFIG_PATH) as f:
-        default_config = yaml.safe_load(f)
+    default_config = load_config_template("default.yaml")
     chat_engine_config = default_config['chat_engine']
 
     loaded_chat_engine = ChatEngine.from_config(chat_engine_config)