refactor: remove agent flavours, move provider to Agent (#2091)

2025-04-09 15:02:47 -04:00
parent a8cbd81c61
commit 513d5c8f5a
30 changed files with 1297 additions and 2272 deletions
@@ -0,0 +1,112 @@
+use indoc::indoc;
+use mcp_core::tool::{Tool, ToolAnnotations};
+use serde_json::json;
+
+pub const PLATFORM_READ_RESOURCE_TOOL_NAME: &str = "platform__read_resource";
+pub const PLATFORM_LIST_RESOURCES_TOOL_NAME: &str = "platform__list_resources";
+pub const PLATFORM_SEARCH_AVAILABLE_EXTENSIONS_TOOL_NAME: &str =
+    "platform__search_available_extensions";
+pub const PLATFORM_ENABLE_EXTENSION_TOOL_NAME: &str = "platform__enable_extension";
+
+pub fn read_resource_tool() -> Tool {
+    Tool::new(
+        PLATFORM_READ_RESOURCE_TOOL_NAME.to_string(),
+        indoc! {r#"
+            Read a resource from an extension.
+
+            Resources allow extensions to share data that provide context to LLMs, such as
+            files, database schemas, or application-specific information. This tool searches for the
+            resource URI in the provided extension, and reads in the resource content. If no extension
+            is provided, the tool will search all extensions for the resource.
+        "#}.to_string(),
+        json!({
+            "type": "object",
+            "required": ["uri"],
+            "properties": {
+                "uri": {"type": "string", "description": "Resource URI"},
+                "extension_name": {"type": "string", "description": "Optional extension name"}
+            }
+        }),
+        Some(ToolAnnotations {
+            title: Some("Read a resource".to_string()),
+            read_only_hint: true,
+            destructive_hint: false,
+            idempotent_hint: false,
+            open_world_hint: false,
+        }),
+    )
+}
+
+pub fn list_resources_tool() -> Tool {
+    Tool::new(
+        PLATFORM_LIST_RESOURCES_TOOL_NAME.to_string(),
+        indoc! {r#"
+            List resources from an extension(s).
+
+            Resources allow extensions to share data that provide context to LLMs, such as
+            files, database schemas, or application-specific information. This tool lists resources
+            in the provided extension, and returns a list for the user to browse. If no extension
+            is provided, the tool will search all extensions for the resource.
+        "#}
+        .to_string(),
+        json!({
+            "type": "object",
+            "properties": {
+                "extension_name": {"type": "string", "description": "Optional extension name"}
+            }
+        }),
+        Some(ToolAnnotations {
+            title: Some("List resources".to_string()),
+            read_only_hint: true,
+            destructive_hint: false,
+            idempotent_hint: false,
+            open_world_hint: false,
+        }),
+    )
+}
+
+pub fn search_available_extensions_tool() -> Tool {
+    Tool::new(
+        PLATFORM_SEARCH_AVAILABLE_EXTENSIONS_TOOL_NAME.to_string(),
+        "Searches for additional extensions available to help complete tasks.
+        Use this tool when you're unable to find a specific feature or functionality you need to complete your task, or when standard approaches aren't working.
+        These extensions might provide the exact tools needed to solve your problem.
+        If you find a relevant one, consider using your tools to enable it.".to_string(),
+        json!({
+            "type": "object",
+            "required": [],
+            "properties": {}
+        }),
+        Some(ToolAnnotations {
+            title: Some("Discover extensions".to_string()),
+            read_only_hint: true,
+            destructive_hint: false,
+            idempotent_hint: false,
+            open_world_hint: false,
+        }),
+    )
+}
+
+pub fn enable_extension_tool() -> Tool {
+    Tool::new(
+        PLATFORM_ENABLE_EXTENSION_TOOL_NAME.to_string(),
+        "Enable extensions to help complete tasks.
+            Enable an extension by providing the extension name.
+            "
+        .to_string(),
+        json!({
+            "type": "object",
+            "required": ["extension_name"],
+            "properties": {
+                "extension_name": {"type": "string", "description": "The name of the extension to enable"}
+            }
+        }),
+        Some(ToolAnnotations {
+            title: Some("Enable extensions".to_string()),
+            read_only_hint: false,
+            destructive_hint: false,
+            idempotent_hint: false,
+            open_world_hint: false,
+        }),
+    )
+}