add more tool

Author: NaccOll

src/core/tools/schemas/__tests__/base-tool-schema.test.ts

@@ -0,0 +1,33 @@
+import { ToolArgs } from "../../prompts/tools/types"
+import { BaseToolSchema } from "./base-tool-schema"
+
+const baseAccessMcpResourceSchema: BaseToolSchema = {
+	name: "access_mcp_resource",
+	description:
+		"Request to access a resource provided by a connected MCP server. Resources represent data sources that can be used as context, such as files, API responses, or system information.",
+	parameters: [
+		{
+			name: "server_name",
+			type: "string",
+			description: "The name of the MCP server providing the resource",
+			required: true,
+		},
+		{
+			name: "uri",
+			type: "string",
+			description: "The URI identifying the specific resource to access",
+			required: true,
+		},
+	],
+}
+
+export const accessMcpResourceSchema: BaseToolSchema = {
+	...baseAccessMcpResourceSchema,
+	customDescription: (args: ToolArgs) => {
+		if (!args.mcpHub) {
+			return undefined
+		}
+		const schema = JSON.parse(JSON.stringify(baseAccessMcpResourceSchema))
+		return schema as BaseToolSchema
+	},
+}

src/core/tools/schemas/__tests__/function-call-converter.test.ts

@@ -18,7 +18,7 @@ export interface BaseToolSchema {
 	name: string
 	description: string
 	parameters: ToolParameter[]
-	customDescription?: (args: ToolArgs) => BaseToolSchema
+	customDescription?: (args: ToolArgs) => BaseToolSchema | undefined
 }
 
 /**
@@ -52,8 +52,7 @@ function toolParamToSchema(param: ToolParameter): any {
 /**
  * Converts a BaseToolSchema to OpenAI function call schema
  */
-export function generateFunctionCallSchema(baseSchema: BaseToolSchema, args?: ToolArgs): any {
-	const schema = baseSchema.customDescription ? baseSchema.customDescription(args || ({} as ToolArgs)) : baseSchema
+export function generateFunctionCallSchema(schema: BaseToolSchema) {
 	const { name, description, parameters } = schema
 	const properties: Record<string, any> = {}
 	const required: string[] = []

src/core/tools/schemas/__tests__/tool-registry.test.ts

@@ -0,0 +1,28 @@
+import { ToolArgs } from "../../prompts/tools/types"
+import { BaseToolSchema } from "./base-tool-schema"
+
+const baseCodebaseSearchSchema: BaseToolSchema = {
+	name: "codebase_search",
+	description:
+		"Find files most relevant to the search query. This is a semantic search tool, so the query should ask for something semantically matching what is needed. If it makes sense to only search in a particular directory, please specify it in the path parameter. Unless there is a clear reason to use your own search query, please just reuse the user's exact query with their wording. Their exact wording/phrasing can often be helpful for the semantic search query. Keeping the same exact question format can also be helpful. IMPORTANT: Queries MUST be in English. Translate non-English queries before searching.",
+	parameters: [
+		{
+			name: "query",
+			type: "string",
+			description:
+				"The search query to find relevant code. You should reuse the user's exact query/most recent message with their wording unless there is a clear reason not to.",
+			required: true,
+		},
+		{
+			name: "path",
+			type: "string",
+			description:
+				"The path to the directory to search in relative to the current working directory. This parameter should only be a directory path, file paths are not supported. Defaults to the current working directory.",
+			required: false,
+		},
+	],
+}
+
+export const codebaseSearchSchema: BaseToolSchema = {
+	...baseCodebaseSearchSchema,
+}