Skip to content

Commit b2f7e61

Browse files
ajroetkerajroetker
committed
feat: regenerate SDK with sparse bool and consolidated index types
1 parent b538e01 commit b2f7e61

17 files changed

Lines changed: 684 additions & 635 deletions

‎src/antfly/client_generated/api/index_management/create_index.py‎

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -5,18 +5,18 @@
55

66
from ... import errors
77
from ...client import AuthenticatedClient, Client
8-
from ...models.bleve_index_v2_config import BleveIndexV2Config
9-
from ...models.embedding_index_config import EmbeddingIndexConfig
8+
from ...models.embeddings_index_config import EmbeddingsIndexConfig
109
from ...models.error import Error
11-
from ...models.graph_index_v0_config import GraphIndexV0Config
10+
from ...models.full_text_index_config import FullTextIndexConfig
11+
from ...models.graph_index_config import GraphIndexConfig
1212
from ...types import Response
1313

1414

1515
def _get_kwargs(
1616
table_name: str,
1717
index_name: str,
1818
*,
19-
body: Union["BleveIndexV2Config", "EmbeddingIndexConfig", "GraphIndexV0Config"],
19+
body: Union["EmbeddingsIndexConfig", "FullTextIndexConfig", "GraphIndexConfig"],
2020
) -> dict[str, Any]:
2121
headers: dict[str, Any] = {}
2222

@@ -26,9 +26,9 @@ def _get_kwargs(
2626
}
2727

2828
_kwargs["json"]: dict[str, Any]
29-
if isinstance(body, BleveIndexV2Config):
29+
if isinstance(body, FullTextIndexConfig):
3030
_kwargs["json"] = body.to_dict()
31-
elif isinstance(body, EmbeddingIndexConfig):
31+
elif isinstance(body, EmbeddingsIndexConfig):
3232
_kwargs["json"] = body.to_dict()
3333
else:
3434
_kwargs["json"] = body.to_dict()
@@ -78,14 +78,14 @@ def sync_detailed(
7878
index_name: str,
7979
*,
8080
client: AuthenticatedClient,
81-
body: Union["BleveIndexV2Config", "EmbeddingIndexConfig", "GraphIndexV0Config"],
81+
body: Union["EmbeddingsIndexConfig", "FullTextIndexConfig", "GraphIndexConfig"],
8282
) -> Response[Union[Any, Error]]:
8383
"""Add an index to a table
8484
8585
Args:
8686
table_name (str):
8787
index_name (str):
88-
body (Union['BleveIndexV2Config', 'EmbeddingIndexConfig', 'GraphIndexV0Config']):
88+
body (Union['EmbeddingsIndexConfig', 'FullTextIndexConfig', 'GraphIndexConfig']):
8989
Configuration for an index
9090
9191
Raises:
@@ -114,14 +114,14 @@ def sync(
114114
index_name: str,
115115
*,
116116
client: AuthenticatedClient,
117-
body: Union["BleveIndexV2Config", "EmbeddingIndexConfig", "GraphIndexV0Config"],
117+
body: Union["EmbeddingsIndexConfig", "FullTextIndexConfig", "GraphIndexConfig"],
118118
) -> Optional[Union[Any, Error]]:
119119
"""Add an index to a table
120120
121121
Args:
122122
table_name (str):
123123
index_name (str):
124-
body (Union['BleveIndexV2Config', 'EmbeddingIndexConfig', 'GraphIndexV0Config']):
124+
body (Union['EmbeddingsIndexConfig', 'FullTextIndexConfig', 'GraphIndexConfig']):
125125
Configuration for an index
126126
127127
Raises:
@@ -145,14 +145,14 @@ async def asyncio_detailed(
145145
index_name: str,
146146
*,
147147
client: AuthenticatedClient,
148-
body: Union["BleveIndexV2Config", "EmbeddingIndexConfig", "GraphIndexV0Config"],
148+
body: Union["EmbeddingsIndexConfig", "FullTextIndexConfig", "GraphIndexConfig"],
149149
) -> Response[Union[Any, Error]]:
150150
"""Add an index to a table
151151
152152
Args:
153153
table_name (str):
154154
index_name (str):
155-
body (Union['BleveIndexV2Config', 'EmbeddingIndexConfig', 'GraphIndexV0Config']):
155+
body (Union['EmbeddingsIndexConfig', 'FullTextIndexConfig', 'GraphIndexConfig']):
156156
Configuration for an index
157157
158158
Raises:
@@ -179,14 +179,14 @@ async def asyncio(
179179
index_name: str,
180180
*,
181181
client: AuthenticatedClient,
182-
body: Union["BleveIndexV2Config", "EmbeddingIndexConfig", "GraphIndexV0Config"],
182+
body: Union["EmbeddingsIndexConfig", "FullTextIndexConfig", "GraphIndexConfig"],
183183
) -> Optional[Union[Any, Error]]:
184184
"""Add an index to a table
185185
186186
Args:
187187
table_name (str):
188188
index_name (str):
189-
body (Union['BleveIndexV2Config', 'EmbeddingIndexConfig', 'GraphIndexV0Config']):
189+
body (Union['EmbeddingsIndexConfig', 'FullTextIndexConfig', 'GraphIndexConfig']):
190190
Configuration for an index
191191
192192
Raises:

‎src/antfly/client_generated/models/__init__.py‎

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -28,8 +28,6 @@
2828
from .bedrock_generator_config import BedrockGeneratorConfig
2929
from .bing_search_config import BingSearchConfig
3030
from .bing_search_config_freshness import BingSearchConfigFreshness
31-
from .bleve_index_v2_config import BleveIndexV2Config
32-
from .bleve_index_v2_stats import BleveIndexV2Stats
3331
from .bool_field_query import BoolFieldQuery
3432
from .boolean_query import BooleanQuery
3533
from .brave_search_config import BraveSearchConfig
@@ -91,8 +89,8 @@
9189
from .edges_response import EdgesResponse
9290
from .embedder_config import EmbedderConfig
9391
from .embedder_provider import EmbedderProvider
94-
from .embedding_index_config import EmbeddingIndexConfig
95-
from .embedding_index_stats import EmbeddingIndexStats
92+
from .embeddings_index_config import EmbeddingsIndexConfig
93+
from .embeddings_index_stats import EmbeddingsIndexStats
9694
from .error import Error
9795
from .eval_config import EvalConfig
9896
from .eval_options import EvalOptions
@@ -116,6 +114,8 @@
116114
from .foreign_column import ForeignColumn
117115
from .foreign_source import ForeignSource
118116
from .foreign_source_type import ForeignSourceType
117+
from .full_text_index_config import FullTextIndexConfig
118+
from .full_text_index_stats import FullTextIndexStats
119119
from .fuzziness_type_1 import FuzzinessType1
120120
from .fuzzy_query import FuzzyQuery
121121
from .generation_step_config import GenerationStepConfig
@@ -131,9 +131,9 @@
131131
from .google_generator_config import GoogleGeneratorConfig
132132
from .google_search_config import GoogleSearchConfig
133133
from .google_search_config_search_type import GoogleSearchConfigSearchType
134-
from .graph_index_v0_config import GraphIndexV0Config
135-
from .graph_index_v0_stats import GraphIndexV0Stats
136-
from .graph_index_v0_stats_edge_types import GraphIndexV0StatsEdgeTypes
134+
from .graph_index_config import GraphIndexConfig
135+
from .graph_index_stats import GraphIndexStats
136+
from .graph_index_stats_edge_types import GraphIndexStatsEdgeTypes
137137
from .graph_node_selector import GraphNodeSelector
138138
from .graph_query import GraphQuery
139139
from .graph_query_params import GraphQueryParams
@@ -313,8 +313,6 @@
313313
"BedrockGeneratorConfig",
314314
"BingSearchConfig",
315315
"BingSearchConfigFreshness",
316-
"BleveIndexV2Config",
317-
"BleveIndexV2Stats",
318316
"BooleanQuery",
319317
"BoolFieldQuery",
320318
"BraveSearchConfig",
@@ -376,8 +374,8 @@
376374
"EdgeTypeConfigTopology",
377375
"EmbedderConfig",
378376
"EmbedderProvider",
379-
"EmbeddingIndexConfig",
380-
"EmbeddingIndexStats",
377+
"EmbeddingsIndexConfig",
378+
"EmbeddingsIndexStats",
381379
"Error",
382380
"EvalConfig",
383381
"EvalOptions",
@@ -401,6 +399,8 @@
401399
"ForeignColumn",
402400
"ForeignSource",
403401
"ForeignSourceType",
402+
"FullTextIndexConfig",
403+
"FullTextIndexStats",
404404
"FuzzinessType1",
405405
"FuzzyQuery",
406406
"GenerationStepConfig",
@@ -416,9 +416,9 @@
416416
"GoogleGeneratorConfig",
417417
"GoogleSearchConfig",
418418
"GoogleSearchConfigSearchType",
419-
"GraphIndexV0Config",
420-
"GraphIndexV0Stats",
421-
"GraphIndexV0StatsEdgeTypes",
419+
"GraphIndexConfig",
420+
"GraphIndexStats",
421+
"GraphIndexStatsEdgeTypes",
422422
"GraphNodeSelector",
423423
"GraphQuery",
424424
"GraphQueryParams",

‎src/antfly/client_generated/models/create_table_request_indexes.py‎

Lines changed: 17 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -5,9 +5,9 @@
55
from attrs import field as _attrs_field
66

77
if TYPE_CHECKING:
8-
from ..models.bleve_index_v2_config import BleveIndexV2Config
9-
from ..models.embedding_index_config import EmbeddingIndexConfig
10-
from ..models.graph_index_v0_config import GraphIndexV0Config
8+
from ..models.embeddings_index_config import EmbeddingsIndexConfig
9+
from ..models.full_text_index_config import FullTextIndexConfig
10+
from ..models.graph_index_config import GraphIndexConfig
1111

1212

1313
T = TypeVar("T", bound="CreateTableRequestIndexes")
@@ -28,19 +28,19 @@ class CreateTableRequestIndexes:
2828
2929
"""
3030

31-
additional_properties: dict[str, Union["BleveIndexV2Config", "EmbeddingIndexConfig", "GraphIndexV0Config"]] = (
31+
additional_properties: dict[str, Union["EmbeddingsIndexConfig", "FullTextIndexConfig", "GraphIndexConfig"]] = (
3232
_attrs_field(init=False, factory=dict)
3333
)
3434

3535
def to_dict(self) -> dict[str, Any]:
36-
from ..models.bleve_index_v2_config import BleveIndexV2Config
37-
from ..models.embedding_index_config import EmbeddingIndexConfig
36+
from ..models.embeddings_index_config import EmbeddingsIndexConfig
37+
from ..models.full_text_index_config import FullTextIndexConfig
3838

3939
field_dict: dict[str, Any] = {}
4040
for prop_name, prop in self.additional_properties.items():
41-
if isinstance(prop, BleveIndexV2Config):
41+
if isinstance(prop, FullTextIndexConfig):
4242
field_dict[prop_name] = prop.to_dict()
43-
elif isinstance(prop, EmbeddingIndexConfig):
43+
elif isinstance(prop, EmbeddingsIndexConfig):
4444
field_dict[prop_name] = prop.to_dict()
4545
else:
4646
field_dict[prop_name] = prop.to_dict()
@@ -49,9 +49,9 @@ def to_dict(self) -> dict[str, Any]:
4949

5050
@classmethod
5151
def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
52-
from ..models.bleve_index_v2_config import BleveIndexV2Config
53-
from ..models.embedding_index_config import EmbeddingIndexConfig
54-
from ..models.graph_index_v0_config import GraphIndexV0Config
52+
from ..models.embeddings_index_config import EmbeddingsIndexConfig
53+
from ..models.full_text_index_config import FullTextIndexConfig
54+
from ..models.graph_index_config import GraphIndexConfig
5555

5656
d = dict(src_dict)
5757
create_table_request_indexes = cls()
@@ -61,26 +61,26 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
6161

6262
def _parse_additional_property(
6363
data: object,
64-
) -> Union["BleveIndexV2Config", "EmbeddingIndexConfig", "GraphIndexV0Config"]:
64+
) -> Union["EmbeddingsIndexConfig", "FullTextIndexConfig", "GraphIndexConfig"]:
6565
try:
6666
if not isinstance(data, dict):
6767
raise TypeError()
68-
componentsschemas_index_config_type_0 = BleveIndexV2Config.from_dict(data)
68+
componentsschemas_index_config_type_0 = FullTextIndexConfig.from_dict(data)
6969

7070
return componentsschemas_index_config_type_0
7171
except: # noqa: E722
7272
pass
7373
try:
7474
if not isinstance(data, dict):
7575
raise TypeError()
76-
componentsschemas_index_config_type_1 = EmbeddingIndexConfig.from_dict(data)
76+
componentsschemas_index_config_type_1 = EmbeddingsIndexConfig.from_dict(data)
7777

7878
return componentsschemas_index_config_type_1
7979
except: # noqa: E722
8080
pass
8181
if not isinstance(data, dict):
8282
raise TypeError()
83-
componentsschemas_index_config_type_2 = GraphIndexV0Config.from_dict(data)
83+
componentsschemas_index_config_type_2 = GraphIndexConfig.from_dict(data)
8484

8585
return componentsschemas_index_config_type_2
8686

@@ -95,11 +95,11 @@ def _parse_additional_property(
9595
def additional_keys(self) -> list[str]:
9696
return list(self.additional_properties.keys())
9797

98-
def __getitem__(self, key: str) -> Union["BleveIndexV2Config", "EmbeddingIndexConfig", "GraphIndexV0Config"]:
98+
def __getitem__(self, key: str) -> Union["EmbeddingsIndexConfig", "FullTextIndexConfig", "GraphIndexConfig"]:
9999
return self.additional_properties[key]
100100

101101
def __setitem__(
102-
self, key: str, value: Union["BleveIndexV2Config", "EmbeddingIndexConfig", "GraphIndexV0Config"]
102+
self, key: str, value: Union["EmbeddingsIndexConfig", "FullTextIndexConfig", "GraphIndexConfig"]
103103
) -> None:
104104
self.additional_properties[key] = value
105105

0 commit comments

Comments
 (0)