From 41ff8d63f931c60a9b1d9b97c6fa01d793b4aedc Mon Sep 17 00:00:00 2001 From: asim-shrestha Date: Tue, 31 Oct 2023 22:08:24 -0700 Subject: [PATCH] =?UTF-8?q?=E2=9C=A8=20Default=20to=20search=20tool?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- next/src/components/index/chat.tsx | 5 ++--- next/src/hooks/useTools.ts | 4 ++-- platform/reworkd_platform/web/api/agent/tools/search.py | 2 +- platform/reworkd_platform/web/api/agent/tools/tools.py | 4 +--- 4 files changed, 6 insertions(+), 9 deletions(-) diff --git a/next/src/components/index/chat.tsx b/next/src/components/index/chat.tsx index cf3aa3ca2a..e1eed4b0da 100644 --- a/next/src/components/index/chat.tsx +++ b/next/src/components/index/chat.tsx @@ -10,7 +10,6 @@ import { ChatMessage } from "../console/ChatMessage"; import ChatWindow from "../console/ChatWindow"; import { ChatWindowTitle } from "../console/ChatWindowTitle"; import Summarize from "../console/SummarizeButton"; -import Expand from "../motions/expand"; import FadeIn from "../motions/FadeIn"; type ChatProps = { @@ -33,7 +32,7 @@ const Chat = (props: ChatProps) => { return ( <> - +
} @@ -63,7 +62,7 @@ const Chat = (props: ChatProps) => { })} - +
{ const obj = z.array(ActiveToolSchema).parse(JSON.parse(data ?? "")); activeTools = allTools.tools.map((db_tool) => { const tool = obj.find((t) => t.name === db_tool.name); - return tool ?? { ...db_tool, active: true }; + return tool ?? { ...db_tool, active: false }; }); } catch (error) { - activeTools = allTools.tools.map((toolModel) => ({ ...toolModel, active: true })); + activeTools = allTools.tools.map((toolModel) => ({ ...toolModel, active: false })); } return activeTools; diff --git a/platform/reworkd_platform/web/api/agent/tools/search.py b/platform/reworkd_platform/web/api/agent/tools/search.py index 1280c24d56..ef9fd2546a 100644 --- a/platform/reworkd_platform/web/api/agent/tools/search.py +++ b/platform/reworkd_platform/web/api/agent/tools/search.py @@ -74,7 +74,7 @@ async def call( CitedSnippet( 0, "\n".join(answer_values), - quote(f"https://www.google.com/search?q={input_str}"), + f"https://www.google.com/search?q={quote(input_str)}", ) ) diff --git a/platform/reworkd_platform/web/api/agent/tools/tools.py b/platform/reworkd_platform/web/api/agent/tools/tools.py index 43691a4927..8921a75927 100644 --- a/platform/reworkd_platform/web/api/agent/tools/tools.py +++ b/platform/reworkd_platform/web/api/agent/tools/tools.py @@ -5,7 +5,6 @@ from reworkd_platform.web.api.agent.tools.code import Code from reworkd_platform.web.api.agent.tools.conclude import Conclude from reworkd_platform.web.api.agent.tools.image import Image -from reworkd_platform.web.api.agent.tools.reason import Reason from reworkd_platform.web.api.agent.tools.search import Search from reworkd_platform.web.api.agent.tools.sidsearch import SID from reworkd_platform.web.api.agent.tools.tool import Tool @@ -30,7 +29,6 @@ def get_external_tools() -> List[Type[Tool]]: return [ # Wikipedia, # TODO: Remove if async doesn't work Image, - Search, Code, SID, ] @@ -38,7 +36,7 @@ def get_external_tools() -> List[Type[Tool]]: def get_default_tools() -> List[Type[Tool]]: return [ - Reason, + Search, Conclude, ]