refactor: add Stringer methods to custom types

Signed-off-by: Milos Gajdos <[email protected]>

cmd/cohere/main.go

@@ -18,9 +18,9 @@ var (
 
 func init() {
 	flag.StringVar(&input, "input", "what is life", "input data")
-	flag.StringVar(&model, "model", string(cohere.EnglishV3), "model name")
-	flag.StringVar(&truncate, "truncate", string(cohere.NoneTrunc), "truncate type")
-	flag.StringVar(&inputType, "input-type", string(cohere.ClusteringInput), "input type")
+	flag.StringVar(&model, "model", cohere.EnglishV3.String(), "model name")
+	flag.StringVar(&truncate, "truncate", cohere.NoneTrunc.String(), "truncate type")
+	flag.StringVar(&inputType, "input-type", cohere.ClusteringInput.String(), "input type")
 }
 
 func main() {

cmd/openai/main.go

@@ -17,8 +17,8 @@ var (
 
 func init() {
 	flag.StringVar(&input, "input", "what is life", "input data")
-	flag.StringVar(&model, "model", string(openai.TextAdaV2), "model name")
-	flag.StringVar(&encoding, "encoding", string(openai.EncodingFloat), "encoding format")
+	flag.StringVar(&model, "model", openai.TextAdaV2.String(), "model name")
+	flag.StringVar(&encoding, "encoding", openai.EncodingFloat.String(), "encoding format")
 }
 
 func main() {

cmd/vertexai/main.go

@@ -20,9 +20,9 @@ var (
 
 func init() {
 	flag.StringVar(&input, "input", "what is life", "input data")
-	flag.StringVar(&model, "model", string(vertexai.EmbedGeckoV2), "model name")
+	flag.StringVar(&model, "model", vertexai.EmbedGeckoV2.String(), "model name")
 	flag.BoolVar(&truncate, "truncate", false, "truncate type")
-	flag.StringVar(&taskType, "task-type", string(vertexai.RetrQueryTask), "task type")
+	flag.StringVar(&taskType, "task-type", vertexai.RetrQueryTask.String(), "task type")
 	flag.StringVar(&title, "title", "", "title: only relevant for retrival document tasks")
 }
 

cohere/cohere.go

@@ -13,6 +13,11 @@ const (
 	MultiLingV2      Model = "embed-multilingual-v2.0"
 )
 
+// String implements stringer.
+func (m Model) String() string {
+	return string(m)
+}
+
 // InputType is an embedding input type.
 type InputType string
 
@@ -23,6 +28,11 @@ const (
 	ClusteringInput     InputType = "clustering"
 )
 
+// String implements stringer.
+func (i InputType) String() string {
+	return string(i)
+}
+
 // Truncate controls input truncating.
 // It controls how the API handles inputs
 // longer than the maximum token length (recommended: <512)
@@ -33,3 +43,8 @@ const (
 	EndTrunc   Truncate = "END"
 	NoneTrunc  Truncate = "NONE"
 )
+
+// String implements stringer.
+func (t Truncate) String() string {
+	return string(t)
+}

openai/openai.go

@@ -7,6 +7,11 @@ const (
 	TextAdaV2 Model = "text-embedding-ada-002"
 )
 
+// String implements stringer.
+func (m Model) String() string {
+	return string(m)
+}
+
 // EncodingFormat for embedding API requests.
 type EncodingFormat string
 
@@ -16,3 +21,8 @@ const (
 	// encoded as base64 string
 	EncodingBase64 EncodingFormat = "base64"
 )
+
+// String implements stringer.
+func (f EncodingFormat) String() string {
+	return string(f)
+}

vertexai/vertexai.go

@@ -14,6 +14,11 @@ const (
 	EmbedMultiGecko Model = "multimodalembedding@001"
 )
 
+// String implements stringer.
+func (m Model) String() string {
+	return string(m)
+}
+
 // TaskType is embedding task type.
 // It can be used to improve the embedding quality
 // when targeting a specific task.
@@ -27,3 +32,8 @@ const (
 	ClassificationTask TaskType = "CLASSIFICATION"
 	ClusteringTask     TaskType = "CLUSTERING"
 )
+
+// String implements stringer.
+func (t TaskType) String() string {
+	return string(t)
+}