Skip to content

Commit 5ed89a2

Browse files
committed
Fix formatting
1 parent eb5d4d5 commit 5ed89a2

1 file changed

Lines changed: 5 additions & 5 deletions

File tree

src/a2a/utils/artifact.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)