aetherforge-platform 1.0.0__py3-none-any.whl

ai-life-assistant-copy/backups/backup_20260404_193836/mcp_tools.py

@@ -0,0 +1,449 @@
+import json
+import os
+from typing import Dict, List, Any, Optional, Callable
+from enum import Enum
+from dataclasses import dataclass
+import sys
+sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
+from mock_shopping_data import search_mock_products, get_price_history, compare_prices
+
+class ToolType(Enum):
+    VISUALIZATION = "visualization"
+    DATA_ANALYSIS = "data_analysis"
+    CALCULATION = "calculation"
+    GENERAL = "general"
+
+@dataclass
+class ToolParameter:
+    name: str
+    type: str
+    description: str
+    required: bool = True
+    default: Any = None
+
+@dataclass
+class ToolDefinition:
+    name: str
+    description: str
+    type: ToolType
+    parameters: List[ToolParameter]
+    handler: Callable
+
+class MCPToolManager:
+    def __init__(self):
+        self.tools: Dict[str, ToolDefinition] = {}
+        self._register_builtin_tools()
+    
+    def _register_builtin_tools(self):
+        self.register_tool(ToolDefinition(
+            name="create_chart",
+            description="Create a visualization chart (bar, line, pie, etc.)",
+            type=ToolType.VISUALIZATION,
+            parameters=[
+                ToolParameter("chart_type", "string", "Type of chart: bar, line, pie, scatter", True),
+                ToolParameter("title", "string", "Chart title", True),
+                ToolParameter("data", "object", "Chart data: {labels: [], values: []}", True),
+                ToolParameter("x_label", "string", "X-axis label", False),
+                ToolParameter("y_label", "string", "Y-axis label", False)
+            ],
+            handler=self._create_chart
+        ))
+        
+        self.register_tool(ToolDefinition(
+            name="create_table",
+            description="Create a data table visualization",
+            type=ToolType.VISUALIZATION,
+            parameters=[
+                ToolParameter("title", "string", "Table title", True),
+                ToolParameter("columns", "array", "Column names: [col1, col2, ...]", True),
+                ToolParameter("rows", "array", "Table rows: [[row1val1, row1val2], ...]", True)
+            ],
+            handler=self._create_table
+        ))
+        
+        self.register_tool(ToolDefinition(
+            name="calculate",
+            description="Perform calculations and show results",
+            type=ToolType.CALCULATION,
+            parameters=[
+                ToolParameter("expression", "string", "Mathematical expression", True),
+                ToolParameter("description", "string", "What this calculation is for", False)
+            ],
+            handler=self._calculate
+        ))
+        
+        self.register_tool(ToolDefinition(
+            name="create_timeline",
+            description="Create a timeline visualization",
+            type=ToolType.VISUALIZATION,
+            parameters=[
+                ToolParameter("title", "string", "Timeline title", True),
+                ToolParameter("events", "array", "Events: [{date: '', title: '', description: ''}, ...]", True)
+            ],
+            handler=self._create_timeline
+        ))
+        
+        self.register_tool(ToolDefinition(
+            name="search_products",
+            description="Search for products across multiple platforms",
+            type=ToolType.DATA_ANALYSIS,
+            parameters=[
+                ToolParameter("keyword", "string", "Product keyword to search", True)
+            ],
+            handler=self._search_products
+        ))
+        
+        self.register_tool(ToolDefinition(
+            name="compare_prices",
+            description="Compare prices of a product across platforms",
+            type=ToolType.DATA_ANALYSIS,
+            parameters=[
+                ToolParameter("product_name", "string", "Name of the product", True)
+            ],
+            handler=self._compare_prices
+        ))
+        
+        self.register_tool(ToolDefinition(
+            name="get_price_history",
+            description="Get price history of a product",
+            type=ToolType.VISUALIZATION,
+            parameters=[
+                ToolParameter("product_id", "string", "Product ID", True),
+                ToolParameter("days", "integer", "Number of days to show (default: 30)", False, 30)
+            ],
+            handler=self._get_price_history
+        ))
+    
+    def register_tool(self, tool: ToolDefinition):
+        self.tools[tool.name] = tool
+    
+    def get_tool(self, name: str) -> Optional[ToolDefinition]:
+        return self.tools.get(name)
+    
+    def list_tools(self, tool_type: Optional[ToolType] = None) -> List[ToolDefinition]:
+        if tool_type:
+            return [t for t in self.tools.values() if t.type == tool_type]
+        return list(self.tools.values())
+    
+    async def execute_tool(self, tool_name: str, params: Dict[str, Any]) -> Dict[str, Any]:
+        tool = self.get_tool(tool_name)
+        if not tool:
+            return {"success": False, "error": f"Tool '{tool_name}' not found"}
+        
+        try:
+            result = await tool.handler(**params)
+            return {"success": True, "result": result}
+        except Exception as e:
+            return {"success": False, "error": str(e)}
+    
+    async def _create_chart(self, chart_type: str, title: str, data: Dict[str, Any], 
+                           x_label: Optional[str] = None, y_label: Optional[str] = None) -> Dict[str, Any]:
+        return {
+            "type": "chart",
+            "chart_type": chart_type,
+            "title": title,
+            "data": data,
+            "x_label": x_label,
+            "y_label": y_label,
+            "html": self._generate_chart_html(chart_type, title, data, x_label, y_label)
+        }
+    
+    async def _create_table(self, title: str, columns: List[str], rows: List[List[Any]]) -> Dict[str, Any]:
+        return {
+            "type": "table",
+            "title": title,
+            "columns": columns,
+            "rows": rows,
+            "html": self._generate_table_html(title, columns, rows)
+        }
+    
+    async def _calculate(self, expression: str, description: Optional[str] = None) -> Dict[str, Any]:
+        try:
+            result = eval(expression, {"__builtins__": {}}, {})
+            return {
+                "type": "calculation",
+                "expression": expression,
+                "description": description,
+                "result": result,
+                "html": self._generate_calculation_html(expression, result, description)
+            }
+        except Exception as e:
+            return {"type": "calculation", "expression": expression, "error": str(e)}
+    
+    async def _create_timeline(self, title: str, events: List[Dict[str, Any]]) -> Dict[str, Any]:
+        return {
+            "type": "timeline",
+            "title": title,
+            "events": events,
+            "html": self._generate_timeline_html(title, events)
+        }
+    
+    async def _search_products(self, keyword: str) -> Dict[str, Any]:
+        results = search_mock_products(keyword)
+        return {
+            "type": "product_list",
+            "keyword": keyword,
+            "results": results["results"],
+            "html": self._generate_product_list_html(keyword, results["results"])
+        }
+    
+    async def _compare_prices(self, product_name: str) -> Dict[str, Any]:
+        comparison = compare_prices(product_name)
+        return {
+            "type": "price_comparison",
+            "product_name": product_name,
+            "comparison": comparison,
+            "html": self._generate_price_comparison_html(product_name, comparison)
+        }
+    
+    async def _get_price_history(self, product_id: str, days: int = 30) -> Dict[str, Any]:
+        history = get_price_history(product_id, days)
+        return {
+            "type": "price_history",
+            "product_id": product_id,
+            "history": history["history"],
+            "html": self._generate_price_history_html(product_id, history["history"])
+        }
+    
+    def _generate_chart_html(self, chart_type: str, title: str, data: Dict[str, Any],
+                            x_label: Optional[str], y_label: Optional[str]) -> str:
+        labels = data.get("labels", [])
+        values = data.get("values", [])
+        
+        return f"""
+<div style="background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); padding: 24px; border-radius: 16px; color: white; margin: 16px 0;">
+    <h3 style="margin: 0 0 16px 0; font-size: 20px;">{title}</h3>
+    <div style="display: flex; flex-wrap: wrap; gap: 8px; margin-bottom: 16px;">
+        <span style="background: rgba(255,255,255,0.2); padding: 4px 12px; border-radius: 20px; font-size: 14px;">
+            {chart_type.capitalize()} Chart
+        </span>
+    </div>
+    <div style="background: rgba(0,0,0,0.2); padding: 16px; border-radius: 12px;">
+        <div style="display: flex; justify-content: space-between; margin-bottom: 8px;">
+            <span>{x_label or 'Labels'}</span>
+            <span>{y_label or 'Values'}</span>
+        </div>
+        {''.join([f'''
+        <div style="display: flex; justify-content: space-between; padding: 8px 0; border-bottom: 1px solid rgba(255,255,255,0.2);">
+            <span>{labels[i] if i < len(labels) else '-'}</span>
+            <span style="font-weight: 600;">{values[i] if i < len(values) else '-'}</span>
+        </div>
+        ''' for i in range(max(len(labels), len(values)))])}
+    </div>
+</div>
+"""
+    
+    def _generate_table_html(self, title: str, columns: List[str], rows: List[List[Any]]) -> str:
+        return f"""
+<div style="background: white; border-radius: 16px; overflow: hidden; box-shadow: 0 4px 20px rgba(0,0,0,0.1); margin: 16px 0;">
+    <div style="background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); padding: 16px 24px; color: white;">
+        <h3 style="margin: 0; font-size: 18px;">{title}</h3>
+    </div>
+    <table style="width: 100%; border-collapse: collapse;">
+        <thead>
+            <tr style="background: #f8f9fa;">
+                {''.join([f'<th style="padding: 12px 16px; text-align: left; border-bottom: 2px solid #e9ecef;">{col}</th>' for col in columns])}
+            </tr>
+        </thead>
+        <tbody>
+            {''.join([f'''
+            <tr style="border-bottom: 1px solid #f1f3f4;">
+                {''.join([f'<td style="padding: 12px 16px;">{cell}</td>' for cell in row])}
+            </tr>
+            ''' for row in rows])}
+        </tbody>
+    </table>
+</div>
+"""
+    
+    def _generate_calculation_html(self, expression: str, result: Any, description: Optional[str]) -> str:
+        return f"""
+<div style="background: linear-gradient(135deg, #11998e 0%, #38ef7d 100%); padding: 24px; border-radius: 16px; color: white; margin: 16px 0;">
+    {f'<p style="margin: 0 0 12px 0; opacity: 0.9;">{description}</p>' if description else ''}
+    <div style="display: flex; align-items: baseline; gap: 16px; flex-wrap: wrap;">
+        <div style="background: rgba(0,0,0,0.2); padding: 12px 20px; border-radius: 12px; font-family: monospace; font-size: 18px;">
+            {expression}
+        </div>
+        <span style="font-size: 24px;">=</span>
+        <div style="background: rgba(255,255,255,0.2); padding: 12px 20px; border-radius: 12px; font-size: 24px; font-weight: 700;">
+            {result}
+        </div>
+    </div>
+</div>
+"""
+    
+    def _generate_timeline_html(self, title: str, events: List[Dict[str, Any]]) -> str:
+        return f"""
+<div style="background: white; padding: 24px; border-radius: 16px; box-shadow: 0 4px 20px rgba(0,0,0,0.1); margin: 16px 0;">
+    <h3 style="margin: 0 0 24px 0; font-size: 20px; background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); -webkit-background-clip: text; -webkit-text-fill-color: transparent;">
+        {title}
+    </h3>
+    <div style="position: relative; padding-left: 24px;">
+        <div style="position: absolute; left: 7px; top: 0; bottom: 0; width: 2px; background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);"></div>
+        {''.join([f'''
+        <div style="margin-bottom: 24px; position: relative;">
+            <div style="position: absolute; left: -28px; top: 4px; width: 16px; height: 16px; background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); border-radius: 50%; border: 3px solid white; box-shadow: 0 2px 8px rgba(0,0,0,0.2);"></div>
+            <div style="background: #f8f9fa; padding: 16px; border-radius: 12px; border-left: 4px solid #667eea;">
+                <div style="font-weight: 600; color: #667eea; margin-bottom: 4px;">{event.get('date', '')}</div>
+                <div style="font-weight: 600; margin-bottom: 4px;">{event.get('title', '')}</div>
+                <div style="color: #6e6e73;">{event.get('description', '')}</div>
+            </div>
+        </div>
+        ''' for event in events])}
+    </div>
+</div>
+"""
+    
+    def _generate_product_list_html(self, keyword: str, products: List[Dict[str, Any]]) -> str:
+        product_cards = ''.join([self._generate_product_card(product) for product in products])
+        return f"""
+<div style="margin: 16px 0;">
+    <h3 style="margin: 0 0 16px 0; font-size: 18px; color: #1d1d1f;">
+        🔍 Results for "{keyword}"
+    </h3>
+    <div style="display: grid; grid-template-columns: repeat(auto-fill, minmax(280px, 1fr)); gap: 16px;">
+        {product_cards}
+    </div>
+</div>
+"""
+    
+    def _generate_product_card(self, product: Dict[str, Any]) -> str:
+        price_change_icon = "⬇️" if product.get("priceChangeDirection") == "down" else "⬆️"
+        price_change_class = "down" if product.get("priceChangeDirection") == "down" else "up"
+        
+        return f"""
+<a href="{product['url']}" target="_blank" style="text-decoration: none; color: inherit;">
+    <div style="background: white; border-radius: 16px; overflow: hidden; box-shadow: 0 2px 12px rgba(0,0,0,0.08); transition: all 0.3s ease; cursor: pointer; border: 1px solid #e6e6e6;" onmouseover="this.style.transform='translateY(-4px)';this.style.boxShadow='0 8px 24px rgba(0,0,0,0.12)';" onmouseout="this.style.transform='translateY(0)';this.style.boxShadow='0 2px 12px rgba(0,0,0,0.08)';">
+        <div style="padding: 16px; display: flex; gap: 16px; align-items: flex-start;">
+            <img src="{product['image']}" alt="{product['title']}" style="width: 88px; height: 88px; object-fit: cover; border-radius: 12px; background: #f5f5f7; flex-shrink: 0;">
+            <div style="flex: 1; min-width: 0;">
+                <div style="font-weight: 600; font-size: 15px; color: #1d1d1f; margin-bottom: 4px; overflow: hidden; text-overflow: ellipsis; display: -webkit-box; -webkit-line-clamp: 2; -webkit-box-orient: vertical;">
+                    {product['title']}
+                </div>
+                <div style="font-size: 13px; color: #6e6e73; margin-bottom: 8px; overflow: hidden; text-overflow: ellipsis; display: -webkit-box; -webkit-line-clamp: 2; -webkit-box-orient: vertical;">
+                    {product.get('description', '')}
+                </div>
+                <div style="display: flex; align-items: baseline; gap: 8px; margin-bottom: 8px;">
+                    <span style="font-size: 20px; font-weight: 700; color: #1d1d1f;">
+                        ${product['price']}
+                    </span>
+                    {f'<span style="font-size: 14px; color: #86868b; text-decoration: line-through;">${product["originalPrice"]}</span>' if product.get('originalPrice') else ''}
+                </div>
+                <div style="display: flex; align-items: center; gap: 12px; flex-wrap: wrap;">
+                    <span style="background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); color: white; padding: 4px 10px; border-radius: 20px; font-size: 12px; font-weight: 500;">
+                        {product['platform']}
+                    </span>
+                    {f'''
+                    <span style="display: flex; align-items: center; gap: 4px; font-size: 14px; color: #ff3b30;">
+                        <i class="fas fa-star"></i>
+                        {product['rating']} ({product['reviewCount']} reviews)
+                    </span>
+                    ''' if product.get('rating') else ''}
+                    {f'''
+                    <span style="color: {'#34c759' if price_change_class == 'down' else '#ff3b30'}; font-size: 14px; font-weight: 500;">
+                        {price_change_icon} {abs(product.get('priceChange', 0))}%
+                    </span>
+                    ''' if product.get('priceChange') else ''}
+                </div>
+            </div>
+        </div>
+        <div style="background: #f5f5f7; padding: 12px 16px; display: flex; justify-content: flex-end; gap: 8px;">
+            <span style="color: #007aff; font-size: 14px; font-weight: 500;">
+                View Deal <i class="fas fa-external-link-alt" style="margin-left: 4px; font-size: 12px;"></i>
+            </span>
+        </div>
+    </div>
+</a>
+"""
+    
+    def _generate_price_comparison_html(self, product_name: str, comparison: Dict[str, Any]) -> str:
+        prices = comparison.get('prices', [])
+        lowest = comparison.get('lowest_price', {})
+        
+        return f"""
+<div style="background: white; border-radius: 16px; overflow: hidden; box-shadow: 0 4px 20px rgba(0,0,0,0.1); margin: 16px 0;">
+    <div style="background: linear-gradient(135deg, #ff9500 0%, #ff6d00 100%); padding: 20px 24px; color: white;">
+        <h3 style="margin: 0 0 8px 0; font-size: 18px;">
+            💸 Price Comparison
+        </h3>
+        <p style="margin: 0; opacity: 0.9; font-size: 14px;">
+            {product_name}
+        </p>
+    </div>
+    <div style="padding: 20px 24px;">
+        <div style="display: flex; flex-direction: column; gap: 12px;">
+            {''.join([f'''
+            <a href="{p['url']}" target="_blank" style="text-decoration: none; color: inherit;">
+                <div style="display: flex; justify-content: space-between; align-items: center; padding: 16px; border-radius: 12px; background: {'#f0fff4' if p == lowest else '#f5f5f7'}; border: 2px solid {'#34c759' if p == lowest else 'transparent'};">
+                    <div style="display: flex; align-items: center; gap: 12px;">
+                        <span style="font-weight: 600; font-size: 16px;">{p['platform']}</span>
+                        {f'<span style="background: #34c759; color: white; padding: 2px 8px; border-radius: 10px; font-size: 11px; font-weight: 600;">LOWEST</span>' if p == lowest else ''}
+                    </div>
+                    <span style="font-size: 20px; font-weight: 700; color: {'#34c759' if p == lowest else '#1d1d1f'};">
+                        ${p['price']}
+                    </span>
+                </div>
+            </a>
+            ''' for p in prices])}
+        </div>
+        {f'''
+        <div style="margin-top: 20px; padding-top: 16px; border-top: 1px solid #e6e6e6;">
+            <div style="display: flex; justify-content: space-between; align-items: center;">
+                <span style="color: #6e6e73;">Maximum Savings</span>
+                <span style="font-size: 24px; font-weight: 700; color: #34c759;">
+                    Save ${comparison.get('savings', 0)}
+                </span>
+            </div>
+        </div>
+        ''' if comparison.get('savings') else ''}
+    </div>
+</div>
+"""
+    
+    def _generate_price_history_html(self, product_id: str, history: List[Dict[str, Any]]) -> str:
+        labels = [item['date'] for item in history]
+        prices = [item['price'] for item in history]
+        min_price = min(prices) if prices else 0
+        max_price = max(prices) if prices else 1
+        
+        return f"""
+<div style="background: white; border-radius: 16px; overflow: hidden; box-shadow: 0 4px 20px rgba(0,0,0,0.1); margin: 16px 0;">
+    <div style="background: linear-gradient(135deg, #5856d6 0%, #af52de 100%); padding: 20px 24px; color: white;">
+        <h3 style="margin: 0 0 8px 0; font-size: 18px;">
+            📊 Price History
+        </h3>
+        <p style="margin: 0; opacity: 0.9; font-size: 14px;">
+            Last {len(history)} days
+        </p>
+    </div>
+    <div style="padding: 24px;">
+        <div style="display: flex; justify-content: space-between; margin-bottom: 16px; padding: 12px; background: #f5f5f7; border-radius: 12px;">
+            <div>
+                <div style="font-size: 12px; color: #6e6e73; margin-bottom: 4px;">Lowest</div>
+                <div style="font-size: 18px; font-weight: 700; color: #34c759;">${min_price}</div>
+            </div>
+            <div>
+                <div style="font-size: 12px; color: #6e6e73; margin-bottom: 4px;">Highest</div>
+                <div style="font-size: 18px; font-weight: 700; color: #ff3b30;">${max_price}</div>
+            </div>
+            <div>
+                <div style="font-size: 12px; color: #6e6e73; margin-bottom: 4px;">Current</div>
+                <div style="font-size: 18px; font-weight: 700; color: #007aff;">${prices[-1] if prices else 0}</div>
+            </div>
+        </div>
+        <div style="height: 120px; display: flex; align-items: flex-end; gap: 4px; padding: 0 8px;">
+            {''.join([f'''
+            <div style="flex: 1; display: flex; flex-direction: column; align-items: center;">
+                <div style="width: 100%; background: linear-gradient(180deg, #5856d6 0%, #af52de 100%); border-radius: 4px 4px 0 0; height: {((p - min_price) / (max_price - min_price) * 100) if max_price > min_price else 50}%; min-height: 8px;"></div>
+                <div style="font-size: 9px; color: #86868b; margin-top: 4px; writing-mode: vertical-rl; transform: rotate(180deg);">
+                    {d.split('-')[2]}
+                </div>
+            </div>
+            ''' for d, p in zip(labels[-15:], prices[-15:])])}
+        </div>
+    </div>
+</div>
+"""
+
+
+mcp_tool_manager = MCPToolManager()

ai-life-assistant-copy/backups/backup_20260404_193836/memory.py

@@ -0,0 +1,92 @@
+import json
+import os
+from typing import List, Dict, Any
+from datetime import datetime
+
+class MemorySystem:
+    def __init__(self, memory_dir: str = "user_memory"):
+        self.memory_dir = memory_dir
+        if not os.path.exists(memory_dir):
+            os.makedirs(memory_dir)
+    
+    def _get_user_file(self, user_id: str) -> str:
+        return os.path.join(self.memory_dir, f"{user_id}.json")
+    
+    def _load_user_memory(self, user_id: str) -> Dict[str, Any]:
+        file_path = self._get_user_file(user_id)
+        if os.path.exists(file_path):
+            with open(file_path, "r", encoding="utf-8") as f:
+                return json.load(f)
+        return {
+            "preferences": {},
+            "history": [],
+            "created_at": datetime.now().isoformat()
+        }
+    
+    def _save_user_memory(self, user_id: str, memory: Dict[str, Any]):
+        file_path = self._get_user_file(user_id)
+        with open(file_path, "w", encoding="utf-8") as f:
+            json.dump(memory, f, ensure_ascii=False, indent=2)
+    
+    def add_preference(self, user_id: str, key: str, value: str):
+        memory = self._load_user_memory(user_id)
+        memory["preferences"][key] = {
+            "value": value,
+            "updated_at": datetime.now().isoformat()
+        }
+        self._save_user_memory(user_id, memory)
+    
+    def get_preferences(self, user_id: str) -> Dict[str, Any]:
+        memory = self._load_user_memory(user_id)
+        return memory.get("preferences", {})
+    
+    def add_history(self, user_id: str, message: str, response: str, intent: str = None):
+        memory = self._load_user_memory(user_id)
+        memory["history"].append({
+            "message": message,
+            "response": response,
+            "intent": intent,
+            "timestamp": datetime.now().isoformat()
+        })
+        if len(memory["history"]) > 50:
+            memory["history"] = memory["history"][-50:]
+        self._save_user_memory(user_id, memory)
+    
+    def get_relevant_memory(self, user_id: str, current_message: str) -> str:
+        memory = self._load_user_memory(user_id)
+        preferences = memory.get("preferences", {})
+        
+        memory_text = ""
+        
+        if preferences:
+            memory_text += "【用户偏好】\n"
+            for key, pref in preferences.items():
+                memory_text += f"- {key}: {pref['value']}\n"
+        
+        recent_history = memory.get("history", [])[-5:]
+        if recent_history:
+            memory_text += "\n【最近对话】\n"
+            for item in recent_history:
+                memory_text += f"- 用户: {item['message']}\n"
+                memory_text += f"- 助手: {item['response'][:100]}...\n"
+        
+        return memory_text if memory_text else "暂无用户记忆"
+    
+    def extract_and_save_preferences(self, user_id: str, message: str, response: str):
+        message_lower = message.lower()
+        
+        preference_keywords = {
+            "喜欢的品牌": ["喜欢", "偏好", "最爱", "常用"],
+            "不喜欢的东西": ["不喜欢", "讨厌", "不要"],
+            "饮食偏好": ["吃素", "不吃辣", "过敏"],
+            "预算范围": ["预算", "大约", "左右"]
+        }
+        
+        for key, keywords in preference_keywords.items():
+            for keyword in keywords:
+                if keyword in message_lower:
+                    self.add_preference(user_id, key, message)
+                    break
+
+
+memory_system = MemorySystem()