relationalai 0.12.3__tar.gz → 0.12.6__tar.gz

{relationalai-0.12.3 → relationalai-0.12.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: relationalai
-Version: 0.12.3
+Version: 0.12.6
 Summary: RelationalAI Library and CLI
 Author-email: RelationalAI <support@relational.ai>
 License-File: LICENSE

{relationalai-0.12.3 → relationalai-0.12.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = 'relationalai'
-version = '0.12.3'
+version = '0.12.6'
 description = 'RelationalAI Library and CLI'
 readme="docs/pypi/README.md"
 authors = [
@@ -94,7 +94,6 @@ allow-direct-references = true
 only-include = ["src"]
 [tool.hatch.build.targets.sdist.force-include]
 "frontend/debugger/dist" = "frontend/debugger/dist"
-"src/relationalai/experimental/paths/pathfinder.rel" = "src/relationalai/experimental/paths/pathfinder.rel"
 
 [tool.hatch.build.targets.wheel]
 packages = ["src/relationalai", "src/relationalai_test_util"]

{relationalai-0.12.3 → relationalai-0.12.6}/src/relationalai/__init__.py

@@ -1,3 +1,7 @@
+"""
+The RelationalAI Python SDK.
+"""
+
 from __future__ import annotations
 import importlib.metadata
 

{relationalai-0.12.3 → relationalai-0.12.6}/src/relationalai/clients/snowflake.py

@@ -83,6 +83,7 @@ FIELD_MAP = {
 VALID_IMPORT_STATES = ["PENDING", "PROCESSING", "QUARANTINED", "LOADED"]
 ENGINE_ERRORS = ["engine is suspended", "create/resume", "engine not found", "no engines found", "engine was deleted"]
 ENGINE_NOT_READY_MSGS = ["engine is in pending", "engine is provisioning"]
+DATABASE_ERRORS = ["database not found"]
 PYREL_ROOT_DB = 'pyrel_root_db'
 
 TERMINAL_TXN_STATES = ["COMPLETED", "ABORTED"]
@@ -281,6 +282,8 @@ def _sanitize_user_name(user: str) -> str:
 def _is_engine_issue(response_message: str) -> bool:
     return any(kw in response_message.lower() for kw in ENGINE_ERRORS + ENGINE_NOT_READY_MSGS)
 
+def _is_database_issue(response_message: str) -> bool:
+    return any(kw in response_message.lower() for kw in DATABASE_ERRORS)
 
 
 #--------------------------------------------------
@@ -298,6 +301,7 @@ class Resources(ResourcesBase):
         dry_run: bool = False,
         reset_session: bool = False,
         generation: Generation | None = None,
+        language: str = "rel",
     ):
         super().__init__(profile, config=config)
         self._token_handler: TokenHandler | None = None
@@ -315,6 +319,8 @@ class Resources(ResourcesBase):
         # self.sources contains fully qualified Snowflake table/view names
         self.sources: set[str] = set()
         self._sproc_models = None
+        self.database = ""
+        self.language = language
         atexit.register(self.cancel_pending_transactions)
 
     @property
@@ -452,6 +458,7 @@ class Resources(ResourcesBase):
             rai_app = self.config.get("rai_app_name", "")
             current_role = self.config.get("role")
             engine = self.get_default_engine_name()
+            engine_size = self.config.get_default_engine_size()
             assert isinstance(rai_app, str), f"rai_app_name must be a string, not {type(rai_app)}"
             assert isinstance(engine, str), f"engine must be a string, not {type(engine)}"
             print("\n")
@@ -460,9 +467,15 @@ class Resources(ResourcesBase):
             if re.search(f"database '{rai_app}' does not exist or not authorized.".lower(), orig_message):
                 exception = SnowflakeAppMissingException(rai_app, current_role)
                 raise exception from None
-            if any(keyword in orig_message for keyword in ENGINE_ERRORS):
+            if _is_engine_issue(orig_message) or _is_database_issue(orig_message):
                 try:
-                    self.auto_create_engine(engine)
+                    self._poll_use_index(
+                        app_name=self.get_app_name(),
+                        sources=self.sources,
+                        model=self.database,
+                        engine_name=engine,
+                        engine_size=engine_size
+                    )
                     return self._exec(code, params, raw=raw, help=help)
                 except EngineNameValidationException as e:
                     raise EngineNameValidationException(engine) from e
@@ -767,7 +780,7 @@ Otherwise, remove it from your '{profile}' configuration profile.
             keep_database = not force and self.config.get("reuse_model", True)
             with debugging.span("release_index", name=name, keep_database=keep_database, language=language):
                 #TODO add headers to release_index
-                response = self._exec(f"call {APP_NAME}.api.release_index('{name}', OBJECT_CONSTRUCT('keep_database', {keep_database}, 'language', '{language}'));")
+                response = self._exec(f"call {APP_NAME}.api.release_index('{name}', OBJECT_CONSTRUCT('keep_database', {keep_database}, 'language', '{language}', 'user_agent', '{get_pyrel_version(self.generation)}'));")
                 if response:
                     result = next(iter(response))
                     obj = json.loads(result["RELEASE_INDEX"])
@@ -788,14 +801,13 @@ Otherwise, remove it from your '{profile}' configuration profile.
             headers = debugging.gen_current_propagation_headers()
             self._exec(f"call {APP_NAME}.api.clone_database('{target_name}', '{source_name}', {nowait_durable}, {headers});")
 
-    def poll_use_index(
+    def _poll_use_index(
         self,
         app_name: str,
         sources: Iterable[str],
         model: str,
         engine_name: str,
         engine_size: str | None = None,
-        language: str = "rel",
         program_span_id: str | None = None,
         headers: Dict | None = None,
     ):
@@ -806,12 +818,42 @@ Otherwise, remove it from your '{profile}' configuration profile.
             model,
             engine_name,
             engine_size,
-            language,
+            self.language,
             program_span_id,
             headers,
             self.generation
         ).poll()
 
+    def maybe_poll_use_index(
+        self,
+        app_name: str,
+        sources: Iterable[str],
+        model: str,
+        engine_name: str,
+        engine_size: str | None = None,
+        program_span_id: str | None = None,
+        headers: Dict | None = None,
+    ):
+        """Only call poll() if there are sources to process and cache is not valid."""
+        sources_list = list(sources)
+        self.database = model
+        if sources_list:
+            poller = UseIndexPoller(
+                self,
+                app_name,
+                sources_list,
+                model,
+                engine_name,
+                engine_size,
+                self.language,
+                program_span_id,
+                headers,
+                self.generation
+            )
+            # If cache is valid (data freshness has not expired), skip polling
+            if not poller.cache.is_valid():
+                return poller.poll()
+
     #--------------------------------------------------
     # Models
     #--------------------------------------------------
@@ -1868,9 +1910,19 @@ Otherwise, remove it from your '{profile}' configuration profile.
             )
         except Exception as e:
             err_message = str(e).lower()
-            if _is_engine_issue(err_message):
-                self.auto_create_engine(engine, headers=headers)
-                self._exec_async_v2(
+            if _is_engine_issue(err_message) or _is_database_issue(err_message):
+                engine_name = engine or self.get_default_engine_name()
+                engine_size = self.config.get_default_engine_size()
+                self._poll_use_index(
+                    app_name=self.get_app_name(),
+                    sources=self.sources,
+                    model=database,
+                    engine_name=engine_name,
+                    engine_size=engine_size,
+                    headers=headers,
+                )
+
+                return self._exec_async_v2(
                     database, engine, raw_code_b64, inputs, readonly, nowait_durable,
                     headers=headers, bypass_index=bypass_index, language='lqp',
                     query_timeout_mins=query_timeout_mins,
@@ -1908,8 +1960,17 @@ Otherwise, remove it from your '{profile}' configuration profile.
             )
         except Exception as e:
             err_message = str(e).lower()
-            if _is_engine_issue(err_message):
-                self.auto_create_engine(engine, headers=headers)
+            if _is_engine_issue(err_message) or _is_database_issue(err_message):
+                engine_name = engine or self.get_default_engine_name()
+                engine_size = self.config.get_default_engine_size()
+                self._poll_use_index(
+                    app_name=self.get_app_name(),
+                    sources=self.sources,
+                    model=database,
+                    engine_name=engine_name,
+                    engine_size=engine_size,
+                    headers=headers,
+                )
                 return self._exec_async_v2(
                     database,
                     engine,
@@ -2972,13 +3033,12 @@ class SnowflakeClient(Client):
 
         query_attrs_dict = json.loads(headers.get("X-Query-Attributes", "{}")) if headers else {}
         with debugging.span("poll_use_index", sources=self.resources.sources, model=model, engine=engine_name, **query_attrs_dict):
-            self.poll_use_index(
+            self.maybe_poll_use_index(
                 app_name=app_name,
                 sources=self.resources.sources,
                 model=model,
                 engine_name=engine_name,
                 engine_size=engine_size,
-                language="rel",
                 program_span_id=program_span_id,
                 headers=headers
             )
@@ -2989,29 +3049,24 @@ class SnowflakeClient(Client):
         if isolated and not self.keep_model:
             atexit.register(self.delete_database)
 
-    # Polling for use_index
-    # if data is ready, break the loop
-    # if data is not ready, print the status of the tables or engines
-    # if data is not ready and there are errors, collect the errors and raise exceptions
-    def poll_use_index(
+    def maybe_poll_use_index(
         self,
         app_name: str,
         sources: Iterable[str],
         model: str,
         engine_name: str,
         engine_size: str | None = None,
-        language: str = "rel",
         program_span_id: str | None = None,
         headers: Dict | None = None,
     ):
+        """Only call _poll_use_index if there are sources to process."""
         assert isinstance(self.resources, Resources)
-        return self.resources.poll_use_index(
+        return self.resources.maybe_poll_use_index(
             app_name=app_name,
             sources=sources,
             model=model,
             engine_name=engine_name,
             engine_size=engine_size,
-            language=language,
             program_span_id=program_span_id,
             headers=headers
         )
@@ -3136,6 +3191,7 @@ class DirectAccessResources(Resources):
         dry_run: bool = False,
         reset_session: bool = False,
         generation: Optional[Generation] = None,
+        language: str = "rel",
     ):
         super().__init__(
             generation=generation,
@@ -3144,11 +3200,13 @@ class DirectAccessResources(Resources):
             connection=connection,
             reset_session=reset_session,
             dry_run=dry_run,
+            language=language,
         )
         self._endpoint_info = ConfigStore(ENDPOINT_FILE)
         self._service_endpoint = ""
         self._direct_access_client = None
         self.generation = generation
+        self.database = ""
 
     @property
     def service_endpoint(self) -> str:
@@ -3226,9 +3284,18 @@ class DirectAccessResources(Resources):
 
                     # fix engine on engine error and retry
                     # Skip auto-retry if skip_auto_create is True to avoid recursion
-                    if _is_engine_issue(message) and not skip_auto_create:
-                        engine = payload.get("engine_name", "") if payload else ""
-                        self.auto_create_engine(engine)
+                    if (_is_engine_issue(message) and not skip_auto_create) or _is_database_issue(message):
+                        engine_name = payload.get("caller_engine_name", "") if payload else ""
+                        engine_name = engine_name or self.get_default_engine_name()
+                        engine_size = self.config.get_default_engine_size()
+                        self._poll_use_index(
+                            app_name=self.get_app_name(),
+                            sources=self.sources,
+                            model=self.database,
+                            engine_name=engine_name,
+                            engine_size=engine_size,
+                            headers=headers,
+                        )
                         response = _send_request()
             except requests.exceptions.ConnectionError as e:
                 if "NameResolutionError" in str(e):
@@ -3356,14 +3423,13 @@ class DirectAccessResources(Resources):
 
             return response.json()
 
-    def poll_use_index(
+    def _poll_use_index(
         self,
         app_name: str,
         sources: Iterable[str],
         model: str,
         engine_name: str,
         engine_size: str | None = None,
-        language: str = "rel",
         program_span_id: str | None = None,
         headers: Dict | None = None,
     ):
@@ -3374,12 +3440,42 @@ class DirectAccessResources(Resources):
             model=model,
             engine_name=engine_name,
             engine_size=engine_size,
-            language=language,
+            language=self.language,
             program_span_id=program_span_id,
             headers=headers,
             generation=self.generation,
         ).poll()
 
+    def maybe_poll_use_index(
+        self,
+        app_name: str,
+        sources: Iterable[str],
+        model: str,
+        engine_name: str,
+        engine_size: str | None = None,
+        program_span_id: str | None = None,
+        headers: Dict | None = None,
+    ):
+        """Only call poll() if there are sources to process and cache is not valid."""
+        sources_list = list(sources)
+        self.database = model
+        if sources_list:
+            poller = DirectUseIndexPoller(
+                self,
+                app_name=app_name,
+                sources=sources_list,
+                model=model,
+                engine_name=engine_name,
+                engine_size=engine_size,
+                language=self.language,
+                program_span_id=program_span_id,
+                headers=headers,
+                generation=self.generation,
+            )
+            # If cache is valid (data freshness has not expired), skip polling
+            if not poller.cache.is_valid():
+                return poller.poll()
+
     def _check_exec_async_status(self, txn_id: str, headers: Dict[str, str] | None = None) -> bool:
         """Check whether the given transaction has completed."""
 
@@ -3522,7 +3618,12 @@ class DirectAccessResources(Resources):
             with debugging.span("release_index", name=name, keep_database=keep_database, language=language):
                 response = self.request(
                     "release_index",
-                    payload={"model_name": name, "keep_database": keep_database, "language": language},
+                    payload={
+                        "model_name": name,
+                        "keep_database": keep_database,
+                        "language": language,
+                        "user_agent": get_pyrel_version(self.generation),
+                    },
                     headers=prop_hdrs,
                 )
                 if (

{relationalai-0.12.3 → relationalai-0.12.6}/src/relationalai/clients/use_index_poller.py

@@ -527,6 +527,9 @@ class UseIndexPoller:
                 else:
                     self.should_check_cdc = False
 
+                if engines and self.init_engine_async:
+                    self.init_engine_async = False
+
                 break_loop = False
                 has_stream_errors = False
                 has_general_errors = False

{relationalai-0.12.3/src/relationalai/semantics/reasoners/graph → relationalai-0.12.6/src/relationalai/experimental}/paths/README.md

@@ -12,8 +12,8 @@ The following is a minimal working example:
 from relationalai.semantics import Model, Integer, select, String, where, define
 from relationalai.semantics.std import strings
 
-from relationalai.semantics.reasoners.graph.paths.graph import Graph
-from relationalai.semantics.reasoners.graph.paths.api import node, edge, path, star, match
+from relationalai.experimental.paths.graph import Graph
+from relationalai.experimental.paths.api import node, edge, path, star, match
 
 model = Model("my_paths")
 

{relationalai-0.12.3/src/relationalai/semantics/reasoners/graph → relationalai-0.12.6/src/relationalai/experimental}/paths/__init__.py

@@ -12,5 +12,3 @@ __all__ = [
     'union',
     'match',
 ]
-
-

{relationalai-0.12.3/src/relationalai/semantics/reasoners/graph → relationalai-0.12.6/src/relationalai/experimental}/paths/examples/basic_example.py

@@ -1,6 +1,6 @@
 from relationalai.semantics import Model, Integer, select, String, define
-from relationalai.semantics.reasoners.graph.paths.graph import Graph
-from relationalai.semantics.reasoners.graph.paths.api import path, star, match
+from relationalai.experimental.paths.graph import Graph
+from relationalai.experimental.paths.api import path, star, match
 
 # Create an automaton that matches paths with pattern: A(A*)B:
 pattern = path("A", star("A"), "B")

{relationalai-0.12.3/src/relationalai/semantics/reasoners/graph → relationalai-0.12.6/src/relationalai/experimental}/paths/examples/paths_benchmark.py

@@ -4,8 +4,8 @@ Example usage of automaton-based pathfinding.
 
 import time
 from relationalai.semantics import Model, Integer, select, String, define
-from relationalai.semantics.reasoners.graph.paths.graph import Graph
-from relationalai.semantics.reasoners.graph.paths.api import node, path, match
+from relationalai.experimental.paths.graph import Graph
+from relationalai.experimental.paths.api import node, path, match
 
 import argparse
 

{relationalai-0.12.3/src/relationalai/semantics/reasoners/graph → relationalai-0.12.6/src/relationalai/experimental}/paths/examples/paths_example.py

@@ -4,8 +4,8 @@ Example usage of automaton-based pathfinding.
 
 import time
 from relationalai.semantics import Model, Integer, select, String, define
-from relationalai.semantics.reasoners.graph.paths.graph import Graph
-from relationalai.semantics.reasoners.graph.paths.api import node, path, match
+from relationalai.experimental.paths.graph import Graph
+from relationalai.experimental.paths.api import node, path, match
 
 import argparse
 

{relationalai-0.12.3/src/relationalai/semantics/reasoners/graph → relationalai-0.12.6/src/relationalai/experimental}/paths/examples/pattern_to_automaton.py

@@ -1,4 +1,4 @@
-from relationalai.semantics.reasoners.graph.paths.api import path, union
+from relationalai.experimental.paths.api import path, union
 from relationalai.semantics import Concept
 
 Node = Concept('Node')

{relationalai-0.12.3/src/relationalai/semantics/reasoners/graph → relationalai-0.12.6/src/relationalai/experimental}/paths/path_algorithms/one_sided_ball_repetition.py

@@ -1,6 +1,6 @@
 # For builder components.
 from relationalai.semantics import Integer, define
-from relationalai.semantics.reasoners.graph.paths.graph import Graph
+from relationalai.experimental.paths.graph import Graph
 
 
 def ball_with_repetition(g:Graph, Source, max_length):

{relationalai-0.12.3/src/relationalai/semantics/reasoners/graph → relationalai-0.12.6/src/relationalai/experimental}/paths/path_algorithms/single.py

@@ -1,8 +1,8 @@
 # For builder components.
 from relationalai.semantics import Integer, where, define, select, min
-from relationalai.semantics.reasoners.graph.paths.graph import Graph
-from relationalai.semantics.reasoners.graph.paths.path_algorithms.one_sided_ball_repetition import ball_with_repetition
-from relationalai.semantics.reasoners.graph.paths.path_algorithms.one_sided_ball_upto import ball_upto
+from relationalai.experimental.paths.graph import Graph
+from relationalai.experimental.paths.path_algorithms.one_sided_ball_repetition import ball_with_repetition
+from relationalai.experimental.paths.path_algorithms.one_sided_ball_upto import ball_upto
 
 # find a (deterministic) path from src to dst inside the given ball
 # where dst is at distance radius from src

{relationalai-0.12.3/src/relationalai/semantics/reasoners/graph → relationalai-0.12.6/src/relationalai/experimental}/paths/path_algorithms/two_sided_balls_repetition.py

@@ -1,6 +1,6 @@
 # For builder components.
 from relationalai.semantics import Integer, define
-from relationalai.semantics.reasoners.graph.paths.graph import Graph
+from relationalai.experimental.paths.graph import Graph
 
 
 def two_balls_repetition(g:Graph, Source, Target, max_length):

{relationalai-0.12.3/src/relationalai/semantics/reasoners/graph → relationalai-0.12.6/src/relationalai/experimental}/paths/path_algorithms/two_sided_balls_upto.py

@@ -1,7 +1,7 @@
 # For builder components.
 from relationalai.semantics import Integer, define, not_, count
-from relationalai.semantics.reasoners.graph.paths.graph import Graph
-from relationalai.semantics.reasoners.graph.paths.utilities.iterators import setup_iteration
+from relationalai.experimental.paths.graph import Graph
+from relationalai.experimental.paths.utilities.iterators import setup_iteration
 
 
 def two_balls_upto(g:Graph, Source, Target, max_length=None):

{relationalai-0.12.3/src/relationalai/semantics/reasoners/graph → relationalai-0.12.6/src/relationalai/experimental}/paths/path_algorithms/usp-old.py

@@ -1,8 +1,8 @@
 # For builder components.
 from relationalai.semantics import Integer, define, max, sum, not_
-from relationalai.semantics.reasoners.graph.paths.graph import Graph
-from relationalai.semantics.reasoners.graph.paths.path_algorithms.one_sided_ball_upto import ball_upto
-from relationalai.semantics.reasoners.graph.paths.path_algorithms.one_sided_ball_repetition import ball_with_repetition
+from relationalai.experimental.paths.graph import Graph
+from relationalai.experimental.paths.path_algorithms.one_sided_ball_upto import ball_upto
+from relationalai.experimental.paths.path_algorithms.one_sided_ball_repetition import ball_with_repetition
 
 
 def compute_usp(g: Graph, Source, Target, max_length = None):

{relationalai-0.12.3/src/relationalai/semantics/reasoners/graph → relationalai-0.12.6/src/relationalai/experimental}/paths/path_algorithms/usp-tuple.py

@@ -1,8 +1,8 @@
 # For builder components.
 from relationalai.semantics import Model, Integer, define, max, sum, select, not_
-from relationalai.semantics.reasoners.graph.paths.graph import Graph
-from relationalai.semantics.reasoners.graph.paths.path_algorithms.one_sided_ball_upto import ball_upto
-from relationalai.semantics.reasoners.graph.paths.path_algorithms.one_sided_ball_repetition import ball_with_repetition
+from relationalai.experimental.paths.graph import Graph
+from relationalai.experimental.paths.path_algorithms.one_sided_ball_upto import ball_upto
+from relationalai.experimental.paths.path_algorithms.one_sided_ball_repetition import ball_with_repetition
 
 
 def compute_usp(g: Graph, Source, Target, max_length = None):

{relationalai-0.12.3/src/relationalai/semantics/reasoners/graph → relationalai-0.12.6/src/relationalai/experimental}/paths/path_algorithms/usp.py

@@ -1,8 +1,8 @@
 # For builder components.
 from relationalai.semantics import Integer, define, sum, not_
-from relationalai.semantics.reasoners.graph.paths.graph import Graph
-from relationalai.semantics.reasoners.graph.paths.path_algorithms.one_sided_ball_upto import ball_upto
-from relationalai.semantics.reasoners.graph.paths.path_algorithms.one_sided_ball_repetition import ball_with_repetition
+from relationalai.experimental.paths.graph import Graph
+from relationalai.experimental.paths.path_algorithms.one_sided_ball_upto import ball_upto
+from relationalai.experimental.paths.path_algorithms.one_sided_ball_repetition import ball_with_repetition
 
 
 def compute_usp(g: Graph, Source, Target, max_length=None):

{relationalai-0.12.3/src/relationalai/semantics/reasoners/graph → relationalai-0.12.6/src/relationalai/experimental}/paths/tests/tests_limit_sp_max_length.py

@@ -1,6 +1,6 @@
 from relationalai.semantics import Model, Integer, define, select, count
-from relationalai.semantics.reasoners.graph.paths.graph import Graph
-from relationalai.semantics.reasoners.graph.paths.path_algorithms.find_paths import find_shortest_paths
+from relationalai.experimental.paths.graph import Graph
+from relationalai.experimental.paths.path_algorithms.find_paths import find_shortest_paths
 
 
 # First test with grid graph

{relationalai-0.12.3/src/relationalai/semantics/reasoners/graph → relationalai-0.12.6/src/relationalai/experimental}/paths/tests/tests_limit_sp_multiple.py

@@ -1,6 +1,6 @@
 from relationalai.semantics import Model, Integer, define, select, count
-from relationalai.semantics.reasoners.graph.paths.graph import Graph
-from relationalai.semantics.reasoners.graph.paths.path_algorithms.find_paths import find_shortest_paths
+from relationalai.experimental.paths.graph import Graph
+from relationalai.experimental.paths.path_algorithms.find_paths import find_shortest_paths
 
 
 # First test with grid graph

{relationalai-0.12.3/src/relationalai/semantics/reasoners/graph → relationalai-0.12.6/src/relationalai/experimental}/paths/tests/tests_limit_sp_single.py

@@ -1,6 +1,6 @@
 from relationalai.semantics import Model, Integer, define, select, count
-from relationalai.semantics.reasoners.graph.paths.graph import Graph
-from relationalai.semantics.reasoners.graph.paths.path_algorithms.find_paths import find_shortest_paths
+from relationalai.experimental.paths.graph import Graph
+from relationalai.experimental.paths.path_algorithms.find_paths import find_shortest_paths
 
 
 # First test with grid graph

{relationalai-0.12.3/src/relationalai/semantics/reasoners/graph → relationalai-0.12.6/src/relationalai/experimental}/paths/tests/tests_limit_walks_multiple.py

@@ -1,6 +1,6 @@
 from relationalai.semantics import Model, Integer, define, select, count
-from relationalai.semantics.reasoners.graph.paths.graph import Graph
-from relationalai.semantics.reasoners.graph.paths.path_algorithms.find_paths import find_walks
+from relationalai.experimental.paths.graph import Graph
+from relationalai.experimental.paths.path_algorithms.find_paths import find_walks
 
 
 # First test with grid graph

{relationalai-0.12.3/src/relationalai/semantics/reasoners/graph → relationalai-0.12.6/src/relationalai/experimental}/paths/tests/tests_limit_walks_single.py

@@ -1,6 +1,6 @@
 from relationalai.semantics import Model, Integer, define, select, count
-from relationalai.semantics.reasoners.graph.paths.graph import Graph
-from relationalai.semantics.reasoners.graph.paths.path_algorithms.find_paths import find_walks
+from relationalai.experimental.paths.graph import Graph
+from relationalai.experimental.paths.path_algorithms.find_paths import find_walks
 
 
 # First test with grid graph

{relationalai-0.12.3/src/relationalai/semantics/reasoners/graph → relationalai-0.12.6/src/relationalai/experimental}/paths/tests/tests_one_sided_ball_repetition_multiple.py

@@ -1,6 +1,6 @@
 from relationalai.semantics import Model, Integer, define, select
-from relationalai.semantics.reasoners.graph.paths.graph import Graph
-from relationalai.semantics.reasoners.graph.paths.path_algorithms.one_sided_ball_repetition import ball_with_repetition
+from relationalai.experimental.paths.graph import Graph
+from relationalai.experimental.paths.path_algorithms.one_sided_ball_repetition import ball_with_repetition
 
 
 # Test with grid graph

{relationalai-0.12.3/src/relationalai/semantics/reasoners/graph → relationalai-0.12.6/src/relationalai/experimental}/paths/tests/tests_one_sided_ball_repetition_single.py

@@ -1,6 +1,6 @@
 from relationalai.semantics import Model, Integer, define, select
-from relationalai.semantics.reasoners.graph.paths.graph import Graph
-from relationalai.semantics.reasoners.graph.paths.path_algorithms.one_sided_ball_repetition import ball_with_repetition
+from relationalai.experimental.paths.graph import Graph
+from relationalai.experimental.paths.path_algorithms.one_sided_ball_repetition import ball_with_repetition
 
 
 # Test with grid graph

{relationalai-0.12.3/src/relationalai/semantics/reasoners/graph → relationalai-0.12.6/src/relationalai/experimental}/paths/tests/tests_one_sided_ball_upto_multiple.py

@@ -1,6 +1,6 @@
 from relationalai.semantics import Model, Integer, define, select
-from relationalai.semantics.reasoners.graph.paths.graph import Graph
-from relationalai.semantics.reasoners.graph.paths.path_algorithms.one_sided_ball_upto import ball_upto
+from relationalai.experimental.paths.graph import Graph
+from relationalai.experimental.paths.path_algorithms.one_sided_ball_upto import ball_upto
 
 
 # First test with grid graph and multiple source and target nodes

{relationalai-0.12.3/src/relationalai/semantics/reasoners/graph → relationalai-0.12.6/src/relationalai/experimental}/paths/tests/tests_one_sided_ball_upto_single.py

@@ -1,6 +1,6 @@
 from relationalai.semantics import Model, Integer, define, select
-from relationalai.semantics.reasoners.graph.paths.graph import Graph
-from relationalai.semantics.reasoners.graph.paths.path_algorithms.one_sided_ball_upto import ball_upto
+from relationalai.experimental.paths.graph import Graph
+from relationalai.experimental.paths.path_algorithms.one_sided_ball_upto import ball_upto
 
 
 # First test with grid graph

{relationalai-0.12.3/src/relationalai/semantics/reasoners/graph → relationalai-0.12.6/src/relationalai/experimental}/paths/tests/tests_single_paths.py

@@ -1,6 +1,6 @@
 from relationalai.semantics import Model, Integer, define, select
-from relationalai.semantics.reasoners.graph.paths.graph import Graph
-from relationalai.semantics.reasoners.graph.paths.path_algorithms.single import single_shortest_path
+from relationalai.experimental.paths.graph import Graph
+from relationalai.experimental.paths.path_algorithms.single import single_shortest_path
 
 
 model = Model("test_single_paths", dry_run=False)

{relationalai-0.12.3/src/relationalai/semantics/reasoners/graph → relationalai-0.12.6/src/relationalai/experimental}/paths/tests/tests_single_walks.py

@@ -1,6 +1,6 @@
 from relationalai.semantics import Model, Integer, define, select
-from relationalai.semantics.reasoners.graph.paths.graph import Graph
-from relationalai.semantics.reasoners.graph.paths.path_algorithms.single import single_walk
+from relationalai.experimental.paths.graph import Graph
+from relationalai.experimental.paths.path_algorithms.single import single_walk
 
 
 model = Model("test_single_paths", dry_run=False)

{relationalai-0.12.3/src/relationalai/semantics/reasoners/graph → relationalai-0.12.6/src/relationalai/experimental}/paths/tests/tests_single_walks_undirected.py

@@ -1,6 +1,6 @@
 from relationalai.semantics import Model, Integer, define, select
-from relationalai.semantics.reasoners.graph.paths.graph import Graph
-from relationalai.semantics.reasoners.graph.paths.path_algorithms.single import single_walk
+from relationalai.experimental.paths.graph import Graph
+from relationalai.experimental.paths.path_algorithms.single import single_walk
 
 
 model = Model("test_single_paths", dry_run=False)