Skip to content

Csv row artifact

CsvRowArtifact

Bases: TextArtifact

Source code in griptape/griptape/artifacts/csv_row_artifact.py 
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
@define(frozen=True)
class CsvRowArtifact(TextArtifact):
    value: dict[str, str] = field(converter=BaseArtifact.value_to_dict)
    delimiter: str = field(default=",", kw_only=True)

    def __add__(self, other: CsvRowArtifact) -> CsvRowArtifact:
        return CsvRowArtifact(self.value | other.value)

    def to_text(self) -> str:
        with io.StringIO() as csvfile:
            writer = csv.DictWriter(
                csvfile, fieldnames=self.value.keys(), quoting=csv.QUOTE_MINIMAL, delimiter=self.delimiter
            )

            writer.writerow(self.value)

            return csvfile.getvalue().strip()

    def to_dict(self) -> dict:
        from griptape.schemas import CsvRowArtifactSchema

        return dict(CsvRowArtifactSchema().dump(self))

    def __bool__(self) -> bool:
        return len(self) > 0

delimiter: str = field(default=',', kw_only=True) class-attribute instance-attribute

value: dict[str, str] = field(converter=BaseArtifact.value_to_dict) class-attribute instance-attribute

__add__(other)

Source code in griptape/griptape/artifacts/csv_row_artifact.py 
13
14
def __add__(self, other: CsvRowArtifact) -> CsvRowArtifact:
    return CsvRowArtifact(self.value | other.value)

__bool__()

Source code in griptape/griptape/artifacts/csv_row_artifact.py 
31
32
def __bool__(self) -> bool:
    return len(self) > 0

to_dict()

Source code in griptape/griptape/artifacts/csv_row_artifact.py 
26
27
28
29
def to_dict(self) -> dict:
    from griptape.schemas import CsvRowArtifactSchema

    return dict(CsvRowArtifactSchema().dump(self))

to_text()

Source code in griptape/griptape/artifacts/csv_row_artifact.py 
16
17
18
19
20
21
22
23
24
def to_text(self) -> str:
    with io.StringIO() as csvfile:
        writer = csv.DictWriter(
            csvfile, fieldnames=self.value.keys(), quoting=csv.QUOTE_MINIMAL, delimiter=self.delimiter
        )

        writer.writerow(self.value)

        return csvfile.getvalue().strip()