aetherforge-platform 1.0.0__py3-none-any.whl

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

@@ -0,0 +1,531 @@
+from langgraph.graph import StateGraph, END
+from typing import TypedDict, List, Optional, Dict, Any
+import datetime
+import json
+import os
+from config import GROK_API_KEY
+from memory_v2 import memory_system_v2, Memory, MemoryType
+from tree_memory import tree_memory_system, InvertedTreeMemory
+from mcp_tools import mcp_tool_manager
+
+def get_user_settings(user_id: str):
+    """获取用户设置"""
+    try:
+        user_settings_file = os.path.join("user_data", f"{user_id}_settings.json")
+        if os.path.exists(user_settings_file):
+            with open(user_settings_file, "r", encoding="utf-8") as f:
+                return json.load(f)
+    except Exception as e:
+        print(f"Error loading user settings: {e}")
+    return {}
+
+def get_user_memory_systems(user_id: str):
+    """获取用户选中的 avatar 的记忆系统"""
+    settings = get_user_settings(user_id)
+    selected_avatar_id = settings.get("selected_avatar_id")
+    
+    if selected_avatar_id:
+        try:
+            from avatar_manager import AvatarManager
+            avatar_manager = AvatarManager()
+            avatar = avatar_manager.load_avatar(selected_avatar_id)
+            if avatar:
+                avatar_memory_v2 = memory_system_v2
+                avatar_memory_tree = InvertedTreeMemory(avatar_id=selected_avatar_id)
+                return avatar_memory_v2, avatar_memory_tree, avatar
+        except Exception as e:
+            print(f"Error loading avatar memory: {e}")
+    
+    return memory_system_v2, tree_memory_system, None
+
+class LifeAssistantState(TypedDict):
+    user_id: str
+    message: str
+    intent: Optional[str]
+    requirements: Optional[Dict[str, Any]]
+    results: List[Dict[str, Any]]
+    cards: List[Dict[str, Any]]
+    visualization: Optional[str]
+    next_action: Optional[str]
+    status: str
+    memory: Dict[str, Any]
+    conversation_history: List[Dict[str, str]]
+
+def parser_node(state: LifeAssistantState) -> LifeAssistantState:
+    """Parser Agent: 解析用户意图"""
+    try:
+        from langchain_groq import ChatGroq
+        from langchain.prompts import ChatPromptTemplate
+        
+        llm = ChatGroq(
+            api_key=GROK_API_KEY,
+            model="mixtral-8x7b-32768"
+        )
+        
+        user_memory_v2, user_memory_tree, avatar = get_user_memory_systems(state["user_id"])
+        memory_v2_context = user_memory_v2.get_relevant_context(state["user_id"], state["message"])
+        memory_tree_context = user_memory_tree.get_relevant_context(state["message"])
+        
+        combined_memory = memory_v2_context
+        if memory_tree_context:
+            combined_memory = f"{memory_v2_context}\n\n{memory_tree_context}"
+        
+        avatar_context = ""
+        if avatar:
+            avatar_context = f"""
+【虚拟形象信息】
+名字: {avatar.name}
+年龄: {avatar.age}
+性格: {', '.join(avatar.personality)}
+描述: {avatar.description}
+"""
+        
+        prompt = ChatPromptTemplate.from_messages([
+            ("system", """你是一个AI生活助手的意图解析器。
+请分析用户的输入，判断用户想要做什么。
+
+可能的意图：
+- shopping: 购物、买东西、推荐商品、找商品
+- food: 外卖、点餐、饿了、想吃东西
+- taxi: 打车、叫车、回家、去某个地方
+- flight: 机票、订机票、旅行
+- weather: 天气、气温、预报
+- calendar: 日程、日历、安排
+- reminder: 提醒、闹钟、通知
+- general: 普通聊天、问候、其他问题
+
+{avatar_info}
+
+{user_memory}
+
+请返回JSON格式，包含：
+- intent: 意图类型（shopping/food/taxi/flight/weather/calendar/reminder/general）
+- requirements: 详细需求（商品名称、数量、预算、时间、地点等）
+
+只返回JSON，不要其他文字。"""),
+            ("human", "{message}")
+        ])
+        
+        chain = prompt | llm
+        response = chain.invoke({
+            "message": state["message"],
+            "user_memory": combined_memory,
+            "avatar_info": avatar_context
+        })
+        
+        try:
+            parsed = json.loads(response.content)
+            state["intent"] = parsed.get("intent", "general")
+            state["requirements"] = parsed.get("requirements", {})
+        except:
+            state["intent"] = "general"
+            state["requirements"] = {}
+            
+    except Exception as e:
+        print(f"Parser error: {e}")
+        state["intent"] = "general"
+        state["requirements"] = {}
+    
+    return state
+
+def router_node(state: LifeAssistantState) -> LifeAssistantState:
+    """Router: 根据意图路由到对应的Agent"""
+    intent = state["intent"]
+    
+    if intent == "shopping":
+        state["next_action"] = "shopping"
+    elif intent == "food":
+        state["next_action"] = "food"
+    elif intent == "taxi":
+        state["next_action"] = "taxi"
+    elif intent == "flight":
+        state["next_action"] = "flight"
+    elif intent == "weather":
+        state["next_action"] = "weather"
+    elif intent == "calendar":
+        state["next_action"] = "calendar"
+    elif intent == "reminder":
+        state["next_action"] = "reminder"
+    else:
+        state["next_action"] = "general"
+    
+    return state
+
+def shopping_agent_node(state: LifeAssistantState) -> LifeAssistantState:
+    """Shopping Agent: 处理购物需求"""
+    from tools.amazon_tool import search_amazon
+    
+    query = state["requirements"].get("query", state["message"])
+    
+    try:
+        results = search_amazon(query)
+        state["cards"] = results.get("cards", [])
+        state["results"] = results.get("items", [])
+        state["status"] = "success"
+    except Exception as e:
+        print(f"Shopping agent error: {e}")
+        state["cards"] = []
+        state["results"] = []
+        state["status"] = "error"
+    
+    return state
+
+def food_agent_node(state: LifeAssistantState) -> LifeAssistantState:
+    """Food Agent: 处理外卖需求"""
+    state["cards"] = []
+    state["results"] = [{"type": "food", "message": "外卖功能即将上线！"}]
+    state["status"] = "success"
+    return state
+
+def taxi_agent_node(state: LifeAssistantState) -> LifeAssistantState:
+    """Taxi Agent: 处理打车需求"""
+    state["cards"] = []
+    state["results"] = [{"type": "taxi", "message": "打车功能即将上线！"}]
+    state["status"] = "success"
+    return state
+
+def flight_agent_node(state: LifeAssistantState) -> LifeAssistantState:
+    """Flight Agent: 处理机票需求"""
+    state["cards"] = []
+    state["results"] = [{"type": "flight", "message": "机票功能即将上线！"}]
+    state["status"] = "success"
+    return state
+
+def weather_agent_node(state: LifeAssistantState) -> LifeAssistantState:
+    """Weather Agent: 处理天气需求"""
+    from tools.weather_tool import get_weather
+    
+    location = state["requirements"].get("location", "北京")
+    
+    try:
+        result = get_weather(location)
+        state["cards"] = result.get("cards", [])
+        state["results"] = [result.get("data", {})]
+        state["status"] = "success"
+    except Exception as e:
+        print(f"Weather agent error: {e}")
+        state["cards"] = []
+        state["results"] = [{"type": "weather", "message": "天气查询失败！"}]
+        state["status"] = "error"
+    
+    return state
+
+def calendar_agent_node(state: LifeAssistantState) -> LifeAssistantState:
+    """Calendar Agent: 处理日程需求"""
+    from tools.calendar_tool import get_events, add_event
+    
+    try:
+        if "add" in state["message"].lower() or "创建" in state["message"] or "添加" in state["message"]:
+            # 添加日程
+            title = state["requirements"].get("title", "")
+            time = state["requirements"].get("time", "")
+            date = state["requirements"].get("date", None)
+            result = add_event(title, time, date)
+        else:
+            # 获取日程
+            date = state["requirements"].get("date", None)
+            result = get_events(date)
+        
+        state["cards"] = result.get("cards", [])
+        state["results"] = [result.get("events", [])]
+        state["status"] = "success"
+    except Exception as e:
+        print(f"Calendar agent error: {e}")
+        state["cards"] = []
+        state["results"] = [{"type": "calendar", "message": "日程处理失败！"}]
+        state["status"] = "error"
+    
+    return state
+
+def reminder_agent_node(state: LifeAssistantState) -> LifeAssistantState:
+    """Reminder Agent: 处理提醒需求"""
+    from tools.reminder_tool import set_reminder, get_reminders
+    
+    try:
+        if "设置" in state["message"] or "添加" in state["message"] or "创建" in state["message"]:
+            # 设置提醒
+            title = state["requirements"].get("title", "")
+            time = state["requirements"].get("time", "")
+            date = state["requirements"].get("date", None)
+            repeat = state["requirements"].get("repeat", None)
+            result = set_reminder(title, time, date, repeat)
+        else:
+            # 获取提醒
+            date = state["requirements"].get("date", None)
+            result = get_reminders(date)
+        
+        state["cards"] = result.get("cards", [])
+        state["results"] = [result.get("reminders", [])]
+        state["status"] = "success"
+    except Exception as e:
+        print(f"Reminder agent error: {e}")
+        state["cards"] = []
+        state["results"] = [{"type": "reminder", "message": "提醒处理失败！"}]
+        state["status"] = "error"
+    
+    return state
+
+def general_agent_node(state: LifeAssistantState) -> LifeAssistantState:
+    """General Agent: 处理普通聊天，自动决定是否调用工具"""
+    try:
+        from langchain_groq import ChatGroq
+        from langchain.prompts import ChatPromptTemplate
+        import json
+        
+        llm = ChatGroq(
+            api_key=GROK_API_KEY,
+            model="mixtral-8x7b-32768"
+        )
+        
+        context = "\n".join([
+            f"{msg['role']}: {msg['content']}" 
+            for msg in state["conversation_history"][-5:]
+        ])
+        
+        user_memory_v2, user_memory_tree, avatar = get_user_memory_systems(state["user_id"])
+        memory_v2_context = user_memory_v2.get_relevant_context(state["user_id"], state["message"])
+        memory_tree_context = user_memory_tree.get_relevant_context(state["message"])
+        
+        combined_memory = memory_v2_context
+        if memory_tree_context:
+            combined_memory = f"{memory_v2_context}\n\n{memory_tree_context}"
+        
+        avatar_context = ""
+        if avatar:
+            avatar_context = f"""
+【虚拟形象信息】
+名字: {avatar.name}
+年龄: {avatar.age}
+性格: {', '.join(avatar.personality)}
+描述: {avatar.description}
+"""
+        
+        tools = mcp_tool_manager.list_tools()
+        
+        prompt = ChatPromptTemplate.from_messages([
+            ("system", """You are a friendly AI life assistant.
+
+Available tools:
+{tools_list}
+
+{avatar_info}
+
+【User background info】
+{user_memory}
+
+Important guidelines:
+1. First, decide if you need to use a tool.
+2. If you need a tool, respond with EXACTLY this JSON format (and nothing else):
+   {{"tool": "tool_name", "params": {{"param1": "value1"}}}}
+3. If you don't need a tool, just respond normally in text.
+4. If you have user background info, use it to provide better help
+5. But, don't just repeat what the user already knows
+6. Provide new insights, suggestions, and ideas
+7. If the user just states a fact, don't always bring it up again
+8. Give the user space to think and discover new things
+9. If you have avatar information, use that persona in your conversation
+
+Please respond in a friendly, concise, natural way, like chatting with a normal person."""),
+            ("human", "Conversation history:\n{context}\n\nUser's latest question: {message}")
+        ])
+        
+        tools_list = "\n".join([
+            f"- {tool['name']}: {tool['description']}\n  Params: {', '.join(tool['params'])}"
+            for tool in tools
+        ])
+        
+        chain = prompt | llm
+        response = chain.invoke({
+            "context": context,
+            "message": state["message"],
+            "user_memory": combined_memory,
+            "tools_list": tools_list,
+            "avatar_info": avatar_context
+        })
+        
+        response_text = response.content.strip()
+        
+        if response_text.startswith('{') and response_text.endswith('}'):
+            try:
+                tool_call = json.loads(response_text)
+                tool_name = tool_call.get("tool")
+                tool_params = tool_call.get("params", {})
+                
+                print(f"🔧 Calling tool: {tool_name} with params: {tool_params}")
+                
+                result = mcp_tool_manager.execute_tool(tool_name, tool_params)
+                
+                if result.get("success"):
+                    state["visualization"] = result.get("result", "")
+                    state["results"] = [{
+                        "type": "text",
+                        "content": "Here's what I found:"
+                    }]
+                else:
+                    state["results"] = [{
+                        "type": "text",
+                        "content": "Sorry, I couldn't complete that. Let me help you another way!"
+                    }]
+            except Exception as e:
+                print(f"Tool call error: {e}")
+                state["results"] = [{
+                    "type": "text",
+                    "content": "Sorry, let's try that again!"
+                }]
+        else:
+            state["results"] = [{
+                "type": "text",
+                "content": response_text
+            }]
+        
+        state["status"] = "success"
+        
+    except Exception as e:
+        print(f"General agent error: {e}")
+        state["results"] = [{
+            "type": "text",
+            "content": "Sorry, I'm a bit busy right now. Please try again later!"
+        }]
+        state["status"] = "error"
+    
+    return state
+
+def should_continue(state: LifeAssistantState) -> str:
+    """决定下一步走向"""
+    if state["next_action"] == "shopping":
+        return "shopping"
+    elif state["next_action"] == "food":
+        return "food"
+    elif state["next_action"] == "taxi":
+        return "taxi"
+    elif state["next_action"] == "flight":
+        return "flight"
+    elif state["next_action"] == "weather":
+        return "weather"
+    elif state["next_action"] == "calendar":
+        return "calendar"
+    elif state["next_action"] == "reminder":
+        return "reminder"
+    else:
+        return "general"
+
+workflow = StateGraph(LifeAssistantState)
+
+workflow.add_node("parser", parser_node)
+workflow.add_node("router", router_node)
+workflow.add_node("shopping", shopping_agent_node)
+workflow.add_node("food", food_agent_node)
+workflow.add_node("taxi", taxi_agent_node)
+workflow.add_node("flight", flight_agent_node)
+workflow.add_node("weather", weather_agent_node)
+workflow.add_node("calendar", calendar_agent_node)
+workflow.add_node("reminder", reminder_agent_node)
+workflow.add_node("general", general_agent_node)
+
+workflow.set_entry_point("parser")
+workflow.add_edge("parser", "router")
+
+workflow.add_conditional_edges(
+    "router",
+    should_continue,
+    {
+        "shopping": "shopping",
+        "food": "food",
+        "taxi": "taxi",
+        "flight": "flight",
+        "weather": "weather",
+        "calendar": "calendar",
+        "reminder": "reminder",
+        "general": "general"
+    }
+)
+
+workflow.add_edge("shopping", END)
+workflow.add_edge("food", END)
+workflow.add_edge("taxi", END)
+workflow.add_edge("flight", END)
+workflow.add_edge("weather", END)
+workflow.add_edge("calendar", END)
+workflow.add_edge("reminder", END)
+workflow.add_edge("general", END)
+
+app = workflow.compile()
+
+def run_chat(message: str, user_id: str = "default", conversation_history: List[Dict[str, str]] = None):
+    """运行聊天流程"""
+    if conversation_history is None:
+        conversation_history = []
+    
+    # 解析命令
+    from commands import CommandParser
+    parser = CommandParser()
+    command_result = parser.parse(message)
+    
+    if command_result["type"] == "command":
+        # 处理命令
+        if command_result["command"] == "help":
+            return {
+                "results": [{"type": "text", "content": parser.get_help()}],
+                "cards": [],
+                "status": "success"
+            }
+        
+        # 直接设置意图和需求
+        initial_state: LifeAssistantState = {
+            "user_id": user_id,
+            "message": message,
+            "intent": command_result["intent"],
+            "requirements": command_result.get("requirements", {}),
+            "results": [],
+            "cards": [],
+            "visualization": None,
+            "next_action": command_result["intent"],
+            "status": "pending",
+            "memory": {},
+            "conversation_history": conversation_history
+        }
+    else:
+        # 正常消息
+        initial_state: LifeAssistantState = {
+            "user_id": user_id,
+            "message": message,
+            "intent": None,
+            "requirements": None,
+            "results": [],
+            "cards": [],
+            "visualization": None,
+            "next_action": None,
+            "status": "pending",
+            "memory": {},
+            "conversation_history": conversation_history
+        }
+    
+    result = app.invoke(initial_state)
+    
+    response_text = ""
+    if result.get("results"):
+        for res in result["results"]:
+            if res.get("type") == "text":
+                response_text = res.get("content", "")
+                break
+    
+    user_memory_v2, user_memory_tree, avatar = get_user_memory_systems(user_id)
+    user_memory_v2.extract_and_save_memory(
+        user_id=user_id,
+        message=message,
+        response=response_text
+    )
+    
+    if avatar:
+        user_memory_tree.extract_and_save_memory(
+            message=message,
+            response=response_text
+        )
+    else:
+        tree_memory_system.extract_and_save_memory(
+            avatar_id=user_id,
+            message=message,
+            response=response_text
+        )
+    
+    return result