Adjust input params (#18)

* Adjust input parameters

* Add operation logging

* Enable customers tool

Author: maarcingebala

src/saleor_mcp/config.py

@@ -8,6 +8,7 @@
 
 LOGLEVEL = os.environ.get("LOGLEVEL", "INFO").upper()
 logging.basicConfig(level=LOGLEVEL)
+logging.getLogger("mcp.server.streamable_http").setLevel(logging.WARNING)
 
 
 def validate_api_url(url, pattern):

src/saleor_mcp/graphql/ListCustomers.graphql

@@ -1,10 +1,17 @@
 query ListCustomers(
   $first: Int
+  $after: String
   $sortBy: UserSortingInput
   $where: CustomerWhereInput
   $search: String
 ) {
-  customers(first: $first, sortBy: $sortBy, where: $where, search: $search) {
+  customers(
+    first: $first
+    after: $after
+    sortBy: $sortBy
+    where: $where
+    search: $search
+  ) {
     pageInfo {
       hasNextPage
       hasPreviousPage

src/saleor_mcp/main.py

@@ -1,17 +1,21 @@
 from fastmcp import FastMCP
+from fastmcp.server.middleware.timing import DetailedTimingMiddleware
 from starlette.requests import Request
 from starlette.responses import FileResponse, JSONResponse
 from starlette.staticfiles import StaticFiles
 
 from saleor_mcp.tools import (
     channels_router,
+    customers_router,
     orders_router,
     products_router,
     utils_router,
 )
 
 mcp = FastMCP("Saleor MCP Server")
+mcp.add_middleware(DetailedTimingMiddleware())
 mcp.mount(channels_router)
+mcp.mount(customers_router)
 mcp.mount(orders_router)
 mcp.mount(products_router)
 mcp.mount(utils_router)

src/saleor_mcp/saleor_client/__init__.py

@@ -138,6 +138,7 @@
     PromotionSortField,
     PromotionTypeEnum,
     PromotionUpdateErrorCode,
+    RefundSettingsErrorCode,
     ReportingPeriod,
     RewardTypeEnum,
     RewardValueTypeEnum,
@@ -454,6 +455,7 @@
     PromotionUpdateInput,
     PromotionWhereInput,
     PublishableChannelListingInput,
+    RefundSettingsUpdateInput,
     ReorderInput,
     SaleChannelListingAddInput,
     SaleChannelListingInput,
@@ -1026,6 +1028,8 @@
     "PromotionUpdateInput",
     "PromotionWhereInput",
     "PublishableChannelListingInput",
+    "RefundSettingsErrorCode",
+    "RefundSettingsUpdateInput",
     "ReorderInput",
     "ReportingPeriod",
     "RewardTypeEnum",

src/saleor_mcp/saleor_client/client.py

@@ -78,15 +78,22 @@ async def list_channels(self, **kwargs: Any) -> ListChannels:
     async def list_customers(
         self,
         first: Union[Optional[int], UnsetType] = UNSET,
+        after: Union[Optional[str], UnsetType] = UNSET,
         sortBy: Union[Optional[UserSortingInput], UnsetType] = UNSET,
         where: Union[Optional[CustomerWhereInput], UnsetType] = UNSET,
         search: Union[Optional[str], UnsetType] = UNSET,
         **kwargs: Any
     ) -> ListCustomers:
         query = gql(
             """
-            query ListCustomers($first: Int, $sortBy: UserSortingInput, $where: CustomerWhereInput, $search: String) {
-              customers(first: $first, sortBy: $sortBy, where: $where, search: $search) {
+            query ListCustomers($first: Int, $after: String, $sortBy: UserSortingInput, $where: CustomerWhereInput, $search: String) {
+              customers(
+                first: $first
+                after: $after
+                sortBy: $sortBy
+                where: $where
+                search: $search
+              ) {
                 pageInfo {
                   hasNextPage
                   hasPreviousPage
@@ -139,6 +146,7 @@ async def list_customers(
         )
         variables: Dict[str, object] = {
             "first": first,
+            "after": after,
             "sortBy": sortBy,
             "where": where,
             "search": search,

src/saleor_mcp/saleor_client/enums.py

@@ -1731,6 +1731,17 @@ class AttributeEntityTypeEnum(str, Enum):
     COLLECTION = "COLLECTION"
 
 
+class ProductTypeKindEnum(str, Enum):
+    NORMAL = "NORMAL"
+    GIFT_CARD = "GIFT_CARD"
+
+
+class VariantAttributeScope(str, Enum):
+    ALL = "ALL"
+    VARIANT_SELECTION = "VARIANT_SELECTION"
+    NOT_VARIANT_SELECTION = "NOT_VARIANT_SELECTION"
+
+
 class AttributeTypeEnum(str, Enum):
     PRODUCT_TYPE = "PRODUCT_TYPE"
     PAGE_TYPE = "PAGE_TYPE"
@@ -1778,17 +1789,6 @@ class AttributeChoicesSortField(str, Enum):
     SLUG = "SLUG"
 
 
-class ProductTypeKindEnum(str, Enum):
-    NORMAL = "NORMAL"
-    GIFT_CARD = "GIFT_CARD"
-
-
-class VariantAttributeScope(str, Enum):
-    ALL = "ALL"
-    VARIANT_SELECTION = "VARIANT_SELECTION"
-    NOT_VARIANT_SELECTION = "NOT_VARIANT_SELECTION"
-
-
 class StockAvailability(str, Enum):
     IN_STOCK = "IN_STOCK"
     OUT_OF_STOCK = "OUT_OF_STOCK"
@@ -2541,6 +2541,12 @@ class GiftCardSettingsErrorCode(str, Enum):
     GRAPHQL_ERROR = "GRAPHQL_ERROR"
 
 
+class RefundSettingsErrorCode(str, Enum):
+    INVALID = "INVALID"
+    REQUIRED = "REQUIRED"
+    GRAPHQL_ERROR = "GRAPHQL_ERROR"
+
+
 class ShippingErrorCode(str, Enum):
     ALREADY_EXISTS = "ALREADY_EXISTS"
     GRAPHQL_ERROR = "GRAPHQL_ERROR"
@@ -2689,6 +2695,7 @@ class TransactionUpdateErrorCode(str, Enum):
 
 class TransactionRequestActionErrorCode(str, Enum):
     INVALID = "INVALID"
+    REQUIRED = "REQUIRED"
     GRAPHQL_ERROR = "GRAPHQL_ERROR"
     NOT_FOUND = "NOT_FOUND"
     MISSING_TRANSACTION_ACTION_REQUEST_WEBHOOK = (

src/saleor_mcp/saleor_client/input_types.py

@@ -107,30 +107,6 @@ class EventDeliveryFilterInput(BaseModel):
     eventType: Optional[WebhookEventTypeEnum] = None
 
 
-class AttributeChoicesSortingInput(BaseModel):
-    direction: OrderDirection
-    field: AttributeChoicesSortField
-
-
-class AttributeValueFilterInput(BaseModel):
-    search: Optional[str] = None
-    ids: Optional[List[str]] = None
-    slugs: Optional[List[str]] = None
-
-
-class AttributeValueWhereInput(BaseModel):
-    ids: Optional[List[str]] = None
-    name: Optional["StringFilterInput"] = None
-    slug: Optional["StringFilterInput"] = None
-    AND: Optional[List["AttributeValueWhereInput"]] = None
-    OR: Optional[List["AttributeValueWhereInput"]] = None
-
-
-class StringFilterInput(BaseModel):
-    eq: Optional[str] = None
-    oneOf: Optional[List[str]] = None
-
-
 class AttributeFilterInput(BaseModel):
     valueRequired: Optional[bool] = None
     isVariantOnly: Optional[bool] = None
@@ -172,6 +148,11 @@ class AttributeWhereInput(BaseModel):
     OR: Optional[List["AttributeWhereInput"]] = None
 
 
+class StringFilterInput(BaseModel):
+    eq: Optional[str] = None
+    oneOf: Optional[List[str]] = None
+
+
 class AttributeInputTypeEnumFilterInput(BaseModel):
     eq: Optional[AttributeInputTypeEnum] = None
     oneOf: Optional[List[AttributeInputTypeEnum]] = None
@@ -192,6 +173,25 @@ class MeasurementUnitsEnumFilterInput(BaseModel):
     oneOf: Optional[List[MeasurementUnitsEnum]] = None
 
 
+class AttributeChoicesSortingInput(BaseModel):
+    direction: OrderDirection
+    field: AttributeChoicesSortField
+
+
+class AttributeValueFilterInput(BaseModel):
+    search: Optional[str] = None
+    ids: Optional[List[str]] = None
+    slugs: Optional[List[str]] = None
+
+
+class AttributeValueWhereInput(BaseModel):
+    ids: Optional[List[str]] = None
+    name: Optional["StringFilterInput"] = None
+    slug: Optional["StringFilterInput"] = None
+    AND: Optional[List["AttributeValueWhereInput"]] = None
+    OR: Optional[List["AttributeValueWhereInput"]] = None
+
+
 class ProductFilterInput(BaseModel):
     isPublished: Optional[bool] = None
     collections: Optional[List[str]] = None
@@ -1098,6 +1098,10 @@ class TimePeriodInputType(BaseModel):
     type: TimePeriodTypeEnum
 
 
+class RefundSettingsUpdateInput(BaseModel):
+    refundReasonReferenceType: str
+
+
 class ShippingMethodChannelListingInput(BaseModel):
     addChannels: Optional[List["ShippingMethodChannelListingAddInput"]] = None
     removeChannels: Optional[List[str]] = None
@@ -1785,6 +1789,7 @@ class OrderReturnFulfillmentLineInput(BaseModel):
 class OrderGrantRefundCreateInput(BaseModel):
     amount: Optional[Any] = None
     reason: Optional[str] = None
+    reasonReference: Optional[str] = None
     lines: Optional[List["OrderGrantRefundCreateLineInput"]] = None
     grantRefundForShipping: Optional[bool] = None
     transactionId: str
@@ -1799,6 +1804,7 @@ class OrderGrantRefundCreateLineInput(BaseModel):
 class OrderGrantRefundUpdateInput(BaseModel):
     amount: Optional[Any] = None
     reason: Optional[str] = None
+    reasonReference: Optional[str] = None
     addLines: Optional[List["OrderGrantRefundUpdateLineAddInput"]] = None
     removeLines: Optional[List[str]] = None
     grantRefundForShipping: Optional[bool] = None
@@ -2370,6 +2376,7 @@ class AttributeCreateInput(BaseModel):
     storefrontSearchPosition: Optional[int] = None
     availableInGrid: Optional[bool] = None
     externalReference: Optional[str] = None
+    referenceTypes: Optional[List[str]] = None
 
 
 class AttributeValueCreateInput(BaseModel):
@@ -2396,6 +2403,7 @@ class AttributeUpdateInput(BaseModel):
     storefrontSearchPosition: Optional[int] = None
     availableInGrid: Optional[bool] = None
     externalReference: Optional[str] = None
+    referenceTypes: Optional[List[str]] = None
 
 
 class AttributeValueUpdateInput(BaseModel):
@@ -2553,9 +2561,9 @@ class PermissionGroupUpdateInput(BaseModel):
     restrictedAccessToChannels: Optional[bool] = None
 
 
-AttributeValueWhereInput.model_rebuild()
 AttributeFilterInput.model_rebuild()
 AttributeWhereInput.model_rebuild()
+AttributeValueWhereInput.model_rebuild()
 ProductFilterInput.model_rebuild()
 AttributeInput.model_rebuild()
 AssignedAttributeValueInput.model_rebuild()

src/saleor_mcp/static/index.html

@@ -249,6 +249,47 @@ <h2 class="card-title">
               </div>
             </div>
 
+            <!-- Customers Tool -->
+            <div class="accordion-item">
+              <button class="accordion-trigger">
+                <div class="accordion-header">
+                  <span class="badge">customers</span>
+                  <span class="tool-name">Customers</span>
+                </div>
+                <svg
+                  class="chevron"
+                  fill="none"
+                  stroke="currentColor"
+                  viewBox="0 0 24 24"
+                >
+                  <path
+                    stroke-linecap="round"
+                    stroke-linejoin="round"
+                    stroke-width="2"
+                    d="M19 9l-7 7-7-7"
+                  />
+                </svg>
+              </button>
+              <div class="accordion-content">
+                <p class="tool-description">
+                  Fetch list of customers from Saleor GraphQL API. This tool
+                  retrieves customer information such as: ID, email, first name,
+                  last name, active status, confirmation status, checkout and
+                  order counts, language code, last login, date joined, and
+                  default shipping/billing addresses.
+                </p>
+                <li>
+                  first: Number of customers to fetch (max 100 per request).
+                </li>
+                <li>
+                  after: Cursor for pagination - fetch customers after this
+                  cursor.
+                </li>
+                <li>sort_by: Sort customers by specific field.</li>
+                <li>where: Filter customers by specific criteria.</li>
+              </div>
+            </div>
+
             <!-- Orders Tool -->
             <div class="accordion-item">
               <button class="accordion-trigger">
@@ -381,7 +422,6 @@ <h4>Arguments:</h4>
                       returned.
                     </li>
                     <li>sort_by: Sort products by specific field.</li>
-                    <li>where: Filter products by specific criteria.</li>
                     <li>search: Search products with full-text search.</li>
                   </ul>
                 </div>

src/saleor_mcp/tools/__init__.py

@@ -1,6 +1,13 @@
 from .channels import channels_router
+from .customers import customers_router
 from .orders import orders_router
 from .products import products_router
 from .utils import utils_router
 
-__all__ = ["channels_router", "orders_router", "products_router", "utils_router"]
+__all__ = [
+    "channels_router",
+    "customers_router",
+    "orders_router",
+    "products_router",
+    "utils_router",
+]