Spaces:
Running
Running
File size: 1,985 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 |
// _ _
// __ _____ __ ___ ___ __ _| |_ ___
// \ \ /\ / / _ \/ _` \ \ / / |/ _` | __/ _ \
// \ V V / __/ (_| |\ V /| | (_| | || __/
// \_/\_/ \___|\__,_| \_/ |_|\__,_|\__\___|
//
// Copyright © 2016 - 2024 Weaviate B.V. All rights reserved.
//
// CONTACT: [email protected]
//
package generate
import (
"fmt"
"github.com/tailor-inc/graphql"
)
func (p *GenerateProvider) additionalGenerateField(classname string) *graphql.Field {
return &graphql.Field{
Args: graphql.FieldConfigArgument{
"singleResult": &graphql.ArgumentConfig{
Description: "Results per object",
Type: graphql.NewInputObject(graphql.InputObjectConfig{
Name: fmt.Sprintf("%sIndividualResultsArg", classname),
Fields: graphql.InputObjectConfigFieldMap{
"prompt": &graphql.InputObjectFieldConfig{
Description: "prompt",
Type: graphql.String,
},
},
}),
DefaultValue: nil,
},
"groupedResult": &graphql.ArgumentConfig{
Description: "Grouped results of all objects",
Type: graphql.NewInputObject(graphql.InputObjectConfig{
Name: fmt.Sprintf("%sAllResultsArg", classname),
Fields: graphql.InputObjectConfigFieldMap{
"task": &graphql.InputObjectFieldConfig{
Description: "task",
Type: graphql.String,
},
"properties": &graphql.InputObjectFieldConfig{
Description: "Properties used for the generation",
Type: graphql.NewList(graphql.String),
DefaultValue: nil,
},
},
}),
DefaultValue: nil,
},
},
Type: graphql.NewObject(graphql.ObjectConfig{
Name: fmt.Sprintf("%sAdditionalGenerate", classname),
Fields: graphql.Fields{
"singleResult": &graphql.Field{Type: graphql.String},
"groupedResult": &graphql.Field{Type: graphql.String},
"error": &graphql.Field{Type: graphql.String},
},
}),
}
}
|