Add missing kw_only

griptape-ai · May 17, 2024 · fc427c7 · fc427c7
1 parent c1e5147
commit fc427c7
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 6 deletions.
diff --git a/griptape/config/anthropic_structure_config.py b/griptape/config/anthropic_structure_config.py
@@ -12,18 +12,24 @@
 @define
 class AnthropicStructureConfig(StructureConfig):
     prompt_driver: AnthropicPromptDriver = field(
-        default=Factory(lambda: AnthropicPromptDriver(model="claude-3-opus-20240229")), metadata={"serializable": True}
+        default=Factory(lambda: AnthropicPromptDriver(model="claude-3-opus-20240229")),
+        metadata={"serializable": True},
+        kw_only=True,
     )
     embedding_driver: VoyageAiEmbeddingDriver = field(
-        default=Factory(lambda: VoyageAiEmbeddingDriver(model="voyage-large-2")), metadata={"serializable": True}
+        default=Factory(lambda: VoyageAiEmbeddingDriver(model="voyage-large-2")),
+        metadata={"serializable": True},
+        kw_only=True,
     )
     vector_store_driver: LocalVectorStoreDriver = field(
         default=Factory(
             lambda: LocalVectorStoreDriver(embedding_driver=VoyageAiEmbeddingDriver(model="voyage-large-2"))
         ),
+        kw_only=True,
         metadata={"serializable": True},
     )
     image_query_driver: AnthropicImageQueryDriver = field(
         default=Factory(lambda: AnthropicImageQueryDriver(model="claude-3-opus-20240229")),
+        kw_only=True,
         metadata={"serializable": True},
     )
diff --git a/griptape/config/google_structure_config.py b/griptape/config/google_structure_config.py
@@ -7,14 +7,17 @@
 @define
 class GoogleStructureConfig(StructureConfig):
     prompt_driver: GooglePromptDriver = field(
-        default=Factory(lambda: GooglePromptDriver(model="gemini-pro")), metadata={"serializable": True}
+        default=Factory(lambda: GooglePromptDriver(model="gemini-pro")), kw_only=True, metadata={"serializable": True}
     )
     embedding_driver: GoogleEmbeddingDriver = field(
-        default=Factory(lambda: GoogleEmbeddingDriver(model="models/embedding-001")), metadata={"serializable": True}
+        default=Factory(lambda: GoogleEmbeddingDriver(model="models/embedding-001")),
+        kw_only=True,
+        metadata={"serializable": True},
     )
     vector_store_driver: LocalVectorStoreDriver = field(
         default=Factory(
             lambda: LocalVectorStoreDriver(embedding_driver=GoogleEmbeddingDriver(model="models/embedding-001"))
         ),
+        kw_only=True,
         metadata={"serializable": True},
     )
diff --git a/griptape/config/openai_structure_config.py b/griptape/config/openai_structure_config.py
@@ -13,22 +13,27 @@
 @define
 class OpenAiStructureConfig(StructureConfig):
     prompt_driver: OpenAiChatPromptDriver = field(
-        default=Factory(lambda: OpenAiChatPromptDriver(model="gpt-4o")), metadata={"serializable": True}
+        default=Factory(lambda: OpenAiChatPromptDriver(model="gpt-4o")), metadata={"serializable": True}, kw_only=True
     )
     image_generation_driver: OpenAiImageGenerationDriver = field(
         default=Factory(lambda: OpenAiImageGenerationDriver(model="dall-e-2", image_size="512x512")),
+        kw_only=True,
         metadata={"serializable": True},
     )
     image_query_driver: OpenAiVisionImageQueryDriver = field(
         default=Factory(lambda: OpenAiVisionImageQueryDriver(model="gpt-4-vision-preview")),
+        kw_only=True,
         metadata={"serializable": True},
     )
     embedding_driver: OpenAiEmbeddingDriver = field(
-        default=Factory(lambda: OpenAiEmbeddingDriver(model="text-embedding-3-small")), metadata={"serializable": True}
+        default=Factory(lambda: OpenAiEmbeddingDriver(model="text-embedding-3-small")),
+        metadata={"serializable": True},
+        kw_only=True,
     )
     vector_store_driver: LocalVectorStoreDriver = field(
         default=Factory(
             lambda: LocalVectorStoreDriver(embedding_driver=OpenAiEmbeddingDriver(model="text-embedding-3-small"))
         ),
+        kw_only=True,
         metadata={"serializable": True},
     )