allow passing skip_verify to autogen constructors (#581)

* allow passing skip_verify to autogen constructors * added flag to examples with a NOTE, also added to docs
letta-ai · Dec 26, 2023 · c0f325c · c0f325c
1 parent 2935b8d
commit c0f325c
Show file tree

Hide file tree

Showing 5 changed files with 11 additions and 1 deletion.
diff --git a/docs/autogen.md b/docs/autogen.md
@@ -23,7 +23,7 @@ config_list_memgpt = [
     {
         "model": "gpt-4",
         "context_window": 8192,
-        "preset": "memgpt_chat",
+        "preset": "memgpt_chat",  # NOTE: you can change the preset here
         # OpenAI specific
         "model_endpoint_type": "openai",
         "openai_key": YOUR_OPENAI_KEY,
@@ -45,6 +45,7 @@ memgpt_autogen_agent = create_memgpt_autogen_agent_from_config(
     system_message=f"Your desired MemGPT persona",
     interface_kwargs=interface_kwargs,
     default_auto_reply="...",
+    skip_verify=False,  # NOTE: you should set this to True if you expect your MemGPT AutoGen agent to call a function other than send_message on the first turn
 )
 ```
 

diff --git a/memgpt/autogen/examples/agent_autoreply.py b/memgpt/autogen/examples/agent_autoreply.py
@@ -169,6 +169,7 @@
         human_input_mode="TERMINATE",
         interface_kwargs=interface_kwargs,
         default_auto_reply="...",  # Set a default auto-reply message here (non-empty auto-reply is required for LM Studio)
+        skip_verify=False,  # NOTE: you should set this to True if you expect your MemGPT AutoGen agent to call a function other than send_message on the first turn
     )
 
 # Begin the group chat with a message from the user

diff --git a/memgpt/autogen/examples/agent_docs.py b/memgpt/autogen/examples/agent_docs.py
@@ -151,6 +151,7 @@
     system_message=f"You are an AI research assistant.\n" f"You are participating in a group chat with a user ({user_proxy.name}).",
     interface_kwargs=interface_kwargs,
     default_auto_reply="...",  # Set a default auto-reply message here (non-empty auto-reply is required for LM Studio)
+    skip_verify=False,  # NOTE: you should set this to True if you expect your MemGPT AutoGen agent to call a function other than send_message on the first turn
 )
 # NOTE: you need to follow steps to load document first: see https://memgpt.readthedocs.io/en/latest/autogen/#loading-documents
 memgpt_agent.load_and_attach("memgpt_research_paper", "directory")

diff --git a/memgpt/autogen/examples/agent_groupchat.py b/memgpt/autogen/examples/agent_groupchat.py
@@ -173,6 +173,7 @@
         f"and a product manager ({pm.name}).",
         interface_kwargs=interface_kwargs,
         default_auto_reply="...",  # Set a default auto-reply message here (non-empty auto-reply is required for LM Studio)
+        skip_verify=False,  # NOTE: you should set this to True if you expect your MemGPT AutoGen agent to call a function other than send_message on the first turn
     )
 
 # Initialize the group chat between the user and two LLM agents (PM and coder)

diff --git a/memgpt/autogen/memgpt_agent.py b/memgpt/autogen/memgpt_agent.py
@@ -28,6 +28,7 @@ def create_memgpt_autogen_agent_from_config(
     nonmemgpt_llm_config: Optional[Union[Dict, bool]] = None,
     default_auto_reply: Optional[Union[str, Dict, None]] = "",
     interface_kwargs: Dict = None,
+    skip_verify: bool = False,
 ):
     """Same function signature as used in base AutoGen, but creates a MemGPT agent
 
@@ -87,6 +88,7 @@ def create_memgpt_autogen_agent_from_config(
         default_auto_reply=default_auto_reply,
         is_termination_msg=is_termination_msg,
         interface_kwargs=interface_kwargs,
+        skip_verify=skip_verify,
     )
 
     if human_input_mode != "ALWAYS":
@@ -95,6 +97,7 @@ def create_memgpt_autogen_agent_from_config(
             default_auto_reply=default_auto_reply,
             is_termination_msg=is_termination_msg,
             interface_kwargs=interface_kwargs,
+            skip_verify=skip_verify,
         )
         if default_auto_reply != "":
             coop_agent2 = UserProxyAgent(
@@ -108,6 +111,7 @@ def create_memgpt_autogen_agent_from_config(
                 default_auto_reply=default_auto_reply,
                 is_termination_msg=is_termination_msg,
                 interface_kwargs=interface_kwargs,
+                skip_verify=skip_verify,
             )
 
         groupchat = GroupChat(
@@ -126,6 +130,7 @@ def create_memgpt_autogen_agent_from_config(
 def create_autogen_memgpt_agent(
     agent_config,
     # interface and persistence manager
+    skip_verify=False,
     interface=None,
     interface_kwargs={},
     persistence_manager=None,
@@ -172,6 +177,7 @@ def create_autogen_memgpt_agent(
         agent=memgpt_agent,
         default_auto_reply=default_auto_reply,
         is_termination_msg=is_termination_msg,
+        skip_verify=skip_verify,
     )
     return autogen_memgpt_agent