[DRAFT] Convert autogenstudio to new serializable config language

python/packages/autogen-core/docs/src/user-guide/agentchat-user-guide/serialize-components.ipynb

@@ -156,27 +156,28 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 2,
+      "execution_count": 19,
       "metadata": {},
       "outputs": [
         {
           "name": "stdout",
           "output_type": "stream",
           "text": [
-            "{\"provider\":\"autogen_agentchat.teams.RoundRobinGroupChat\",\"component_type\":\"team\",\"version\":1,\"component_version\":1,\"description\":null,\"config\":{\"participants\":[{\"provider\":\"autogen_agentchat.agents.AssistantAgent\",\"component_type\":\"agent\",\"version\":1,\"component_version\":1,\"config\":{\"name\":\"assistant\",\"model_client\":{\"provider\":\"autogen_ext.models.openai.OpenAIChatCompletionClient\",\"component_type\":\"model\",\"version\":1,\"component_version\":1,\"config\":{\"model\":\"gpt-4o\"}},\"handoffs\":[{\"target\":\"flights_refunder\",\"description\":\"Handoff to flights_refunder.\",\"name\":\"transfer_to_flights_refunder\",\"message\":\"Transferred to flights_refunder, adopting the role of flights_refunder immediately.\"},{\"target\":\"user\",\"description\":\"Handoff to user.\",\"name\":\"transfer_to_user\",\"message\":\"Transferred to user, adopting the role of user immediately.\"}],\"model_context\":{\"provider\":\"autogen_core.model_context.UnboundedChatCompletionContext\",\"component_type\":\"chat_completion_context\",\"version\":1,\"component_version\":1,\"config\":{}},\"description\":\"An agent that provides assistance with ability to use tools.\",\"system_message\":\"Use tools to solve tasks.\",\"reflect_on_tool_use\":false,\"tool_call_summary_format\":\"{result}\"}}],\"termination_condition\":{\"provider\":\"autogen_agentchat.conditions.MaxMessageTermination\",\"component_type\":\"termination\",\"version\":1,\"component_version\":1,\"config\":{\"max_messages\":2}}}}\n"
+            "{\"provider\":\"autogen_agentchat.teams.RoundRobinGroupChat\",\"component_type\":\"team\",\"version\":1,\"component_version\":1,\"description\":null,\"config\":{\"participants\":[{\"provider\":\"autogen_agentchat.agents.AssistantAgent\",\"component_type\":\"agent\",\"version\":1,\"component_version\":1,\"config\":{\"name\":\"assistant\",\"model_client\":{\"provider\":\"autogen_ext.models.openai.OpenAIChatCompletionClient\",\"component_type\":\"model\",\"version\":1,\"component_version\":1,\"config\":{\"model\":\"gpt-4o\",\"api_key\":\"YOUR_API_KEY\"}},\"handoffs\":[{\"target\":\"flights_refunder\",\"description\":\"Handoff to flights_refunder.\",\"name\":\"transfer_to_flights_refunder\",\"message\":\"Transferred to flights_refunder, adopting the role of flights_refunder immediately.\"},{\"target\":\"user\",\"description\":\"Handoff to user.\",\"name\":\"transfer_to_user\",\"message\":\"Transferred to user, adopting the role of user immediately.\"}],\"model_context\":{\"provider\":\"autogen_core.model_context.UnboundedChatCompletionContext\",\"component_type\":\"chat_completion_context\",\"version\":1,\"component_version\":1,\"config\":{}},\"description\":\"An agent that provides assistance with ability to use tools.\",\"system_message\":\"Use tools to solve tasks.\",\"reflect_on_tool_use\":false,\"tool_call_summary_format\":\"{result}\"}}],\"termination_condition\":{\"provider\":\"autogen_agentchat.conditions.MaxMessageTermination\",\"component_type\":\"termination\",\"version\":1,\"component_version\":1,\"config\":{\"max_messages\":2}}}}\n"
           ]
         }
       ],
       "source": [
-        "from autogen_agentchat.agents import AssistantAgent, UserProxyAgent\n",
+        "from autogen_agentchat.agents import AssistantAgent\n",
         "from autogen_agentchat.conditions import MaxMessageTermination\n",
         "from autogen_agentchat.teams import RoundRobinGroupChat\n",
+        "from autogen_core import ComponentBase\n",
         "from autogen_ext.models.openai import OpenAIChatCompletionClient\n",
         "\n",
         "# Create an agent that uses the OpenAI GPT-4o model.\n",
         "model_client = OpenAIChatCompletionClient(\n",
         "    model=\"gpt-4o\",\n",
-        "    # api_key=\"YOUR_API_KEY\",\n",
+        "    api_key=\"YOUR_API_KEY\",\n",
         ")\n",
         "agent = AssistantAgent(\n",
         "    name=\"assistant\",\n",
@@ -189,7 +190,7 @@
         "team = RoundRobinGroupChat(participants=[agent], termination_condition=MaxMessageTermination(2))\n",
         "\n",
         "team_config = team.dump_component()  # dump component\n",
-        "print(team_config.model_dump_json())"
+        "print(team_config.model_dump_json())\n"
       ]
     },
     {

python/packages/autogen-core/src/autogen_core/_component_config.py

@@ -7,7 +7,7 @@
 from pydantic import BaseModel
 from typing_extensions import Self, TypeVar
 
-ComponentType = Literal["model", "agent", "tool", "termination", "token_provider"] | str
+ComponentType = Literal["model", "agent", "tool", "team", "termination", "token_provider"] | str
 ConfigT = TypeVar("ConfigT", bound=BaseModel)
 FromConfigT = TypeVar("FromConfigT", bound=BaseModel, contravariant=True)
 ToConfigT = TypeVar("ToConfigT", bound=BaseModel, covariant=True)

python/packages/autogen-studio/autogenstudio/__init__.py

@@ -1,5 +1,5 @@
 from .database.db_manager import DatabaseManager
-from .datamodel import Agent, AgentConfig, Model, ModelConfig, Team, TeamConfig, Tool, ToolConfig
+from .datamodel import Agent, Model, Team, Tool
 from .teammanager import TeamManager
 from .version import __version__
 
@@ -9,10 +9,6 @@
     "DatabaseManager",
     "Team",
     "Agent",
-    "ToolConfig",
-    "ModelConfig",
-    "TeamConfig",
-    "AgentConfig",
     "TeamManager",
     "__version__",
 ]