From 2761a5c3619616b2c1367bc63c05b2f73bf4d00e Mon Sep 17 00:00:00 2001
From: BillSchumacher <34168009+BillSchumacher@users.noreply.github.com>
Date: Sat, 15 Apr 2023 22:18:55 -0500
Subject: [PATCH] Add post_prompt hook

---
 autogpt/config/ai_config.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/autogpt/config/ai_config.py b/autogpt/config/ai_config.py
index d2641b82..23be7286 100644
--- a/autogpt/config/ai_config.py
+++ b/autogpt/config/ai_config.py
@@ -4,7 +4,7 @@ A module that contains the AIConfig class object that contains the configuration
 """
 import os
 from pathlib import Path
-from typing import List, Optional, Type
+from typing import List, Optional
 import yaml
 
 from autogpt.prompts.generator import PromptGenerator
@@ -110,6 +110,8 @@ class AIConfig:
         )
 
         from autogpt.prompts.prompt import build_default_prompt_generator
+        from autogpt.config import Config
+        cfg = Config()
         # Construct full prompt
         full_prompt = (
             f"You are {self.ai_name}, {self.ai_role}\n{prompt_start}\n\nGOALS:\n\n"
@@ -119,6 +121,8 @@ class AIConfig:
         if prompt_generator is None:
             prompt_generator = build_default_prompt_generator()
         prompt_generator.goals = self.ai_goals
+        for plugin in cfg.plugins:
+            prompt_generator = plugin.post_prompt(prompt_generator)
         self.prompt_generator = prompt_generator
         full_prompt += f"\n\n{prompt_generator.generate_prompt_string()}"
         return full_prompt