Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 1 addition & 3 deletions robosystems_client/clients/graph_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,6 @@ class GraphInfo:
class MaterializationOptions:
"""Options for graph materialization operations"""

ignore_errors: bool = True
rebuild: bool = False
force: bool = False
materialize_embeddings: bool = False
Expand Down Expand Up @@ -317,7 +316,7 @@ def materialize(

Args:
graph_id: Graph database identifier
options: Materialization options (ignore_errors, rebuild, force, timeout)
options: Materialization options (rebuild, force, timeout)

Returns:
MaterializationResult with detailed execution information
Expand All @@ -340,7 +339,6 @@ def materialize(
graph_id=graph_id,
client=client,
body=MaterializeOp(
ignore_errors=options.ignore_errors,
rebuild=options.rebuild,
force=options.force,
materialize_embeddings=options.materialize_embeddings,
Expand Down
9 changes: 0 additions & 9 deletions robosystems_client/models/materialize_op.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,15 +18,13 @@ class MaterializeOp:
Attributes:
force (bool | Unset): Force materialization even if already up to date Default: False.
rebuild (bool | Unset): Rebuild the graph from scratch, dropping existing data Default: False.
ignore_errors (bool | Unset): Continue past non-fatal row errors Default: True.
dry_run (bool | Unset): Validate tables without writing to the graph Default: False.
source (None | str | Unset): Materialization source: 'extensions' for OLTP, omit for DuckDB staging tables
materialize_embeddings (bool | Unset): Generate vector embeddings during materialization Default: False.
"""

force: bool | Unset = False
rebuild: bool | Unset = False
ignore_errors: bool | Unset = True
dry_run: bool | Unset = False
source: None | str | Unset = UNSET
materialize_embeddings: bool | Unset = False
Expand All @@ -37,8 +35,6 @@ def to_dict(self) -> dict[str, Any]:

rebuild = self.rebuild

ignore_errors = self.ignore_errors

dry_run = self.dry_run

source: None | str | Unset
Expand All @@ -56,8 +52,6 @@ def to_dict(self) -> dict[str, Any]:
field_dict["force"] = force
if rebuild is not UNSET:
field_dict["rebuild"] = rebuild
if ignore_errors is not UNSET:
field_dict["ignore_errors"] = ignore_errors
if dry_run is not UNSET:
field_dict["dry_run"] = dry_run
if source is not UNSET:
Expand All @@ -74,8 +68,6 @@ def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:

rebuild = d.pop("rebuild", UNSET)

ignore_errors = d.pop("ignore_errors", UNSET)

dry_run = d.pop("dry_run", UNSET)

def _parse_source(data: object) -> None | str | Unset:
Expand All @@ -92,7 +84,6 @@ def _parse_source(data: object) -> None | str | Unset:
materialize_op = cls(
force=force,
rebuild=rebuild,
ignore_errors=ignore_errors,
dry_run=dry_run,
source=source,
materialize_embeddings=materialize_embeddings,
Expand Down
3 changes: 0 additions & 3 deletions tests/test_materialization_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,6 @@ def test_materialization_options_defaults(self):
"""Test MaterializationOptions default values."""
options = MaterializationOptions()

assert options.ignore_errors is True
assert options.rebuild is False
assert options.force is False
assert options.materialize_embeddings is False
Expand All @@ -32,15 +31,13 @@ def test_materialization_options_custom(self):
"""Test MaterializationOptions with custom values."""
progress_fn = Mock()
options = MaterializationOptions(
ignore_errors=False,
rebuild=True,
force=True,
materialize_embeddings=True,
on_progress=progress_fn,
timeout=300,
)

assert options.ignore_errors is False
assert options.rebuild is True
assert options.force is True
assert options.materialize_embeddings is True
Expand Down
Loading