@@ -79,7 +79,7 @@ def new_data_artifact(
7979 )
8080
8181
82- def get_artifact_text (artifact : Artifact , delimiter : str = ' \n ' ) -> str :
82+ def get_artifact_text (artifact : Artifact , delimiter : str = " \n " ) -> str :
8383 """Extracts and joins all text content from an Artifact's parts.
8484
8585 Args:
@@ -102,7 +102,7 @@ class ArtifactStreamer:
102102
103103 streamer = ArtifactStreamer(context_id, task_id, name='response')
104104
105- async for chunk in llm .stream(prompt):
105+ async for chunk in model .stream(prompt):
106106 await event_queue.enqueue_event(streamer.append(chunk))
107107
108108 await event_queue.enqueue_event(streamer.finalize())
@@ -118,7 +118,7 @@ def __init__(
118118 self ,
119119 context_id : str ,
120120 task_id : str ,
121- name : str = ' response' ,
121+ name : str = " response" ,
122122 artifact_id : str | None = None ,
123123 ) -> None :
124124 self ._context_id = context_id
@@ -145,7 +145,7 @@ def append(self, text: str) -> TaskArtifactUpdateEvent:
145145 artifact_id = self ._artifact_id ,
146146 name = self ._name ,
147147 parts = [Part (text = text )],
148- )
148+ ),
149149 )
150150
151151 def finalize (self ) -> TaskArtifactUpdateEvent :
@@ -164,5 +164,5 @@ def finalize(self) -> TaskArtifactUpdateEvent:
164164 artifact_id = self ._artifact_id ,
165165 name = self ._name ,
166166 parts = [],
167- )
167+ ),
168168 )
0 commit comments