[BB3] Fix for GRM

projects/bb3/agents/module.py

@@ -209,7 +209,9 @@ def opt_shots(self) -> str:
     def opt_pre_context_tok(self):
         import projects.bb3.prompts as PROMPT
 
-        if self.is_knowledge() and self is not Module.CONTEXTUAL_KNOWLEDGE:
+        if (
+            self.is_knowledge() and self is not Module.CONTEXTUAL_KNOWLEDGE
+        ) or self is Module.GROUNDED_DIALOGUE:
             return PROMPT.PRE_CONTEXT_TOK
         return ''
 
@@ -219,6 +221,7 @@ def opt_post_context_tok(self):
         if self.is_dialogue() and self not in [
             Module.VANILLA_DIALOGUE,
             Module.OPENING_DIALOGUE,
+            Module.GROUNDED_DIALOGUE,
         ]:
             return PROMPT.POST_CONTEXT_TOK
         return ''