google_drivers_config

Suggest edits

Bases: DriversConfig

Source Code in griptape/configs/drivers/google_drivers_config.py
@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
@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
@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
@lazy_property()
def vector_store_driver(self) -> LocalVectorStoreDriver:
    return LocalVectorStoreDriver(embedding_driver=GoogleEmbeddingDriver(model="models/embedding-004"))
← Prev

drivers_config

↑ Up

drivers

Next →

openai_drivers_config

Could this page be better? Report a problem or suggest an addition!