Enha: model name to config
This commit is contained in:
@@ -1,6 +1,7 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"grailbench/config"
|
||||
"log/slog"
|
||||
"os"
|
||||
"testing"
|
||||
@@ -11,7 +12,7 @@ func TestDeepSeekParser_ParseBytes(t *testing.T) {
|
||||
logger := slog.New(slog.NewTextHandler(os.Stderr, nil))
|
||||
|
||||
// Create a new deepSeekParser
|
||||
parser := NewDeepSeekParser(logger)
|
||||
parser := NewDeepSeekParser(logger, config.LoadConfigOrDefault("config.toml"))
|
||||
|
||||
// Test case 1: Regular text response
|
||||
t.Run("RegularTextResponse", func(t *testing.T) {
|
||||
@@ -119,7 +120,7 @@ func TestOpenRouterParser_ParseBytes(t *testing.T) {
|
||||
logger := slog.New(slog.NewTextHandler(os.Stderr, nil))
|
||||
|
||||
// Create a new openRouterParser
|
||||
parser := NewOpenRouterParser(logger)
|
||||
parser := NewOpenRouterParser(logger, config.LoadConfigOrDefault("config.toml"))
|
||||
|
||||
// Test case 1: Regular text response
|
||||
t.Run("RegularTextResponse", func(t *testing.T) {
|
||||
@@ -189,4 +190,5 @@ func TestOpenRouterParser_ParseBytes(t *testing.T) {
|
||||
t.Errorf("Expected %s, got %s", expected, result)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
|
Reference in New Issue
Block a user