Retriever - dify

go get github.com/cloudwego/eino-ext/components/retriever/dify

package main

import (
	"context"
	"fmt"
	"log"
	"os"

	"github.com/cloudwego/eino-ext/components/retriever/dify"
)



func main() {
	APIKey := os.Getenv("DIFY_DATASET_API_KEY")
	Endpoint := os.Getenv("DIFY_ENDPOINT")
	DatasetID := os.Getenv("DIFY_DATASET_ID")
	ctx := context.Background()

	// create Dify Retriever
	ret, err := dify.NewRetriever(ctx, &dify.RetrieverConfig{
		APIKey:    APIKey,
		Endpoint:  Endpoint,
		DatasetID: DatasetID,
	})
	if err != nil {
		log.Fatalf("Failed to create retriever: %v", err)
	}

	// do search
	docs, err := ret.Retrieve(ctx, "test query")
	if err != nil {
		log.Fatalf("Failed to retrieve: %v", err)
	}

	// print docs
	for _, doc := range docs {
		fmt.Printf("doc id: %s\n", doc.ID)
		fmt.Printf("doc content: %s\n", doc.Content)
		fmt.Printf("score: %v\n\n", doc.Score())
	}
}

type RetrieverConfig struct {
    APIKey string   // Dify Datasets API key
    Endpoint string // Endpoint of the Dify API, default: https://api.dify.ai/v1
    DatasetID string    // DatasetID of the Dify datasets
    RetrievalModel *RetrievalModel  // Retrieval model configuration 
    Timeout time.Duration   // HTTP connection timeout
}

type RetrievalModel struct {
    SearchMethod          SearchMethod    // Search method
    RerankingEnable      *bool           // Enable reranking
    RerankingMode        *string         // Reranking mode
    RerankingModel       *RerankingModel // Reranking model settings
    Weights              *float64        // Search weights
    TopK                 *int            // Number of documents to retrieve
    ScoreThresholdEnabled *bool          // Enable score threshold
    ScoreThreshold       *float64        // Minimum score threshold
}

docID := dify.GetOrgDocID(doc)
docName := dify.GetOrgDocName(doc)
keywords := dify.GetKeywords(doc)