Fix

Upsonic · Apr 7, 2024 · 430c8f6 · 430c8f6
1 parent d67ab0b
commit 430c8f6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/upsonic/remote/on_prem.py b/upsonic/remote/on_prem.py
@@ -1351,6 +1351,7 @@ def langchain(self, prefix=None, version=None):
 
 
     def autogen(self, caller, executor, prefix=None, version=None):
+        import autogen
         for each in self.get_all():
             the_true_name = each
             the_true_name = the_true_name.replace("_", ".")
@@ -1368,7 +1369,7 @@ def autogen(self, caller, executor, prefix=None, version=None):
                         autogen.agentchat.register_function(
                             the_function,
                             caller=caller,
-                            executor=user_proxy,
+                            executor=executor,
                             description=the_document,
                             name=the_true_name
                         )