andreas/llama-swap
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add proxy.Process to manage upstream proxy logic

Browse Source
This commit is contained in:
Benson Wong
2024-11-17 16:41:15 -08:00
parent a8e5ee13b9
commit 36a31f450f
4 changed files with 249 additions and 202 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
proxy/config_test.go
View File

@@ -91,18 +91,21 @@ func TestFindConfig(t *testing.T) {
}
// Test finding a model by its name
modelConfig, found := config.FindConfig("model1")
modelConfig, modelId, found := config.FindConfig("model1")
assert.True(t, found)
assert.Equal(t, "model1", modelId)
assert.Equal(t, config.Models["model1"], modelConfig)
// Test finding a model by its alias
modelConfig, found = config.FindConfig("m1")
modelConfig, modelId, found = config.FindConfig("m1")
assert.True(t, found)
assert.Equal(t, "model1", modelId)
assert.Equal(t, config.Models["model1"], modelConfig)
// Test finding a model that does not exist
modelConfig, found = config.FindConfig("model3")
modelConfig, modelId, found = config.FindConfig("model3")
assert.False(t, found)
assert.Equal(t, "", modelId)
assert.Equal(t, ModelConfig{}, modelConfig)
}