Skip to content

Commit 25d4c4d

Browse files
committed
Fix formatting
1 parent 7c73f6b commit 25d4c4d

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
@@ -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

Comments
 (0)