44 lines
1.4 KiB
Go
44 lines
1.4 KiB
Go
|
package vectordb
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/mlogclub/simple/common/structs"
|
||
|
|
"github.com/spf13/cast"
|
||
|
|
)
|
||
|
|
|
||
|
|
type ChunkPayload struct {
|
||
|
|
KnowledgeBaseID int64 `json:"knowledge_base_id"`
|
||
|
|
DocumentID int64 `json:"document_id"`
|
||
|
|
DocumentTitle string `json:"document_title"`
|
||
|
|
FaqID int64 `json:"faq_id"`
|
||
|
|
FaqQuestion string `json:"faq_question"`
|
||
|
|
ChunkNo int `json:"chunk_no"`
|
||
|
|
ChunkType string `json:"chunk_type"`
|
||
|
|
SectionPath string `json:"section_path"`
|
||
|
|
Title string `json:"title"`
|
||
|
|
Content string `json:"content"`
|
||
|
|
Provider string `json:"provider"`
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p ChunkPayload) ToMap() map[string]any {
|
||
|
|
return structs.StructToMap(p)
|
||
|
|
}
|
||
|
|
|
||
|
|
func ChunkPayloadFromMap(data map[string]any) ChunkPayload {
|
||
|
|
if data == nil {
|
||
|
|
return ChunkPayload{}
|
||
|
|
}
|
||
|
|
return ChunkPayload{
|
||
|
|
KnowledgeBaseID: cast.ToInt64(data["knowledge_base_id"]),
|
||
|
|
DocumentID: cast.ToInt64(data["document_id"]),
|
||
|
|
DocumentTitle: cast.ToString(data["document_title"]),
|
||
|
|
FaqID: cast.ToInt64(data["faq_id"]),
|
||
|
|
FaqQuestion: cast.ToString(data["faq_question"]),
|
||
|
|
ChunkNo: cast.ToInt(data["chunk_no"]),
|
||
|
|
ChunkType: cast.ToString(data["chunk_type"]),
|
||
|
|
SectionPath: cast.ToString(data["section_path"]),
|
||
|
|
Title: cast.ToString(data["title"]),
|
||
|
|
Content: cast.ToString(data["content"]),
|
||
|
|
Provider: cast.ToString(data["provider"]),
|
||
|
|
}
|
||
|
|
}
|