Skip to content

Commit

Permalink
#149: Removed the broken test
Browse files Browse the repository at this point in the history
  • Loading branch information
roma-glushko committed Mar 2, 2024
1 parent 20c48b8 commit 1cb2ccf
Showing 1 changed file with 0 additions and 46 deletions.
46 changes: 0 additions & 46 deletions pkg/routers/router_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -288,49 +288,3 @@ func TestLangRouter_ChatStream(t *testing.T) {
require.NoError(t, err)
}
}

func TestLangRouter_ChatStream_NoStreamSupport(t *testing.T) {
budget := health.NewErrorBudget(3, health.SEC)
latConfig := latency.DefaultConfig()

langModels := []providers.LanguageModel{
providers.NewLangModel(
"first",
providers.NewProviderMock([]providers.ResponseMock{{Msg: "1"}}, false),
*budget,
*latConfig,
1,
),
providers.NewLangModel(
"second",
providers.NewProviderMock([]providers.ResponseMock{{Msg: "1"}, {Msg: "2"}}, true),
*budget,
*latConfig,
1,
),
}

models := make([]providers.Model, 0, len(langModels))
for _, model := range langModels {
models = append(models, model)
}

router := LangRouter{
routerID: "test_stream_router",
Config: &LangRouterConfig{},
retry: retry.NewExpRetry(3, 2, 1*time.Second, nil),
routing: routing.NewPriority(models),
models: langModels,
telemetry: telemetry.NewTelemetryMock(),
}

ctx := context.Background()
req := schemas.NewChatFromStr("tell me a dad joke")
respC := make(chan schemas.ChatResponse)

for i := 0; i < 2; i++ {
err := router.ChatStream(ctx, req, respC)

require.NoError(t, err)
}
}

0 comments on commit 1cb2ccf

Please sign in to comment.