Enha: model name to config

This commit is contained in:
Grail Finder
2025-09-06 14:41:21 +03:00
parent 0276000bfa
commit c75ac433d4
5 changed files with 44 additions and 33 deletions

View File

@@ -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)
}
})
}
}