소스 검색

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)