superagent-ai · homanp · Jun 8, 2023 · Jun 8, 2023 · Jun 8, 2023 · Jun 8, 2023
diff --git a/.env.example b/.env.example
@@ -10,6 +10,7 @@ ANTHROPIC_API_KEY=
 COHERE_API_KEY=
 BING_SUBSCRIPTION_KEY=
 BING_SEARCH_URL=
+WOLFRAM_ALPHA_APPID=
 
 #For Azure OpenAI
 AZURE_API_KEY=

diff --git a/README.md b/README.md
@@ -77,5 +77,3 @@ We appreciate your contributions and aim to make it easy for anyone to create an
 ## Support
 
 We appreciate all the support you can give us, either with contributions, feedback, bug reports or feature requests. Drop a star and share Superagent to the world!
-
-[![Star History Chart](https://api.star-history.com/svg?repos=homanp/superagent&type=Date)](https://superagent.sh)
diff --git a/app/lib/agents/base.py b/app/lib/agents/base.py
@@ -15,7 +15,7 @@
     agent_template,
     default_chat_prompt,
 )
-from app.lib.tools import get_search_tool
+from app.lib.tools import get_search_tool, get_wolfram_alpha_tool
 
 
 class AgentBase:
@@ -73,7 +73,10 @@ def _get_tool(self) -> Any:
             if self.tool.type == "SEARCH":
                 tools = get_search_tool()
 
-                return tools
+            if self.tool.type == "WOLFRAM_ALPHA":
+                tools = get_wolfram_alpha_tool()
+
+            return tools
 
         except Exception:
             return None

diff --git a/app/lib/tools.py b/app/lib/tools.py
@@ -3,6 +3,7 @@
 from decouple import config
 from langchain.agents import Tool
 from langchain.utilities import BingSearchAPIWrapper
+from langchain.utilities.wolfram_alpha import WolframAlphaAPIWrapper
 
 
 def get_search_tool() -> List:
@@ -19,3 +20,16 @@ def get_search_tool() -> List:
     ]
 
     return tools
+
+
+def get_wolfram_alpha_tool() -> List:
+    wolfram = WolframAlphaAPIWrapper()
+    tools = [
+        Tool(
+            name="Wolfram Alpha",
+            func=wolfram.run,
+            description="useful for when you need to do computation",
+        )
+    ]
+
+    return tools
diff --git a/poetry.lock b/poetry.lock
diff --git a/prisma/migrations/20230608154711_tool_wolfram_alpha/migration.sql b/prisma/migrations/20230608154711_tool_wolfram_alpha/migration.sql
@@ -0,0 +1,2 @@
+-- AlterEnum
+ALTER TYPE "ToolType" ADD VALUE 'WOLFRAM_ALPHA';
diff --git a/prisma/schema.prisma b/prisma/schema.prisma
@@ -29,6 +29,7 @@ enum DocumentType {
 enum ToolType {
   BROWSER
   SEARCH
+  WOLFRAM_ALPHA
 }
 
 model User {

diff --git a/pyproject.toml b/pyproject.toml
@@ -30,6 +30,7 @@ pinecone-client = "^2.2.1"
 tiktoken = "^0.4.0"
 pypdf = "^3.8.1"
 cohere = "^4.5.1"
+wolframalpha = "^5.0.0"
 
 
 [build-system]
Original file line number	Diff line number	Diff line change
Expand Up		@@ -77,5 +77,3 @@ We appreciate your contributions and aim to make it easy for anyone to create an
		## Support

		We appreciate all the support you can give us, either with contributions, feedback, bug reports or feature requests. Drop a star and share Superagent to the world!

		[![Star History Chart](https://api.star-history.com/svg?repos=homanp/superagent&type=Date)](https://superagent.sh)
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,2 @@
		-- AlterEnum
		ALTER TYPE "ToolType" ADD VALUE 'WOLFRAM_ALPHA';
-Original file line number
+Diff line change
@@ Expand Up / @@ -29,6 +29,7 @@ enum DocumentType { @@
     enum ToolType {
       BROWSER
       SEARCH
+      WOLFRAM_ALPHA
     }
     model User {
@@ Expand Down @@