"""推理引擎API路由"""

from fastapi import APIRouter, HTTPException
from pydantic import BaseModel

router = APIRouter()

# 导入推理引擎
import sys
from pathlib import Path
sys.path.insert(0, str(Path(__file__).parent.parent / 'scripts' / 'utils'))
try:
    from reasoning_engine import ReasoningEngine
except ImportError:
    # 备用导入方式
    import importlib.util
    reasoning_engine_path = Path(__file__).parent.parent / 'scripts' / 'utils' / 'reasoning_engine.py'
    spec = importlib.util.spec_from_file_location("reasoning_engine", reasoning_engine_path)
    reasoning_engine = importlib.util.module_from_spec(spec)
    spec.loader.exec_module(reasoning_engine)
    ReasoningEngine = reasoning_engine.ReasoningEngine

class ReasoningQuery(BaseModel):
    question: str

@router.post("/reasoning/query")
async def reasoning_query(query: ReasoningQuery):
    """多跳推理查询"""
    try:
        engine = ReasoningEngine()
        result = engine.multi_hop_query(query.question)
        return {"status": "success", **result}
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))

@router.get("/reasoning/examples")
async def reasoning_examples():
    """推理查询示例"""
    return {
        "status": "success",
        "examples": [
            {
                "question": "查找与CYP71D184同组的基因",
                "description": "多跳推理：基因→分组→同组基因",
                "hops": 2
            },
            {
                "question": "P450分组统计比较",
                "description": "聚合统计：所有分组的统计分析",
                "hops": 1
            },
            {
                "question": "最大的分组是哪个",
                "description": "极值查询：查找最大分组",
                "hops": 1
            }
        ]
    }