Global Config

griptape/utils/chat.py

@@ -25,15 +25,18 @@
     )
 
     def default_output_fn(self, text: str) -> None:
-        from griptape.config import config
+        from griptape.tasks.prompt_task import PromptTask
 
-        if config.drivers.prompt.stream:
+        streaming_tasks = [
+            task for task in self.structure.tasks if isinstance(task, PromptTask) and task.prompt_driver.stream
+        ]
+        if streaming_tasks:
             print(text, end="", flush=True)  # noqa: T201
         else:
             print(text)  # noqa: T201
 
     def start(self) -> None:
         from griptape.config import config
 
         if self.intro_text:
             self.output_fn(self.intro_text)

griptape/utils/stream.py

@@ -34,10 +34,13 @@ class Stream:
 
     @structure.validator  # pyright: ignore[reportAttributeAccessIssue]
     def validate_structure(self, _: Attribute, structure: Structure) -> None:
-        from griptape.config import config
+        from griptape.tasks import PromptTask
 
-        if not config.drivers.prompt.stream:
-            raise ValueError("prompt driver does not have streaming enabled, enable with stream=True")
+        streaming_tasks = [
+            task for task in structure.tasks if isinstance(task, PromptTask) and task.prompt_driver.stream
+        ]
+        if not streaming_tasks:
+            raise ValueError("Structure does not have any streaming tasks, enable with stream=True")
 
     _event_queue: Queue[BaseEvent] = field(default=Factory(lambda: Queue()))
 

tests/unit/utils/test_stream.py

@@ -2,19 +2,17 @@
 
 import pytest
 
-from griptape.config import config
-from griptape.structures import Agent
+from griptape.structures import Agent, Pipeline
 from griptape.utils import Stream
 
 
 class TestStream:
     @pytest.fixture(params=[True, False])
     def agent(self, request):
-        config.drivers.prompt.stream = request.param
-        return Agent()
+        return Agent(stream=request.param)
 
     def test_init(self, agent):
-        if config.drivers.prompt.stream:
+        if agent.stream:
             chat_stream = Stream(agent)
 
             assert chat_stream.structure == agent
@@ -29,3 +27,9 @@ def test_init(self, agent):
         else:
             with pytest.raises(ValueError):
                 Stream(agent)
+
+    def test_validate_structure_invalid(self):
+        pipeline = Pipeline(tasks=[])
+
+        with pytest.raises(ValueError):
+            Stream(pipeline)