Limit profile data returned to prevent from context window blow up (#87)

ecekyn · web-flow · commit e7b6742f3e30 · 2025-10-09T09:12:45.000+01:00
diff --git a/tools/get_profiles.go b/tools/get_profiles.go
@@ -33,6 +33,11 @@ func GetProfilesHandler(ctx context.Context, arguments GetProfileHandlerArgs) (*
 	if err != nil {
 		return nil, fmt.Errorf("error getting profiles: %v", err)
 	}
+
+	if len(body) > 200000 {
+		return nil, fmt.Errorf("response too large, please refine your query to get a smaller response")
+	}
+
 	return mcpgolang.NewToolResponse(mcpgolang.NewTextContent(fmt.Sprintf("%s", string(body)))), nil
 }
 

Original file line number	Diff line number	Diff line change
`@@ -33,6 +33,11 @@ func GetProfilesHandler(ctx context.Context, arguments GetProfileHandlerArgs) (*`
`33`	`33`	`if err != nil {`
`34`	`34`	`return nil, fmt.Errorf("error getting profiles: %v", err)`
`35`	`35`	`}`
	`36`	`+`
	`37`	`+ if len(body) > 200000 {`
	`38`	`+ return nil, fmt.Errorf("response too large, please refine your query to get a smaller response")`
	`39`	`+ }`
	`40`	`+`
`36`	`41`	`return mcpgolang.NewToolResponse(mcpgolang.NewTextContent(fmt.Sprintf("%s", string(body)))), nil`
`37`	`42`	`}`
`38`	`43`