diff --git a/app/api/controller/aiagent/generate_pipeline_step.go b/app/api/controller/aiagent/generate_pipeline_step.go index f5b44adbc..c7904ceb7 100644 --- a/app/api/controller/aiagent/generate_pipeline_step.go +++ b/app/api/controller/aiagent/generate_pipeline_step.go @@ -44,13 +44,7 @@ func (c *Controller) GeneratePipelineStep( RepoRef: in.RepoRef, } - // do permission check on repo here? - repo, err := c.repoStore.FindByRef(ctx, in.RepoRef) - if err != nil { - return nil, fmt.Errorf("failed to find repo by ref: %w", err) - } - - output, err := c.intelligenceService.GenerateStep(ctx, generateRequest, repo) + output, err := c.intelligenceService.GenerateStep(ctx, generateRequest) if err != nil { return nil, fmt.Errorf("generate pipeline: %w", err) } diff --git a/app/services/aiagent/aiagent.go b/app/services/aiagent/aiagent.go index 50be41100..0d870b387 100644 --- a/app/services/aiagent/aiagent.go +++ b/app/services/aiagent/aiagent.go @@ -92,23 +92,14 @@ func (s *HarnessIntelligence) Generate( func (s *HarnessIntelligence) GenerateStep( ctx context.Context, - req *types.PipelineStepGenerateRequest, - repo *types.Repository) (*types.PipelineStepGenerateResponse, error) { - if req.RepoRef == "" { - return nil, fmt.Errorf("no repo ref specified") - } - + req *types.PipelineStepGenerateRequest) (*types.PipelineStepGenerateResponse, error) { conversationID := uuid.New() chatRequest := &genai.ChatRequest{ Prompt: req.Prompt, ConversationID: conversationID.String(), ConversationRaw: "", - Context: genai.GenerateAIContext( - genai.RepoRef{ - Ref: repo.Path, - }, - ), - Capabilities: s.cr.Capabilities(), + Context: genai.GenerateAIContext(), + Capabilities: s.cr.Capabilities(), } resp, err := s.CapabilitiesLoop(ctx, chatRequest) diff --git a/app/services/capabilities/wire.go b/app/services/capabilities/wire.go index b634761bc..c589013ff 100644 --- a/app/services/capabilities/wire.go +++ b/app/services/capabilities/wire.go @@ -15,9 +15,6 @@ package capabilities import ( - "github.com/harness/gitness/app/store" - "github.com/harness/gitness/git" - "github.com/google/wire" ) @@ -31,10 +28,10 @@ func panicOnErr(err error) { } } -func ProvideCapabilities(repoStore store.RepoStore, git git.Interface) (*Registry, error) { +func ProvideCapabilities() (*Registry, error) { registry := NewRegistry() - panicOnErr(registry.RegisterListFilesCapability(ListFiles(repoStore, git))) - panicOnErr(registry.RegisterGetFileCapability(GetFile(repoStore, git))) + // panicOnErr(registry.RegisterListFilesCapability(ListFiles(repoStore, git))) + // panicOnErr(registry.RegisterGetFileCapability(GetFile(repoStore, git))) panicOnErr(registry.RegisterDisplayYamlCapability(DisplayYaml())) return registry, nil } diff --git a/cmd/gitness/wire_gen.go b/cmd/gitness/wire_gen.go index 8c89edfcd..6ef6c8a00 100644 --- a/cmd/gitness/wire_gen.go +++ b/cmd/gitness/wire_gen.go @@ -422,7 +422,7 @@ func initSystem(ctx context.Context, config *types.Config) (*server.System, erro migrateWebhook := migrate.ProvideWebhookImporter(webhookConfig, transactor, webhookStore) migrateLabel := migrate.ProvideLabelImporter(transactor, labelStore, labelValueStore, spaceStore) migrateController := migrate2.ProvideController(authorizer, publicaccessService, gitInterface, provider, pullReq, rule, migrateWebhook, migrateLabel, resourceLimiter, auditService, repoIdentifier, transactor, spaceStore, repoStore) - registry, err := capabilities.ProvideCapabilities(repoStore, gitInterface) + registry, err := capabilities.ProvideCapabilities() if err != nil { return nil, err }