fix: minor glitches in quickstart 01

Signed-off-by: Xavier Vergés <github@verg.es>

Author: xverges

dapr_agents/workflow/runners/agent.py

@@ -587,7 +587,7 @@ async def _get_status(instance_id: str) -> dict:
         )
         logger.info("Mounted default workflow status endpoint at %s", status_path)
 
-    def shutdown(self, agent: Optional[AgentComponents]) -> None:
+    def shutdown(self, agent: Optional[AgentComponents] = None) -> None:
         """
         Unwire subscriptions and close owned clients.
 

quickstarts/01-dapr-agents-fundamentals/05_agent_memory.py

@@ -1,4 +1,5 @@
 import asyncio
+from pathlib import Path
 
 from dapr_agents.llm import DaprChatClient
 
@@ -21,7 +22,7 @@ async def main() -> None:
         memory=AgentMemoryConfig(
             store=ConversationDaprStateMemory(
                 store_name="conversation-statestore",
-                session_id="03-agent-with-memory",
+                session_id=Path(__file__).stem,
             )
         ),
     )

quickstarts/01-dapr-agents-fundamentals/06_durable_agent_http.py

@@ -1,3 +1,5 @@
+from pathlib import Path
+
 from dapr_agents.llm import DaprChatClient
 
 from dapr_agents import DurableAgent
@@ -21,7 +23,7 @@ def main() -> None:
         memory=AgentMemoryConfig(
             store=ConversationDaprStateMemory(
                 store_name="conversation-statestore",
-                session_id="04-durable-agent-http",
+                session_id=Path(__file__).stem,
             )
         ),
         # This is where the execution state is stored

quickstarts/01-dapr-agents-fundamentals/07_durable_agent_pubsub.py

@@ -1,4 +1,5 @@
 import asyncio
+from pathlib import Path
 
 from dapr_agents.llm import DaprChatClient
 
@@ -16,7 +17,7 @@
 from function_tools import slow_weather_func
 
 
-def main() -> None:
+async def main() -> None:
     weather_agent = DurableAgent(
         name="WeatherAgent",
         role="Weather Assistant",
@@ -28,7 +29,7 @@ def main() -> None:
         memory=AgentMemoryConfig(
             store=ConversationDaprStateMemory(
                 store_name="conversation-statestore",
-                session_id="05-durable-agent-sub",
+                session_id=Path(__file__).stem,
             )
         ),
         # This is where the execution state is stored
@@ -50,12 +51,10 @@ def main() -> None:
     runner = AgentRunner()
     try:
         runner.subscribe(weather_agent)
+        await wait_for_shutdown()
     finally:
         runner.shutdown()
 
 
 if __name__ == "__main__":
-    try:
-        main()
-    except KeyboardInterrupt:
-        print("\nInterrupted by user. Exiting gracefully...")
+    asyncio.run(main())

quickstarts/01-dapr-agents-fundamentals/09_workflow_agents.py

@@ -1,3 +1,4 @@
+from pathlib import Path
 import time
 
 import dapr.ext.workflow as wf
@@ -46,7 +47,7 @@ def get_customer_info(customer_name: str) -> str:
     memory=AgentMemoryConfig(
         store=ConversationDaprStateMemory(
             store_name="conversation-statestore",
-            session_id="04-support-triage",
+            session_id=f"{Path(__file__).stem}-triage",
         )
     ),
 )
@@ -62,7 +63,7 @@ def get_customer_info(customer_name: str) -> str:
     memory=AgentMemoryConfig(
         store=ConversationDaprStateMemory(
             store_name="conversation-statestore",
-            session_id="04-support-expert",
+            session_id=f"{Path(__file__).stem}-expert",
         )
     ),
 )

quickstarts/01-dapr-agents-fundamentals/10_durable_agent_tracing.py

@@ -1,5 +1,6 @@
 import asyncio
 import logging
+from pathlib import Path
 
 from dapr_agents.llm import DaprChatClient
 
@@ -28,7 +29,7 @@ async def main() -> None:
         memory=AgentMemoryConfig(
             store=ConversationDaprStateMemory(
                 store_name="conversation-statestore",
-                session_id="08-durable-agent-trace",
+                session_id=Path(__file__).stem,
             )
         ),
         state=AgentStateConfig(

quickstarts/01-dapr-agents-fundamentals/README.md

@@ -73,7 +73,7 @@ metadata:
     value: "{{OPENAI_API_KEY}}"
 ```
 
-Replace `OPENAI_API_KEY` with your actual OpenAI API key.
+Replace `OPENAI_API_KEY` with your actual OpenAI API key. If you are using a different provider, make sure that you keep the name of the component as `llm-provider`.
 
 ---