Function TestModelIntent_Info_WrapsSchema

coderd/x/chatd/mcpclient/mcpclient_test.go:1291–1325

(t *testing.T)

}

func TestModelIntent_Info_WrapsSchema(t *testing.T) {
	t.Parallel()
	ctx := context.Background()
	logger := slogtest.Make(t, &slogtest.Options{IgnoreErrors: true})

	ts := newTestMCPServer(t, echoTool())

	cfg := makeConfig("intent-srv", ts.URL)
	cfg.ModelIntent = true

	tools, cleanup := mcpclient.ConnectAll(ctx, logger, []database.MCPServerConfig{cfg}, nil, uuid.Nil, nil, nil)
	t.Cleanup(cleanup)
	require.Len(t, tools, 1)

	info := tools[0].Info()

	// Top-level schema should have model_intent and properties.
	_, hasModelIntent := info.Parameters["model_intent"]
	_, hasProperties := info.Parameters["properties"]
	assert.True(t, hasModelIntent, "schema should contain model_intent")
	assert.True(t, hasProperties, "schema should contain properties")

	// Required should include both.
	assert.Contains(t, info.Required, "model_intent")
	assert.Contains(t, info.Required, "properties")

	// The original "input" parameter should be nested under
	// properties.properties.
	propsObj, ok := info.Parameters["properties"].(map[string]any)
	require.True(t, ok)
	innerProps, ok := propsObj["properties"].(map[string]any)
	require.True(t, ok)
	_, hasInput := innerProps["input"]
	assert.True(t, hasInput, "original 'input' param should be nested")
}

func TestModelIntent_Info_NoWrapWhenDisabled(t *testing.T) {
	t.Parallel()