feat: optimize model service apis

Signed-off-by: bjwswang <[email protected]>
kubeagi · Dec 27, 2023 · 9fafe44 · 9fafe44
1 parent 9c2752f
commit 9fafe44
Show file tree

Hide file tree

Showing 14 changed files with 557 additions and 1,144 deletions.
diff --git a/apiserver/graph/generated/generated.go b/apiserver/graph/generated/generated.go
diff --git a/apiserver/graph/generated/models_gen.go b/apiserver/graph/generated/models_gen.go
diff --git a/apiserver/graph/impl/modelservice.resolvers.go b/apiserver/graph/impl/modelservice.resolvers.go
diff --git a/apiserver/graph/schema/modelservice.gql b/apiserver/graph/schema/modelservice.gql
@@ -9,45 +9,13 @@ mutation createModelService($input: CreateModelServiceInput!) {
             creator
             displayName
             description
+            providerType
             types
             apiType
             creationTimestamp
             updateTimestamp
-            llmResource {
-                name
-                namespace
-                labels
-                annotations
-                displayName
-                description
-                baseUrl
-                models
-                provider
-                type
-                updateTimestamp
-                status
-                message
-            }
-            embedderResource {
-                name
-                namespace
-                labels
-                annotations
-                displayName
-                description
-                type
-                baseUrl
-                models
-                provider
-                updateTimestamp
-                status
-                message
-            }
-            resource {
-                cpu
-                nvidiaGPU
-                memory
-            }
+            status
+            message
         }
     }
 }
@@ -63,45 +31,13 @@ mutation updateModelService($input: UpdateModelServiceInput) {
             creator
             displayName
             description
+            providerType
             types
             apiType
             creationTimestamp
             updateTimestamp
-            llmResource {
-                name
-                namespace
-                labels
-                annotations
-                displayName
-                description
-                baseUrl
-                models
-                provider
-                type
-                updateTimestamp
-                status
-                message
-            }
-            embedderResource {
-                name
-                namespace
-                labels
-                annotations
-                displayName
-                description
-                type
-                baseUrl
-                models
-                provider
-                updateTimestamp
-                status
-                message
-            }
-            resource {
-                cpu
-                nvidiaGPU
-                memory
-            }
+            status
+            message
         }
     }
 }
@@ -112,60 +48,6 @@ mutation deleteModelServices($input: DeleteCommonInput) {
     }
 }
 
-query getModelService($name: String!, $namespace: String!, $apiType: String!) {
-    ModelService {
-        getModelService(name: $name, namespace: $namespace, apiType: $apiType) {
-            id
-            name
-            namespace
-            labels
-            annotations
-            creator
-            displayName
-            description
-            types
-            apiType
-            creationTimestamp
-            updateTimestamp 
-            llmResource {
-                name
-                namespace
-                labels
-                annotations
-                displayName
-                description
-                baseUrl
-                models
-                provider
-                type
-                updateTimestamp
-                status
-                message
-            }
-            embedderResource{
-                name
-                namespace
-                labels
-                annotations
-                displayName
-                description
-                baseUrl
-                models
-                provider
-                type
-                updateTimestamp
-                status
-                message
-            }
-            resource {
-                cpu
-                memory
-                nvidiaGPU
-            }
-        }
-    }
-}
-
 query listModelServices($input: ListModelService) {
     ModelService {
         listModelServices(input: $input) {
@@ -182,45 +64,13 @@ query listModelServices($input: ListModelService) {
                     creator
                     displayName
                     description
+                    providerType
                     types
                     apiType
                     creationTimestamp
-                    updateTimestamp 
-                    llmResource {
-                        name
-                        namespace
-                        labels
-                        annotations
-                        displayName
-                        description
-                        baseUrl
-                        models
-                        provider
-                        type
-                        updateTimestamp
-                        status
-                        message
-                    }
-                    embedderResource{
-                        name
-                        namespace
-                        labels
-                        annotations
-                        displayName
-                        description
-                        baseUrl
-                        models
-                        provider
-                        type
-                        updateTimestamp
-                        status
-                        message
-                    }
-                    resource {
-                        cpu
-                        memory
-                        nvidiaGPU
-                    }
+                    updateTimestamp
+                    status
+                    message
                 }
             }
         }