Ajout de SEARXNG pour les recherches web

2026-10-05 19:05:24 +02:00
1 parent 87eeb41364
commit 345129c7bf
@@ -0,0 +1,151 @@
+package searxng
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"log/slog"
+	"net/http"
+	"net/url"
+	"strings"
+
+	"trankilou.fr/lassistanoque/backend/internal/domain"
+)
+
+type SearxngConfig struct {
+	URL string `json:"url"`
+}
+
+type SearxngTool struct {
+}
+
+func (t *SearxngTool) Name() string {
+	return "searxng"
+}
+
+func (t *SearxngTool) Description() string {
+	return "Search the web using a SearXNG instance. Returns a list of relevant results with title, URL, and snippet."
+}
+
+func (t *SearxngTool) Params() []domain.ToolParam {
+	return []domain.ToolParam{
+		{JsonName: "url", Type: "string", Mandatory: true, Description: "Searxng URL"},
+	}
+}
+
+func (t *SearxngTool) Definition(ctx context.Context) *domain.ToolDefinition {
+	return &domain.ToolDefinition{
+		Type: "function",
+		Function: &domain.ToolFunction{
+			Name:        t.Name(),
+			Description: t.Description(),
+			Parameters: map[string]any{
+				"type": "object",
+				"properties": map[string]any{
+					"query": map[string]any{
+						"type":        "string",
+						"description": "The search query",
+					},
+					"categories": map[string]any{
+						"type":        "string",
+						"description": "Comma-separated list of categories: general, news, images, science, it, social_media. Defaults to 'general'.",
+					},
+					"language": map[string]any{
+						"type":        "string",
+						"description": "Language code for results, e.g. 'fr', 'en'. Defaults to 'all'.",
+					},
+				},
+				"required": []string{"query"},
+			},
+		},
+	}
+}
+
+type searxngResponse struct {
+	Results []searxngResult `json:"results"`
+}
+
+type searxngResult struct {
+	Title   string `json:"title"`
+	URL     string `json:"url"`
+	Content string `json:"content"`
+	Engine  string `json:"engine"`
+}
+
+func (t *SearxngTool) Execute(ctx context.Context, input []byte, configuration string) ([]byte, error) {
+
+	slog.Debug("[searxng]", "input", string(input))
+
+	var params struct {
+		Query      string `json:"query"`
+		Categories string `json:"categories"`
+		Language   string `json:"language"`
+	}
+	if err := json.Unmarshal(input, &params); err != nil {
+		return nil, fmt.Errorf("invalid parameters: %w", err)
+	}
+
+	var toolCfg SearxngConfig
+	err := json.Unmarshal([]byte(configuration), &toolCfg)
+	if err != nil {
+		return nil, fmt.Errorf("searxng tool not configured: %w", err)
+	}
+	if toolCfg.URL == "" {
+		return nil, fmt.Errorf("searxng tool config missing 'url' field")
+	}
+	baseURL := toolCfg.URL
+
+	categories := "general"
+	if params.Categories != "" {
+		categories = params.Categories
+	}
+	language := "all"
+	if params.Language != "" {
+		language = params.Language
+	}
+
+	reqURL := fmt.Sprintf("%s/search?q=%s&format=json&categories=%s&language=%s",
+		strings.TrimRight(baseURL, "/"),
+		url.QueryEscape(params.Query),
+		url.QueryEscape(categories),
+		url.QueryEscape(language),
+	)
+
+	req, err := http.NewRequestWithContext(ctx, http.MethodGet, reqURL, nil)
+	if err != nil {
+		return nil, fmt.Errorf("failed to build request: %w", err)
+	}
+	req.Header.Set("Accept", "application/json")
+
+	resp, err := http.DefaultClient.Do(req)
+	if err != nil {
+		return nil, fmt.Errorf("searxng request failed: %w", err)
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != http.StatusOK {
+		return nil, fmt.Errorf("searxng returned status %d", resp.StatusCode)
+	}
+
+	var searchResp searxngResponse
+	if err := json.NewDecoder(resp.Body).Decode(&searchResp); err != nil {
+		return nil, fmt.Errorf("failed to decode response: %w", err)
+	}
+
+	if len(searchResp.Results) == 0 {
+		return []byte("No results found."), nil
+	}
+
+	var sb strings.Builder
+	fmt.Fprintf(&sb, "Search results for \"%s\":\n\n", params.Query)
+	for i, r := range searchResp.Results {
+		fmt.Fprintf(&sb, "%d. %s\n", i+1, r.Title)
+		fmt.Fprintf(&sb, "   URL: %s\n", r.URL)
+		if r.Content != "" {
+			fmt.Fprintf(&sb, "   %s\n", r.Content)
+		}
+		fmt.Fprintf(&sb, "\n")
+	}
+
+	return []byte(sb.String()), nil
+}