Skip to content

google_drivers_config

GoogleDriversConfig

Bases: DriversConfig

Source code in griptape/configs/drivers/google_drivers_config.py 
10
11
12
13
14
15
16
17
18
19
20
21
22
@define
class GoogleDriversConfig(DriversConfig):
    @lazy_property()
    def prompt_driver(self) -> GooglePromptDriver:
        return GooglePromptDriver(model="gemini-2.0-flash")

    @lazy_property()
    def embedding_driver(self) -> GoogleEmbeddingDriver:
        return GoogleEmbeddingDriver(model="models/embedding-004")

    @lazy_property()
    def vector_store_driver(self) -> LocalVectorStoreDriver:
        return LocalVectorStoreDriver(embedding_driver=GoogleEmbeddingDriver(model="models/embedding-004"))

embedding_driver()

Source code in griptape/configs/drivers/google_drivers_config.py 
16
17
18
@lazy_property()
def embedding_driver(self) -> GoogleEmbeddingDriver:
    return GoogleEmbeddingDriver(model="models/embedding-004")

prompt_driver()

Source code in griptape/configs/drivers/google_drivers_config.py 
12
13
14
@lazy_property()
def prompt_driver(self) -> GooglePromptDriver:
    return GooglePromptDriver(model="gemini-2.0-flash")

vector_store_driver()

Source code in griptape/configs/drivers/google_drivers_config.py 
20
21
22
@lazy_property()
def vector_store_driver(self) -> LocalVectorStoreDriver:
    return LocalVectorStoreDriver(embedding_driver=GoogleEmbeddingDriver(model="models/embedding-004"))