From 0b92c3127d48695f9a48a74719a7cfcf03440c16 Mon Sep 17 00:00:00 2001
From: Arsenii Shatokhin <arseny9795@gmail.com>
Date: Sun, 14 Jan 2024 09:40:47 +0400
Subject: [PATCH] Genesis agency notebook

---
 notebooks/genesis_agency.ipynb | 43 ++++++++++++++++++----------------
 1 file changed, 23 insertions(+), 20 deletions(-)

diff --git a/notebooks/genesis_agency.ipynb b/notebooks/genesis_agency.ipynb
index 1751ae27..2160b84c 100644
--- a/notebooks/genesis_agency.ipynb
+++ b/notebooks/genesis_agency.ipynb
@@ -7,8 +7,8 @@
    "metadata": {
     "collapsed": true,
     "ExecuteTime": {
-     "end_time": "2024-01-11T04:14:26.736327Z",
-     "start_time": "2024-01-11T04:14:26.723533Z"
+     "end_time": "2024-01-11T05:42:05.498363Z",
+     "start_time": "2024-01-11T05:42:05.482640Z"
     }
    },
    "outputs": [],
@@ -30,8 +30,8 @@
    "metadata": {
     "collapsed": false,
     "ExecuteTime": {
-     "end_time": "2024-01-11T04:14:27.063630Z",
-     "start_time": "2024-01-11T04:14:26.881702Z"
+     "end_time": "2024-01-11T05:42:05.793832Z",
+     "start_time": "2024-01-11T05:42:05.621095Z"
     }
    },
    "id": "85f1c8fb6aab6421",
@@ -41,14 +41,25 @@
    "cell_type": "code",
    "outputs": [],
    "source": [
-    "from agency_swarm.agents.genesis import AgentCreator, ToolCreator\n",
+    "from agency_swarm import set_openai_key\n",
+    "set_openai_key(\"YOUR_OPENAI_API_KEY\")"
+   ],
+   "metadata": {
+    "collapsed": false
+   },
+   "id": "1ee6f9992fb209c4"
+  },
+  {
+   "cell_type": "code",
+   "outputs": [],
+   "source": [
     "from agency_swarm.agency.genesis import GenesisAgency"
    ],
    "metadata": {
     "collapsed": false,
     "ExecuteTime": {
-     "end_time": "2024-01-11T04:14:27.763148Z",
-     "start_time": "2024-01-11T04:14:27.284715Z"
+     "end_time": "2024-01-11T05:42:06.237671Z",
+     "start_time": "2024-01-11T05:42:05.796187Z"
     }
    },
    "id": "6cafc0d9a51ed4c9",
@@ -69,21 +80,13 @@
     }
    ],
    "source": [
-    "agent_creator = AgentCreator()\n",
-    "tool_creator = ToolCreator()\n",
-    "\n",
-    "test_agency = GenesisAgency(\n",
-    "#     agency_chart=[\n",
-    "#     agent_creator,\n",
-    "#     [agent_creator, tool_creator]\n",
-    "# ]\n",
-    ")"
+    "test_agency = GenesisAgency()"
    ],
    "metadata": {
     "collapsed": false,
     "ExecuteTime": {
-     "end_time": "2024-01-11T04:14:32.205462Z",
-     "start_time": "2024-01-11T04:14:29.026670Z"
+     "end_time": "2024-01-11T05:42:09.869069Z",
+     "start_time": "2024-01-11T05:42:06.498881Z"
     }
    },
    "id": "64b988e65898f6b3",
@@ -124,8 +127,8 @@
    "metadata": {
     "collapsed": false,
     "ExecuteTime": {
-     "end_time": "2024-01-11T04:14:33.706633Z",
-     "start_time": "2024-01-11T04:14:32.205904Z"
+     "end_time": "2024-01-11T05:42:12.145710Z",
+     "start_time": "2024-01-11T05:42:09.869494Z"
     }
    },
    "id": "c95ce4524b2b3ab4",