sku-availability-mcp 1.0.0__py3-none-any.whl

sku_availability_mcp/__init__.py

@@ -0,0 +1,14 @@
+"""
+SKU Availability MCP Server
+Real-time inventory availability queries via Confluent Cloud ksqlDB
+"""
+
+__version__ = "1.0.0"
+__author__ = "Your Name"
+__email__ = "your.email@example.com"
+
+from .server import mcp, get_sku_availability
+
+__all__ = ["mcp", "get_sku_availability"]
+
+# Made with Bob

sku_availability_mcp/__main__.py

@@ -0,0 +1,10 @@
+"""
+Entry point for running the SKU Availability MCP server as a module
+"""
+
+from .server import mcp
+
+if __name__ == "__main__":
+    mcp.run()
+
+# Made with Bob

sku_availability_mcp/server.py

@@ -0,0 +1,149 @@
+#!/usr/bin/env python3
+"""
+FastMCP Server for SKU Availability
+Provides tools to query inventory availability from ksqlDB
+"""
+
+import requests
+import json
+import os
+from typing import Optional
+from dotenv import load_dotenv
+from requests.auth import HTTPBasicAuth
+from fastmcp import FastMCP
+import math 
+
+# Load environment variables
+load_dotenv()
+
+# ksqlDB Configuration
+KSQLDB_ENDPOINT = "https://pksqlc-mzv0x2.us-east1.gcp.confluent.cloud:443"
+KSQLDB_API_KEY = os.getenv('KSQLDB_API_KEY')
+KSQLDB_API_SECRET = os.getenv('KSQLDB_API_SECRET')
+
+# Create FastMCP server
+mcp = FastMCP("SKU Availability Server")
+
+def validate_config():
+    """Validate that all required configuration is present"""
+    required_vars = ['KSQLDB_ENDPOINT', 'KSQLDB_API_KEY', 'KSQLDB_API_SECRET']
+    missing_vars = [var for var in required_vars if not os.getenv(var)]
+    
+    if missing_vars:
+        raise ValueError(f"Missing required environment variables: {', '.join(missing_vars)}")
+    
+    if 'xxxxx' in str(KSQLDB_ENDPOINT) or 'your-ksqldb-api-key' in str(KSQLDB_API_KEY):
+        raise ValueError("Please update .env file with your actual ksqlDB credentials")
+
+def query_ksqldb(query: str) -> list:
+    """
+    Execute a query against ksqlDB
+    
+    Args:
+        query: SQL query to execute
+    
+    Returns:
+        List of result records
+    """
+    validate_config()
+    
+    url = f"{KSQLDB_ENDPOINT}/query-stream"
+    headers = {
+        'Content-Type': 'application/vnd.ksql.v1+json'
+    }
+    
+    payload = {
+        "sql": query,
+        "properties": {}
+    }
+    
+    try:
+        # Ensure credentials are not None
+        if not KSQLDB_API_KEY or not KSQLDB_API_SECRET:
+            raise ValueError("KSQLDB_API_KEY and KSQLDB_API_SECRET must be set")
+        
+        response = requests.post(
+            url,
+            headers=headers,
+            json=payload,
+            auth=HTTPBasicAuth(KSQLDB_API_KEY, KSQLDB_API_SECRET),
+            stream=True,
+            timeout=30
+        )
+        
+        if response.status_code != 200:
+            raise Exception(f"HTTP {response.status_code}: {response.text}")
+        
+        results = []
+        
+        # Read streaming response
+        for line in response.iter_lines():
+            if line:
+                try:
+                    data = json.loads(line.decode('utf-8'))
+                    
+                    # Skip header row
+                    if 'header' in data or 'columnNames' in data or 'columnTypes' in data:
+                        continue
+                    
+                    # Process data row
+                    if isinstance(data, list) and len(data) >= 3:
+                        results.append({
+                            'sku': data[0],
+                            'branch': data[1],
+                            'available_quantity': data[2]
+                        })
+                    
+                except json.JSONDecodeError:
+                    continue
+                except Exception:
+                    continue
+        
+        return results
+        
+    except Exception as e:
+        raise Exception(f"Failed to query ksqlDB: {str(e)}")
+
+@mcp.tool()
+def get_sku_availability(sku: str = "", branch: str = "") -> str:
+    """
+    Get real-time inventory availability for SKUs across branches.
+    
+    Query the INVENTORY_AVAILABILITY table which aggregates inventory transactions.
+    Returns current stock levels calculated from all inventory adds and sales.
+    
+    Args:
+        sku: Optional SKU filter (e.g., 'LAPTOP-DELL-XPS-15'). Leave empty to get all SKUs.
+        branch: Optional branch filter (e.g., 'DubaiMall', 'MallOfEgypt'). Leave empty to get all branches.
+    
+    Returns:
+        JSON string with availability records
+    """
+    try:
+        # Build query - treat empty strings as no filter
+        if sku and branch:
+            query = f"SELECT * FROM INVENTORY_AVAILABILITY WHERE SKU='{sku}' AND BRANCH='{branch}';"
+        elif sku:
+            query = f"SELECT * FROM INVENTORY_AVAILABILITY WHERE SKU='{sku}';"
+        elif branch:
+            query = f"SELECT * FROM INVENTORY_AVAILABILITY WHERE BRANCH='{branch}';"
+        else:
+            query = "SELECT * FROM INVENTORY_AVAILABILITY;"
+        
+        # Execute query
+        results = query_ksqldb(query)
+        
+        # Format response
+        if not results:
+            return json.dumps({"message": "No inventory records found", "results": []})
+        
+        return json.dumps({"results": results}, indent=2)
+        
+    except Exception as e:
+        return json.dumps({"error": str(e)})
+
+
+if __name__ == "__main__":
+    mcp.run()
+
+# Made with Bob

sku_availability_mcp-1.0.0.dist-info/METADATA

@@ -0,0 +1,144 @@
+Metadata-Version: 2.4
+Name: sku-availability-mcp
+Version: 1.0.0
+Summary: MCP server for real-time SKU availability queries via Confluent Cloud ksqlDB
+Home-page: https://github.com/yourusername/sku-availability-mcp
+Author: Bob AI Assistant
+Author-email: Bob AI Assistant <bob@example.com>
+License: MIT
+Project-URL: Homepage, https://github.com/yourusername/sku-availability-mcp
+Project-URL: Bug Reports, https://github.com/yourusername/sku-availability-mcp/issues
+Project-URL: Source, https://github.com/yourusername/sku-availability-mcp
+Keywords: mcp,model-context-protocol,confluent,kafka,ksqldb,inventory
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: fastmcp>=2.14.4
+Requires-Dist: requests>=2.32.0
+Requires-Dist: python-dotenv>=1.0.0
+Dynamic: author
+Dynamic: home-page
+Dynamic: license-file
+Dynamic: requires-python
+
+# SKU Availability MCP Server
+
+A Model Context Protocol (MCP) server that provides real-time inventory availability queries from Confluent Cloud ksqlDB.
+
+## Features
+
+- 🔍 Real-time SKU availability queries
+- ☁️ Integration with Confluent Cloud ksqlDB
+- ⚡ FastMCP-based server implementation
+- 🎯 Filter by SKU and/or branch
+- 🔌 Easy integration with watsonx Orchestrate
+
+## Installation
+
+```bash
+pip install sku-availability-mcp
+```
+
+## Configuration
+
+Set the following environment variables:
+
+```bash
+export KSQLDB_API_KEY="your-api-key"
+export KSQLDB_API_SECRET="your-api-secret"
+export KSQLDB_ENDPOINT="https://your-ksqldb-endpoint.confluent.cloud:443"
+```
+
+Or create a `.env` file:
+
+```env
+KSQLDB_API_KEY=your-api-key
+KSQLDB_API_SECRET=your-api-secret
+KSQLDB_ENDPOINT=https://your-ksqldb-endpoint.confluent.cloud:443
+```
+
+## Usage
+
+### Run as a module
+
+```bash
+python -m sku_availability_mcp
+```
+
+### Run as a command
+
+```bash
+sku-availability-mcp
+```
+
+### Use with watsonx Orchestrate
+
+Add the toolkit using the orchestrate CLI:
+
+```bash
+orchestrate toolkits add \
+  --kind mcp \
+  --name "sku-availability-toolkit" \
+  --description "Real-time inventory availability from Confluent Cloud" \
+  --language python \
+  --package "sku-availability-mcp" \
+  --tools "*"
+```
+
+## Available Tools
+
+### get_sku_availability
+
+Query real-time inventory availability for SKUs across branches.
+
+**Parameters:**
+- `sku` (optional): SKU filter (e.g., 'LAPTOP-DELL-XPS-15')
+- `branch` (optional): Branch filter (e.g., 'DubaiMall', 'MallOfEgypt')
+
+**Returns:**
+JSON object with availability records containing:
+- `sku`: Product SKU
+- `branch`: Store branch name
+- `available_quantity`: Current available quantity
+
+**Example:**
+
+```python
+# Get all inventory
+get_sku_availability()
+
+# Get specific SKU
+get_sku_availability(sku="LAPTOP-DELL-XPS-15")
+
+# Get inventory for specific branch
+get_sku_availability(branch="DubaiMall")
+
+# Get specific SKU at specific branch
+get_sku_availability(sku="LAPTOP-DELL-XPS-15", branch="DubaiMall")
+```
+
+## Architecture
+
+This MCP server connects to Confluent Cloud ksqlDB to query the `INVENTORY_AVAILABILITY` table, which aggregates real-time inventory transactions from Kafka topics.
+
+## License
+
+MIT License
+
+## Contributing
+
+Contributions are welcome! Please feel free to submit a Pull Request.
+
+## Support
+
+For issues and questions, please open an issue on GitHub.

sku_availability_mcp-1.0.0.dist-info/RECORD

@@ -0,0 +1,9 @@
+sku_availability_mcp/__init__.py,sha256=LuKtg5t4TEjbKijzb6lu8pk8w_UvD5akQjApP4657mc,296
+sku_availability_mcp/__main__.py,sha256=g736Zntdeh80a7-YvzYEvoaYPCYO5qgRLfW17Yg3Byc,160
+sku_availability_mcp/server.py,sha256=T1UtuhppE-QXWj4yE3ppUjk73ZEhgPMnNzWVwswEYVw,4657
+sku_availability_mcp-1.0.0.dist-info/licenses/LICENSE,sha256=wbZ-cDWp28xmjCM1RUzgBJCLJN3XnZTvqIBGsf2DSJQ,1076
+sku_availability_mcp-1.0.0.dist-info/METADATA,sha256=uM_wATkg2NICDFa0mMvI6djMBpkccnNJD4NERmPnLDs,3791
+sku_availability_mcp-1.0.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+sku_availability_mcp-1.0.0.dist-info/entry_points.txt,sha256=-4KsNSLIZb-dl5vWJTv_9b6yIXCGCJrVcme-mopVpVQ,74
+sku_availability_mcp-1.0.0.dist-info/top_level.txt,sha256=wn9Ka1Xxku2p8WiGRzrM6P0vtIrB2SHTevs7imKGCNk,21
+sku_availability_mcp-1.0.0.dist-info/RECORD,,

sku_availability_mcp-1.0.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.10.2)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

sku_availability_mcp-1.0.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+sku-availability-mcp = sku_availability_mcp.server:main

sku_availability_mcp-1.0.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 SKU Availability MCP
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

sku_availability_mcp-1.0.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+sku_availability_mcp