Rename: GenericAgent -> ResponsiveAgent

microsoft · Jul 28, 2023 · 56a5f84 · 56a5f84
1 parent 24d865a
commit 56a5f84
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 7 deletions.
diff --git a/flaml/autogen/agent/opti_guide.py b/flaml/autogen/agent/opti_guide.py
@@ -28,7 +28,7 @@
 
 from .agent import Agent
 from .assistant_agent import AssistantAgent
-from .generic_agent import GenericAgent
+from .responsive_agent import ResponsiveAgent
 from .user_proxy_agent import UserProxyAgent
 
 # %% System Messages
@@ -78,7 +78,7 @@
 
 
 # %%
-class OptiGuideAgent(GenericAgent):
+class OptiGuideAgent(ResponsiveAgent):
     """(Experimental) OptiGuide is an agent to write Python code and to answer
       users questions for supply chain-related coding project.
 

diff --git a/flaml/autogen/agentchat/__init__.py b/flaml/autogen/agentchat/__init__.py
@@ -1,7 +1,7 @@
 from .agent import Agent
-from .responsive_agent import ResponsiveAgent
 from .assistant_agent import AssistantAgent
+from .opti_guide import OptiGuideAgent
+from .responsive_agent import ResponsiveAgent
 from .user_proxy_agent import UserProxyAgent
 
-
-__all__ = ["Agent", "ResponsiveAgent", "AssistantAgent", "UserProxyAgent"]
+__all__ = ["Agent", "ResponsiveAgent", "AssistantAgent", "UserProxyAgent", "OptiGuideAgent"]
diff --git a/notebook/autogen_optiguide.ipynb b/notebook/autogen_optiguide.ipynb
@@ -123,7 +123,7 @@
     "\n",
     "# import flaml and autogen\n",
     "from flaml import oai\n",
-    "from flaml.autogen.agent import Agent, GenericAgent\n",
+    "from flaml.autogen.agent import Agent, ResponsiveAgent\n",
     "from flaml.autogen.agent.opti_guide import OptiGuideAgent"
    ]
   },
@@ -272,7 +272,7 @@
     "                  source_code=code,\n",
     "                  example_qa=\"\")\n",
     "\n",
-    "user = GenericAgent(\"user\", max_consecutive_auto_reply=0,\n",
+    "user = ResponsiveAgent(\"user\", max_consecutive_auto_reply=0,\n",
     "                         human_input_mode=\"NEVER\", code_execution_config=False)\n"
    ]
   },

diff --git a/setup.py b/setup.py
@@ -148,6 +148,7 @@
             "joblib<1.3.0",  # temp solution for joblib 1.3.0 issue, no need once https://github.com/joblib/joblib-spark/pull/48 is merged
         ],
         "autozero": ["scikit-learn", "pandas", "packaging"],
+        "optiguide": ["gurobipy", "eventlet", "termcolor"],
     },
     classifiers=[
         "Programming Language :: Python :: 3",