@@ -707,6 +707,7 @@ func TestPatternAnalysisWorker_GlobInheritsUnanimousMapping(t *testing.T) {
|
||||
for _, ep := range exacts {
|
||||
ep.ThirdPartyID = &thirdPartyID
|
||||
}
|
||||
|
||||
exacts[1].Description = "Google Analytics measurement cookie"
|
||||
|
||||
require.NoError(t, client.WithTx(ctx, func(ctx context.Context, tx pg.Tx) error {
|
||||
|
||||
@@ -91,6 +91,7 @@ func loadCommonTrackerPattern(
|
||||
t.Helper()
|
||||
|
||||
var reloaded coredata.CommonTrackerPattern
|
||||
|
||||
require.NoError(t, client.WithConn(ctx, func(ctx context.Context, conn pg.Querier) error {
|
||||
return reloaded.LoadByID(ctx, conn, id)
|
||||
}))
|
||||
|
||||
@@ -177,28 +177,34 @@ func sanitizeRequest(ctx context.Context, logger *log.Logger, req *ChatCompletio
|
||||
sanitized := *req
|
||||
|
||||
dropped := make([]string, 0, 5)
|
||||
|
||||
if dropTemperature {
|
||||
sanitized.Temperature = nil
|
||||
|
||||
dropped = append(dropped, "temperature")
|
||||
}
|
||||
|
||||
if dropTopP {
|
||||
sanitized.TopP = nil
|
||||
|
||||
dropped = append(dropped, "top_p")
|
||||
}
|
||||
|
||||
if dropFrequencyPenalty {
|
||||
sanitized.FrequencyPenalty = nil
|
||||
|
||||
dropped = append(dropped, "frequency_penalty")
|
||||
}
|
||||
|
||||
if dropPresencePenalty {
|
||||
sanitized.PresencePenalty = nil
|
||||
|
||||
dropped = append(dropped, "presence_penalty")
|
||||
}
|
||||
|
||||
if dropStop {
|
||||
sanitized.StopSequences = nil
|
||||
|
||||
dropped = append(dropped, "stop")
|
||||
}
|
||||
|
||||
|
||||
4
pkg/thirdparty/resolver_test.go
vendored
4
pkg/thirdparty/resolver_test.go
vendored
@@ -137,6 +137,7 @@ func TestResolveOrCreateCommonThirdParty(t *testing.T) {
|
||||
party := seedCatalogThirdParty(t, ctx, client, name, slug.Make(name))
|
||||
|
||||
var got *gid.GID
|
||||
|
||||
require.NoError(t, client.WithTx(ctx, func(ctx context.Context, tx pg.Tx) error {
|
||||
id, err := ResolveOrCreateCommonThirdParty(
|
||||
ctx,
|
||||
@@ -166,6 +167,7 @@ func TestResolveOrCreateCommonThirdParty(t *testing.T) {
|
||||
require.Equal(t, slug.Make(name), slug.Make(variant))
|
||||
|
||||
var got *gid.GID
|
||||
|
||||
require.NoError(t, client.WithTx(ctx, func(ctx context.Context, tx pg.Tx) error {
|
||||
id, err := ResolveOrCreateCommonThirdParty(
|
||||
ctx,
|
||||
@@ -189,6 +191,7 @@ func TestResolveOrCreateCommonThirdParty(t *testing.T) {
|
||||
expectedSlug := slug.Make(name)
|
||||
|
||||
var got *gid.GID
|
||||
|
||||
require.NoError(t, client.WithTx(ctx, func(ctx context.Context, tx pg.Tx) error {
|
||||
id, err := ResolveOrCreateCommonThirdParty(
|
||||
ctx,
|
||||
@@ -213,6 +216,7 @@ func TestResolveOrCreateCommonThirdParty(t *testing.T) {
|
||||
})
|
||||
|
||||
var created coredata.CommonThirdParty
|
||||
|
||||
require.NoError(t, client.WithConn(ctx, func(ctx context.Context, conn pg.Querier) error {
|
||||
return created.LoadByID(ctx, conn, *got)
|
||||
}))
|
||||
|
||||
Reference in New Issue
Block a user