@@ -75,7 +75,7 @@ def new_data_artifact(
7575 )
7676
7777
78- def get_artifact_text (artifact : Artifact , delimiter : str = ' \n ' ) -> str :
78+ def get_artifact_text (artifact : Artifact , delimiter : str = " \n " ) -> str :
7979 """Extracts and joins all text content from an Artifact's parts.
8080
8181 Args:
@@ -98,7 +98,7 @@ class ArtifactStreamer:
9898
9999 streamer = ArtifactStreamer(context_id, task_id, name='response')
100100
101- async for chunk in llm .stream(prompt):
101+ async for chunk in model .stream(prompt):
102102 await event_queue.enqueue_event(streamer.append(chunk))
103103
104104 await event_queue.enqueue_event(streamer.finalize())
@@ -114,7 +114,7 @@ def __init__(
114114 self ,
115115 context_id : str ,
116116 task_id : str ,
117- name : str = ' response' ,
117+ name : str = " response" ,
118118 artifact_id : str | None = None ,
119119 ) -> None :
120120 self ._context_id = context_id
@@ -141,7 +141,7 @@ def append(self, text: str) -> TaskArtifactUpdateEvent:
141141 artifact_id = self ._artifact_id ,
142142 name = self ._name ,
143143 parts = [Part (root = TextPart (text = text ))],
144- )
144+ ),
145145 )
146146
147147 def finalize (self ) -> TaskArtifactUpdateEvent :
@@ -160,5 +160,5 @@ def finalize(self) -> TaskArtifactUpdateEvent:
160160 artifact_id = self ._artifact_id ,
161161 name = self ._name ,
162162 parts = [],
163- )
163+ ),
164164 )
0 commit comments