@probelabs/probe 0.6.0-rc229 → 0.6.0-rc231

package/build/tools/common.js

@@ -386,6 +386,30 @@ User: Check system info
 </examples>
 `;
 
+export const googleSearchToolDefinition = `
+## gemini_google_search (Gemini Built-in)
+Description: Web search powered by Google. This is a built-in Gemini capability that automatically searches the web when the model needs current information. The model decides when to search and integrates results directly into its response with source citations.
+
+This tool is invoked automatically by the model — you do NOT need to use XML tool calls for it. Simply ask questions that require up-to-date or real-world information and the model will search the web as needed.
+
+Capabilities:
+- Real-time web search with grounded citations
+- Automatic query generation and result synthesis
+- Source attribution with URLs
+`;
+
+export const urlContextToolDefinition = `
+## gemini_url_context (Gemini Built-in)
+Description: URL content reader powered by Google. This is a built-in Gemini capability that automatically fetches and analyzes the content of URLs mentioned in the conversation. When you include URLs in your message, the model can read and understand their content.
+
+This tool is invoked automatically by the model — you do NOT need to use XML tool calls for it. Simply include URLs in your message and the model will fetch and analyze their content.
+
+Capabilities:
+- Fetch and read web page content from URLs in the prompt
+- Supports up to 20 URLs per request
+- Processes HTML content (does not execute JavaScript)
+`;
+
 export const searchDescription = 'Search code in the repository. Free-form questions are accepted, but Elasticsearch-style keyword queries work best. Use this tool first for any code-related questions.';
 export const queryDescription = 'Search code using ast-grep structural pattern matching. Use this tool to find specific code structures like functions, classes, or methods.';
 export const extractDescription = 'Extract code blocks from files based on file paths and optional line numbers. Use this tool to see complete context after finding relevant files.';

package/build/utils/path-validation.js

@@ -55,9 +55,9 @@ export function safeRealpath(inputPath) {
  * - Does NOT restrict access to specific directories (that's the responsibility
  *   of higher-level components like ProbeAgent with allowedFolders)
  *
- * @param {string} inputPath - The path to validate
+ * @param {string} inputPath - The path to validate (can be a file or directory; file paths are resolved to their parent directory)
  * @param {string} [defaultPath] - Default path to use if inputPath is not provided
- * @returns {Promise<string>} Normalized absolute path
+ * @returns {Promise<string>} Normalized absolute directory path. If inputPath is a file, returns its parent directory.
  * @throws {PathError} If the path is invalid or doesn't exist
  */
 export async function validateCwdPath(inputPath, defaultPath = process.cwd()) {
@@ -72,6 +72,32 @@ export async function validateCwdPath(inputPath, defaultPath = process.cwd()) {
 	try {
 		const stats = await fs.stat(normalizedPath);
 		if (!stats.isDirectory()) {
+			// If the path is a file, resolve to its parent directory
+			// This handles cases where a file path is passed as cwd
+			// Use safeRealpath to resolve symlinks before extracting parent directory
+			const resolvedPath = safeRealpath(normalizedPath);
+			const dirPath = path.dirname(resolvedPath);
+			try {
+				const dirStats = await fs.stat(dirPath);
+				if (dirStats.isDirectory()) {
+					return safeRealpath(dirPath);
+				}
+			} catch (dirError) {
+				if (dirError.code === 'ENOENT') {
+					throw new PathError(`Parent directory does not exist for file: ${normalizedPath}`, {
+						suggestion: 'The specified path is a file whose parent directory does not exist.',
+						details: { path: normalizedPath, parentPath: dirPath, type: 'file' }
+					});
+				}
+				if (dirError.code === 'EACCES') {
+					throw new PathError(`Permission denied accessing parent directory: ${dirPath}`, {
+						recoverable: false,
+						suggestion: 'Permission denied accessing the parent directory of the specified file.',
+						details: { path: normalizedPath, parentPath: dirPath, type: 'file' }
+					});
+				}
+				throw dirError;
+			}
 			throw new PathError(`Path is not a directory: ${normalizedPath}`, {
 				suggestion: 'The specified path is a file, not a directory. Please provide a directory path for searching.',
 				details: { path: normalizedPath, type: 'file' }