shishantbiswas/coder
1
0
Fork 0
mirror of https://github.com/coder/coder.git synced 2026-06-02 20:48:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: route TransportFor by provider name (#25650)

Browse Source 
Delegate `aibridge` routing responsibility to the in-memory transport
layer.

Co-authored-by: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
This commit is contained in:
Danny Kopping
2026-05-25 18:04:12 +02:00
committed by GitHub
parent 0a45f96d30
commit 8652ef3e3b
5 changed files with 101 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files
coderd/aibridge_test.go
+7 -8
View File
@@ -6,7 +6,6 @@ import (
"net/http/httptest"
"testing"
"github.com/google/uuid"
"github.com/stretchr/testify/require"
"github.com/coder/coder/v2/coderd/aibridge"
@@ -23,12 +22,12 @@ type stubTransportFactory struct {
}
type callRecord struct {
providerID uuid.UUID
source aibridge.Source
providerName string
source aibridge.Source
}
func (f *stubTransportFactory) TransportFor(providerID uuid.UUID, source aibridge.Source) (http.RoundTripper, error) {
f.calls <- callRecord{providerID: providerID, source: source}
func (f *stubTransportFactory) TransportFor(providerName string, source aibridge.Source) (http.RoundTripper, error) {
f.calls <- callRecord{providerName: providerName, source: source}
return &handlerRoundTripper{handler: f.handler}, nil
}
@@ -71,14 +70,14 @@ func TestAIBridgeTransportFactory_Registration(t *testing.T) {
loaded := api.AIBridgeTransportFactory.Load()
require.NotNil(t, loaded)
providerID := uuid.New()
rt, err := (*loaded).TransportFor(providerID, aibridge.SourceAgents)
providerName := "openai"
rt, err := (*loaded).TransportFor(providerName, aibridge.SourceAgents)
require.NoError(t, err)
require.NotNil(t, rt)
select {
case got := <-stub.calls:
require.Equal(t, providerID, got.providerID)
require.Equal(t, providerName, got.providerName)
require.Equal(t, aibridge.SourceAgents, got.source)
default:
t.Fatal("factory was not invoked")