nucliadb 6.2.1.post3365__py3-none-any.whl → 6.2.1.post3377__py3-none-any.whl

nucliadb/learning_proxy.py

@@ -236,11 +236,7 @@ async def _retriable_proxied_request(
     params: dict[str, Any],
 ) -> httpx.Response:
     return await client.request(
-        method=method.upper(),
-        url=url,
-        params=params,
-        content=content,
-        headers=headers,
+        method=method.upper(), url=url, params=params, content=content, headers=headers
     )
 
 
@@ -414,12 +410,7 @@ class DummyClient(httpx.AsyncClient):
         return self.get_config(*args, **kwargs)
 
     async def request(  # type: ignore
-        self,
-        method: str,
-        url: str,
-        params=None,
-        content=None,
-        headers=None,
+        self, method: str, url: str, params=None, content=None, headers=None, *args, **kwargs
     ) -> httpx.Response:
         return self._handle_request(method, url, params=params, content=content, headers=headers)
 

nucliadb/reader/api/v1/learning_config.py

@@ -17,8 +17,11 @@
 # You should have received a copy of the GNU Affero General Public License
 # along with this program. If not, see <http://www.gnu.org/licenses/>.
 #
+from typing import Dict
+
 from fastapi import Request
 from fastapi_versioning import version
+from nuclia_models.config.proto import ExtractConfig
 
 from nucliadb.learning_proxy import learning_config_proxy
 from nucliadb.models.responses import HTTPClientError
@@ -143,3 +146,40 @@ async def get_schema_for_configuration_creation(
     if not is_onprem_nucliadb():
         return HTTPClientError(status_code=404, detail="Endpoint not available for Hosted NucliaDB")
     return await learning_config_proxy(request, "GET", f"/schema")
+
+
+@api.get(
+    path=f"/{KB_PREFIX}/{{kbid}}/extract_strategies",
+    status_code=200,
+    summary="Learning extract strategies",
+    description="Get available extract strategies ",
+    response_model=Dict[str, ExtractConfig],
+    tags=["Extract Strategies"],
+)
+@requires(NucliaDBRoles.READER)
+@version(1)
+async def get_extract_strategies(
+    request: Request,
+    kbid: str,
+):
+    return await learning_config_proxy(request, "GET", f"/extract_strategies/{kbid}")
+
+
+@api.get(
+    path=f"/{KB_PREFIX}/{{kbid}}/extract_strategies/strategy/{{strategy_id}}",
+    status_code=200,
+    summary="Extract strategy configuration",
+    description="Get extract strategy for a given id",
+    response_model=None,
+    tags=["Extract Strategies"],
+)
+@requires(NucliaDBRoles.READER)
+@version(1)
+async def get_extract_strategy_from_id(
+    request: Request,
+    kbid: str,
+    strategy_id: str,
+):
+    return await learning_config_proxy(
+        request, "GET", f"/extract_strategies/{kbid}/strategies/{strategy_id}"
+    )

nucliadb/search/search/predict_proxy.py

@@ -36,6 +36,10 @@ class PredictProxiedEndpoints(str, Enum):
     TOKENS = "tokens"
     CHAT = "chat"
     REPHRASE = "rephrase"
+    RUN_AGENTS_TEXT = "run-agents-text"
+    SUMMARIZE = "summarize"
+    RERANK = "rerank"
+    REMI = "remi"
 
 
 async def predict_proxy(

nucliadb/writer/api/v1/learning_config.py

@@ -19,6 +19,7 @@
 #
 from fastapi import Request
 from fastapi_versioning import version
+from nuclia_models.config.proto import ExtractConfig
 
 from nucliadb.learning_proxy import learning_config_proxy
 from nucliadb.writer.api.v1.router import KB_PREFIX, api
@@ -58,3 +59,41 @@ async def patch_configuration(
     kbid: str,
 ):
     return await learning_config_proxy(request, "PATCH", f"/config/{kbid}")
+
+
+@api.post(
+    path=f"/{KB_PREFIX}/{{kbid}}/extract_strategies",
+    status_code=200,
+    summary="Add a extract strategy to a KB",
+    description="Add a extract strategy to a KB",
+    response_model=str,
+    tags=["Extract Strategies"],
+)
+@requires(NucliaDBRoles.WRITER)
+@version(1)
+async def add_strategy(
+    request: Request,
+    kbid: str,
+    item: ExtractConfig,
+):
+    return await learning_config_proxy(request, "POST", f"/extract_strategies/{kbid}")
+
+
+@api.delete(
+    path=f"/{KB_PREFIX}/{{kbid}}/extract_strategies/strategy/{{strategy_id}}",
+    status_code=204,
+    summary="Remove a extract strategy from a KB",
+    description="Removes a extract strategy from a KB",
+    response_model=None,
+    tags=["Extract Strategies"],
+)
+@requires(NucliaDBRoles.WRITER)
+@version(1)
+async def delete_strategy(
+    request: Request,
+    kbid: str,
+    strategy_id: str,
+):
+    return await learning_config_proxy(
+        request, "POST", f"/extract_strategies/{kbid}/strategies/{strategy_id}"
+    )

{nucliadb-6.2.1.post3365.dist-info → nucliadb-6.2.1.post3377.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: nucliadb
-Version: 6.2.1.post3365
+Version: 6.2.1.post3377
 Summary: NucliaDB
 Author-email: Nuclia <nucliadb@nuclia.com>
 License: AGPL
@@ -20,11 +20,11 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3 :: Only
 Requires-Python: <4,>=3.9
 Description-Content-Type: text/markdown
-Requires-Dist: nucliadb-telemetry[all]>=6.2.1.post3365
-Requires-Dist: nucliadb-utils[cache,fastapi,storages]>=6.2.1.post3365
-Requires-Dist: nucliadb-protos>=6.2.1.post3365
-Requires-Dist: nucliadb-models>=6.2.1.post3365
-Requires-Dist: nidx-protos>=6.2.1.post3365
+Requires-Dist: nucliadb-telemetry[all]>=6.2.1.post3377
+Requires-Dist: nucliadb-utils[cache,fastapi,storages]>=6.2.1.post3377
+Requires-Dist: nucliadb-protos>=6.2.1.post3377
+Requires-Dist: nucliadb-models>=6.2.1.post3377
+Requires-Dist: nidx-protos>=6.2.1.post3377
 Requires-Dist: nucliadb-admin-assets>=1.0.0.post1224
 Requires-Dist: nuclia-models>=0.24.2
 Requires-Dist: uvicorn

{nucliadb-6.2.1.post3365.dist-info → nucliadb-6.2.1.post3377.dist-info}/RECORD

@@ -33,7 +33,7 @@ migrations/pg/0003_catalog_kbid_index.py,sha256=uKq_vtnuf73GVf0mtl2rhzdk_czAoEU1
 migrations/pg/__init__.py,sha256=cp15ZcFnHvpcu_5-aK2A4uUyvuZVV_MJn4bIXMa20ks,835
 nucliadb/__init__.py,sha256=_abCmDJ_0ku483Os4UAjPX7Nywm39cQgAV_DiyjsKeQ,891
 nucliadb/health.py,sha256=UIxxA4oms4HIsCRZM_SZsdkIZIlgzmOxw-qSHLlWuak,3465
-nucliadb/learning_proxy.py,sha256=rQ9gOLy_NwcVgsSi4jyYYHFdo6Vnb-1tEJ4kz2PIo_4,19411
+nucliadb/learning_proxy.py,sha256=Gf76qXxjl1lrHEFaCpOUfjjf0ab6eGLNxLMJz3-M_mo,19354
 nucliadb/metrics_exporter.py,sha256=Rz6G7V_C_GTZCFzd0xEtIfixtZgUuffnr4rDKCbXXWM,5595
 nucliadb/openapi.py,sha256=wDiw0dVEvTpJvbatkJ0JZLkKm9RItZT5PWRHjqRfqTA,2272
 nucliadb/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -170,7 +170,7 @@ nucliadb/reader/api/v1/__init__.py,sha256=ieP8lsCCwG66Jupv8II5MSTj6nh3eCtLcF4utH
 nucliadb/reader/api/v1/download.py,sha256=rGv1c5CjrJheDgGwAnNWy76A_4V2shqqHYvwmKGFlpk,10758
 nucliadb/reader/api/v1/export_import.py,sha256=x4VBNDFjnlY1nIt5kdq0eZTB_DeRzGzT8T7uB7wUhNU,6448
 nucliadb/reader/api/v1/knowledgebox.py,sha256=Uu-yPB8KKZt1VaFrFNMMaXOvLsclBJDK9dzZ9lF2ctI,3645
-nucliadb/reader/api/v1/learning_config.py,sha256=w5a_miobPGakKmyLXkmuz1e6tXyIMSas-qLngqzVQTw,4436
+nucliadb/reader/api/v1/learning_config.py,sha256=CZ7pFXzBZkJE2dXbC1wArszJw_ZLpuEb6gnsz2MKEz0,5525
 nucliadb/reader/api/v1/resource.py,sha256=SFIv_vpgkdJQv7L_UgYZS5FvubipJ0ligpExGDjKHV0,14064
 nucliadb/reader/api/v1/router.py,sha256=eyNmEGSP9zHkCIG5XlAXl6sukq950B7gFT3X2peMtIE,1011
 nucliadb/reader/api/v1/services.py,sha256=B8fD4zcUydZ5Fl1DDxfXGNuW6C8rhGYc920O5Zy5Yv8,11716
@@ -222,7 +222,7 @@ nucliadb/search/search/merge.py,sha256=i_PTBFRqC5iTTziOMEltxLIlmokIou5hjjgR4BnoL
 nucliadb/search/search/metrics.py,sha256=81X-tahGW4n2CLvUzCPdNxNClmZqUWZjcVOGCUHoiUM,2872
 nucliadb/search/search/paragraphs.py,sha256=pNAEiYqJGGUVcEf7xf-PFMVqz0PX4Qb-WNG-_zPGN2o,7799
 nucliadb/search/search/pgcatalog.py,sha256=IaNK4dAxdXs38PoIkTdgqMDuZDjeiOtcXn3LeaT-OMw,8855
-nucliadb/search/search/predict_proxy.py,sha256=xBlh6kjuQpWRq7KsBx4pEl2PtnwljjQIiYMaTWpcCSA,3015
+nucliadb/search/search/predict_proxy.py,sha256=IFI3v_ODz2_UU1XZnyaD391fE7-2C0npSmj_HmDvzS4,3123
 nucliadb/search/search/query.py,sha256=vSnnqJPB5iHI7kugaL6boH1l4j5HV6ln4EKbskJnFWw,27346
 nucliadb/search/search/rank_fusion.py,sha256=tRGo_KlsFsVx1CQEy1iqQ6f0T1Dq1kf0axDXHuuzvvM,6946
 nucliadb/search/search/rerankers.py,sha256=3vep4EOVNeDJGsMdx-1g6Ar4ZGJG3IHym3HkxnbwtAQ,7321
@@ -314,7 +314,7 @@ nucliadb/writer/api/v1/__init__.py,sha256=akI9A_jloNLb0dU4T5zjfdyvmSAiDeIdjAlzNx
 nucliadb/writer/api/v1/export_import.py,sha256=Cv4DNtqoR_x2LMNx80C9ehyaCaFIxhq2eutEk2uzcZg,8249
 nucliadb/writer/api/v1/field.py,sha256=OsWOYA0WQ6onE5Rkl20QIEdtrSi7Jgnu62fUt90Ziy8,17503
 nucliadb/writer/api/v1/knowledgebox.py,sha256=MLeIuym4jPrJgfy1NTcN9CpUGwuBiqDHMcx0hY9DR7g,9530
-nucliadb/writer/api/v1/learning_config.py,sha256=GaYaagjBrVG9ZxrWQyVQfqGMQV3tAJjqJ5CStaKhktU,2058
+nucliadb/writer/api/v1/learning_config.py,sha256=i9fkyZfHYLFEQGqdF98yWnucIES8HzMyh9OtBlxRooo,3115
 nucliadb/writer/api/v1/resource.py,sha256=A8fAHlN5XFsg6XFYKhfWJS8czgNH6yXr-PsnUqz2WUE,18757
 nucliadb/writer/api/v1/router.py,sha256=RjuoWLpZer6Kl2BW_wznpNo6XL3BOpdTGqXZCn3QrrQ,1034
 nucliadb/writer/api/v1/services.py,sha256=THnBnRxiHrEZPpBTL-E-vplEUfcD-fZpuslKRonM6xs,10286
@@ -336,8 +336,8 @@ nucliadb/writer/tus/local.py,sha256=7jYa_w9b-N90jWgN2sQKkNcomqn6JMVBOVeDOVYJHto,
 nucliadb/writer/tus/s3.py,sha256=vF0NkFTXiXhXq3bCVXXVV-ED38ECVoUeeYViP8uMqcU,8357
 nucliadb/writer/tus/storage.py,sha256=ToqwjoYnjI4oIcwzkhha_MPxi-k4Jk3Lt55zRwaC1SM,2903
 nucliadb/writer/tus/utils.py,sha256=MSdVbRsRSZVdkaum69_0wku7X3p5wlZf4nr6E0GMKbw,2556
-nucliadb-6.2.1.post3365.dist-info/METADATA,sha256=g_650xn-beF0Qeo5H3cP5BlhSObncN0TJz-QOgA8VDk,4291
-nucliadb-6.2.1.post3365.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-nucliadb-6.2.1.post3365.dist-info/entry_points.txt,sha256=XqGfgFDuY3zXQc8ewXM2TRVjTModIq851zOsgrmaXx4,1268
-nucliadb-6.2.1.post3365.dist-info/top_level.txt,sha256=hwYhTVnX7jkQ9gJCkVrbqEG1M4lT2F_iPQND1fCzF80,20
-nucliadb-6.2.1.post3365.dist-info/RECORD,,
+nucliadb-6.2.1.post3377.dist-info/METADATA,sha256=qkAN9pDnp84ZNF8ywzbwKqVjROHktWkM_KOYhfzzYK0,4291
+nucliadb-6.2.1.post3377.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+nucliadb-6.2.1.post3377.dist-info/entry_points.txt,sha256=XqGfgFDuY3zXQc8ewXM2TRVjTModIq851zOsgrmaXx4,1268
+nucliadb-6.2.1.post3377.dist-info/top_level.txt,sha256=hwYhTVnX7jkQ9gJCkVrbqEG1M4lT2F_iPQND1fCzF80,20
+nucliadb-6.2.1.post3377.dist-info/RECORD,,