Skip to content

Amazon bedrock structure config

AmazonBedrockStructureConfig

Bases: BaseStructureConfig

Source code in griptape/config/amazon_bedrock_structure_config.py 
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
@define()
class AmazonBedrockStructureConfig(BaseStructureConfig):
    global_drivers: StructureGlobalDriversConfig = field(
        default=Factory(
            lambda: StructureGlobalDriversConfig(
                prompt_driver=AmazonBedrockPromptDriver(
                    model="anthropic.claude-3-sonnet-20240229-v1:0",
                    stream=False,
                    prompt_model_driver=BedrockClaudePromptModelDriver(),
                ),
                image_generation_driver=AmazonBedrockImageGenerationDriver(
                    model="amazon.titan-image-generator-v1",
                    image_generation_model_driver=BedrockTitanImageGenerationModelDriver(),
                ),
                image_query_driver=AmazonBedrockImageQueryDriver(
                    model="anthropic.claude-3-sonnet-20240229-v1:0",
                    image_query_model_driver=BedrockClaudeImageQueryModelDriver(),
                ),
                embedding_driver=AmazonBedrockTitanEmbeddingDriver(model="amazon.titan-embed-text-v1"),
                vector_store_driver=LocalVectorStoreDriver(
                    embedding_driver=AmazonBedrockTitanEmbeddingDriver(model="amazon.titan-embed-text-v1")
                ),
            )
        ),
        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=self.global_drivers.vector_store_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=AmazonBedrockPromptDriver(model='anthropic.claude-3-sonnet-20240229-v1:0', stream=False, prompt_model_driver=BedrockClaudePromptModelDriver()), image_generation_driver=AmazonBedrockImageGenerationDriver(model='amazon.titan-image-generator-v1', image_generation_model_driver=BedrockTitanImageGenerationModelDriver()), image_query_driver=AmazonBedrockImageQueryDriver(model='anthropic.claude-3-sonnet-20240229-v1:0', image_query_model_driver=BedrockClaudeImageQueryModelDriver()), embedding_driver=AmazonBedrockTitanEmbeddingDriver(model='amazon.titan-embed-text-v1'), vector_store_driver=LocalVectorStoreDriver(embedding_driver=AmazonBedrockTitanEmbeddingDriver(model='amazon.titan-embed-text-v1')))), 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=self.global_drivers.vector_store_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