Skip to content

Openai structure config

OpenAiStructureConfig

Bases: StructureConfig

Source code in griptape/config/openai_structure_config.py 
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
49
50
51
52
53
54
@define
class OpenAiStructureConfig(StructureConfig):
    prompt_driver: BasePromptDriver = field(
        default=Factory(lambda: OpenAiChatPromptDriver(model="gpt-4o")), metadata={"serializable": True}, kw_only=True
    )
    image_generation_driver: BaseImageGenerationDriver = field(
        default=Factory(lambda: OpenAiImageGenerationDriver(model="dall-e-2", image_size="512x512")),
        kw_only=True,
        metadata={"serializable": True},
    )
    image_query_driver: BaseImageQueryDriver = field(
        default=Factory(lambda: OpenAiImageQueryDriver(model="gpt-4o")), kw_only=True, metadata={"serializable": True}
    )
    embedding_driver: BaseEmbeddingDriver = field(
        default=Factory(lambda: OpenAiEmbeddingDriver(model="text-embedding-3-small")),
        metadata={"serializable": True},
        kw_only=True,
    )
    vector_store_driver: BaseVectorStoreDriver = field(
        default=Factory(
            lambda: LocalVectorStoreDriver(embedding_driver=OpenAiEmbeddingDriver(model="text-embedding-3-small"))
        ),
        kw_only=True,
        metadata={"serializable": True},
    )
    text_to_speech_driver: BaseTextToSpeechDriver = field(
        default=Factory(lambda: OpenAiTextToSpeechDriver(model="tts")), kw_only=True, metadata={"serializable": True}
    )
    audio_transcription_driver: BaseAudioTranscriptionDriver = field(
        default=Factory(lambda: OpenAiAudioTranscriptionDriver(model="whisper-1")),
        kw_only=True,
        metadata={"serializable": True},
    )

audio_transcription_driver: BaseAudioTranscriptionDriver = field(default=Factory(lambda: OpenAiAudioTranscriptionDriver(model='whisper-1')), kw_only=True, metadata={'serializable': True}) class-attribute instance-attribute

embedding_driver: BaseEmbeddingDriver = field(default=Factory(lambda: OpenAiEmbeddingDriver(model='text-embedding-3-small')), metadata={'serializable': True}, kw_only=True) class-attribute instance-attribute

image_generation_driver: BaseImageGenerationDriver = field(default=Factory(lambda: OpenAiImageGenerationDriver(model='dall-e-2', image_size='512x512')), kw_only=True, metadata={'serializable': True}) class-attribute instance-attribute

image_query_driver: BaseImageQueryDriver = field(default=Factory(lambda: OpenAiImageQueryDriver(model='gpt-4o')), kw_only=True, metadata={'serializable': True}) class-attribute instance-attribute

prompt_driver: BasePromptDriver = field(default=Factory(lambda: OpenAiChatPromptDriver(model='gpt-4o')), metadata={'serializable': True}, kw_only=True) class-attribute instance-attribute

text_to_speech_driver: BaseTextToSpeechDriver = field(default=Factory(lambda: OpenAiTextToSpeechDriver(model='tts')), kw_only=True, metadata={'serializable': True}) class-attribute instance-attribute

vector_store_driver: BaseVectorStoreDriver = field(default=Factory(lambda: LocalVectorStoreDriver(embedding_driver=OpenAiEmbeddingDriver(model='text-embedding-3-small'))), kw_only=True, metadata={'serializable': True}) class-attribute instance-attribute