Files
ragflow/internal/handler/plugin.go

56 lines
1.7 KiB
Go
Raw Normal View History

//
// Copyright 2026 The InfiniFlow Authors. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
package handler
import (
"github.com/gin-gonic/gin"
"ragflow/internal/common"
"ragflow/internal/service"
)
// PluginHandler serves the /plugin/* HTTP routes.
type PluginHandler struct {
pluginService *service.PluginService
}
// NewPluginHandler creates a plugin handler.
func NewPluginHandler(pluginService *service.PluginService) *PluginHandler {
return &PluginHandler{
pluginService: pluginService,
}
}
// ListLLMTools handles GET /v1/plugin/tools.
//
// @Summary List LLM tool plugins
// @Description Return the metadata of every embedded LLM tool plugin. Matches
// @Description the response of the Python GET /v1/plugin/tools endpoint.
// @Tags plugin
// @Produce json
// @Security ApiKeyAuth
// @Success 200 {object} map[string]interface{}
// @Router /v1/plugin/tools [get]
func (h *PluginHandler) ListLLMTools(c *gin.Context) {
if _, errorCode, errorMessage := GetUser(c); errorCode != common.CodeSuccess {
common.ErrorWithCode(c, errorCode, errorMessage)
return
}
Fix internal/handler test failures, align response with Python contract, and re-enable handler/storage/agent tests in CI (#17554) ## Summary Fixes the 6 pre-existing failures in `internal/handler`, aligns one response field with the Python API contract, and re-enables packages in CI that were previously excluded because of (or unrelated to) those failures. ### Test/handler fixes - **plugin.go**: return `"success"` (lowercase) instead of `"SUCCESS"` so the response envelope matches the Python backend, keeping backend-swap transparent. - **search_handler_test.go**: expect lowercase `"no authorization"` to match the service error message, which mirrors Python. - **agent_test.go**: seed versions with `CreateTime` instead of `UpdateTime` so `ListVersions` ordering (`create_time DESC`) is exercised correctly. - **agent_wait_for_user_test.go**: a clean run may emit only the `[DONE]` frame; relax the SSE assertion to require a non-empty stream ending in `[DONE]`. - **bot_test.go**: align attachment-download assertions with actual behavior (`Content-Disposition: attachment; filename="file"`, default `application/octet-stream`), matching Python `resolve_attachment_content_type`. ### CI - **tests.yml / sep-tests.yml**: - Remove the `grep -v '/internal/handler$'` filter — the 6 failures that motivated it are now fixed. - Remove the `grep -v '/internal/storage$'` filter — `internal/storage` tests self-skip via `t.Skipf` when MinIO is unavailable, so the package is safe to run in CI. - Remove the `grep -v '/internal/agent$'` filter (only present in the tests.yml infinity job) — the exclusion was undocumented and inconsistent with the other jobs that already run `internal/agent`. - Keep the `internal/tokenizer` exclusion: it is a genuine environmental dependency (dict files at `/usr/share/infinity/resource`, absent in the Go test environment). ## Test plan - `build.sh --test ./internal/handler/` is green (previously 6 FAIL). - `build.sh --test ./internal/storage/ ./internal/agent/` should pass; the MinIO-backed `internal/storage` tests skip gracefully without a MinIO server. - After this PR, `internal/handler`, `internal/storage`, and `internal/agent` run again in CI unit-test jobs; `internal/tokenizer` stays excluded. 🤖 Generated with [Claude Code](https://claude.com/claude-code)
2026-07-30 16:59:39 +08:00
common.SuccessWithData(c, h.pluginService.ListLLMTools(), "success")
}