semantic

package main

import (
	"fmt"

	"github.com/jonwraymond/tooldiscovery/index"
	"github.com/jonwraymond/tooldiscovery/semantic"
)

func main() {
	// Convert from index.SearchDoc to semantic.Document
	searchDoc := index.SearchDoc{
		ID:      "github:create-issue",
		DocText: "create-issue github create issue tracker",
		Summary: index.Summary{
			ID:               "github:create-issue",
			Name:             "create-issue",
			Namespace:        "github",
			ShortDescription: "Create a new GitHub issue",
			Tags:             []string{"github", "issues"},
		},
	}

	doc := semantic.DocumentFromSearchDoc(searchDoc)
	fmt.Println("ID:", doc.ID)
	fmt.Println("Name:", doc.Name)
	fmt.Println("Namespace:", doc.Namespace)
}

Output:

ID: github:create-issue
Name: create-issue
Namespace: github

package main

import (
	"fmt"

	"github.com/jonwraymond/tooldiscovery/semantic"
)

func main() {
	docs := []semantic.Document{
		{ID: "git:status", Namespace: "git"},
		{ID: "git:commit", Namespace: "git"},
		{ID: "docker:ps", Namespace: "docker"},
	}

	filtered := semantic.FilterByNamespace(docs, "git")
	fmt.Println("Git documents:", len(filtered))
}

Output:

Git documents: 2

package main

import (
	"fmt"

	"github.com/jonwraymond/tooldiscovery/semantic"
)

func main() {
	docs := []semantic.Document{
		{ID: "tool1", Tags: []string{"vcs", "git"}},
		{ID: "tool2", Tags: []string{"containers", "docker"}},
		{ID: "tool3", Tags: []string{"vcs", "svn"}},
	}

	// Find all documents with "vcs" tag
	filtered := semantic.FilterByTags(docs, []string{"vcs"})
	fmt.Println("VCS documents:", len(filtered))
}

Output:

VCS documents: 2

package main

import (
	"context"
	"fmt"

	"github.com/jonwraymond/tooldiscovery/semantic"
)

func main() {
	// Create an index and add documents
	idx := semantic.NewInMemoryIndex()
	ctx := context.Background()

	docs := []semantic.Document{
		{ID: "git:status", Name: "status", Namespace: "git", Description: "Show working tree status", Tags: []string{"vcs"}},
		{ID: "git:commit", Name: "commit", Namespace: "git", Description: "Record changes to repository", Tags: []string{"vcs"}},
		{ID: "docker:ps", Name: "ps", Namespace: "docker", Description: "List containers", Tags: []string{"containers"}},
	}

	for _, doc := range docs {
		_ = idx.Add(ctx, doc)
	}

	// Create a searcher with BM25 strategy
	strategy := semantic.NewBM25Strategy(nil) // nil uses default scorer
	searcher := semantic.NewSearcher(idx, strategy)

	// Search for git-related tools
	results, _ := searcher.Search(ctx, "git status")
	fmt.Println("Found:", len(results), "results")
	if len(results) > 0 {
		fmt.Println("Top result:", results[0].Document.ID)
	}
}

Output:

Found: 3 results
Top result: git:status

package main

import (
	"context"
	"fmt"

	"github.com/jonwraymond/tooldiscovery/semantic"
)

func main() {
	// Create mock embedder for demonstration
	embedder := &mockEmbedder{}

	// Create individual strategies
	bm25 := semantic.NewBM25Strategy(nil)
	embedding := semantic.NewEmbeddingStrategy(embedder)

	// Combine with hybrid strategy (0.7 = 70% BM25, 30% embedding)
	hybrid, err := semantic.NewHybridStrategy(bm25, embedding, 0.7)
	if err != nil {
		fmt.Println("Error:", err)
		return
	}

	// Use the hybrid strategy
	doc := semantic.Document{
		ID:          "test",
		Name:        "search_files",
		Description: "Search for files in the filesystem",
	}

	ctx := context.Background()
	score, _ := hybrid.Score(ctx, "search files", doc.Normalized())
	fmt.Printf("Hybrid score: %.2f\n", score)
}

// mockEmbedder is a simple embedder for examples
type mockEmbedder struct{}

func (m *mockEmbedder) Embed(_ context.Context, _ string) ([]float32, error) {

	return []float32{1.0, 0.0, 0.0}, nil
}

Output:

Hybrid score: 1.70