orchestrator-core 4.6.2__py3-none-any.whl → 4.6.3rc1__py3-none-any.whl

{orchestrator_core-4.6.2.dist-info → orchestrator_core-4.6.3rc1.dist-info}/RECORD

@@ -1,9 +1,9 @@
-orchestrator/__init__.py,sha256=GBSQUsDdHmxliqH9rPmCznK2vxTmQpobA4SmVWCg_q4,1454
+orchestrator/__init__.py,sha256=A8BOfSSiukGCfYFuZpv79RCPqojLwZU_jhE2xj4MIjU,1457
 orchestrator/agentic_app.py,sha256=ouiyyZiS4uS6Lox2DtbGGRnb2njJBMSHpSAGe-T5rX0,3028
 orchestrator/app.py,sha256=w8ubXaaogwjmwLM0TXqZaLkAhmaOTWzVlwiYbi5mHeE,13203
 orchestrator/exception_handlers.py,sha256=UsW3dw8q0QQlNLcV359bIotah8DYjMsj2Ts1LfX4ClY,1268
 orchestrator/llm_settings.py,sha256=giqxZmwO8sXyNF5Zc7-O9JYd7PLN065pKQYpJYOpXcE,2816
-orchestrator/log_config.py,sha256=1cPl_OXT4tEUyNxG8cwIWXrmadUm1E81vq0mdtrV-v4,1912
+orchestrator/log_config.py,sha256=aCbIzKzDnVwF0zVoQ9TlvuHxn0uhJWRjNL72Hc5pcss,2031
 orchestrator/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 orchestrator/security.py,sha256=iXFxGxab54aav7oHEKLAVkTgrQMJGHy6IYLojEnD7gI,2422
 orchestrator/settings.py,sha256=mvs1VhBYth6Zp55HsNroML4DU1jiq5SkVM47_BLgcIo,4662
@@ -24,7 +24,7 @@ orchestrator/api/api_v1/endpoints/processes.py,sha256=z_bantT415cCuAw0viVrsi7Hqu
 orchestrator/api/api_v1/endpoints/product_blocks.py,sha256=kZ6ywIOsS_S2qGq7RvZ4KzjvaS1LmwbGWR37AKRvWOw,2146
 orchestrator/api/api_v1/endpoints/products.py,sha256=BfFtwu9dZXEQbtKxYj9icc73GKGvAGMR5ytyf41nQlQ,3081
 orchestrator/api/api_v1/endpoints/resource_types.py,sha256=gGyuaDyOD0TAVoeFGaGmjDGnQ8eQQArOxKrrk4MaDzA,2145
-orchestrator/api/api_v1/endpoints/search.py,sha256=Zk3kDICF3ruaz0VUJH2q5GbbccLSdt9x4I5SN00M9EM,8133
+orchestrator/api/api_v1/endpoints/search.py,sha256=R_OzfJfquoaTeGBwXTbomh16aYy0ael6_Xn3WkKfyjg,8575
 orchestrator/api/api_v1/endpoints/settings.py,sha256=5s-k169podZjgGHUbVDmSQwpY_3Cs_Bbf2PPtZIkBcw,6184
 orchestrator/api/api_v1/endpoints/subscription_customer_descriptions.py,sha256=1_6LtgQleoq3M6z_W-Qz__Bj3OFUweoPrUqHMwSH6AM,3288
 orchestrator/api/api_v1/endpoints/subscriptions.py,sha256=7KaodccUiMkcVnrFnK2azp_V_-hGudcIyhov5WwVGQY,9810
@@ -104,10 +104,11 @@ orchestrator/cli/helpers/__init__.py,sha256=GyHNfEFCGKQwRiN6rQmvSRH2iYX7npjMZn97
 orchestrator/cli/helpers/input_helpers.py,sha256=pv5GTMuIWLzBE_bKNhn1XD_gxoqB0s1ZN4cnKkIIu5I,1139
 orchestrator/cli/helpers/print_helpers.py,sha256=b3ePg6HfBLKPYBBVr5XOA__JnFEMI5HBjbjov3CP8Po,859
 orchestrator/cli/search/__init__.py,sha256=K15_iW9ogR7xtX7qHDal4H09tmwVGnOBZWyPBLWhuzc,1274
+orchestrator/cli/search/display.py,sha256=PKy9sOTpq0WUdSfY2khLrIQ1OdAfsyl95ogF1Z6Dae0,3629
 orchestrator/cli/search/index_llm.py,sha256=RWPkFz5bxiznjpN1vMsSWeqcvYKB90DLL4pXQ92QJNI,2239
 orchestrator/cli/search/resize_embedding.py,sha256=iJdM7L6Kyq4CzRjXHWLwpGRiMnKK7xZ9133C0THebBE,4847
-orchestrator/cli/search/search_explore.py,sha256=SDn1DMN8a4roSPodIHl-KrNAvdHo5jTDUvMUFLVh1P4,8602
-orchestrator/cli/search/speedtest.py,sha256=4YGd2JsOHozuI3YMphO_GAJLJZ8KyL72ERhf_1RwABI,4812
+orchestrator/cli/search/search_explore.py,sha256=LNAn6w13Q69fpv8CHcicHAbabrctrjGvwTjjJyC0AZY,8447
+orchestrator/cli/search/speedtest.py,sha256=J_l-8WxgN3YnqmwnbRhDyVbeqtvk3d2SfIpRBOJuhvE,4840
 orchestrator/config/__init__.py,sha256=GyHNfEFCGKQwRiN6rQmvSRH2iYX7npjMZn97n8XzmLU,571
 orchestrator/config/assignee.py,sha256=9mFFe9hoi2NCkXFOKL2pU2aveBzcZhljSvqUnh55vrk,780
 orchestrator/db/__init__.py,sha256=I9QSWgqnenShixO_Kseuo0ZtM1epLAOIRpucxccrQDk,3625
@@ -115,7 +116,7 @@ orchestrator/db/database.py,sha256=MU_w_e95ho2dVb2JDnt_KFYholx___XDkiQXbc8wCkI,1
 orchestrator/db/helpers.py,sha256=L8kEdnSSNGnUpZhdeGx2arCodakWN8vSpKdfjoLuHdY,831
 orchestrator/db/listeners.py,sha256=UBPYcH0FE3a7aZQu_D0O_JMXpXIRYXC0gjSAvlv5GZo,1142
 orchestrator/db/loaders.py,sha256=ez6JzQ3IKVkC_oLAkVlIIiI8Do7hXbdcPKCvUSLxRog,7962
-orchestrator/db/models.py,sha256=vgYRMBiMzwDvePZBBkMrvflljfYHsAGNo6u6jUtC0QY,31328
+orchestrator/db/models.py,sha256=AeBHLm0PLtoSAHoXzxjy9JsDQA5JhPvS7Ro55ynu_tk,31275
 orchestrator/db/filters/__init__.py,sha256=RUj6P0XxEBhYj0SN5wH5-Vf_Wt_ilZR_n9DSar5m9oM,371
 orchestrator/db/filters/filters.py,sha256=55RtpQwM2rhrk4A6CCSeSXoo-BT9GnQoNTryA8CtLEg,5020
 orchestrator/db/filters/process.py,sha256=xvGhyfo_MZ1xhLvFC6yULjcT4mJk0fKc1glJIYgsWLE,4018
@@ -272,54 +273,59 @@ orchestrator/schemas/process.py,sha256=UACBNt-4g4v9Y528u-gZ-Wk7YxwJHhnI4cEu5CtQm
 orchestrator/schemas/product.py,sha256=MhMCh058ZuS2RJq-wSmxIPUNlhQexxXIx3DSz2OmOh4,1570
 orchestrator/schemas/product_block.py,sha256=kCqvm6qadHpegMr9aWI_fYX-T7mS-5S-ldPxnGQZg7M,1519
 orchestrator/schemas/resource_type.py,sha256=VDju4XywcDDLxdpbWU62RTvR9QF8x_GRrpTlN_NE8uI,1064
-orchestrator/schemas/search.py,sha256=DOW-zwxr6VZVYbVsvft2mwFLv9M_nDegX33F6Bx1tJE,1548
+orchestrator/schemas/search.py,sha256=d_Vs1qU9Z5zuXN4pDk6jrVwiUXRKZ93U-tHW5Zfrw-w,1546
+orchestrator/schemas/search_requests.py,sha256=j2X98eLRTr_dYeGcIc78iPyKoYaAgvl1NQjYV4L1CGY,1925
 orchestrator/schemas/subscription.py,sha256=-jXyHZIed9Xlia18ksSDyenblNN6Q2yM2FlGELyJ458,3423
 orchestrator/schemas/subscription_descriptions.py,sha256=Ft_jw1U0bf9Z0U8O4OWfLlcl0mXCVT_qYVagBP3GbIQ,1262
 orchestrator/schemas/workflow.py,sha256=StVoRGyNT2iIeq3z8BIlTPt0bcafzbeYxXRrCucR6LU,2146
 orchestrator/search/__init__.py,sha256=2uhTQexKx-cdBP1retV3CYSNCs02s8WL3fhGvupRGZk,571
-orchestrator/search/export.py,sha256=_0ncVpTqN6AoQfW3WX0fWnDQX3hBz6ZGC31Beu4PVwQ,6678
-orchestrator/search/llm_migration.py,sha256=tJAfAoykMFIROQrKBKpAbDaGYDLKcmevKWjYrsBmuAY,6703
+orchestrator/search/llm_migration.py,sha256=afDqdo5t-L6gLduKSSVKuAegznPdIsQn4YIoAAFraC8,6767
 orchestrator/search/agent/__init__.py,sha256=_O4DN0MSTUtr4olhyE0-2hsb7x3f_KURMCYjg8jV4QA,756
 orchestrator/search/agent/agent.py,sha256=iWa4_achqh5zRIfcJvjmY3hmDuFVGpV_PxjgttLdokU,2075
+orchestrator/search/agent/handlers.py,sha256=z1KqeUUn7cYNBW39femwYnZOqRiYm1885Cx4TWVNyCY,4476
 orchestrator/search/agent/json_patch.py,sha256=_Z5ULhLyeuOuy-Gr_DJR4eA-wo9F78qySKUt5F_SQvQ,1892
-orchestrator/search/agent/prompts.py,sha256=6EPubiSLFyICIeinfVUF6miU1nS2QTAhqgzm-l5O3PI,5810
-orchestrator/search/agent/state.py,sha256=WhvZu7N0NhD1DD5mfZSUAzYN4mu8dDyvQ4Tz9I-hLtg,1364
-orchestrator/search/agent/tools.py,sha256=m4Krtb-Qmep-JkbJ9-RC7QqKa0CuQJM6-Z6_PN-b8HU,14706
+orchestrator/search/agent/prompts.py,sha256=aNxdJz7M1RPg_Py5v3Wvc2Kzo_tshkfHqNlF6S7YQv4,6975
+orchestrator/search/agent/state.py,sha256=YBrpHb8ok2RPwoLwBUWHYaGms_uj6nz51le2vbDjJwA,1113
+orchestrator/search/agent/tools.py,sha256=hMovAKJ03EKsmGbdS363kbx1-BX4YJB6MsQUzZkEA38,17144
+orchestrator/search/agent/validation.py,sha256=-U7g9Tgw6wH_tli0X0wrVKmiP9Id08sISq3kRrJjFNQ,2926
+orchestrator/search/aggregations/__init__.py,sha256=Re5XDxvbRCT5DUFTQbHeDCyync1RL7yHh-D3Bm28CXg,921
+orchestrator/search/aggregations/base.py,sha256=8zIdZx9cLxgFg6PePHzUNdMhHejPpb-hKL6lirDJlyQ,6916
 orchestrator/search/core/__init__.py,sha256=q5G0z3nKjIHKFs1PkEG3nvTUy3Wp4kCyBtCbqUITj3A,579
 orchestrator/search/core/embedding.py,sha256=n16H5fZRlfn91wI8PfZPa1R39HwQd8T1nwlDOzcOUBU,2823
 orchestrator/search/core/exceptions.py,sha256=S_ZMEhrqsQBVqJ559FQ5J6tZU6BYLiU65AGWgSvgv_k,1159
-orchestrator/search/core/types.py,sha256=0U_m4ZmPwvL77hIx9yk7UyvkE8HoiRvEnGdY4mDLzCo,8853
+orchestrator/search/core/types.py,sha256=eEar-1MFKiOqagKp9e2DLEZtGQQGQR_Ia7fiuRzIVoc,8953
 orchestrator/search/core/validators.py,sha256=zktY5A3RTBmfdARJoxoz9rnnyTZj7L30Kbmh9UTQz2o,1204
 orchestrator/search/docs/index.md,sha256=zKzE2fbtHDfYTKaHg628wAsqCTOJ5yFUWV0ucFH3pAg,863
 orchestrator/search/docs/running_local_text_embedding_inference.md,sha256=OR0NVZMb8DqpgXYxlwDUrJwfRk0bYOk1-LkDMqsV6bU,1327
-orchestrator/search/filters/__init__.py,sha256=Yutr21lv8RtZf5OKaBozlYufgmmV2QHuzAPPjvUamLE,1222
+orchestrator/search/filters/__init__.py,sha256=pmnHyq5SrqxS6eEiyhSIify776W9BIZ0gjOZFuYZ0nU,1335
 orchestrator/search/filters/base.py,sha256=lUr0eW0zi4oIMVUHuRD3GAQ9xEbHiFUl_EfAI6ABPVo,12456
 orchestrator/search/filters/date_filters.py,sha256=0a6nbUTK647_Qf4XXZMLDvBLVjF5Qqy9eJ-9SrTGaGg,3040
-orchestrator/search/filters/definitions.py,sha256=wl2HiXlTWXQN4JmuSq2SBuhTMvyIeonTtUZoCrJAK6M,4093
+orchestrator/search/filters/definitions.py,sha256=k30Dp1bEr3CWMeuIcF1wPgmmF3rxI9Urx-sCaPfaE3c,4607
 orchestrator/search/filters/ltree_filters.py,sha256=1OOmM5K90NsGBQmTqyoDlphdAOGd9r2rmz1rNItm8yk,2341
-orchestrator/search/filters/numeric_filter.py,sha256=lcOAOpPNTwA0SW8QPiMOs1oKTYZLwGDQSrwFydXgMUU,2774
+orchestrator/search/filters/numeric_filter.py,sha256=do52w5Dmb5Rt4ipfX1iEObNcdymbWHtgS2HI1Otq-JQ,2771
 orchestrator/search/indexing/__init__.py,sha256=Or78bizNPiuNOgwLGJQ0mspCF1G_gSe5C9Ap7qi0MZk,662
 orchestrator/search/indexing/indexer.py,sha256=2qqDe6IlKfz1exh0xLBmpPdkTqbapLnTJORZneM6tmw,15320
 orchestrator/search/indexing/registry.py,sha256=N4YOUhNJfY6iBwPnn76tDcZaOVYJMA2SxwWhdBE85Xs,3716
 orchestrator/search/indexing/tasks.py,sha256=vmS1nnprPF74yitS0xGpP1dhSDis2nekMYF0v_jduDE,2478
 orchestrator/search/indexing/traverse.py,sha256=NKkKSri-if1d1vwzTQlDCF0hvBdB2IbWWuMdPrQ78Jg,14330
-orchestrator/search/retrieval/__init__.py,sha256=WpbC65DrI1AlNPvp5QDZXUZzGnzZ8Av0-9fH95eWvs0,763
-orchestrator/search/retrieval/builder.py,sha256=af_m4I-p_8eJ85t2B99jxecYlzFaY4F0CZwh4YFf_Us,4669
-orchestrator/search/retrieval/engine.py,sha256=RjaH70ChVU2etx5I6PuePcwbVGhxVVw_zG2ZlOviOPQ,7635
-orchestrator/search/retrieval/exceptions.py,sha256=oHoLGLLxxmVcV-W36uK0V-Pn4vf_iw6hajpQbap3NqI,3588
-orchestrator/search/retrieval/pagination.py,sha256=mJMesuHaxYvJggy8MsxRmfof_D-LuDVc_lGQ_jhqWkQ,3148
-orchestrator/search/retrieval/query_state.py,sha256=Gxhb7JcS40Qv2w9wvCp6_CPokGOQoxByyjc-AhK1LRI,2301
-orchestrator/search/retrieval/utils.py,sha256=svhF9YfMClq2MVPArS3ir3pg5_e_bremquv_l6tTsOQ,4597
-orchestrator/search/retrieval/validation.py,sha256=AjhttVJWlZDaT1_pUL_LaypQV11U21JpTCE4OwnpoqA,5849
+orchestrator/search/query/__init__.py,sha256=nCjvK_n2WQdV_ACrncFXEfnvLcHtuI__J7KLlFIaQvo,2437
+orchestrator/search/query/builder.py,sha256=kgnJ93TOCm8UTL5k09nWLsG4NXAlvFFa65gbciOwZ8E,10153
+orchestrator/search/query/engine.py,sha256=TFdV_sSoSXCSDSpyhVA2S6YaJysDSW2WtPj7duAyomk,5745
+orchestrator/search/query/exceptions.py,sha256=DrkNzXVbQAOi28FTHKimf_eTrXmhYwXrH986QhfQLPU,4941
+orchestrator/search/query/export.py,sha256=_0ncVpTqN6AoQfW3WX0fWnDQX3hBz6ZGC31Beu4PVwQ,6678
+orchestrator/search/query/mixins.py,sha256=BdVDzCOFDXT6N9LI_WrbVzGrk61UNplX-UZPvD0rEV0,3019
+orchestrator/search/query/queries.py,sha256=j1uKSQgF_ifVaDJaxjs4h2z48KqGVEIKCXOoJ7Ur9Mk,3805
+orchestrator/search/query/results.py,sha256=8k3o7F5EESuNOdNRPsmv0klPAG9YlPisCR4iGtewiXg,8662
+orchestrator/search/query/state.py,sha256=fMSBJs39kZTkpDE2T4h4x0x-51GqUvzAuePg2YUbO6I,3220
+orchestrator/search/query/validation.py,sha256=m0xJ71A0Qa5hm8b71zKRjSVpPrnkG7LbqPu4lv_GboI,8260
+orchestrator/search/retrieval/__init__.py,sha256=q5G0z3nKjIHKFs1PkEG3nvTUy3Wp4kCyBtCbqUITj3A,579
+orchestrator/search/retrieval/pagination.py,sha256=kcUzq1QQk4GrZq02M4hsKwAelUo1qDeCqsXImLUK6DA,3006
 orchestrator/search/retrieval/retrievers/__init__.py,sha256=1bGmbae0GYRM6e1vxf0ww79NaTSmfOMe9S0pPVmh3CM,897
-orchestrator/search/retrieval/retrievers/base.py,sha256=GyVbFVsZwjolyVfgGv3kYD22dxRhvRbUI7h9BjeK1DU,4174
+orchestrator/search/retrieval/retrievers/base.py,sha256=Xfjo-LD_fltKy0xhBBUSsmm2v2TNygDotWQ0wTTyvVE,4155
 orchestrator/search/retrieval/retrievers/fuzzy.py,sha256=PLp_ANRLzmtGQP1t9X4jt43_JLKDnOxWU2xqlexSH1U,3779
 orchestrator/search/retrieval/retrievers/hybrid.py,sha256=l-7J4qct0h28wSi0KvdFJw2lyh3jyobbrCbg0PuX-4I,11141
 orchestrator/search/retrieval/retrievers/semantic.py,sha256=36ky_A_LNWs13IYe809qy1RPrd0Fab-G-9pf2ZDARhA,3905
 orchestrator/search/retrieval/retrievers/structured.py,sha256=13TxC52fpNGXHnPX40J2GczRYFk8LAvWn2a0HWZCd2Q,1426
-orchestrator/search/schemas/__init__.py,sha256=q5G0z3nKjIHKFs1PkEG3nvTUy3Wp4kCyBtCbqUITj3A,579
-orchestrator/search/schemas/parameters.py,sha256=rrRLF_3p41SDEzX5yt-IPQWawGAcSzpeOVROhqulnbc,4834
-orchestrator/search/schemas/results.py,sha256=5UU4mmTrIu--RDW4nrl6_EpdYCMf_S-kj0OyR0aaOHQ,2096
 orchestrator/services/__init__.py,sha256=GyHNfEFCGKQwRiN6rQmvSRH2iYX7npjMZn97n8XzmLU,571
 orchestrator/services/fixed_inputs.py,sha256=kyz7s2HLzyDulvcq-ZqefTw1om86COvyvTjz0_5CmgI,876
 orchestrator/services/input_state.py,sha256=6BZOpb3cHpO18K-XG-3QUIV9pIM25_ufdODrp5CmXG4,2390
@@ -373,7 +379,7 @@ orchestrator/workflows/tasks/resume_workflows.py,sha256=T3iobSJjVgiupe0rClD34kUZ
 orchestrator/workflows/tasks/validate_product_type.py,sha256=lo2TX_MZOfcOmYFjLyD82FrJ5AAN3HOsE6BhDVFuy9Q,3210
 orchestrator/workflows/tasks/validate_products.py,sha256=GZJBoFF-WMphS7ghMs2-gqvV2iL1F0POhk0uSNt93n0,8510
 orchestrator/workflows/translations/en-GB.json,sha256=ST53HxkphFLTMjFHonykDBOZ7-P_KxksktZU3GbxLt0,846
-orchestrator_core-4.6.2.dist-info/licenses/LICENSE,sha256=b-aA5OZQuuBATmLKo_mln8CQrDPPhg3ghLzjPjLn4Tg,11409
-orchestrator_core-4.6.2.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
-orchestrator_core-4.6.2.dist-info/METADATA,sha256=UaZQ04AM0BKhwHPTf6YyqG_xkPLPzGLRnPz_REMyC3M,6351
-orchestrator_core-4.6.2.dist-info/RECORD,,
+orchestrator_core-4.6.3rc1.dist-info/licenses/LICENSE,sha256=b-aA5OZQuuBATmLKo_mln8CQrDPPhg3ghLzjPjLn4Tg,11409
+orchestrator_core-4.6.3rc1.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
+orchestrator_core-4.6.3rc1.dist-info/METADATA,sha256=6-UwigL10OGwkHzvuZOq_ilbYhJoS_7Mmo1Y1Jpub-I,6362
+orchestrator_core-4.6.3rc1.dist-info/RECORD,,

orchestrator/search/retrieval/builder.py

@@ -1,127 +0,0 @@
-# Copyright 2019-2025 SURF, GÉANT.
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#    http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-from collections import defaultdict
-from typing import Sequence
-
-from sqlalchemy import Select, String, cast, func, select
-from sqlalchemy.engine import Row
-
-from orchestrator.db.models import AiSearchIndex
-from orchestrator.search.core.types import EntityType, FieldType, FilterOp, UIType
-from orchestrator.search.filters import LtreeFilter
-from orchestrator.search.schemas.parameters import BaseSearchParameters
-from orchestrator.search.schemas.results import ComponentInfo, LeafInfo
-
-
-def create_path_autocomplete_lquery(prefix: str) -> str:
-    """Create the lquery pattern for a multi-level path autocomplete search."""
-    return f"{prefix}*.*"
-
-
-def build_candidate_query(params: BaseSearchParameters) -> Select:
-    """Build the base query for retrieving candidate entities.
-
-    Constructs a `SELECT` statement that retrieves distinct `entity_id` values
-    from the index table for the given entity type, applying any structured
-    filters from the provided search parameters.
-
-    Args:
-        params (BaseSearchParameters): The search parameters containing the entity type and optional filters.
-
-    Returns:
-        Select: The SQLAlchemy `Select` object representing the query.
-    """
-
-    stmt = (
-        select(AiSearchIndex.entity_id, AiSearchIndex.entity_title)
-        .where(AiSearchIndex.entity_type == params.entity_type.value)
-        .distinct()
-    )
-
-    if params.filters is not None:
-        entity_id_col = AiSearchIndex.entity_id
-        stmt = stmt.where(
-            params.filters.to_expression(
-                entity_id_col,
-                entity_type_value=params.entity_type.value,
-            )
-        )
-
-    return stmt
-
-
-def build_paths_query(entity_type: EntityType, prefix: str | None = None, q: str | None = None) -> Select:
-    """Build the query for retrieving paths and their value types for leaves/components processing."""
-    stmt = select(AiSearchIndex.path, AiSearchIndex.value_type).where(AiSearchIndex.entity_type == entity_type.value)
-
-    if prefix:
-        lquery_pattern = create_path_autocomplete_lquery(prefix)
-        ltree_filter = LtreeFilter(op=FilterOp.MATCHES_LQUERY, value=lquery_pattern)
-        stmt = stmt.where(ltree_filter.to_expression(AiSearchIndex.path, path=""))
-
-    stmt = stmt.group_by(AiSearchIndex.path, AiSearchIndex.value_type)
-
-    if q:
-        score = func.similarity(cast(AiSearchIndex.path, String), q)
-        stmt = stmt.order_by(score.desc(), AiSearchIndex.path)
-    else:
-        stmt = stmt.order_by(AiSearchIndex.path)
-
-    return stmt
-
-
-def process_path_rows(rows: Sequence[Row]) -> tuple[list[LeafInfo], list[ComponentInfo]]:
-    """Process query results to extract leaves and components information.
-
-    Parameters
-    ----------
-    rows : Sequence[Row]
-        Database rows containing path and value_type information
-
-    Returns:
-    -------
-    tuple[list[LeafInfo], list[ComponentInfo]]
-        Processed leaves and components
-    """
-    leaves_dict: dict[str, set[UIType]] = defaultdict(set)
-    leaves_paths_dict: dict[str, set[str]] = defaultdict(set)
-    components_set: set[str] = set()
-
-    for row in rows:
-        path, value_type = row
-
-        path_str = str(path)
-        path_segments = path_str.split(".")
-
-        # Remove numeric segments
-        clean_segments = [seg for seg in path_segments if not seg.isdigit()]
-
-        if clean_segments:
-            # Last segment is a leaf
-            leaf_name = clean_segments[-1]
-            ui_type = UIType.from_field_type(FieldType(value_type))
-            leaves_dict[leaf_name].add(ui_type)
-            leaves_paths_dict[leaf_name].add(path_str)
-
-            # All segments except the first/last are components
-            for component in clean_segments[1:-1]:
-                components_set.add(component)
-
-    leaves = [
-        LeafInfo(name=leaf, ui_types=list(types), paths=sorted(leaves_paths_dict[leaf]))
-        for leaf, types in leaves_dict.items()
-    ]
-    components = [ComponentInfo(name=component, ui_types=[UIType.COMPONENT]) for component in sorted(components_set)]
-
-    return leaves, components

orchestrator/search/retrieval/engine.py

@@ -1,197 +0,0 @@
-# Copyright 2019-2025 SURF, GÉANT.
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#    http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-from collections.abc import Sequence
-
-import structlog
-from sqlalchemy.engine.row import RowMapping
-from sqlalchemy.orm import Session
-
-from orchestrator.search.core.embedding import QueryEmbedder
-from orchestrator.search.core.types import FilterOp, SearchMetadata
-from orchestrator.search.filters import FilterTree, LtreeFilter
-from orchestrator.search.schemas.parameters import BaseSearchParameters
-from orchestrator.search.schemas.results import MatchingField, SearchResponse, SearchResult
-
-from .builder import build_candidate_query
-from .pagination import PageCursor
-from .query_state import SearchQueryState
-from .retrievers import Retriever
-from .utils import generate_highlight_indices
-
-logger = structlog.get_logger(__name__)
-
-
-def _format_response(
-    db_rows: Sequence[RowMapping], search_params: BaseSearchParameters, metadata: SearchMetadata
-) -> SearchResponse:
-    """Format database query results into a `SearchResponse`.
-
-    Converts raw SQLAlchemy `RowMapping` objects into `SearchResult` instances,
-    including highlight metadata if present in the database results.
-
-    Args:
-        db_rows (Sequence[RowMapping]): The rows returned from the executed SQLAlchemy query.
-        search_params (BaseSearchParameters): The search parameters, including query text and filters.
-        metadata (SearchMetadata): Metadata about the search execution.
-
-    Returns:
-        SearchResponse: A list of `SearchResult` objects containing entity IDs, scores,
-        and optional highlight information.
-    """
-
-    if not db_rows:
-        return SearchResponse(results=[], metadata=metadata)
-
-    user_query = search_params.query
-
-    results = []
-    for row in db_rows:
-        matching_field = None
-
-        if (
-            user_query
-            and (text := row.get(Retriever.HIGHLIGHT_TEXT_LABEL))
-            and (path := row.get(Retriever.HIGHLIGHT_PATH_LABEL))
-        ):
-            if not isinstance(text, str):
-                text = str(text)
-            if not isinstance(path, str):
-                path = str(path)
-
-            highlight_indices = generate_highlight_indices(text, user_query) or None
-            matching_field = MatchingField(text=text, path=path, highlight_indices=highlight_indices)
-
-        elif not user_query and search_params.filters and metadata.search_type == "structured":
-            # Structured search (filter-only)
-            matching_field = _extract_matching_field_from_filters(search_params.filters)
-
-        entity_title = row.get("entity_title", "")
-        if not isinstance(entity_title, str):
-            entity_title = str(entity_title) if entity_title is not None else ""
-
-        results.append(
-            SearchResult(
-                entity_id=str(row.entity_id),
-                entity_type=search_params.entity_type,
-                entity_title=entity_title,
-                score=row.score,
-                perfect_match=row.get("perfect_match", 0),
-                matching_field=matching_field,
-            )
-        )
-    return SearchResponse(results=results, metadata=metadata)
-
-
-def _extract_matching_field_from_filters(filters: FilterTree) -> MatchingField | None:
-    """Extract the first path filter to use as matching field for structured searches."""
-    leaves = filters.get_all_leaves()
-    if len(leaves) != 1:
-        return None
-
-    pf = leaves[0]
-
-    if isinstance(pf.condition, LtreeFilter):
-        op = pf.condition.op
-        # Prefer the original component/pattern (validator may set path="*" and move the value)
-        display = str(getattr(pf.condition, "value", "") or pf.path)
-
-        # There can be no match for abscence.
-        if op == FilterOp.NOT_HAS_COMPONENT:
-            return None
-
-        return MatchingField(text=display, path=display, highlight_indices=[(0, len(display))])
-
-    # Everything thats not Ltree
-    val = getattr(pf.condition, "value", "")
-    text = "" if val is None else str(val)
-    return MatchingField(text=text, path=pf.path, highlight_indices=[(0, len(text))])
-
-
-async def _execute_search_internal(
-    search_params: BaseSearchParameters,
-    db_session: Session,
-    limit: int,
-    cursor: PageCursor | None = None,
-    query_embedding: list[float] | None = None,
-) -> SearchResponse:
-    """Internal function to execute search with specified parameters.
-
-    Args:
-        search_params: The search parameters specifying vector, fuzzy, or filter criteria.
-        db_session: The active SQLAlchemy session for executing the query.
-        limit: Maximum number of results to return.
-        cursor: Optional pagination cursor.
-        query_embedding: Optional pre-computed query embedding to use instead of generating a new one.
-
-    Returns:
-        SearchResponse with results and embedding (for internal use).
-    """
-    if not search_params.vector_query and not search_params.filters and not search_params.fuzzy_term:
-        logger.warning("No search criteria provided (vector_query, fuzzy_term, or filters).")
-        return SearchResponse(results=[], metadata=SearchMetadata.empty())
-
-    candidate_query = build_candidate_query(search_params)
-
-    if search_params.vector_query and not query_embedding:
-
-        query_embedding = await QueryEmbedder.generate_for_text_async(search_params.vector_query)
-
-    retriever = await Retriever.route(search_params, cursor, query_embedding)
-    logger.debug("Using retriever", retriever_type=retriever.__class__.__name__)
-
-    final_stmt = retriever.apply(candidate_query)
-    final_stmt = final_stmt.limit(limit)
-    logger.debug(final_stmt)
-    result = db_session.execute(final_stmt).mappings().all()
-
-    response = _format_response(result, search_params, retriever.metadata)
-    # Store embedding in response for agent to save to DB
-    response.query_embedding = query_embedding
-    return response
-
-
-async def execute_search(
-    search_params: BaseSearchParameters,
-    db_session: Session,
-    cursor: PageCursor | None = None,
-    query_embedding: list[float] | None = None,
-) -> SearchResponse:
-    """Execute a search and return ranked results."""
-    return await _execute_search_internal(search_params, db_session, search_params.limit, cursor, query_embedding)
-
-
-async def execute_search_for_export(
-    query_state: SearchQueryState,
-    db_session: Session,
-) -> list[dict]:
-    """Execute a search for export and fetch flattened entity data.
-
-    Args:
-        query_state: Query state containing parameters and query_embedding.
-        db_session: The active SQLAlchemy session for executing the query.
-
-    Returns:
-        List of flattened entity records suitable for export.
-    """
-    from orchestrator.search.export import fetch_export_data
-
-    search_response = await _execute_search_internal(
-        search_params=query_state.parameters,
-        db_session=db_session,
-        limit=query_state.parameters.export_limit,
-        query_embedding=query_state.query_embedding,
-    )
-
-    entity_ids = [res.entity_id for res in search_response.results]
-    return fetch_export_data(query_state.parameters.entity_type, entity_ids)

orchestrator/search/schemas/parameters.py

@@ -1,133 +0,0 @@
-# Copyright 2019-2025 SURF, GÉANT.
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#    http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-import uuid
-from typing import Any, ClassVar, Literal
-
-from pydantic import BaseModel, ConfigDict, Field, TypeAdapter
-
-from orchestrator.search.core.types import ActionType, EntityType
-from orchestrator.search.filters import FilterTree
-
-
-class BaseSearchParameters(BaseModel):
-    """Base model with common search parameters."""
-
-    DEFAULT_EXPORT_LIMIT: ClassVar[int] = 1000
-    MAX_EXPORT_LIMIT: ClassVar[int] = 10000
-
-    action: ActionType = Field(default=ActionType.SELECT, description="The action to perform.")
-    entity_type: EntityType
-
-    filters: FilterTree | None = Field(default=None, description="A list of structured filters to apply to the search.")
-
-    query: str | None = Field(
-        default=None, description="Unified search query - will be processed into vector_query and/or fuzzy_term"
-    )
-
-    limit: int = Field(default=10, ge=1, le=30, description="Maximum number of search results to return.")
-    export_limit: int = Field(
-        default=DEFAULT_EXPORT_LIMIT, ge=1, le=MAX_EXPORT_LIMIT, description="Maximum number of results to export."
-    )
-    model_config = ConfigDict(extra="forbid")
-
-    @classmethod
-    def create(cls, **kwargs: Any) -> "SearchParameters":
-        """Create the correct search parameter subclass instance based on entity_type."""
-        from orchestrator.search.schemas.parameters import SearchParameters
-
-        adapter: TypeAdapter = TypeAdapter(SearchParameters)
-        return adapter.validate_python(kwargs)
-
-    @property
-    def vector_query(self) -> str | None:
-        """Extract vector query from unified query field."""
-        if not self.query:
-            return None
-        try:
-            uuid.UUID(self.query)
-            return None  # It's a UUID, so disable vector search.
-        except ValueError:
-            return self.query
-
-    @property
-    def fuzzy_term(self) -> str | None:
-        """Extract fuzzy term from unified query field."""
-        if not self.query:
-            return None
-
-        words = self.query.split()
-        # Only use fuzzy for single words
-        # otherwise, trigram operator filters out too much.
-        return self.query if len(words) == 1 else None
-
-
-class SubscriptionSearchParameters(BaseSearchParameters):
-    entity_type: Literal[EntityType.SUBSCRIPTION] = Field(
-        default=EntityType.SUBSCRIPTION, description="The type of entity to search."
-    )
-    model_config = ConfigDict(
-        json_schema_extra={
-            "title": "SearchSubscriptions",
-            "description": "Search subscriptions based on specific criteria.",
-            "examples": [
-                {
-                    "filters": {
-                        "op": "AND",
-                        "children": [
-                            {"path": "subscription.status", "condition": {"op": "eq", "value": "provisioning"}},
-                            {"path": "subscription.end_date", "condition": {"op": "gte", "value": "2025-01-01"}},
-                        ],
-                    }
-                }
-            ],
-        }
-    )
-
-
-class ProductSearchParameters(BaseSearchParameters):
-    entity_type: Literal[EntityType.PRODUCT] = Field(
-        default=EntityType.PRODUCT, description="The type of entity to search."
-    )
-    model_config = ConfigDict(
-        json_schema_extra={
-            "title": "SearchProducts",
-            "description": "Search products based on specific criteria.",
-            "examples": [
-                {
-                    "filters": [
-                        {"path": "product.product_type", "condition": {"op": "eq", "value": "Shop"}},
-                    ]
-                }
-            ],
-        }
-    )
-
-
-class WorkflowSearchParameters(BaseSearchParameters):
-    entity_type: Literal[EntityType.WORKFLOW] = Field(
-        default=EntityType.WORKFLOW, description="The type of entity to search."
-    )
-
-
-class ProcessSearchParameters(BaseSearchParameters):
-    """Search parameters specifically for PROCESS entities."""
-
-    entity_type: Literal[EntityType.PROCESS] = Field(
-        default=EntityType.PROCESS, description="The type of entity to search."
-    )
-
-
-SearchParameters = (
-    SubscriptionSearchParameters | ProductSearchParameters | WorkflowSearchParameters | ProcessSearchParameters
-)