Added vertex ai prompt attribute

Author: richardwooding

client/vertexai/instance.go

@@ -1,6 +1,6 @@
 package vertexai
 
 type predictRequest struct {
-	Instances  []instance `json:"instances"`
-	Parameters parameters `json:"parameters"`
+	Instances  []map[string]interface{} `json:"instances"`
+	Parameters parameters               `json:"parameters"`
 }

client/vertexai/predictrequest.go

@@ -7,6 +7,7 @@ import (
 	"github.com/spandigitial/codeassistant/client"
 	"github.com/spandigitial/codeassistant/client/debugger"
 	"github.com/spandigitial/codeassistant/model"
+	"github.com/spf13/viper"
 	"io"
 	"net/http"
 	"time"
@@ -74,14 +75,15 @@ func (c *Client) Completion(commandInstance *model.CommandInstance, messageParts
 		TopK:            topK,
 	}
 
+	prompt := commandInstance.JoinedPromptsContent("\n")
 	request := predictRequest{
-		Instances: []instance{{
-			Content: commandInstance.JoinedPromptsContent("\n"),
+		Instances: []map[string]interface{}{{
+			viper.GetString("vertexAiPromptAttribute"): prompt,
 		}},
 		Parameters: parameters,
 	}
 
-	c.debugger.Message(debugger.SentPrompt, request.Instances[0].Content)
+	c.debugger.Message(debugger.SentPrompt, prompt)
 
 	requestBytes, err := json.Marshal(request)
 	if err != nil {

client/vertexai/vertexaiclient.go

@@ -94,6 +94,10 @@ func init() {
 	if err := viper.BindPFlag("vertexAiModel", rootCmd.PersistentFlags().Lookup("vertexAiModel")); err != nil {
 		log.Fatal("Unable to find flag vertexAiModel", err)
 	}
+	rootCmd.PersistentFlags().String("vertexAiPromptAttribute", "content", "Model to use if not specified")
+	if err := viper.BindPFlag("vertexAiPromptAttribute", rootCmd.PersistentFlags().Lookup("vertexAiPromptAttribute")); err != nil {
+		log.Fatal("Unable to find flag vertexAiModel", err)
+	}
 	rootCmd.PersistentFlags().String("vertexAiLocation", "us-central1", "Locstion to use if not specified")
 	if err := viper.BindPFlag("vertexAiLocation", rootCmd.PersistentFlags().Lookup("vertexAiLocation")); err != nil {
 		log.Fatal("Unable to find flag vertexAiLocation", err)