Need to add selected index to GenerateCells request.

jlewi · Oct 2, 2024 · 104a0ab · 104a0ab
1 parent 8592f71
commit 104a0ab
Show file tree

Hide file tree

Showing 6 changed files with 101 additions and 84 deletions.
diff --git a/app/pkg/agent/agent.go b/app/pkg/agent/agent.go
@@ -450,7 +450,8 @@ func (a *Agent) GenerateCells(ctx context.Context, req *connect.Request[v1alpha1
 		return nil, err
 	}
 	agentReq := &v1alpha1.GenerateRequest{
-		Doc: doc,
+		Doc:           doc,
+		SelectedIndex: req.Msg.GetSelectedIndex(),
 	}
 
 	// Call the agent

diff --git a/protos/buf.lock b/protos/buf.lock
@@ -8,5 +8,5 @@ deps:
     commit: e7f8d366f5264595bcc4cd4139af9973
     digest: b5:0cd69a689ee320ed815663d57d1bc3a1d6823224a7a717d46fee3a68197c25a6f5f932c0b0e49f8370c70c247a6635969a6a54af5345cafd51e0667298768aca
   - name: buf.build/stateful/runme
-    commit: 20a8540bddaf485682d5bef7e619d15f
-    digest: b5:627fd8553c9470db441f4e338a5a628e43f3a684c62d41678f13765840e4189540e9df547eaae3c006722513811725fc70b5a6206628c71db80da3a8c2994507
+    commit: 45813f39881a42538128c8d9757878a4
+    digest: b5:669c5838be7211c2fb8a663716813eb19682db4fe1dcec3e076c3aa8579d511aceb9f0ba5016294f205d7ec51bbe8e5bbc45c52bea8a1e01b6fdbc3871c43b5a
diff --git a/protos/foyle/v1alpha1/agent.proto b/protos/foyle/v1alpha1/agent.proto
@@ -114,6 +114,8 @@ message StreamGenerateResponse {
 
 message GenerateCellsRequest {
   runme.parser.v1.Notebook notebook = 1;
+  // The index of the selected cell.
+  int32 selected_index = 2;
 }
 
 message GenerateCellsResponse {

diff --git a/protos/go/foyle/logs/conversion.zap.go b/protos/go/foyle/logs/conversion.zap.go
diff --git a/protos/go/foyle/v1alpha1/agent.pb.go b/protos/go/foyle/v1alpha1/agent.pb.go
diff --git a/protos/go/foyle/v1alpha1/agent.zap.go b/protos/go/foyle/v1alpha1/agent.zap.go