fix prompts section

mark3labs · Dec 11, 2024 · bfec18e · bfec18e
1 parent 711de0c
commit bfec18e
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 6 deletions.
diff --git a/README.md b/README.md
@@ -558,7 +558,12 @@ Prompts are reusable templates that help LLMs interact with your server effectiv
 
 ```go
 // Simple greeting prompt
-s.AddPrompt("greeting", func(args map[string]string) (*mcp.GetPromptResult, error) {
+s.AddPrompt(mcp.NewPrompt("greeting",
+    mcp.WithPromptDescription("A friendly greeting prompt"),
+    mcp.WithArgument("name",
+        mcp.ArgumentDescription("Name of the person to greet"),
+    ),
+), func(args map[string]string) (*mcp.GetPromptResult, error) {
     name := args["name"]
     if name == "" {
         name = "friend"
@@ -616,7 +621,13 @@ s.AddPrompt(mcp.NewPrompt("code_review",
 })
 
 // Database query builder prompt
-s.AddPrompt("query_builder", func(args map[string]string) (*mcp.GetPromptResult, error) {
+s.AddPrompt(mcp.NewPrompt("query_builder",
+    mcp.WithPromptDescription("SQL query builder assistance"),
+    mcp.WithArgument("table",
+        mcp.ArgumentDescription("Name of the table to query"),
+        mcp.RequiredArgument(),
+    ),
+), func(args map[string]string) (*mcp.GetPromptResult, error) {
     tableName := args["table"]
     if tableName == "" {
         return nil, fmt.Errorf("table name is required")

diff --git a/examples/server/everything/stdio/main.go b/examples/server/everything/stdio/main.go
@@ -52,10 +52,9 @@ func NewMCPServer() *MCPServer {
 		"test://static/resource/{id}",
 		s.handleResourceTemplate,
 	)
-	s.server.AddPrompt(mcp.Prompt{
-		Name:        string(SIMPLE),
-		Description: "A simple prompt",
-	}, s.handleSimplePrompt)
+	s.server.AddPrompt(mcp.NewPrompt(string(SIMPLE),
+		mcp.WithPromptDescription("A simple prompt"),
+	), s.handleSimplePrompt)
 	s.server.AddPrompt(mcp.NewPrompt(string(COMPLEX),
 		mcp.WithPromptDescription("A complex prompt"),
 		mcp.WithArgument("temperature",