add migration script

Author: Peteredwardj

src/cli.py

@@ -15,6 +15,7 @@
 from src.agent_factory import AgentFactory
 from src.helpers import print_h_bar
 from src.langgraph.langgraph_agent import LangGraphAgent
+from src.migration_script import migrate_config
 
 # Configure logging
 logging.basicConfig(level=logging.INFO, format='%(message)s')
@@ -156,6 +157,17 @@ def _initialize_commands(self) -> None:
             )
         )
 
+        # Migrate agent config command
+        self._register_command(
+            Command(
+                name="migrate",
+                description="Migrate agent configuration from old to new format.",
+                tips=["Use this command to convert old agent JSON files to the new format"],
+                handler=self.migrate_agent_config,
+                aliases=['migrate-config']
+            )
+        )
+
         ################## CONNECTIONS ################## 
         # List actions command
         self._register_command(
@@ -544,6 +556,17 @@ def chat_session(self, input_list: List[str]) -> None:
 
             except KeyboardInterrupt:
                 break
+            
+    def migrate_agent_config(self, input_list: List[str]) -> None:
+        """Migrate agent configuration from old to new format."""
+        print_h_bar()
+        
+        agent_file_name = input("Enter the name of the agent file to migrate: ").strip()
+        agent_type_input = input("Enter the agent type (a for autonomous, l for legacy): ").strip().lower()
+        
+        agent_type = "autonomous" if agent_type_input == "a" else "legacy"
+        
+        migrate_config(agent_file_name, agent_type)
 
     def exit(self, input_list: List[str]) -> None:
         """Exit the CLI gracefully"""

src/migration_script.py

@@ -0,0 +1,53 @@
+import json
+
+#Easily convert the old JSON format to the new agent configuration format
+
+def migrate_old_to_new(old_json, conversion_type="legacy"):
+    new_json = {
+        "config": {
+            "name": old_json["name"],
+            "type": conversion_type,
+            "loop_delay": old_json["loop_delay"],
+        },
+        "llms": {
+            "character": {
+                "name": old_json["name"],
+                "bio": old_json["bio"],
+                "traits": old_json["traits"],
+                "examples": old_json["examples"],
+                "example_accounts": old_json["example_accounts"],
+                "model_provider": "openai",
+                "model": "gpt-3.5-turbo",
+            }
+        },
+        "connections": [],
+    }
+
+    if conversion_type == "autonomous":
+        new_json["llms"]["executor"] = {
+            "model_provider": "openai",  # openai or anthropic for langchain
+            "model": "gpt-4o-mini",
+        }
+    elif conversion_type == "legacy":
+        new_json["config"]["time_based_multipliers"] = old_json["time_based_multipliers"]
+        new_json["tasks"] = old_json.get("tasks", [])
+
+    new_json["connections"] = [
+        {"name": item["name"], "config": {k: v for k, v in item.items() if k != "name"}}
+        for item in old_json["config"]
+    ]
+
+    return new_json
+
+def migrate_config(agent_file_name,agent_type):
+    with open("agents/{}.json".format(agent_file_name), "r") as f:
+        old_data = json.load(f)
+
+    new_data = migrate_old_to_new(old_data, agent_type)
+
+    # Save new JSON to a file
+    new_file_name = "agents/{}-migrated.json".format(agent_file_name)
+    with open(new_file_name, "w") as f:
+        json.dump(new_data, f, indent=2)
+
+    print("Migration complete. New agent configuration of type {} saved to {}".format(agent_type,new_file_name))