mirror of
https://github.com/coder/coder.git
synced 2026-06-03 04:58:23 +00:00
Paweł Banaszewski
e00e85765b
This PR merges code from `coder/aibridge` repository into `coder/coder`. It was split into 4 PRs for easier review but stacked PRs will need to be merged into this PR so all checks pass. * https://github.com/coder/coder/pull/24190 -> raw code copy (this PR, before merging PRs on top of it, it was just 1 commit: https://github.com/coder/coder/commit/70d33f33200c7e77df910957595715f81f9bec24) * https://github.com/coder/coder/pull/24570 -> update imports in `coder/coder` to use copied code * https://github.com/coder/coder/pull/24586 -> linter fixes and CI integration (also added README.md) * https://github.com/coder/coder/pull/24571 -> added exclude to scripts/check_emdash.sh check Original PR message (before PR squash): Moves coder/aibridge code into coder/coder repository. Omitted files: - `go.mod`, `go.sum`, `.gitignore`, `.github/workflows/ci.yml,` `Makefile`, `LICENSE`, `README.md` (modified README.md is added later) - `.github`, `example`, `buildinfo,` `scripts` directories Simple verification script (will list omitted files) ``` tmp=$(mktemp -d) echo "$tmp" git clone --depth=1 https://github.com/coder/aibridge "$tmp/aibridge" git clone --depth=1 --branch pb/aibridge-code-move https://github.com/coder/coder "$tmp/coder" diff -rq --exclude=.git "$tmp/aibridge" "$tmp/coder/aibridge" # rm -rf "$tmp" ```
58 lines
1.4 KiB
Go
58 lines
1.4 KiB
Go
package intercept_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/coder/coder/v2/aibridge/context"
|
|
"github.com/coder/coder/v2/aibridge/intercept"
|
|
"github.com/coder/coder/v2/aibridge/recorder"
|
|
)
|
|
|
|
func TestNilActor(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
require.Nil(t, intercept.ActorHeadersAsOpenAIOpts(nil))
|
|
require.Nil(t, intercept.ActorHeadersAsAnthropicOpts(nil))
|
|
}
|
|
|
|
func TestBasic(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
actorID := uuid.NewString()
|
|
actor := &context.Actor{
|
|
ID: actorID,
|
|
}
|
|
|
|
// We can't peek inside since these opts require an internal type to apply onto.
|
|
// All we can do is check the length.
|
|
// See TestActorHeaders for an integration test.
|
|
oaiOpts := intercept.ActorHeadersAsOpenAIOpts(actor)
|
|
require.Len(t, oaiOpts, 1)
|
|
antOpts := intercept.ActorHeadersAsAnthropicOpts(actor)
|
|
require.Len(t, antOpts, 1)
|
|
}
|
|
|
|
func TestBasicAndMetadata(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
actorID := uuid.NewString()
|
|
actor := &context.Actor{
|
|
ID: actorID,
|
|
Metadata: recorder.Metadata{
|
|
"This": "That",
|
|
"And": "The other",
|
|
},
|
|
}
|
|
|
|
// We can't peek inside since these opts require an internal type to apply onto.
|
|
// All we can do is check the length.
|
|
// See TestActorHeaders for an integration test.
|
|
oaiOpts := intercept.ActorHeadersAsOpenAIOpts(actor)
|
|
require.Len(t, oaiOpts, 1+len(actor.Metadata))
|
|
antOpts := intercept.ActorHeadersAsAnthropicOpts(actor)
|
|
require.Len(t, antOpts, 1+len(actor.Metadata))
|
|
}
|