#!/usr/bin/env python3
"""
LangChain + LangGraph 脚手架的基础用法示例
"""

import sys
import os
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

from workflows.workflow_manager import WorkflowManager, WorkflowType


def example_conversation():
    """对话工作流示例"""
    print("=== Conversation Workflow Example ===")
    
    # 使用 config.ini 中的默认模型
    manager = WorkflowManager()
    
    # 第一条消息
    result1 = manager.execute_workflow(
        WorkflowType.CONVERSATION,
        "你好，你能帮我做什么？"
    )
    
    print(f"Session ID: {result1['session_id']}")
    print(f"Response: {result1['result']['messages'][-1].content}")
    
    # 同一会话中的第二条消息
    result2 = manager.execute_workflow(
        WorkflowType.CONVERSATION,
        "你的优势是什么？",
        session_id=result1['session_id']
    )
    
    print(f"Second response: {result2['result']['messages'][-1].content}")

    # 使用不同模型的示例
    # print("\n--- Using a different model (gpt-3.5-turbo) ---")
    # try:
    #     manager_alt_model = WorkflowManager(default_model_section='gpt-3.5-turbo')
    #     result_alt = manager_alt_model.execute_workflow(
    #         WorkflowType.CONVERSATION,
    #         "Hi, what model are you?"
    #     )
    #     print(f"Response from gpt-3.5-turbo: {result_alt['result']['messages'][-1].content}")
    # except ValueError as e:
    #     print(f"Could not run alternate model example: {e}")
    #     print("Please ensure you have a [gpt-3.5-turbo] section in your config.ini")
    #
    # print("\n")


def example_tool_usage():
    """工具工作流示例"""
    print("=== Tool Workflow Example ===")
    
    manager = WorkflowManager()
    
    # 使用计算器工具
    result = manager.execute_workflow(
        WorkflowType.TOOL_USING,
        "计算 25 * 4 + 10"
    )
    
    print(f"Session ID: {result['session_id']}")
    
    # 提取工具消息与回复
    for message in result['result']['messages']:
        if hasattr(message, 'tool_calls') and message.tool_calls:
            print(f"Tool call: {message.tool_calls}")
        elif hasattr(message, 'content') and message.content:
            print(f"Response: {message.content}")
    
    print("\n")


def list_available_workflows():
    """列出可用工作流"""
    print("=== Available Workflows ===")
    
    manager = WorkflowManager()
    workflows = manager.get_available_workflows()
    
    for workflow in workflows:
        print(f"- {workflow}")
    
    print("\n")


if __name__ == "__main__":
    # 检查配置是否有效
    try:
        from config import Config
        Config.validate_config()
        print("✅ Configuration is valid")
        print("\n")
        
        # 运行示例
        list_available_workflows()
        example_conversation()
        example_tool_usage()
        
    except Exception as e:
        print(f"❌ Configuration error: {e}")
        print("\nPlease make sure to:")
        print("1. Copy 'config/config.ini.example' to 'config/config.ini'")
        print("2. Set your API key in the 'config/config.ini' file")
        print("3. Install dependencies: pip install -r requirements.txt")