Channel automation additions

api/api.go

@@ -137,13 +137,14 @@ func (a *API) ServeHTTP(c *plugin.Context, w http.ResponseWriter, r *http.Reques
 	llmBridgeRoute.Use(a.interPluginAuthorizationRequired)
 
 	// Discovery endpoints
-	llmBridgeRoute.GET("/agents", a.handleGetAgents)
-	llmBridgeRoute.GET("/services", a.handleGetServices)
+	llmBridgeRoute.GET("/agents", a.validateUserIDQuery, a.handleGetAgents)
+	llmBridgeRoute.GET("/agents/:agent/tools", a.validateAgentParam, a.validateUserIDQuery, a.handleGetAgentTools)
+	llmBridgeRoute.GET("/services", a.validateUserIDQuery, a.handleGetServices)
 
 	// Completion endpoints
 	completionRoute := llmBridgeRoute.Group("/completion")
-	completionRoute.POST("/agent/:agent", a.handleAgentCompletionStreaming)
-	completionRoute.POST("/agent/:agent/nostream", a.handleAgentCompletionNoStream)
+	completionRoute.POST("/agent/:agent", a.validateAgentParam, a.handleAgentCompletionStreaming)
+	completionRoute.POST("/agent/:agent/nostream", a.validateAgentParam, a.handleAgentCompletionNoStream)
 	completionRoute.POST("/service/:service", a.handleServiceCompletionStreaming)
 	completionRoute.POST("/service/:service/nostream", a.handleServiceCompletionNoStream)
 

api/api_channel_license_test.go

@@ -66,11 +66,10 @@ func TestChannelAnalysisLicenseMiddleware(t *testing.T) {
 			})
 
 			e.api.licenseChecker = enterprise.NewLicenseChecker(e.client)
-			e.mockAPI.On("GetConfig").Return(&model.Config{}).Maybe()
 			if test.licensed {
-				e.mockAPI.On("GetLicense").Return(&model.License{SkuShortName: "advanced"}).Maybe()
+				e.OverrideLicense(&model.License{SkuShortName: "advanced"})
 			} else {
-				e.mockAPI.On("GetLicense").Return(&model.License{}).Maybe()
+				e.OverrideLicense(&model.License{})
 			}
 
 			e.mockAPI.On("GetChannel", "channelid").Return(&model.Channel{