mirror of
https://github.com/hanzoai/team.git
synced 2026-09-30 01:57:35 +00:00
zeekay
8feb24ea1f
The server module named github.com/hanzoai/team/server/v8 — a path that resolves to the retired workspace's repository, not this one — while server/public still published github.com/mattermost/mattermost/server/public. Both are github.com/hanzoteam/server/... now, which is where this code actually lives. That second half is what made the binary unbootable. Upstream's public path stayed claimed by upstream, mattermost-plugin-ai imported it, and the build linked public/model twice; public/plugin's gob.Register then ran twice under one name and panicked at init, before main. Reverting the rename traded the brand for a working binary. Rebranding the plugin alongside it needs no such trade: the agents fork is github.com/hanzoteam/agents/v2 against our public module, so no module named mattermost/mattermost is in the graph at all. Measured: `go list -deps ./cmd/mattermost` names exactly one public/model, and it is ours.
52 lines
1.7 KiB
Go
52 lines
1.7 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package app
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/hanzoteam/server/server/public/model"
|
|
)
|
|
|
|
func TestBuildRewriteSystemPrompt(t *testing.T) {
|
|
basePrompt := model.RewriteSystemPrompt
|
|
|
|
t.Run("uses_user_locale", func(t *testing.T) {
|
|
prompt := buildRewriteSystemPrompt("en_CA")
|
|
require.True(t, strings.HasPrefix(prompt, basePrompt))
|
|
require.Contains(t, prompt, "User locale: en_CA.")
|
|
})
|
|
|
|
t.Run("returns_base_prompt_when_no_locale", func(t *testing.T) {
|
|
prompt := buildRewriteSystemPrompt("")
|
|
require.Equal(t, basePrompt, prompt)
|
|
})
|
|
}
|
|
|
|
func TestRewriteMessage(t *testing.T) {
|
|
t.Run("sets_structured_output_schema_on_bridge_request", func(t *testing.T) {
|
|
bridge := &testAgentsBridge{
|
|
completeFn: func(sessionUserID, agentID string, req BridgeCompletionRequest) (string, error) {
|
|
return `{"rewritten_text":"Rewritten message"}`, nil
|
|
},
|
|
}
|
|
|
|
th := Setup(t, WithAgentsBridge(bridge)).InitBasic(t)
|
|
ctx := th.Context.WithSession(&model.Session{UserId: th.BasicUser.Id})
|
|
|
|
response, appErr := th.App.RewriteMessage(ctx, model.NewId(), "original message", model.RewriteActionImproveWriting, "", "")
|
|
require.Nil(t, appErr)
|
|
require.NotNil(t, response)
|
|
assert.Equal(t, "Rewritten message", response.RewrittenText)
|
|
require.Len(t, bridge.completeCalls, 1)
|
|
assert.Equal(t, BridgeOperationRewrite, bridge.completeCalls[0].request.Operation)
|
|
assert.Equal(t, string(model.RewriteActionImproveWriting), bridge.completeCalls[0].request.OperationSubType)
|
|
assert.Equal(t, rewriteResponseJSONSchema, bridge.completeCalls[0].request.JSONOutputFormat)
|
|
})
|
|
}
|