Fix the dall e invocation process

2023-12-04 10:39:56 +08:00
parent 83149b5590
commit 3dc4fe8e4c
7 changed files with 20 additions and 9 deletions
@@ -477,6 +477,8 @@ class BaseAIAgent(abc.ABC):
        org_msg:AgentMsg,
        stack_limit = 5
    ) -> ComputeTaskResult:
+        from ..frame.compute_kernel import ComputeKernel
+        
        arguments = None
        try:
            func_name = inner_func_call_node.get("name")