Merge pull request kubeagi#387 from bjwswang/gql

feat: add llm_config in data process config
nkwangleiGIT · Dec 15, 2023 · 256b51a · 256b51a
2 parents 1f650c3 + f3b4b32
commit 256b51a
Show file tree

Hide file tree

Showing 5 changed files with 154 additions and 14 deletions.
diff --git a/graphql-server/go-server/graph/generated/generated.go b/graphql-server/go-server/graph/generated/generated.go
diff --git a/graphql-server/go-server/graph/generated/models_gen.go b/graphql-server/go-server/graph/generated/models_gen.go
diff --git a/graphql-server/go-server/graph/impl/dataprocessing.resolvers.go b/graphql-server/go-server/graph/impl/dataprocessing.resolvers.go
diff --git a/graphql-server/go-server/graph/schema/dataprocessing.graphqls b/graphql-server/go-server/graph/schema/dataprocessing.graphqls
@@ -55,6 +55,16 @@ input FileItem {
 # 数据处理配置条目
 input DataProcessConfigItem {
   type: String!
+  llm_config: LLMConfigItem
+}
+
+# LLM for 数据处理配置条目 
+input LLMConfigItem {
+  name: String
+  namespace: String
+  model: String
+  temperature: String
+  top_k: String
 }
 
 input DeleteDataProcessInput {

diff --git a/graphql-server/go-server/pkg/dataprocessing/dataprocessing.go b/graphql-server/go-server/pkg/dataprocessing/dataprocessing.go
@@ -20,6 +20,8 @@ import (
 	"net/http"
 	"sync"
 
+	"k8s.io/client-go/dynamic"
+
 	"github.com/kubeagi/arcadia/graphql-server/go-server/graph/generated"
 )
 
@@ -36,7 +38,7 @@ func Init(dataprocessingURL string) {
 	})
 }
 
-func ListDataprocessing(ctx context.Context, obj *generated.DataProcessQuery, input *generated.AllDataProcessListByPageInput) (*generated.PaginatedDataProcessItem, error) {
+func ListDataprocessing(ctx context.Context, c dynamic.Interface, obj *generated.DataProcessQuery, input *generated.AllDataProcessListByPageInput) (*generated.PaginatedDataProcessItem, error) {
 	// prepare http request
 	jsonParams, err := json.Marshal(input)
 	if err != nil {
@@ -64,7 +66,7 @@ func ListDataprocessing(ctx context.Context, obj *generated.DataProcessQuery, in
 	return pagedData, nil
 }
 
-func ListDataprocessingByCount(ctx context.Context, obj *generated.DataProcessQuery, input *generated.AllDataProcessListByCountInput) (*generated.CountDataProcessItem, error) {
+func ListDataprocessingByCount(ctx context.Context, c dynamic.Interface, obj *generated.DataProcessQuery, input *generated.AllDataProcessListByCountInput) (*generated.CountDataProcessItem, error) {
 	// prepare http request
 	jsonParams, err := json.Marshal(input)
 	if err != nil {
@@ -92,7 +94,7 @@ func ListDataprocessingByCount(ctx context.Context, obj *generated.DataProcessQu
 	return countData, nil
 }
 
-func DataProcessSupportType(ctx context.Context, obj *generated.DataProcessQuery) (*generated.DataProcessSupportType, error) {
+func DataProcessSupportType(ctx context.Context, c dynamic.Interface, obj *generated.DataProcessQuery) (*generated.DataProcessSupportType, error) {
 	// prepare http request
 	req, err := http.NewRequest("POST", url+"/text-process-type", nil)
 	if err != nil {
@@ -116,7 +118,9 @@ func DataProcessSupportType(ctx context.Context, obj *generated.DataProcessQuery
 	return data, nil
 }
 
-func CreateDataProcessTask(ctx context.Context, obj *generated.DataProcessMutation, input *generated.AddDataProcessInput) (*generated.DataProcessResponse, error) {
+func CreateDataProcessTask(ctx context.Context, c dynamic.Interface, obj *generated.DataProcessMutation, input *generated.AddDataProcessInput) (*generated.DataProcessResponse, error) {
+	// create complete http payload to data processing service
+
 	// prepare http request
 	jsonParams, err := json.Marshal(input)
 	if err != nil {
@@ -144,7 +148,7 @@ func CreateDataProcessTask(ctx context.Context, obj *generated.DataProcessMutati
 	return data, nil
 }
 
-func DeleteDataProcessTask(ctx context.Context, obj *generated.DataProcessMutation, input *generated.DeleteDataProcessInput) (*generated.DataProcessResponse, error) {
+func DeleteDataProcessTask(ctx context.Context, c dynamic.Interface, obj *generated.DataProcessMutation, input *generated.DeleteDataProcessInput) (*generated.DataProcessResponse, error) {
 	// prepare http request
 	jsonParams, err := json.Marshal(input)
 	if err != nil {
@@ -172,7 +176,7 @@ func DeleteDataProcessTask(ctx context.Context, obj *generated.DataProcessMutati
 	return data, nil
 }
 
-func DataProcessDetails(ctx context.Context, obj *generated.DataProcessQuery, input *generated.DataProcessDetailsInput) (*generated.DataProcessDetails, error) {
+func DataProcessDetails(ctx context.Context, c dynamic.Interface, obj *generated.DataProcessQuery, input *generated.DataProcessDetailsInput) (*generated.DataProcessDetails, error) {
 	// prepare http request
 	jsonParams, err := json.Marshal(input)
 	if err != nil {