fix a bug

LazyAGI · Sep 9, 2024 · 71a6fbf · 71a6fbf
1 parent f1f4272
commit 71a6fbf
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/lazyllm/engine/engine.py b/lazyllm/engine/engine.py
@@ -102,6 +102,7 @@ def get_args(cls, key, value, builder_key=None):
 
 class ServerGraph(lazyllm.ModuleBase):
     def __init__(self, g: lazyllm.graph, server: Node, web: Node):
+        super().__init__()
         self._g = lazyllm.ActionModule(g)
         if server:
             if server.args.get('port'): raise NotImplementedError('Port is not supported now')

diff --git a/lazyllm/engine/lightengine.py b/lazyllm/engine/lightengine.py
@@ -43,7 +43,7 @@ def update(self, nodes: List[Dict] = [], changed_nodes: List[Dict] = [],
                edges: List[Dict] = [], changed_resources: List[Dict] = [],
                gid: Optional[str] = None, name: Optional[str] = None):
         for r in changed_resources:
-            if r['kind'] not in ('server', 'web'):
+            if r['kind'] in ('server', 'web'):
                 raise NotImplementedError('Web and Api server are not allowed now')
             self.update_node(r)
         for n in changed_nodes: self.update_node(n)