Skip to content

Google structure config

GoogleStructureConfig

Bases: BaseStructureConfig

Source code in griptape/config/google_structure_config.py 
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
@define
class GoogleStructureConfig(BaseStructureConfig):
    global_drivers: StructureGlobalDriversConfig = field(
        default=Factory(
            lambda: StructureGlobalDriversConfig(
                prompt_driver=GooglePromptDriver(model="gemini-pro"),
                embedding_driver=GoogleEmbeddingDriver(model="models/embedding-001"),
                vector_store_driver=LocalVectorStoreDriver(
                    embedding_driver=GoogleEmbeddingDriver(model="models/embedding-001")
                ),
            )
        ),
        kw_only=True,
        metadata={"serializable": True},
    )
    task_memory: StructureTaskMemoryConfig = field(
        default=Factory(
            lambda self: StructureTaskMemoryConfig(
                query_engine=StructureTaskMemoryQueryEngineConfig(
                    prompt_driver=self.global_drivers.prompt_driver,
                    vector_store_driver=LocalVectorStoreDriver(embedding_driver=self.global_drivers.embedding_driver),
                ),
                extraction_engine=StructureTaskMemoryExtractionEngineConfig(
                    csv=StructureTaskMemoryExtractionEngineCsvConfig(prompt_driver=self.global_drivers.prompt_driver),
                    json=StructureTaskMemoryExtractionEngineJsonConfig(prompt_driver=self.global_drivers.prompt_driver),
                ),
                summary_engine=StructureTaskMemorySummaryEngineConfig(prompt_driver=self.global_drivers.prompt_driver),
            ),
            takes_self=True,
        ),
        kw_only=True,
        metadata={"serializable": True},
    )

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

task_memory: StructureTaskMemoryConfig = field(default=Factory(lambda self: StructureTaskMemoryConfig(query_engine=StructureTaskMemoryQueryEngineConfig(prompt_driver=self.global_drivers.prompt_driver, vector_store_driver=LocalVectorStoreDriver(embedding_driver=self.global_drivers.embedding_driver)), extraction_engine=StructureTaskMemoryExtractionEngineConfig(csv=StructureTaskMemoryExtractionEngineCsvConfig(prompt_driver=self.global_drivers.prompt_driver), json=StructureTaskMemoryExtractionEngineJsonConfig(prompt_driver=self.global_drivers.prompt_driver)), summary_engine=StructureTaskMemorySummaryEngineConfig(prompt_driver=self.global_drivers.prompt_driver)), takes_self=True), kw_only=True, metadata={'serializable': True}) class-attribute instance-attribute