refactor: text_embedding result to list[int]

langgenius · Nov 22, 2024 · 33e51cd · 33e51cd
1 parent 5b6c71c
commit 33e51cd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/internal/types/entities/model_entities/text_embedding.go b/internal/types/entities/model_entities/text_embedding.go
@@ -19,5 +19,5 @@ type TextEmbeddingResult struct {
 }
 
 type GetTextEmbeddingNumTokensResponse struct {
-	NumTokens int `json:"num_tokens" validate:"required"`
+	NumTokens []int `json:"num_tokens" validate:"required"`
 }
diff --git a/internal/types/exception/generic.go b/internal/types/exception/generic.go
@@ -18,6 +18,7 @@ func (e *genericError) Error() string {
 }
 
 func (e *genericError) ToResponse() *entities.Response {
+	// TODO: using struct instead, currently, for compatibility with old code
 	errorMsg := parser.MarshalJson(e)
 
 	return entities.NewDaemonErrorResponse(e.code, errorMsg)