Make RunContext internal (#386)

* Make RunContext internal * Mypy
microsoft · Aug 21, 2024 · ed08905 · ed08905
1 parent 09ceef4
commit ed08905
Show file tree

Hide file tree

Showing 40 changed files with 2,360 additions and 2,329 deletions.
diff --git a/python/docs/src/cookbook/langgraph-agent.ipynb b/python/docs/src/cookbook/langgraph-agent.ipynb
diff --git a/python/docs/src/cookbook/llamaindex-agent.ipynb b/python/docs/src/cookbook/llamaindex-agent.ipynb
diff --git a/python/docs/src/cookbook/openai-assistant-agent.ipynb b/python/docs/src/cookbook/openai-assistant-agent.ipynb
diff --git a/python/docs/src/getting-started/agent-and-agent-runtime.ipynb b/python/docs/src/getting-started/agent-and-agent-runtime.ipynb
diff --git a/python/docs/src/getting-started/message-and-communication.ipynb b/python/docs/src/getting-started/message-and-communication.ipynb
@@ -136,10 +136,10 @@
     }
    ],
    "source": [
-    "run_context = runtime.start()\n",
+    "runtime.start()\n",
     "await runtime.send_message(TextMessage(content=\"Hello, World!\", source=\"User\"), agent)\n",
     "await runtime.send_message(ImageMessage(url=\"https://example.com/image.jpg\", source=\"User\"), agent)\n",
-    "await run_context.stop_when_idle()"
+    "await runtime.stop_when_idle()"
    ]
   },
   {
@@ -241,10 +241,10 @@
     "runtime = SingleThreadedAgentRuntime()\n",
     "await runtime.register(\"inner_agent\", lambda: InnerAgent(\"InnerAgent\"))\n",
     "await runtime.register(\"outer_agent\", lambda: OuterAgent(\"OuterAgent\", \"InnerAgent\"))\n",
-    "run_context = runtime.start()\n",
+    "runtime.start()\n",
     "outer = AgentId(\"outer_agent\", \"default\")\n",
     "await runtime.send_message(Message(content=\"Hello, World!\"), outer)\n",
-    "await run_context.stop_when_idle()"
+    "await runtime.stop_when_idle()"
    ]
   },
   {
@@ -344,9 +344,9 @@
     "await runtime.register(\"receiving_agent\", lambda: ReceivingAgent(\"Receiving Agent\"))\n",
     "await runtime.add_subscription(TypeSubscription(\"default\", \"broadcasting_agent\"))\n",
     "await runtime.add_subscription(TypeSubscription(\"default\", \"receiving_agent\"))\n",
-    "run_context = runtime.start()\n",
+    "runtime.start()\n",
     "await runtime.send_message(Message(\"Hello, World!\"), AgentId(\"broadcasting_agent\", \"default\"))\n",
-    "await run_context.stop_when_idle()"
+    "await runtime.stop()"
    ]
   },
   {
@@ -380,9 +380,9 @@
     "await runtime.register(\"receiving_agent\", lambda: ReceivingAgent(\"Receiving Agent\"))\n",
     "await runtime.add_subscription(TypeSubscription(\"default\", \"broadcasting_agent\"))\n",
     "await runtime.add_subscription(TypeSubscription(\"default\", \"receiving_agent\"))\n",
-    "run_context = runtime.start()\n",
+    "runtime.start()\n",
     "await runtime.publish_message(Message(\"Hello, World! From the runtime!\"), topic_id=TopicId(\"default\", \"default\"))\n",
-    "await run_context.stop_when_idle()"
+    "await runtime.stop_when_idle()"
    ]
   },
   {

diff --git a/python/docs/src/getting-started/model-clients.ipynb b/python/docs/src/getting-started/model-clients.ipynb
@@ -331,13 +331,13 @@
     "    ),\n",
     ")\n",
     "# Start the runtime processing messages.\n",
-    "run_context = runtime.start()\n",
+    "runtime.start()\n",
     "# Send a message to the agent and get the response.\n",
     "message = Message(\"Hello, what are some fun things to do in Seattle?\")\n",
     "response = await runtime.send_message(message, AgentId(\"simple-agent\", \"default\"))\n",
     "print(response.content)\n",
     "# Stop the runtime processing messages.\n",
-    "await run_context.stop()"
+    "await runtime.stop()"
    ]
   }
  ],

diff --git a/python/docs/src/getting-started/multi-agent-design-patterns.ipynb b/python/docs/src/getting-started/multi-agent-design-patterns.ipynb
@@ -509,14 +509,14 @@
     "    lambda: CoderAgent(model_client=OpenAIChatCompletionClient(model=\"gpt-4o-mini\")),\n",
     ")\n",
     "await runtime.add_subscription(TypeSubscription(\"default\", \"ReviewerAgent\"))\n",
-    "run_context = runtime.start()\n",
+    "runtime.start()\n",
     "await runtime.publish_message(\n",
     "    message=CodeWritingTask(task=\"Write a function to find the sum of all even numbers in a list.\"),\n",
     "    topic_id=TopicId(\"default\", \"default\"),\n",
     ")\n",
     "\n",
     "# Keep processing messages until idle.\n",
-    "await run_context.stop_when_idle()"
+    "await runtime.stop_when_idle()"
    ]
   },
   {