Skip to content

Add materialization_embeddings option to materialization API #83

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
jfrench9 merged 1 commit into from
Apr 8, 2026
Merged

Add materialization_embeddings option to materialization API #83

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
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
2 changes: 2 additions & 0 deletions robosystems_client/extensions/materialization_client.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,7 @@ class MaterializationOptions:
ignore_errors: bool = True
rebuild: bool = False
force: bool = False
materialize_embeddings: bool = False
on_progress: Optional[Callable[[str], None]] = None
timeout: Optional[int] = 600 # 10 minute default timeout

Expand Down Expand Up @@ -112,6 +113,7 @@ def materialize(
ignore_errors=options.ignore_errors,
rebuild=options.rebuild,
force=options.force,
materialize_embeddings=options.materialize_embeddings,
)

from ..client import AuthenticatedClient
Expand Down
11 changes: 11 additions & 0 deletions robosystems_client/models/materialize_request.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -22,13 +22,17 @@ class MaterializeRequest:
source (None | str | Unset): Data source for materialization. Auto-detected from graph type if not specified.
'staged' materializes from uploaded files (generic graphs). 'extensions' materializes from the extensions OLTP
database (entity graphs).
materialize_embeddings (bool | Unset): Include embedding columns in materialization and build HNSW vector
indexes in the graph database. When false (default), embedding columns are NULLed out to save space. Set to true
for graphs that need vector search. 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

def to_dict(self) -> dict[str, Any]:
force = self.force
Expand All @@ -45,6 +49,8 @@ def to_dict(self) -> dict[str, Any]:
else:
source = self.source

materialize_embeddings = self.materialize_embeddings

field_dict: dict[str, Any] = {}

field_dict.update({})
Expand All @@ -58,6 +64,8 @@ def to_dict(self) -> dict[str, Any]:
field_dict["dry_run"] = dry_run
if source is not UNSET:
field_dict["source"] = source
if materialize_embeddings is not UNSET:
field_dict["materialize_embeddings"] = materialize_embeddings

return field_dict

Expand All @@ -81,12 +89,15 @@ def _parse_source(data: object) -> None | str | Unset:

source = _parse_source(d.pop("source", UNSET))

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

materialize_request = cls(
force=force,
rebuild=rebuild,
ignore_errors=ignore_errors,
dry_run=dry_run,
source=source,
materialize_embeddings=materialize_embeddings,
)

return materialize_request
Loading