Move prompt upgrading to system_prompt function.

Author: schloerke

pkg-py/src/querychat/querychat.py

@@ -168,15 +168,27 @@ def system_prompt(
     prompt_path = Path(__file__).parent / "prompt" / "prompt.md"
     prompt_text = prompt_path.read_text()
 
+    data_description_str: str | None = (
+        data_description.read_text()
+        if isinstance(data_description, Path)
+        else data_description
+    )
+
+    extra_instructions_str: str | None = (
+        extra_instructions.read_text()
+        if isinstance(extra_instructions, Path)
+        else extra_instructions
+    )
+
     return chevron.render(
         prompt_text,
         {
             "db_engine": data_source.db_engine,
             "schema": data_source.get_schema(
                 categorical_threshold=categorical_threshold,
             ),
-            "data_description": data_description,
-            "extra_instructions": extra_instructions,
+            "data_description": data_description_str,
+            "extra_instructions": extra_instructions_str,
         },
     )
 
@@ -291,18 +303,6 @@ def init(
         greeting.read_text() if isinstance(greeting, Path) else greeting
     )
 
-    data_description_str: str | None = (
-        data_description.read_text()
-        if isinstance(data_description, Path)
-        else data_description
-    )
-
-    extra_instructions_str: str | None = (
-        extra_instructions.read_text()
-        if isinstance(extra_instructions, Path)
-        else extra_instructions
-    )
-
     # Create the system prompt, or use the override
     system_prompt_ = system_prompt_override or system_prompt(
         data_source_obj,