text_artifact

text_artifact#

Bases: BaseArtifact “)

Source Code in griptape/artifacts/text_artifact.py

@define
class TextArtifact(BaseArtifact):
    value: str = field(converter=str, metadata={"serializable": True})
    embedding: Optional[list[float]] = field(default=None, kw_only=True)

    def __add__(self, other: BaseArtifact) -> TextArtifact:
        return TextArtifact(self.value + other.value)

    def __bool__(self) -> bool:
        return bool(self.value.strip())

    def to_text(self) -> str:
        return self.value

    def generate_embedding(self, driver: BaseEmbeddingDriver) -> list[float]:
        embedding = driver.embed(str(self.value))

        if self.embedding is None:
            self.embedding = []
        self.embedding.clear()
        self.embedding.extend(embedding)

        return self.embedding

    def token_count(self, tokenizer: BaseTokenizer) -> int:
        return tokenizer.count_tokens(str(self.value))

  • embedding = field(default=None, kw_only=True) class-attribute instance-attribute

  • value = field(converter=str, metadata={'serializable': True}) class-attribute instance-attribute

add(other)#

Source Code in griptape/artifacts/text_artifact.py

def __add__(self, other: BaseArtifact) -> TextArtifact:
    return TextArtifact(self.value + other.value)

bool()#

Source Code in griptape/artifacts/text_artifact.py

def __bool__(self) -> bool:
    return bool(self.value.strip())

generate_embedding(driver)#

Source Code in griptape/artifacts/text_artifact.py

def generate_embedding(self, driver: BaseEmbeddingDriver) -> list[float]:
    embedding = driver.embed(str(self.value))

    if self.embedding is None:
        self.embedding = []
    self.embedding.clear()
    self.embedding.extend(embedding)

    return self.embedding

to_text()#

Source Code in griptape/artifacts/text_artifact.py

def to_text(self) -> str:
    return self.value

token_count(tokenizer)#

Source Code in griptape/artifacts/text_artifact.py

def token_count(self, tokenizer: BaseTokenizer) -> int:
    return tokenizer.count_tokens(str(self.value))
Contents