Fix various errors

Signed-off-by: Bryan Frimin <bryan@getprobo.com>

pkg/server/api/mcp/v1/add_vendor.go

@@ -44,6 +44,7 @@ func (r *resolver) AddVendor(
 			BusinessAssociateAgreementURL: args.BusinessAssociateAgreementURL,
 			SubprocessorsListURL:          args.SubprocessorsListURL,
 			Certifications:                args.Certifications,
+			Countries:                     args.Countries,
 			SecurityPageURL:               args.SecurityPageURL,
 			TrustPageURL:                  args.TrustPageURL,
 			TermsOfServiceURL:             args.TermsOfServiceURL,

pkg/server/api/mcp/v1/list_organizations.go

@@ -30,10 +30,8 @@ var (
 		Description: "List all organizations the user has access to",
 		Annotations: &mcp.ToolAnnotations{ReadOnlyHint: true},
 		InputSchema: &jsonschema.Schema{
-			Type: "object",
-			Properties: map[string]*jsonschema.Schema{
-				"organizationID": {Type: "string"},
-			},
+			Type:       "object",
+			Properties: map[string]*jsonschema.Schema{},
 		},
 	}
 )

pkg/server/api/mcp/v1/types/vendor.go

@@ -103,19 +103,19 @@ type (
 var (
 	ListVendorsInputSchema = &jsonschema.Schema{
 		Type:     "object",
-		Required: []string{"organizationID"},
+		Required: []string{"organization_id"},
 		Properties: map[string]*jsonschema.Schema{
-			"organizationID": {Type: "string"},
+			"organization_id": {Type: "string"},
 			"filter": {
 				Type: "object",
 				Properties: map[string]*jsonschema.Schema{
-					"snapshotID": {Type: "string"},
+					"snapshot_id": {Type: "string"},
 				},
 			},
-			"orderBy": {
+			"order_field": {
 				Types: []string{"object", "null"},
 				Properties: map[string]*jsonschema.Schema{
-					"field":     {Type: "string", Enum: []any{"CREATED_AT"}},
+					"field":     {Type: "string", Enum: []any{"CREATED_AT", "UPDATED_AT", "NAME"}},
 					"direction": OrderByDirectionSchema,
 				},
 			},
@@ -170,7 +170,7 @@ var (
 			"description":                      NullableStringSchema,
 			"category":                         VendorCategorySchema,
 			"headquarter_address":              NullableStringSchema,
-			"legalName":                        NullableStringSchema,
+			"legal_name":                       NullableStringSchema,
 			"website_url":                      NullableStringSchema,
 			"privacy_policy_url":               NullableStringSchema,
 			"service_level_agreement_url":      NullableStringSchema,