Skip to content

google_drivers_config

GoogleDriversConfig

Bases: DriversConfig

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

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

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

embedding_driver()

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

prompt_driver()

Source code in griptape/configs/drivers/google_drivers_config.py 
14
15
16
@lazy_property()
def prompt_driver(self) -> GooglePromptDriver:
    return GooglePromptDriver(model="gemini-1.5-pro")

vector_store_driver()

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