gemini-cli is stupid

1 files changed, 15 insertions, 6 deletions

diff --git a/convertToPB.go b/convertToPB.go
index e9399b6..13faad5 100644
--- a/convertToPB.go
+++ b/convertToPB.go
@@ -21,15 +21,24 @@ func convertToPB(resp *genai.GenerateContentResponse) *chatpb.ChatEntry {
 				PartType: &chatpb.Part_Text{Text: part.Text},
 			})
 		}
-		if part.FunctionCall != nil {
-			fmt.Printf("Gemini API requested to execute command: %s\n", part.FunctionCall.Name)
-			// This is a simplified conversion of args.
-			// A more robust implementation would handle different value types.
+		if fc := part.FunctionCall; fc != nil {
+			fmt.Printf("Gemini API requested to execute command: %s\n", fc.Name)
 			entry.Parts = append(entry.Parts, &chatpb.Part{
 				PartType: &chatpb.Part_FunctionCall{
 					FunctionCall: &chatpb.FunctionCall{
-						Name: part.FunctionCall.Name,
-						Args: &chatpb.ArgsInfo{}, // TODO: Properly map args if needed
+						Name: fc.Name,
+						Args: &chatpb.ArgsInfo{}, // TODO: Properly map args from fc.Args map[string]any
+					},
+				},
+			})
+		}
+		if fr := part.FunctionResponse; fr != nil {
+			// Convert the FunctionResponse to the protobuf equivalent
+			entry.Parts = append(entry.Parts, &chatpb.Part{
+				PartType: &chatpb.Part_FunctionResponse{
+					FunctionResponse: &chatpb.FunctionResponse{
+						Name: fr.Name,
+						// TODO: Properly map the response content
 					},
 				},
 			})