Skip to content

Google structure config

GoogleStructureConfig

Bases: StructureConfig

Source code in griptape/config/google_structure_config.py 
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
@define
class GoogleStructureConfig(StructureConfig):
    prompt_driver: BasePromptDriver = field(
        default=Factory(lambda: GooglePromptDriver(model="gemini-pro")), kw_only=True, metadata={"serializable": True}
    )
    embedding_driver: BaseEmbeddingDriver = field(
        default=Factory(lambda: GoogleEmbeddingDriver(model="models/embedding-001")),
        kw_only=True,
        metadata={"serializable": True},
    )
    vector_store_driver: BaseVectorStoreDriver = field(
        default=Factory(
            lambda: LocalVectorStoreDriver(embedding_driver=GoogleEmbeddingDriver(model="models/embedding-001"))
        ),
        kw_only=True,
        metadata={"serializable": True},
    )

embedding_driver: BaseEmbeddingDriver = field(default=Factory(lambda: GoogleEmbeddingDriver(model='models/embedding-001')), kw_only=True, metadata={'serializable': True}) class-attribute instance-attribute

prompt_driver: BasePromptDriver = field(default=Factory(lambda: GooglePromptDriver(model='gemini-pro')), kw_only=True, metadata={'serializable': True}) class-attribute instance-attribute

vector_store_driver: BaseVectorStoreDriver = field(default=Factory(lambda: LocalVectorStoreDriver(embedding_driver=GoogleEmbeddingDriver(model='models/embedding-001'))), kw_only=True, metadata={'serializable': True}) class-attribute instance-attribute