[Bugfix] fix config validation for google llm config (#1088)

Co-authored-by: Deven Patel <deven298@yahoo.com>
2023-12-30 16:25:41 +05:30
parent a54dde0509
commit a304ded500
2 changed files with 9 additions and 2 deletions
--- a/embedchain/utils.py
+++ b/embedchain/utils.py
@@ -420,6 +420,8 @@ def validate_config(config_data):
                    Optional("model"): Optional(str),
                    Optional("deployment_name"): Optional(str),
                    Optional("api_key"): str,
+                    Optional("title"): str,
+                    Optional("task_type"): str,
                },
            },
            Optional("embedding_model"): {
@@ -428,6 +430,8 @@ def validate_config(config_data):
                    Optional("model"): str,
                    Optional("deployment_name"): str,
                    Optional("api_key"): str,
+                    Optional("title"): str,
+                    Optional("task_type"): str,
                },
            },
            Optional("chunker"): {