test: add rest of quickstarts plus wip starting with 05 ones (#272)

* test: add rest of quickstarts plus wip starting with 05 ones

Signed-off-by: Samantha Coyle <[email protected]>

* fix: update so rest of tests work

Signed-off-by: Samantha Coyle <[email protected]>

* fix: tox -e ruff

Signed-off-by: Samantha Coyle <[email protected]>

* style: stream all logs by default for starters

Signed-off-by: Samantha Coyle <[email protected]>

* style: appease flake8:

Signed-off-by: Samantha Coyle <[email protected]>

* fix: run all int tests in ci wf

Signed-off-by: Samantha Coyle <[email protected]>

* style: tox -e ruff

Signed-off-by: Samantha Coyle <[email protected]>

* style: use helpers

Signed-off-by: Samantha Coyle <[email protected]>

* style: update comments

Signed-off-by: Samantha Coyle <[email protected]>

* style: tox -e ruff

Signed-off-by: Samantha Coyle <[email protected]>

* fix: rm pr req check

Signed-off-by: Samantha Coyle <[email protected]>

---------

Signed-off-by: Samantha Coyle <[email protected]>

Author: sicoyle

.github/workflows/integration-tests.yaml

@@ -23,9 +23,7 @@ jobs:
     # Team membership will be verified for issue_comment events only
     if: |
       (github.event_name == 'issue_comment' &&
-       github.event.issue.pull_request &&
        contains(github.event.comment.body, '/ok-to-test')) ||
-      github.event_name == 'pull_request' ||
       github.event_name == 'workflow_dispatch'
     steps:
     - name: Simple check to verify user is authorized
@@ -106,8 +104,8 @@ jobs:
         OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
       run: |
         
-        uv run pytest -m integration -v tests/integration/quickstarts/test_01_*.py tests/integration/quickstarts/test_02_*.py
-        # Run all integration tests in parallel (one file per quickstart directory)
+        uv run pytest -m integration -v tests/integration/quickstarts
+        # TODO: In future, run all integration tests in parallel (one file per quickstart directory)
         # -n auto uses all available CPUs, or specify -n 4 for fixed number
         # pytest -m integration -v -n auto tests/integration/quickstarts/
 

dapr_agents/agents/orchestrators/random.py

@@ -272,7 +272,13 @@ async def _broadcast() -> None:
             logger.exception("Failed to publish broadcast message.")
 
     def _select_random_speaker_activity(self, ctx: wf.WorkflowActivityContext) -> str:
-        """Pick a random agent from the registry, avoiding the most recent speaker when possible."""
+        """
+        Pick a random agent from the registry, avoiding the most recent speaker when possible.
+
+        TODO: This method selects from all registered agents in the team, including
+        orchestrators. In the future, we may want to add health checks or availability
+        checks to ensure selected agents are actually running and ready to receive triggers.
+        """
         try:
             agents_metadata = self.list_team_agents(
                 include_self=False, team=self.effective_team()

dapr_agents/agents/orchestrators/roundrobin.py

@@ -255,6 +255,10 @@ def _select_next_speaker_activity(
     ) -> str:
         """
         Deterministic round-robin selection based on turn number.
+
+        TODO: This method selects from all registered agents in the team, including
+        orchestrators. In the future, we may want to add health checks or availability
+        checks to ensure selected agents are actually running and ready to receive triggers.
         """
         turn = int(payload.get("turn", 1))
         try:

quickstarts/05-multi-agent-workflows/components/registrystore_llm.yaml

@@ -0,0 +1,12 @@
+apiVersion: dapr.io/v1alpha1
+kind: Component
+metadata:
+  name: agentregistrystore_llm
+spec:
+  type: state.redis
+  version: v1
+  metadata:
+  - name: redisHost
+    value: localhost:6379
+  - name: keyPrefix
+    value: none

quickstarts/05-multi-agent-workflows/components/registrystore_random.yaml

@@ -0,0 +1,12 @@
+apiVersion: dapr.io/v1alpha1
+kind: Component
+metadata:
+  name: agentregistrystore_random
+spec:
+  type: state.redis
+  version: v1
+  metadata:
+  - name: redisHost
+    value: localhost:6379
+  - name: keyPrefix
+    value: none

quickstarts/05-multi-agent-workflows/components/registrystore_roundrobin.yaml

@@ -0,0 +1,12 @@
+apiVersion: dapr.io/v1alpha1
+kind: Component
+metadata:
+  name: agentregistrystore_roundrobin
+spec:
+  type: state.redis
+  version: v1
+  metadata:
+  - name: redisHost
+    value: localhost:6379
+  - name: keyPrefix
+    value: none

quickstarts/05-multi-agent-workflows/services/frodo/app.py

@@ -2,6 +2,7 @@
 
 import asyncio
 import logging
+import os
 
 from dotenv import load_dotenv
 
@@ -35,7 +36,9 @@ async def main() -> None:
     """
     # Shared infra (registry)
     registry = AgentRegistryConfig(
-        store=StateStoreService(store_name="agentregistrystore"),
+        store=StateStoreService(
+            store_name=os.getenv("REGISTRY_STATE_STORE", "agentregistrystore")
+        ),
         team_name="fellowship",
     )
 

quickstarts/05-multi-agent-workflows/services/gandalf/app.py

@@ -2,6 +2,7 @@
 
 import asyncio
 import logging
+import os
 
 from dotenv import load_dotenv
 
@@ -35,7 +36,9 @@ async def main() -> None:
     """
     # Shared infra (registry)
     registry = AgentRegistryConfig(
-        store=StateStoreService(store_name="agentregistrystore"),
+        store=StateStoreService(
+            store_name=os.getenv("REGISTRY_STATE_STORE", "agentregistrystore")
+        ),
         team_name="fellowship",
     )
 

quickstarts/05-multi-agent-workflows/services/legolas/app.py

@@ -2,6 +2,7 @@
 
 import asyncio
 import logging
+import os
 
 from dotenv import load_dotenv
 
@@ -35,7 +36,9 @@ async def main() -> None:
     """
     # Shared infra (registry)
     registry = AgentRegistryConfig(
-        store=StateStoreService(store_name="agentregistrystore"),
+        store=StateStoreService(
+            store_name=os.getenv("REGISTRY_STATE_STORE", "agentregistrystore")
+        ),
         team_name="fellowship",
     )
 

quickstarts/05-multi-agent-workflows/services/sam/app.py

@@ -2,6 +2,7 @@
 
 import asyncio
 import logging
+import os
 
 from dotenv import load_dotenv
 
@@ -40,7 +41,9 @@ async def main() -> None:
     """
     # Shared infra (registry)
     registry = AgentRegistryConfig(
-        store=StateStoreService(store_name="agentregistrystore"),
+        store=StateStoreService(
+            store_name=os.getenv("REGISTRY_STATE_STORE", "agentregistrystore")
+        ),
         team_name="fellowship",
     )