Movatterモバイル変換

[0]ホーム
URL:

Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

 Sign up
Appearance settings

Commit7f3955a

Browse files
deansheatherdannykopping
authored andcommitted
add metadata column to interceptions
1 parent0971409 commit7f3955a

File tree

9 files changed

+36
-18
lines changed

9 files changed

+36
-18
lines changed

‎coderd/aibridgedserver/aibridgedserver.go‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -127,6 +127,7 @@ func (s *Server) RecordInterception(ctx context.Context, in *proto.RecordInterce
127127
InitiatorID:initID,
128128
Provider:in.Provider,
129129
Model:in.Model,
130+
Metadata:marshalMetadata(ctx,s.logger,in.GetMetadata()),
130131
StartedAt:in.StartedAt.AsTime(),
131132
})
132133
iferr!=nil {

‎coderd/aibridgedserver/aibridgedserver_test.go‎

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -158,7 +158,7 @@ func TestAuthorization(t *testing.T) {
158158
CreatedAt:now,
159159
UpdatedAt:now,
160160
LoginType:database.LoginTypePassword,
161-
Scope:database.APIKeyScopeAll,
161+
Scopes:[]database.APIKeyScope{database.APIKeyScopeAll},
162162
TokenName:"",
163163
}
164164
iftc.key=="" {
@@ -340,6 +340,13 @@ func TestGetMCPServerAccessTokensBatch(t *testing.T) {
340340
funcTestRecordInterception(t*testing.T) {
341341
t.Parallel()
342342

343+
var (
344+
metadataProto=map[string]*anypb.Any{
345+
"key":mustMarshalAny(t,&structpb.Value{Kind:&structpb.Value_StringValue{StringValue:"value"}}),
346+
}
347+
metadataJSON=`{"key":"value"}`
348+
)
349+
343350
testRecordMethod(t,
344351
func(srv*aibridgedserver.Server,ctx context.Context,req*proto.RecordInterceptionRequest) (*proto.RecordInterceptionResponse,error) {
345352
returnsrv.RecordInterception(ctx,req)
@@ -352,6 +359,7 @@ func TestRecordInterception(t *testing.T) {
352359
InitiatorId:uuid.NewString(),
353360
Provider:"anthropic",
354361
Model:"claude-4-opus",
362+
Metadata:metadataProto,
355363
StartedAt:timestamppb.Now(),
356364
},
357365
setupMocks:func(t*testing.T,db*dbmock.MockStore,req*proto.RecordInterceptionRequest) {
@@ -366,6 +374,7 @@ func TestRecordInterception(t *testing.T) {
366374
!assert.Equal(t,initiatorID,p.InitiatorID,"initiator ID")||
367375
!assert.Equal(t,req.GetProvider(),p.Provider,"provider")||
368376
!assert.Equal(t,req.GetModel(),p.Model,"model")||
377+
!assert.JSONEq(t,metadataJSON,string(p.Metadata),"metadata")||
369378
!assert.WithinDuration(t,req.GetStartedAt().AsTime(),p.StartedAt,time.Second,"started at") {
370379
returnfalse
371380
}

‎coderd/database/dump.sql‎

Lines changed: 2 additions & 1 deletion
Some generated files are not rendered by default. Learn more aboutcustomizing how changed files appear on GitHub.
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
ALTERTABLE aibridge_interceptions DROP COLUMN metadata;
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
ALTERTABLE aibridge_interceptions ADD COLUMN metadata JSONB DEFAULTNULL;

‎coderd/database/models.go‎

Lines changed: 5 additions & 4 deletions
Some generated files are not rendered by default. Learn more aboutcustomizing how changed files appear on GitHub.

‎coderd/database/queries.sql.go‎

Lines changed: 13 additions & 9 deletions
Some generated files are not rendered by default. Learn more aboutcustomizing how changed files appear on GitHub.

‎coderd/database/queries/aibridge.sql‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
-- name: InsertAIBridgeInterception :one
2-
INSERT INTO aibridge_interceptions (id, initiator_id, provider, model, started_at)
3-
VALUES (@id::uuid, @initiator_id::uuid, @provider, @model, @started_at)
2+
INSERT INTO aibridge_interceptions (id, initiator_id, provider, model,metadata,started_at)
3+
VALUES (@id::uuid, @initiator_id::uuid, @provider, @model,COALESCE(@metadata::jsonb,'{}'::jsonb),@started_at)
44
RETURNING*;
55

66
-- name: InsertAIBridgeTokenUsage :exec

‎coderd/mcp/mcp_e2e_test.go‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1232,7 +1232,7 @@ func TestMCPHTTP_E2E_ChatGPTEndpoint(t *testing.T) {
12321232
template:=coderdtest.CreateTemplate(t,coderClient,user.OrganizationID,version.ID)
12331233

12341234
// Create MCP client pointing to the ChatGPT endpoint
1235-
mcpURL:=api.AccessURL.String()+"/api/experimental/mcp/http?toolset=chatgpt"
1235+
mcpURL:=api.AccessURL.String()+mcpserver.MCPEndpoint+"?toolset=chatgpt"
12361236

12371237
// Configure client with authentication headers using RFC 6750 Bearer token
12381238
mcpClient,err:=mcpclient.NewStreamableHttpClient(mcpURL,

0 commit comments

Comments
 (0)
[8]ページ先頭
©2009-2025 Movatter.jp