Fix

[matrix-shop-bot.git] / skill.py

diff --git a/skill.py b/skill.py
index bc99032ddda8b11ba82f00f903429e52c5c5f4f5..3a1b1cbde80b015c99434702d557c289543391e1 100644 (file)
--- a/skill.py
+++ b/skill.py
@@ -19,7 +19,40 @@ mcp = FastMCP("Shop Bot")
 def add_item(name: str) -> str:
     return '\n'.join(Data().add_item(name))
 
- 
+@mcp.tool
+def buy(items:str) -> str:
+    return '\n'.join(Data().purchase_items(items))
+
+@mcp.tool
+def list_bought(store:str) -> str:
+    return '\n'.join(Data().list_purchases(store))
+
+#@mcp.tool
+#def clear(store_name: str, clear_all:bool=False) -> 'list[str]':
+#    return Data().clear_list(store_name, clear_all)
+
+
+@mcp.tool
+def shop_at(store_name:str) -> str:
+    '''Select store to shop at'''
+    return '\n'.join(Data().get_shopping_list(store_name))
+
+
+@mcp.tool
+def list() -> str:
+    '''Use previous store'''
+    return '\n'.join(Data().get_last_shopping_list())
+
+
+#@mcp.tool
+#def list_stores() -> 'list[str]':
+#    return Data().get_stores()
+
+
+#@mcp.tool
+#def list_purchases(store: str) -> 'list[str]':
+#    return Data().list_purchases(store)
+
 # Configure CORS for browser-based clients
 middleware = [
     Middleware(
@@ -36,51 +69,9 @@ middleware = [
     )
 ]
 
-# if __name__ == "__main__":
-
-
 
-#@mcp.tool
-#def buy(items:str) -> 'list[str]':
-#    return Data().purchase_items(items)
-#
-#@mcp.tool
-#def list_bought(store:str) -> 'list[str]':
-#    return Data().list_purchases(store)
-#
-#@mcp.tool
-#def clear(store_name: str, clear_all:bool=False) -> 'list[str]':
-#    return Data().clear_list(store_name, clear_all)
-#
-#
-#@mcp.tool
-#def shop_at(store_name:str) -> 'list[str]':
-#    '''Select store to shop at'''
-#    return Data().get_shopping_list(store_name)
-#
-#
-#@mcp.tool
-#def list() -> 'list[str]':
-#    '''Use previous store'''
-#    return Data().get_last_shopping_list()
-#
-#
-#@mcp.tool
-#def list_stores() -> 'list[str]':
-#    return Data().get_stores()
-#
-#
-#@mcp.tool
-#def list_purchases(store: str) -> 'list[str]':
-#    return Data().list_purchases(store)
 
 if __name__ == "__main__":
-#   mcp.run(
-#           transport="http", 
-#           host=MY_IP, 
-#           stateless_http=True,
-#           port=9000)
-
 
        # Create the ASGI app
     mcp_app = mcp.http_app(