Method call_tool

ddddocr/api/mcp.py:121–214

调用MCP工具

(request: MCPRequest)

        
        @self.router.post("/call")
        async def call_tool(request: MCPRequest):
            """调用MCP工具"""
            try:
                method = request.method
                params = request.params
                
                if method == "ddddocr_initialize":
                    from .models import InitializeRequest
                    init_request = InitializeRequest(**params)
                    result = self.service.initialize(init_request)
                    
                elif method == "ddddocr_ocr":
                    from .models import OCRRequest
                    ocr_request = OCRRequest(**params)
                    
                    if not self.service.ocr_instance:
                        raise HTTPException(status_code=400, detail="OCR功能未初始化")
                    
                    # 解码base64图片
                    image_data = base64.b64decode(ocr_request.image)
                    
                    # 设置字符集范围
                    if ocr_request.charset_range is not None:
                        self.service.ocr_instance.set_ranges(ocr_request.charset_range)
                    
                    # 执行OCR识别
                    result = self.service.ocr_instance.classification(
                        image_data,
                        png_fix=ocr_request.png_fix,
                        probability=ocr_request.probability,
                        color_filter_colors=ocr_request.color_filter_colors,
                        color_filter_custom_ranges=ocr_request.color_filter_custom_ranges
                    )
                    
                elif method == "ddddocr_detection":
                    from .models import DetectionRequest
                    det_request = DetectionRequest(**params)
                    
                    if not self.service.det_instance:
                        raise HTTPException(status_code=400, detail="目标检测功能未初始化")
                    
                    # 解码base64图片
                    image_data = base64.b64decode(det_request.image)
                    
                    # 执行目标检测
                    result = self.service.det_instance.detection(image_data)
                    
                elif method == "ddddocr_slide_match":
                    from .models import SlideMatchRequest
                    slide_request = SlideMatchRequest(**params)
                    
                    if not self.service.slide_instance:
                        raise HTTPException(status_code=500, detail="滑块功能未初始化")
                    
                    # 解码base64图片
                    target_data = base64.b64decode(slide_request.target_image)
                    background_data = base64.b64decode(slide_request.background_image)