Skip to content

Commit 469570e

Browse files
committed
Tag client name and version
1 parent 9d7231b commit 469570e

File tree

2 files changed

+8
-0
lines changed

2 files changed

+8
-0
lines changed

contrib/mark3labs/mcp-go/mcpgo.go

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -80,6 +80,11 @@ func newHooks() *hooks {
8080

8181
func (h *hooks) onBeforeInitialize(ctx context.Context, id any, request *mcp.InitializeRequest) {
8282
taskSpan, _ := llmobs.StartTaskSpan(ctx, "mcp.initialize", llmobs.WithIntegration("mark3labs/mcp-go"))
83+
84+
clientName := request.Params.ClientInfo.Name
85+
clientVersion := request.Params.ClientInfo.Version
86+
taskSpan.Annotate(llmobs.WithAnnotatedTags(map[string]string{"client_name": clientName, "client_version": clientName + "_" + clientVersion}))
87+
8388
h.spanCache.Store(id, taskSpan)
8489
}
8590

contrib/mark3labs/mcp-go/mcpgo_test.go

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -76,6 +76,9 @@ func TestIntegrationSessionInitialize(t *testing.T) {
7676
assert.Equal(t, "mcp.initialize", taskSpan.Name)
7777
assert.Equal(t, "task", taskSpan.Meta["span.kind"])
7878

79+
assert.Contains(t, taskSpan.Tags, "client_name:test-client")
80+
assert.Contains(t, taskSpan.Tags, "client_version:test-client_1.0.0")
81+
7982
assert.Contains(t, taskSpan.Meta, "input")
8083
assert.Contains(t, taskSpan.Meta, "output")
8184

0 commit comments

Comments
 (0)