浏览代码

refactor: text_embedding result to list[int]

Yeuoly 8 月之前
父节点
当前提交
33e51cd75e
共有 2 个文件被更改,包括 2 次插入1 次删除
  1. 1 1
      internal/types/entities/model_entities/text_embedding.go
  2. 1 0
      internal/types/exception/generic.go

+ 1 - 1
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"`
 }

+ 1 - 0
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)