feat: added env var KONTEXT_COPILOT_DEFAULT_SYS_PROMPT to set differe… (

#50) * feat: added env var KONTEXT_COPILOT_DEFAULT_SYS_PROMPT to set different system propmt. * docs: added codeowners for pr review.
kontext-tech · Sep 10, 2024 · 87f5c3d · 87f5c3d
1 parent ddd7765
commit 87f5c3d
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 1 deletion.
diff --git a/CODEOWNERS b/CODEOWNERS
@@ -0,0 +1 @@
+@kontext-tech/kontext-copilot-reviewer
diff --git a/kontext_copilot/copilot/_prompt_factory.py b/kontext_copilot/copilot/_prompt_factory.py
@@ -51,7 +51,7 @@ def create_system_prompt(
         logger.info("Creating system prompt for data source: %s", data_source.name)
 
         prompt_model = PromptFactory.get_prompt_template_by_id(
-            "system-prompt-da-metadata"
+            os.getenv("KONTEXT_COPILOT_DEFAULT_SYS_PROMPT", "system-prompt-da-metadata")
         )
 
         logger.debug("Prompt template: %s", prompt_model)

diff --git a/kontext_copilot/copilot/prompts.json b/kontext_copilot/copilot/prompts.json
@@ -16,6 +16,14 @@
          "user_input": null,
          "system_defined": true
       },
+      {
+         "id": "system-prompt-da-metadata-refined",
+         "name": "System Prompt - Data Analytics (using Metadata)",
+         "prompt": null,
+         "system_prompt": "./prompts/system-prompt-da-metadata-refined.md",
+         "user_input": null,
+         "system_defined": true
+      },
       {
          "id": "sentiment-analysis",
          "name": "Sentiment Analysis",