Add templating agentstack config validation

Author: tcdent

agentstack/generation/files.py

@@ -3,6 +3,7 @@
 import json
 from pathlib import Path
 from pydantic import BaseModel
+from agentstack.utils import get_version
 
 
 DEFAULT_FRAMEWORK = "crewai"
@@ -34,12 +35,21 @@ class ConfigFile(BaseModel):
         Whether the user has opted out of telemetry.
     default_model: Optional[str]
         The default model to use when generating agent configurations.
+    agentstack_version: Optional[str]
+        The version of agentstack used to generate the project.
+    template: Optional[str]
+        The template used to generate the project.
+    template_version: Optional[str]
+        The version of the template system used to generate the project.
     """
 
     framework: str = DEFAULT_FRAMEWORK
     tools: list[str] = []
     telemetry_opt_out: Optional[bool] = None
     default_model: Optional[str] = None
+    agentstack_version: Optional[str] = get_version()
+    template: Optional[str] = None
+    template_version: Optional[str] = None
 
     def __init__(self, path: Union[str, Path, None] = None):
         path = Path(path) if path else Path.cwd()

tests/test_generation_files.py

@@ -7,6 +7,7 @@
     verify_agentstack_project,
     get_framework,
     get_telemetry_opt_out,
+    get_version,
 )
 
 BASE_PATH = Path(__file__).parent
@@ -19,6 +20,9 @@ def test_read_config(self):
         assert config.tools == []
         assert config.telemetry_opt_out is None
         assert config.default_model is None
+        assert config.agentstack_version == get_version()
+        assert config.template is None
+        assert config.template_version is None
 
     def test_write_config(self):
         try:
@@ -30,6 +34,9 @@ def test_write_config(self):
                 config.tools = ["tool1", "tool2"]
                 config.telemetry_opt_out = True
                 config.default_model = "openai/gpt-4o"
+                config.agentstack_version = "0.2.1"
+                config.template = "default"
+                config.template_version = "1"
 
             tmp_data = open(BASE_PATH / "tmp/agentstack.json").read()
             assert (
@@ -41,7 +48,10 @@ def test_write_config(self):
         "tool2"
     ],
     "telemetry_opt_out": true,
-    "default_model": "openai/gpt-4o"
+    "default_model": "openai/gpt-4o",
+    "agentstack_version": "0.2.1",
+    "template": "default",
+    "template_version": "1"
 }"""
             )
         except Exception as e: