Skip to content

Base multi model image query driver

BaseMultiModelImageQueryDriver

Bases: BaseImageQueryDriver, ABC

Image Query Driver for platforms like Amazon Bedrock that host many LLM models.

Instances of this Image Query Driver require a Image Query Model Driver which is used to structure the image generation request in the format required by the model and to process the output.

Attributes:

Name Type Description
model str

Model name to use

image_query_model_driver BaseImageQueryModelDriver

Image Model Driver to use.

Source code in griptape/drivers/image_query/base_multi_model_image_query_driver.py
@define
class BaseMultiModelImageQueryDriver(BaseImageQueryDriver, ABC):
    """Image Query Driver for platforms like Amazon Bedrock that host many LLM models.

    Instances of this Image Query Driver require a Image Query Model Driver which is used to structure the
    image generation request in the format required by the model and to process the output.

    Attributes:
        model: Model name to use
        image_query_model_driver: Image Model Driver to use.
    """

    model: str = field(kw_only=True, metadata={"serializable": True})
    image_query_model_driver: BaseImageQueryModelDriver = field(kw_only=True, metadata={"serializable": True})

image_query_model_driver: BaseImageQueryModelDriver = field(kw_only=True, metadata={'serializable': True}) class-attribute instance-attribute

model: str = field(kw_only=True, metadata={'serializable': True}) class-attribute instance-attribute