Merge pull request #262 from zqq454224016/main

feat: add embedder and knowledgebase gql
kubeagi · Nov 22, 2023 · a70c276 · a70c276
2 parents d0cd5ef + 8af5ab7
commit a70c276
Show file tree

Hide file tree

Showing 2 changed files with 248 additions and 0 deletions.
diff --git a/graphql-server/go-server/graph/schema/embedder.gql b/graphql-server/go-server/graph/schema/embedder.gql
@@ -0,0 +1,108 @@
+# create
+mutation createEmbedder($input: CreateEmbedderInput!) {
+  Embedder {
+    createEmbedder(input: $input) {
+      name
+      namespace
+      labels
+      annotations
+      displayName
+      description
+      endpoint {
+        url
+        authSecret {
+          kind
+          Name
+        }
+        insecure
+      }
+      serviceType
+      updateTimestamp
+    }
+  }
+}
+
+# update
+mutation updateEmbedder($input: UpdateEmbedderInput) {
+  Embedder {
+    updateEmbedder(input: $input) {
+      name
+      namespace
+      labels
+      annotations
+      displayName
+      description
+      endpoint {
+        url
+        authSecret {
+          kind
+          Name
+        }
+        insecure
+      }
+      serviceType
+      updateTimestamp
+    }
+  }
+}
+
+# delete
+mutation deleteEmbedder($input: DeleteEmbedderInput!) {
+  Embedder {
+    deleteEmbedder(input: $input)
+  }
+}
+
+# list embedders
+query listEmbedders($input: ListEmbedderInput!) {
+  Embedder {
+    listEmbedders(input: $input) {
+    totalCount
+    hasNextPage
+    nodes {
+      ... on Embedder {
+        name
+        namespace
+        labels
+        annotations
+        displayName
+        description
+        endpoint {
+          url
+          authSecret {
+            kind
+            Name
+          }
+          insecure
+        }
+        serviceType
+        updateTimestamp
+      }
+      }
+    }
+  }
+}
+
+# get embedder
+query getEmbedder($name: String!, $namespace: String!) {
+  Embedder {
+    getEmbedder(name: $name, namespace: $namespace) {
+        name
+        namespace
+        labels
+        annotations
+        displayName
+        description
+        endpoint {
+          url
+          authSecret {
+            kind
+            Name
+          }
+          insecure
+        }
+        serviceType
+        updateTimestamp
+    }
+  }
+}
diff --git a/graphql-server/go-server/graph/schema/knowledgebase.gql b/graphql-server/go-server/graph/schema/knowledgebase.gql
@@ -0,0 +1,140 @@
+# list
+query listKnowledgeBases($input: ListKnowledgeBaseInput!){
+  KnowledgeBase {
+    listKnowledgeBases(input: $input) {
+      totalCount
+      hasNextPage
+      nodes {
+        ... on KnowledgeBase {
+          name
+          namespace
+          labels
+          annotations
+          creator
+          displayName
+          description
+          embedder {
+            kind
+            Name
+          }
+          vectorstore{
+            kind
+            Name
+          }
+          fileGroups {
+            ... on file{
+                source {
+                    kind
+                    Name
+                }
+                path
+            }
+          }
+        }
+      }
+    }
+  }
+}
+
+# get
+query getKnowledgeBase($name: String!, $namespace: String!) {
+  KnowledgeBase {
+    getKnowledgeBase(name: $name, namespace: $namespace){
+          name
+          namespace
+          labels
+          annotations
+          creator
+          displayName
+          description
+          embedder {
+            kind
+            Name
+          }
+          vectorstore{
+            kind
+            Name
+          }
+          fileGroups {
+            ... on file{
+                source {
+                    kind
+                    Name
+                }
+                path
+            }
+          }
+    }
+  }
+}
+
+# create
+mutation createKnowledgeBase($input: CreateKnowledgeBaseInput!) {
+  KnowledgeBase {
+    createKnowledgeBase(input: $input) {
+          name
+          namespace
+          labels
+          annotations
+          creator
+          displayName
+          description
+          embedder {
+            kind
+            Name
+          }
+          vectorstore{
+            kind
+            Name
+          }
+          fileGroups {
+            ... on file{
+                source {
+                    kind
+                    Name
+                }
+                path
+            }
+          }
+    }
+  }
+}
+
+# update
+mutation updateKnowledgeBase($input: UpdateKnowledgeBaseInput) {
+  KnowledgeBase {
+    updateKnowledgeBase(input: $input){
+          name
+          namespace
+          labels
+          annotations
+          creator
+          displayName
+          description
+          embedder {
+            kind
+            Name
+          }
+          vectorstore{
+            kind
+            Name
+          }
+          fileGroups {
+            ... on file{
+                source {
+                    kind
+                    Name
+                }
+                path
+            }
+          }
+    }
+  }
+}
+
+# delete
+mutation deleteKnowledgeBase($input: DeleteKnowledgeBaseInput) {
+  KnowledgeBase {
+    deleteKnowledgeBase(input: $input)
+  }
+}