@@ -81,7 +81,7 @@ def new_data_artifact(
8181 )
8282
8383
84- def get_artifact_text (artifact : Artifact , delimiter : str = ' \n ' ) -> str :
84+ def get_artifact_text (artifact : Artifact , delimiter : str = " \n " ) -> str :
8585 """Extracts and joins all text content from an Artifact's parts.
8686
8787 Args:
@@ -104,7 +104,7 @@ class ArtifactStreamer:
104104
105105 streamer = ArtifactStreamer(context_id, task_id, name='response')
106106
107- async for chunk in llm .stream(prompt):
107+ async for chunk in model .stream(prompt):
108108 await event_queue.enqueue_event(streamer.append(chunk))
109109
110110 await event_queue.enqueue_event(streamer.finalize())
@@ -120,7 +120,7 @@ def __init__(
120120 self ,
121121 context_id : str ,
122122 task_id : str ,
123- name : str = ' response' ,
123+ name : str = " response" ,
124124 artifact_id : str | None = None ,
125125 ) -> None :
126126 self ._context_id = context_id
@@ -147,7 +147,7 @@ def append(self, text: str) -> TaskArtifactUpdateEvent:
147147 artifact_id = self ._artifact_id ,
148148 name = self ._name ,
149149 parts = [Part (root = TextPart (text = text ))],
150- )
150+ ),
151151 )
152152
153153 def finalize (self ) -> TaskArtifactUpdateEvent :
@@ -166,5 +166,5 @@ def finalize(self) -> TaskArtifactUpdateEvent:
166166 artifact_id = self ._artifact_id ,
167167 name = self ._name ,
168168 parts = [],
169- )
169+ ),
170170 )
0 commit comments