zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-08-06 23:51:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

Go: introduce content_length and max_output (#17807)

Browse Source 
Signed-off-by: Jin Hai <haijin.chn@gmail.com>
This commit is contained in:
Jin Hai
2026-08-04 19:08:31 +08:00
committed by GitHub
parent f707cca074
commit c1f960cd47
19 changed files with 254 additions and 148 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
internal/entity/models/nvidia_test.go
View File

@@ -62,7 +62,7 @@ func TestParseNvidiaModelListPrefersPresetMetadata(t *testing.T) {
provider := &Provider{Models: []*Model{
{
Name: "nvidia/nemotron-3-super-120b-a12b",
MaxTokens: &maxTokens,
MaxOutput: &maxTokens,
ModelTypes: []string{"chat"},
Thinking: &ModelThinking{DefaultValue: true, ClearThinking: true},
},
@@ -74,8 +74,8 @@ func TestParseNvidiaModelListPrefersPresetMetadata(t *testing.T) {
if len(models) != 1 {
t.Fatalf("len(models) = %d, want 1", len(models))
}
if models[0].MaxTokens == nil || *models[0].MaxTokens != maxTokens {
t.Fatalf("MaxTokens = %v, want %d", models[0].MaxTokens, maxTokens)
if models[0].MaxOutput == nil || *models[0].MaxOutput != maxTokens {
t.Fatalf("MaxOutput = %v, want %d", models[0].MaxOutput, maxTokens)
}
if models[0].Thinking == nil || !models[0].Thinking.DefaultValue {
t.Fatalf("Thinking = %#v, want preset metadata", models[0].Thinking)