File size: 2,674 Bytes
b110593
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
//                           _       _
// __      _____  __ ___   ___  __ _| |_ ___
// \ \ /\ / / _ \/ _` \ \ / / |/ _` | __/ _ \
//  \ V  V /  __/ (_| |\ V /| | (_| | ||  __/
//   \_/\_/ \___|\__,_| \_/ |_|\__,_|\__\___|
//
//  Copyright © 2016 - 2024 Weaviate B.V. All rights reserved.
//
//  CONTACT: [email protected]
//

package articles

import (
	"github.com/go-openapi/strfmt"
	"github.com/google/uuid"
	"github.com/weaviate/weaviate/entities/models"
	"github.com/weaviate/weaviate/entities/schema"
)

func ArticlesClass() *models.Class {
	return &models.Class{
		Class: "Article",
		Properties: []*models.Property{
			{
				Name:         "title",
				DataType:     schema.DataTypeText.PropString(),
				Tokenization: models.PropertyTokenizationWhitespace,
			},
			{
				Name:     "hasParagraphs",
				DataType: []string{"Paragraph"},
			},
		},
	}
}

func ParagraphsClass() *models.Class {
	return &models.Class{
		Class: "Paragraph",
		Properties: []*models.Property{
			{
				Name:     "contents",
				DataType: schema.DataTypeText.PropString(),
			},
		},
		Vectorizer: "none",
	}
}

type Article models.Object

func (a *Article) WithID(id strfmt.UUID) *Article {
	a.ID = id
	return a
}

func (a *Article) WithReferences(refs ...*models.SingleRef) *Article {
	props := a.Properties.(map[string]interface{})
	props["hasParagraphs"] = models.MultipleRef(refs)
	return a
}

func (a *Article) WithTitle(title string) *Article {
	props := a.Properties.(map[string]interface{})
	props["title"] = title
	return a
}

func (a *Article) WithTenant(tk string) *Article {
	a.Tenant = tk
	return a
}

func (a *Article) Object() *models.Object {
	obj := models.Object(*a)
	return &obj
}

func NewArticle() *Article {
	return &Article{
		Class:      "Article",
		ID:         strfmt.UUID(uuid.NewString()),
		Properties: make(map[string]interface{}),
	}
}

type Paragraph models.Object

func (p *Paragraph) WithID(id strfmt.UUID) *Paragraph {
	p.ID = id
	return p
}

func (p *Paragraph) WithContents(contents string) *Paragraph {
	props := p.Properties.(map[string]interface{})
	props["contents"] = contents
	return p
}

func (p *Paragraph) WithVector(vec []float32) *Paragraph {
	p.Vector = vec
	return p
}

func (p *Paragraph) WithTenant(tk string) *Paragraph {
	p.Tenant = tk
	return p
}

func (p *Paragraph) Object() *models.Object {
	obj := models.Object(*p)
	return &obj
}

func NewParagraph() *Paragraph {
	return &Paragraph{
		Class:      "Paragraph",
		ID:         strfmt.UUID(uuid.NewString()),
		Properties: make(map[string]interface{}),
	}
}