35 lines
719 B
Go
35 lines
719 B
Go
package tools
|
|
|
|
import (
|
|
"ai_scheduler/internal/config"
|
|
"ai_scheduler/internal/entitys"
|
|
"testing"
|
|
)
|
|
|
|
func TestKnowledgeBaseTool_Execute(t *testing.T) {
|
|
|
|
kb := NewKnowledgeBaseTool(config.ToolConfig{})
|
|
channel := make(chan entitys.ResponseData)
|
|
err := kb.Execute(channel, nil, nil)
|
|
if err != nil {
|
|
t.Errorf("Execute() error = %v", err)
|
|
}
|
|
|
|
}
|
|
|
|
// session
|
|
func TestKnowledgeBaseTool_Submit(t *testing.T) {
|
|
|
|
apiKey := "sk-EfnUANKMj3DUOiEPJZ5xS8SGMsbO6be_qYAg9uZ8T3zyoFM-"
|
|
baseId := "kb-00000001"
|
|
host := "http://117.175.169.61:10000"
|
|
|
|
sessionId, err := GetKnowledgeBaseSession(host, baseId, apiKey)
|
|
if err != nil {
|
|
t.Errorf("GetKnowledgeBaseSession() error = %v", err)
|
|
}
|
|
|
|
t.Log("sessionId:", sessionId)
|
|
|
|
}
|