reformulate query if the query is duplicate #69
Conversation
| @@ -150,6 +152,29 @@ def run(self, task: str) -> str: | |||
| action = self.output_parser.parse(assistant_reply) | |||
| print("action:", action) | |||
| tools = {t.name: t for t in self.tools} | |||
| if action == previous_action: | |||
| if action.name == "Search" or action.name == "Context Search": | |||
There was a problem hiding this comment.
I recommend we drop this condition since we may want to check duplication for all tools. (We don't need to worry about the finish tool for now as it will be separated and it will also break the loop)
| if action == previous_action: | ||
| if action.name == "Search" or action.name == "Context Search": | ||
| print("Action name: ", action.name, "\nStart reformulating the query") | ||
| instruction = ( |
There was a problem hiding this comment.
If I understand correctly, this prompt only asks for a different input for the same tool. Is this intended?
| f"Reformulated query:\n\n" | ||
| ) | ||
| openai.api_key = environ.get("OPENAI_KEY") | ||
| response = openai.Completion.create( |
There was a problem hiding this comment.
We may want to use self.llm so that we don't need to provide extra configurations of the LLM here
apps/slackbot/task_agent.py
Outdated
| @@ -150,6 +152,29 @@ def run(self, task: str) -> str: | |||
| action = self.output_parser.parse(assistant_reply) | |||
| print("action:", action) | |||
| tools = {t.name: t for t in self.tools} | |||
| if action == previous_action: | |||
| if action.name == "Search" or action.name == "Context Search": | |||
| print("Action name: ", action.name, "\nStart reformulating the query") | |||
There was a problem hiding this comment.
remove test print stmts
|
As many of these changes do not influence the main functionality, I will merge this PR and make these required changes as new issues. |
|
please rebase this @YujingYang666777 |
No description provided.