mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-07-21 23:21:04 +08:00
e87008e26c
## Summary - Enable synchronous and streaming tool calls for Aliyun models. - Preserve provider-specific chat endpoints and prevent repeated qwen-flash tool calls. - Restrict retrieval tool inputs to model-provided query parameters. ## Testing - `bash build.sh --test ./internal/entity/models ./internal/agent/component ./internal/agent/tool` - Manual frontend UI testing passed.
48 lines
1.3 KiB
Go
48 lines
1.3 KiB
Go
package component
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/cloudwego/eino/schema"
|
|
|
|
"ragflow/internal/entity/models"
|
|
)
|
|
|
|
func TestNewChatModelDriverPreservesProviderChatSuffix(t *testing.T) {
|
|
if err := models.InitProviderManager("../../../conf/models"); err != nil {
|
|
t.Fatalf("InitProviderManager: %v", err)
|
|
}
|
|
|
|
requestPath := make(chan string, 1)
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
requestPath <- r.URL.Path
|
|
w.Header().Set("Content-Type", "application/json")
|
|
_, _ = w.Write([]byte(`{"choices":[{"message":{"content":"ok"}}]}`))
|
|
}))
|
|
defer server.Close()
|
|
|
|
driver, err := newChatModelDriver("Tongyi-Qianwen", server.URL)
|
|
if err != nil {
|
|
t.Fatalf("newChatModelDriver: %v", err)
|
|
}
|
|
apiKey := "test-key"
|
|
modelName := "qwen-flash"
|
|
chatModel := models.NewEinoChatModel(
|
|
models.NewChatModel(driver, &modelName, &models.APIConfig{ApiKey: &apiKey}),
|
|
nil,
|
|
)
|
|
response, err := chatModel.Generate(context.Background(), []*schema.Message{schema.UserMessage("hi")})
|
|
if err != nil {
|
|
t.Fatalf("Generate: %v", err)
|
|
}
|
|
if response.Content != "ok" {
|
|
t.Errorf("content = %q, want ok", response.Content)
|
|
}
|
|
if got := <-requestPath; got != "/compatible-mode/v1/chat/completions" {
|
|
t.Errorf("request path = %q, want /compatible-mode/v1/chat/completions", got)
|
|
}
|
|
}
|