mirror of
https://github.com/Ed1s0nZ/CyberStrikeAI.git
synced 2026-05-07 10:06:41 +02:00
公明
24 lines
532 B
Go
24 lines
532 B
Go
package knowledge
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
func TestBuildKnowledgeRetrieveChain_Compile(t *testing.T) {
|
|
r := NewRetriever(nil, nil, &RetrievalConfig{TopK: 3, SimilarityThreshold: 0.5}, zap.NewNop())
|
|
_, err := BuildKnowledgeRetrieveChain(context.Background(), r)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestBuildKnowledgeRetrieveChain_NilRetriever(t *testing.T) {
|
|
_, err := BuildKnowledgeRetrieveChain(context.Background(), nil)
|
|
if err == nil {
|
|
t.Fatal("expected error for nil retriever")
|
|
}
|
|
}
|