Skip to content

Cohere structure config

CohereStructureConfig

Bases: StructureConfig

Source code in griptape/config/cohere_structure_config.py 
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
@define
class CohereStructureConfig(StructureConfig):
    api_key: str = field(metadata={"serializable": False}, kw_only=True)

    prompt_driver: BasePromptDriver = field(
        default=Factory(lambda self: CoherePromptDriver(model="command-r", api_key=self.api_key), takes_self=True),
        metadata={"serializable": True},
        kw_only=True,
    )
    embedding_driver: BaseEmbeddingDriver = field(
        default=Factory(
            lambda self: CohereEmbeddingDriver(
                model="embed-english-v3.0", api_key=self.api_key, input_type="search_document"
            ),
            takes_self=True,
        ),
        metadata={"serializable": True},
        kw_only=True,
    )
    vector_store_driver: BaseVectorStoreDriver = field(
        default=Factory(lambda self: LocalVectorStoreDriver(embedding_driver=self.embedding_driver), takes_self=True),
        kw_only=True,
        metadata={"serializable": True},
    )

api_key: str = field(metadata={'serializable': False}, kw_only=True) class-attribute instance-attribute

embedding_driver: BaseEmbeddingDriver = field(default=Factory(lambda self: CohereEmbeddingDriver(model='embed-english-v3.0', api_key=self.api_key, input_type='search_document'), takes_self=True), metadata={'serializable': True}, kw_only=True) class-attribute instance-attribute

prompt_driver: BasePromptDriver = field(default=Factory(lambda self: CoherePromptDriver(model='command-r', api_key=self.api_key), takes_self=True), metadata={'serializable': True}, kw_only=True) class-attribute instance-attribute

vector_store_driver: BaseVectorStoreDriver = field(default=Factory(lambda self: LocalVectorStoreDriver(embedding_driver=self.embedding_driver), takes_self=True), kw_only=True, metadata={'serializable': True}) class-attribute instance-attribute