fix: elasticsearch form schema

ToucanToco · Mar 14, 2024 · d277250 · d277250
1 parent 122108d
commit d277250
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -2,6 +2,10 @@
 
 ## Unreleased
 
+### Fixed
+
+- Elasticsearch: force `widget="json"` on `body` so the form is properly filled when updating a data source
+
 ## [6.1.0] 2024-03-13
 
 ### Changed

diff --git a/toucan_connectors/elasticsearch/elasticsearch_connector.py b/toucan_connectors/elasticsearch/elasticsearch_connector.py
@@ -1,6 +1,6 @@
 from copy import deepcopy
 from enum import Enum
-from typing import List, Union
+from typing import List
 from urllib.parse import urlparse
 
 import pandas as pd
@@ -112,9 +112,9 @@ class SearchMethod(str, Enum):
 
 
 class ElasticsearchDataSource(ToucanDataSource):
-    search_method: SearchMethod
-    index: str = None
-    body: Union[dict, list]
+    search_method: SearchMethod = Field(SearchMethod.search, title="Search method")
+    index: str | None = Field(None, title="Index")
+    body: dict | list = Field({}, description="Body of elasticsearch query", widget="json")
 
 
 class ElasticsearchConnector(ToucanConnector, data_source_model=ElasticsearchDataSource):