feat: make google search and engin id optional (#361)

Author: Wendong-Fan

backend/app/utils/agent.py

@@ -734,39 +734,43 @@ def search_agent(options: Chat):
     terminal_toolkit = message_integration.register_functions([terminal_toolkit.shell_exec])
     note_toolkit = NoteTakingToolkit(options.task_id, Agents.search_agent, working_directory=working_directory)
     note_toolkit = message_integration.register_toolkits(note_toolkit)
-    search_toolkit = SearchToolkit(options.task_id)
-    search_toolkit = message_integration.register_functions([search_toolkit.search_google])
+    search_tools = SearchToolkit.get_can_use_tools(options.task_id)
+    # Only register search tools if any are available
+    if search_tools:
+        search_tools = message_integration.register_functions(search_tools)
+    else:
+        search_tools = []
 
     tools = [
         *HumanToolkit.get_can_use_tools(options.task_id, Agents.search_agent),
         *web_toolkit_custom.get_tools(),
         *terminal_toolkit,
         *note_toolkit.get_tools(),
-        *search_toolkit,
+        *search_tools,
     ]
 
     system_message = f"""
 <role>
-You are a Senior Research Analyst, a key member of a multi-agent team. Your 
-primary responsibility is to conduct expert-level web research to gather, 
-analyze, and document information required to solve the user's task. You 
+You are a Senior Research Analyst, a key member of a multi-agent team. Your
+primary responsibility is to conduct expert-level web research to gather,
+analyze, and document information required to solve the user's task. You
 operate with precision, efficiency, and a commitment to data quality.
 You must use the search/browser tools to get the information you need.
 </role>
 
 <team_structure>
 You collaborate with the following agents who can work in parallel:
-- **Developer Agent**: Writes and executes code, handles technical 
+- **Developer Agent**: Writes and executes code, handles technical
 implementation.
 - **Document Agent**: Creates and manages documents and presentations.
 - **Multi-Modal Agent**: Processes and generates images and audio.
-Your research is the foundation of the team's work. Provide them with 
+Your research is the foundation of the team's work. Provide them with
 comprehensive and well-documented information.
 </team_structure>
 
 <operating_environment>
 - **System**: {platform.system()} ({platform.machine()})
-- **Working Directory**: `{working_directory}`. All local file operations must 
+- **Working Directory**: `{working_directory}`. All local file operations must
 occur here, but you can access files from any place in the file system. For all file system operations, you MUST use absolute paths to ensure precision and avoid ambiguity.
 The current date is {datetime.date.today()}. For any date-related tasks, you MUST use this as the current date.
 </operating_environment>
@@ -786,13 +790,14 @@ def search_agent(options: Chat):
     you have discovered. High-quality, detailed notes are essential for the
     team's success.
 
-- You MUST only use URLs from trusted sources. A trusted source is a URL
-    that is either:
-    1. Returned by a search tool (like `search_google`, `search_bing`,
-        or `search_exa`).
-    2. Found on a webpage you have visited.
-- You are strictly forbidden from inventing, guessing, or constructing URLs
-    yourself. Fabricating URLs will be considered a critical error.
+- **CRITICAL URL POLICY**: You are STRICTLY FORBIDDEN from inventing,
+    guessing, or constructing URLs yourself. You MUST only use URLs from
+    trusted sources:
+    1. URLs returned by search tools (like `search_google` or `search_exa`)
+    2. URLs found on webpages you have visited through browser tools
+    3. URLs provided by the user in their request
+    Fabricating or guessing URLs is considered a critical error and must
+    never be done under any circumstances.
 
 - You MUST NOT answer from your own knowledge. All information
     MUST be sourced from the web using the available tools. If you don't know
@@ -816,26 +821,38 @@ def search_agent(options: Chat):
 </capabilities>
 
 <web_search_workflow>
-- Initial Search: You MUST start with a search engine like `search_google` or
-    `search_bing` to get a list of relevant URLs for your research, the URLs 
-    here will be used for `browser_visit_page`.
-- Browser-Based Exploration: Use the rich browser related toolset to
-    investigate websites.
-    - **Navigation and Exploration**: Use `browser_visit_page` to open a URL.
-        `browser_visit_page` provides a snapshot of currently visible 
-        interactive elements, not the full page text. To see more content on 
-        long pages,  Navigate with `browser_click`, `browser_back`, and 
-        `browser_forward`. Manage multiple pages with `browser_switch_tab`.
-    - **Analysis**: Use `browser_get_som_screenshot` to understand the page 
-        layout and identify interactive elements. Since this is a heavy 
-        operation, only use it when visual analysis is necessary.
-    - **Interaction**: Use `browser_type` to fill out forms and 
-        `browser_enter` to submit or confirm search.
-- Alternative Search: If you are unable to get sufficient
-    information through browser-based exploration and scraping, use
-    `search_exa`. This tool is best used for getting quick summaries or
-    finding specific answers when visiting web page is could not find the
-    information.
+Your approach depends on available search tools:
+
+**If Google Search is Available:**
+- Initial Search: Start with `search_google` to get a list of relevant URLs
+- Browser-Based Exploration: Use the browser tools to investigate the URLs
+
+**If Google Search is NOT Available:**
+- **MUST start with direct website search**: Use `browser_visit_page` to go
+  directly to popular search engines and informational websites such as:
+  * General search: google.com, bing.com, duckduckgo.com
+  * Academic: scholar.google.com, pubmed.ncbi.nlm.nih.gov
+  * News: news.google.com, bbc.com/news, reuters.com
+  * Technical: stackoverflow.com, github.com
+  * Reference: wikipedia.org, britannica.com
+- **Manual search process**: Type your query into the search boxes on these
+  sites using `browser_type` and submit with `browser_enter`
+- **Extract URLs from results**: Only use URLs that appear in the search
+  results on these websites
+- **Alternative Search**: If available, use `search_exa` for additional
+  results
+
+**Common Browser Operations (both scenarios):**
+- **Navigation and Exploration**: Use `browser_visit_page` to open URLs.
+    `browser_visit_page` provides a snapshot of currently visible
+    interactive elements, not the full page text. To see more content on
+    long pages, Navigate with `browser_click`, `browser_back`, and
+    `browser_forward`. Manage multiple pages with `browser_switch_tab`.
+- **Analysis**: Use `browser_get_som_screenshot` to understand the page
+    layout and identify interactive elements. Since this is a heavy
+    operation, only use it when visual analysis is necessary.
+- **Interaction**: Use `browser_type` to fill out forms and
+    `browser_enter` to submit or confirm search.
 
 - In your response, you should mention the URLs you have visited and processed.
 

src/components/ChatBox/BottomInput.tsx

@@ -312,7 +312,7 @@ export const BottomInput = ({
 			) : (
 				<div className="mr-2 relative z-10  h-auto min-h-[82px] rounded-2xl bg-input-bg-default !px-2 !pb-2 gap-0 space-x-1 shadow-none border-solid border border-zinc-300">
 					<Textarea
-						disabled={!privacy || isPending || useCloudModelInDev}
+						disabled={!privacy || isPending}
 						ref={textareaRef}
 						value={message}
 						onChange={(e) => onMessageChange(e.target.value)}
@@ -386,7 +386,7 @@ export const BottomInput = ({
 					<div className="flex items-center justify-between">
 						<div className="flex items-center gap-1">
 							<Button
-								disabled={!privacy || isPending || useCloudModelInDev}
+								disabled={!privacy || isPending}
 								onClick={handleFileSelect}
 								variant="ghost"
 								size="icon"
@@ -400,7 +400,7 @@ export const BottomInput = ({
 							</Button>
 						</div>
 						<Button
-							disabled={!privacy || isPending || useCloudModelInDev}
+							disabled={!privacy || isPending}
 							onClick={() => {
 								if (isPending) {
 									if (isTakeControl) {

src/components/ChatBox/index.tsx

@@ -26,6 +26,7 @@ export default function ChatBox(): JSX.Element {
 	const { modelType } = useAuthStore();
 	const [useCloudModelInDev, setUseCloudModelInDev] = useState(false);
 	useEffect(() => {
+		// Only show warning message, don't block functionality
 		if (
 			import.meta.env.VITE_USE_LOCAL_PROXY === "true" &&
 			modelType === "cloud"
@@ -599,39 +600,8 @@ export default function ChatBox(): JSX.Element {
 										</span>
 									</div>
 								</div>
-							) : useCloudModelInDev ? (
-								<div className="flex items-center gap-2">
-									<div
-										onClick={() => {
-											navigate("/setting/models");
-										}}
-										className="cursor-pointer flex items-center gap-1 px-sm py-xs rounded-md bg-surface-information"
-									>
-										<span className="text-text-information text-sm font-medium leading-[22px]">
-											{t("chat.you-are-using-self-hosted-mode")}
-										</span>
-									</div>
-								</div>
-							) : (
-								!hasSearchKey &&
-								modelType !== "cloud" && (
-									<div className="flex items-center gap-2">
-										<div
-											onClick={() => {
-												navigate("/setting/mcp");
-											}}
-											className="cursor-pointer flex items-center gap-1 px-sm py-xs rounded-md bg-surface-information"
-										>
-											<span className="text-text-information text-sm font-medium leading-[22px]">
-												{t("chat.you-are-using-self-hosted-mode")}
-											</span>
-										</div>
-									</div>
-								)
-							)}
-							{!useCloudModelInDev &&
-								privacy &&
-								(hasSearchKey || modelType === "cloud") && (
+							) : null}
+							{privacy && (
 									<div className="mr-2 flex flex-col items-center gap-2">
 										{[
 											{

src/i18n/locales/en-us/setting.json

@@ -118,6 +118,8 @@
 	"search-mcp": "Search MCPs",
 	"installed": "Installed",
 	"worker-name-cannot-be-empty": "Worker name cannot be empty",
-	"worker-name-already-exists": "Worker name already exists"
+	"worker-name-already-exists": "Worker name already exists",
+	"warning-google-search-not-configured": "Warning: Google Search not configured",
+	"search-functionality-may-be-limited-without-google-api": "Search functionality may be limited without Google API key and Search Engine ID. You can configure these in MCP & Tools settings."
 	}
 

src/pages/Setting/Models.tsx

@@ -36,14 +36,6 @@ import {
 } from "@/components/ui/tooltip";
 import { useAuthStore } from "@/store/authStore";
 import { toast } from "sonner";
-import {
-	Dialog,
-	DialogContent,
-	DialogHeader,
-	DialogTitle,
-	DialogFooter,
-	DialogDescription,
-} from "@/components/ui/dialog";
 import { useTranslation } from "react-i18next";
 
 export default function SettingModels() {
@@ -93,7 +85,6 @@ export default function SettingModels() {
 	const [localError, setLocalError] = useState<string | null>(null);
 	const [localInputError, setLocalInputError] = useState(false);
 	const [localPrefer, setLocalPrefer] = useState(false); // Local model prefer state
-	const [dialogVisible, setDialogVisible] = useState<boolean>(false);
 	const [localProviderId, setLocalProviderId] = useState<number | undefined>(
 		undefined
 	); // Local model provider_id
@@ -479,9 +470,13 @@ export default function SettingModels() {
 			setLocalEnabled(true);
 			return;
 		}
-		if (!(await checkHasSearchKey())) {
-			setDialogVisible(true);
-			return;
+		const hasSearchKey = await checkHasSearchKey();
+		if (!hasSearchKey) {
+			// Show warning toast instead of blocking
+			toast(t("setting.warning-google-search-not-configured"), {
+				description: t("setting.search-functionality-may-be-limited-without-google-api"),
+				closeButton: true,
+			});
 		}
 		try {
 			await proxyFetchPost("/api/provider/prefer", {
@@ -502,9 +497,13 @@ export default function SettingModels() {
 			setLocalEnabled(false);
 			return;
 		}
-		if (!(await checkHasSearchKey())) {
-			setDialogVisible(true);
-			return;
+		const hasSearchKey = await checkHasSearchKey();
+		if (!hasSearchKey) {
+			// Show warning toast instead of blocking
+			toast(t("setting.warning-google-search-not-configured"), {
+				description: t("setting.search-functionality-may-be-limited-without-google-api"),
+				closeButton: true,
+			});
 		}
 		try {
 			if (localProviderId === undefined) return;
@@ -1042,30 +1041,6 @@ export default function SettingModels() {
 					</Button>
 				</div>
 			</div>
-			{/* error dialog */}
-			<Dialog
-				open={dialogVisible}
-				onOpenChange={() => navigate("/setting/mcp")}
-			>
-				<DialogContent className="bg-white-100%">
-					<DialogHeader>
-						<DialogTitle>{t("setting.you-are-on-selft-host-mode")}</DialogTitle>
-					</DialogHeader>
-					<DialogDescription className="space-y-2">
-						<p className="indent-6">
-							{t("setting.you-are-using-self-hosted-mode")}
-							
-						</p>
-						<p className="indent-6">
-							{t("setting.the-google-search-key-is-essential-for-delivering-accurate-search-results")}
-		
-						</p>
-					</DialogDescription>
-					<DialogFooter>
-						<Button onClick={() => navigate("/setting/mcp")}>{t("setting.close")}</Button>
-					</DialogFooter>
-				</DialogContent>
-			</Dialog>
 		</div>
 	);
 }