Refactor prompt template initialization and update black dependency

Refactor the initialization of the sidekick prompt template for better readability. Remove version pinning for the black dependency in requirements-dev.txt.
TechNickAI · Jul 17, 2024 · 9e26289 · 9e26289
1 parent 56ef0f8
commit 9e26289
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/aicodebot/prompts.py b/aicodebot/prompts.py
@@ -449,7 +449,9 @@ def get_prompt(command, structured_output=False):
             "commit": PromptTemplate(template=COMMIT_TEMPLATE, input_variables=["diff_context", "languages"]),
             "debug": PromptTemplate(template=DEBUG_TEMPLATE, input_variables=["command_output", "languages"]),
             "fun_fact": PromptTemplate(template=FUN_FACT_TEMPLATE, input_variables=["topic"]),
-            "sidekick": PromptTemplate(template=SIDEKICK_TEMPLATE, input_variables=["task", "context", "languages"]),
+            "sidekick": PromptTemplate(
+                template=SIDEKICK_TEMPLATE, input_variables=["task", "context", "languages"]
+            ),
         }
 
         try:

diff --git a/requirements/requirements-dev.txt b/requirements/requirements-dev.txt
@@ -6,7 +6,7 @@
 aicodebot
 
 # For committing, formatting, and dependency management
-black==24.4.2
+black
 isort
 pip-tools
 pre-commit