Run stream thread using with_contextvars

CHANGELOG.md

@@ -22,6 +22,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - `StructureVisualizer.base_url` for setting the base URL on the url generated by `StructureVisualizer.to_url()`.
 - `StructureVisualizer.query_params` for setting query parameters on the url generated by `StructureVisualizer.to_url()`.
 - Parsing `ActionCallDeltaMessageContent`s with empty string `partial_input`s.
+- `Stream` util not properly propagating thread contextvars.
 
 ## [1.0.0] - 2024-12-09
 

griptape/utils/stream.py

@@ -17,6 +17,7 @@
     FinishStructureRunEvent,
     TextChunkEvent,
 )
+from griptape.utils.contextvars_utils import with_contextvars
 
 if TYPE_CHECKING:
     from collections.abc import Iterator
@@ -44,7 +45,7 @@ class Stream:
     _event_queue: Queue[BaseEvent] = field(default=Factory(lambda: Queue()))
 
     def run(self, *args) -> Iterator[TextArtifact]:
-        t = Thread(target=self._run_structure, args=args)
+        t = Thread(target=with_contextvars(self._run_structure), args=args)
         t.start()
 
         action_str = ""