Engines
__all__ = ['BaseQueryEngine', 'VectorQueryEngine', 'BaseSummaryEngine', 'PromptSummaryEngine', 'BaseExtractionEngine', 'CsvExtractionEngine', 'JsonExtractionEngine', 'BaseImageGenerationEngine', 'PromptImageGenerationEngine', 'VariationImageGenerationEngine', 'InpaintingImageGenerationEngine', 'OutpaintingImageGenerationEngine', 'ImageQueryEngine']
module-attribute
BaseExtractionEngine
Bases: ABC
Source code in griptape/engines/extraction/base_extraction_engine.py
chunk_joiner: str = field(default='\n\n', kw_only=True)
class-attribute
instance-attribute
chunker: BaseChunker = field(default=Factory(lambda self: TextChunker(tokenizer=self.prompt_driver.tokenizer, max_tokens=self.max_chunker_tokens), takes_self=True), kw_only=True)
class-attribute
instance-attribute
max_chunker_tokens: int
property
max_token_multiplier: float = field(default=0.5, kw_only=True)
class-attribute
instance-attribute
min_response_tokens: int
property
prompt_driver: BasePromptDriver = field(kw_only=True)
class-attribute
instance-attribute
extract(text, *, rulesets=None, **kwargs)
abstractmethod
validate_max_token_multiplier(_, max_token_multiplier)
Source code in griptape/engines/extraction/base_extraction_engine.py
BaseImageGenerationEngine
Bases: ABC
Source code in griptape/engines/image/base_image_generation_engine.py
image_generation_driver: BaseImageGenerationDriver = field(kw_only=True)
class-attribute
instance-attribute
BaseQueryEngine
Bases: ABC
Source code in griptape/engines/query/base_query_engine.py
load_artifacts(namespace)
abstractmethod
query(query, namespace=None, *, rulesets=None, top_n=None, filter=None)
abstractmethod
upsert_text_artifact(artifact, namespace=None)
abstractmethod
BaseSummaryEngine
Bases: ABC
Source code in griptape/engines/summary/base_summary_engine.py
summarize_artifacts(artifacts, *, rulesets=None)
abstractmethod
CsvExtractionEngine
Bases: BaseExtractionEngine
Source code in griptape/engines/extraction/csv_extraction_engine.py
template_generator: J2 = field(default=Factory(lambda: J2('engines/extraction/csv_extraction.j2')), kw_only=True)
class-attribute
instance-attribute
extract(text, *, rulesets=None, column_names=None, **kwargs)
Source code in griptape/engines/extraction/csv_extraction_engine.py
text_to_csv_rows(text, column_names)
Source code in griptape/engines/extraction/csv_extraction_engine.py
ImageQueryEngine
Source code in griptape/engines/image_query/image_query_engine.py
image_query_driver: BaseImageQueryDriver = field(kw_only=True)
class-attribute
instance-attribute
InpaintingImageGenerationEngine
Bases: BaseImageGenerationEngine
Source code in griptape/engines/image/inpainting_image_generation_engine.py
run(prompts, *args, image, mask, negative_prompts=None, rulesets=None, negative_rulesets=None, **kwargs)
Source code in griptape/engines/image/inpainting_image_generation_engine.py
JsonExtractionEngine
Bases: BaseExtractionEngine
Source code in griptape/engines/extraction/json_extraction_engine.py
template_generator: J2 = field(default=Factory(lambda: J2('engines/extraction/json_extraction.j2')), kw_only=True)
class-attribute
instance-attribute
extract(text, *, rulesets=None, template_schema=None, **kwargs)
Source code in griptape/engines/extraction/json_extraction_engine.py
OutpaintingImageGenerationEngine
Bases: BaseImageGenerationEngine
Source code in griptape/engines/image/outpainting_image_generation_engine.py
run(prompts, *args, image, mask, negative_prompts=None, rulesets=None, negative_rulesets=None, **kwargs)
Source code in griptape/engines/image/outpainting_image_generation_engine.py
PromptImageGenerationEngine
Bases: BaseImageGenerationEngine
Source code in griptape/engines/image/prompt_image_generation_engine.py
run(prompts, *args, negative_prompts=None, rulesets=None, negative_rulesets=None, **kwargs)
Source code in griptape/engines/image/prompt_image_generation_engine.py
PromptSummaryEngine
Bases: BaseSummaryEngine
Source code in griptape/engines/summary/prompt_summary_engine.py
chunk_joiner: str = field(default='\n\n', kw_only=True)
class-attribute
instance-attribute
chunker: BaseChunker = field(default=Factory(lambda self: TextChunker(tokenizer=self.prompt_driver.tokenizer, max_tokens=self.max_chunker_tokens), takes_self=True), kw_only=True)
class-attribute
instance-attribute
max_chunker_tokens: int
property
max_token_multiplier: float = field(default=0.5, kw_only=True)
class-attribute
instance-attribute
min_response_tokens: int
property
prompt_driver: BasePromptDriver = field(kw_only=True)
class-attribute
instance-attribute
template_generator: J2 = field(default=Factory(lambda: J2('engines/summary/prompt_summary.j2')), kw_only=True)
class-attribute
instance-attribute
summarize_artifacts(artifacts, *, rulesets=None)
summarize_artifacts_rec(artifacts, summary=None, rulesets=None)
Source code in griptape/engines/summary/prompt_summary_engine.py
validate_allowlist(_, max_token_multiplier)
Source code in griptape/engines/summary/prompt_summary_engine.py
VariationImageGenerationEngine
Bases: BaseImageGenerationEngine
Source code in griptape/engines/image/variation_image_generation_engine.py
run(prompts, *args, image, negative_prompts=None, rulesets=None, negative_rulesets=None, **kwargs)
Source code in griptape/engines/image/variation_image_generation_engine.py
VectorQueryEngine
Bases: BaseQueryEngine