tools 6

config.yml

@@ -18,4 +18,9 @@ default_presence_penalty: 0
 ### Logging ###
 wandb_enabled: false
 
+### Tools ###
+tools_enabled: true
+active_tools:
+  - 'generate_image'
+
 working_codebase_dirname: 'generated_code' # A folder that exists 

helpers.py

@@ -1,6 +1,12 @@
 import yaml
 import matplotlib.pyplot as plt
 
+# Load the configuration
+with open('config.yml', 'r') as f:
+    config = yaml.safe_load(f)
+
+tools_enabled = config['tools_enabled']
+
 def load_pipeline(file_path):
     """Load a pipeline configuration from a YAML file."""
     print("\033[93mLoading pipeline...\033[00m")
@@ -13,6 +19,11 @@ def load_system_prompt(module_name):
     with open('general_system.txt', 'r') as file:
         system_prompt = file.read().replace('\n', '')
 
+    if tools_enabled:
+        with open(f'tools/tool_prompt.txt', 'r') as file:
+            tool_prompt = file.read().replace('\n', '')
+        system_prompt += '\n\n' + tool_prompt + '\n'
+
     with open(f'system_prompts/{module_name}.txt', 'r') as file:
         module_prompt = file.read().replace('\n', '')