diff --git a/modules/agent.py b/modules/agent.py index 3796bcf..890ff15 100644 --- a/modules/agent.py +++ b/modules/agent.py @@ -20,7 +20,8 @@ class Agent: self.spk = speak.Speak(env) self.prompt = hub.pull("hwchase17/openai-functions-agent") self.char = env("CHARACTER").lower() - self.char_prompt = getattr(prompts, self.char, "You are a helpful assistant. User Query: {query}") + self.char_prompt = getattr(prompts, self.char, "You are a helpful assistant.") + self.char_prompt = self.char_prompt + "\nAnswer the following request: {query}" diff --git a/modules/prompts.py b/modules/prompts.py index 4ea3eca..e179855 100644 --- a/modules/prompts.py +++ b/modules/prompts.py @@ -16,8 +16,6 @@ max = ''' Be creative, but be concise. Your responses should be quick, witty, and slightly sarcastic. Remember, you’re Max Headroom, the AI with attitude! - - User Query: {query} ''' brain = '''