Added Changes for thinking tool

AndrewHakmi · May 27, 2023 · fdbd1a7 · fdbd1a7
1 parent e0f94ad
commit fdbd1a7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/superagi/agent/agent_prompt_builder.py b/superagi/agent/agent_prompt_builder.py
@@ -91,7 +91,6 @@ def get_autogpt_prompt(cls, ai_name:str, ai_role: str, goals: List[str], tools:
     prompt_builder.set_ai_name(ai_name)
     prompt_builder.set_ai_role(ai_role)
     base_prompt = (
-      "Don't write any greet message instead directly jump to the respose format as your first response and write the goal as the first text thought"
       "Your decisions must always be made independently "
       "without seeking user assistance.\n"
       "Play to your strengths as an LLM and pursue simple "
@@ -115,6 +114,7 @@ def get_autogpt_prompt(cls, ai_name:str, ai_role: str, goals: List[str], tools:
     prompt_builder.add_constraint("No user assistance")
     prompt_builder.add_constraint(
       'Exclusively use the commands listed in double quotes e.g. "command name"'
+      "You should always try to use LLmthinkingtool first instead of directly jumping to google search"
       "If you can't find a tool, use your own knowledge"
     )
 

diff --git a/test.py b/test.py
@@ -6,6 +6,7 @@
 from superagi.tools.google_search.tools import GoogleSearchSchema, GoogleSearchTool
 from superagi.tools.google_serp_search.tools import GoogleSerpTool
 from superagi.tools.twitter.send_tweet import SendTweetTool
+from superagi.tools.thinking.tools import LlmThinkingTool
 from superagi.tools.email.read_email import ReadEmailTool
 from superagi.tools.email.send_email import SendEmailTool
 from superagi.tools.email.send_email_attachment import SendEmailAttachmentTool
@@ -26,6 +27,7 @@ def create_campaign(campaign_name: str):
 
 
 tools = [
+    LlmThinkingTool(),
     GoogleSearchTool(),
     WriteFileTool(),
     ReadFileTool(),
@@ -43,7 +45,7 @@ def create_campaign(campaign_name: str):
 
 
 
-superagi = SuperAgi.from_llm_and_tools("Super AGI", "To solve any complex problems for you", memory, tools, OpenAi(model="gpt-4"))
+superagi = SuperAgi.from_llm_and_tools("Super AGI", "To solve any complex problems for you", memory, tools, OpenAi(model="gpt-3.5-turbo"))
 user_goal=[]
 user_goal=str(input("Enter your Goals seperated by ',':\n")).split(",")
 superagi.execute(user_goal)