x

2025-03-12 01:19:35 +08:00
parent 160040f3e1
commit 78cce92ce3
70 changed files with 10134 additions and 1 deletions
--- a/internal/mcp/prompt.go
+++ b/internal/mcp/prompt.go
@@ -0,0 +1,137 @@
+package mcp
+
+// Document: https://modelcontextprotocol.io/docs/concepts/prompts
+
+const (
+	// Client => Server
+	MethodPromptsList = "prompts/list"
+	MethodPromptsGet  = "prompts/get"
+)
+
+// Prompt
+//
+//	{
+//		name: string;              // Unique identifier for the prompt
+//		description?: string;      // Human-readable description
+//		arguments?: [              // Optional list of arguments
+//			{
+//				name: string;          // Argument identifier
+//				description?: string;  // Argument description
+//				required?: boolean;    // Whether argument is required
+//			}
+//		]
+//	}
+type Prompt struct {
+	Name        string           `json:"name"`
+	Description string           `json:"description,omitempty"`
+	Arguments   []PromptArgument `json:"arguments,omitempty"`
+}
+
+type PromptArgument struct {
+	Name        string `json:"name"`
+	Description string `json:"description,omitempty"`
+	Required    bool   `json:"required,omitempty"`
+}
+
+// ListPrompts
+//
+//	{
+//		prompts: [
+//			{
+//				name: "analyze-code",
+//				description: "Analyze code for potential improvements",
+//				arguments: [
+//					{
+//						name: "language",
+//						description: "Programming language",
+//						required: true
+//					}
+//				]
+//			}
+//		]
+//	}
+type PromptsListResponse struct {
+	Prompts []Prompt `json:"prompts"`
+}
+
+// Use Prompt
+// Request
+//
+//	{
+//		method: "prompts/get",
+//		params: {
+//			name: "analyze-code",
+//			arguments: {
+//				language: "python"
+//			}
+//		}
+//	}
+//
+// Response
+//
+//	{
+//		description: "Analyze Python code for potential improvements",
+//		messages: [
+//			{
+//				role: "user",
+//				content: {
+//					type: "text",
+//					text: "Please analyze the following Python code for potential improvements:\n\n```python\ndef calculate_sum(numbers):\n    total = 0\n    for num in numbers:\n        total = total + num\n    return total\n\nresult = calculate_sum([1, 2, 3, 4, 5])\nprint(result)\n```"
+//				}
+//			}
+//		]
+//	}
+type PromptsGetRequest struct {
+	Name      string `json:"name"`
+	Arguments M      `json:"arguments"`
+}
+
+type PromptsGetResponse struct {
+	Description string          `json:"description"`
+	Messages    []PromptMessage `json:"messages"`
+}
+
+type PromptMessage struct {
+	Role    string        `json:"role"`
+	Content PromptContent `json:"content"`
+}
+
+type PromptContent struct {
+	Type     string      `json:"type"`
+	Text     string      `json:"text,omitempty"`
+	Resource interface{} `json:"resource,omitempty"` // Resource or ResourceTemplate
+}
+
+// {
+// 	"messages": [
+// 		{
+// 			"role": "user",
+// 			"content": {
+// 				"type": "text",
+// 				"text": "Analyze these system logs and the code file for any issues:"
+// 			}
+// 		},
+// 		{
+// 			"role": "user",
+// 			"content": {
+// 				"type": "resource",
+// 				"resource": {
+// 					"uri": "logs://recent?timeframe=1h",
+// 					"text": "[2024-03-14 15:32:11] ERROR: Connection timeout in network.py:127\n[2024-03-14 15:32:15] WARN: Retrying connection (attempt 2/3)\n[2024-03-14 15:32:20] ERROR: Max retries exceeded",
+// 					"mimeType": "text/plain"
+// 				}
+// 			}
+// 		},
+// 		{
+// 			"role": "user",
+// 			"content": {
+// 				"type": "resource",
+// 				"resource": {
+// 					"uri": "file:///path/to/code.py",
+// 					"text": "def connect_to_service(timeout=30):\n    retries = 3\n    for attempt in range(retries):\n        try:\n            return establish_connection(timeout)\n        except TimeoutError:\n            if attempt == retries - 1:\n                raise\n            time.sleep(5)\n\ndef establish_connection(timeout):\n    # Connection implementation\n    pass",
+// 					"mimeType": "text/x-python"
+// 				}
+// 			}
+// 		}
+// 	]
+// }