get total count of wikidata organizations

robert-koch-institut · Jul 16, 2024 · 561e341 · 561e341
1 parent b4c1da3
commit 561e341
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 11 deletions.
diff --git a/mex/backend/auxiliary/wikidata.py b/mex/backend/auxiliary/wikidata.py
@@ -11,7 +11,10 @@
     transform_seed_primary_sources_to_extracted_primary_sources,
 )
 from mex.common.types import TextLanguage
-from mex.common.wikidata.extract import search_organizations_by_label
+from mex.common.wikidata.extract import (
+    get_count_of_found_organizations_by_label,
+    search_organizations_by_label,
+)
 from mex.common.wikidata.transform import (
     transform_wikidata_organizations_to_extracted_organizations,
 )
@@ -37,6 +40,7 @@ def search_organization_in_wikidata(
     Returns:
         Paginated list of ExtractedOrganization
     """
+    total_orgs = get_count_of_found_organizations_by_label(q, lang)
     organizations = search_organizations_by_label(q, offset, limit, lang)
 
     extracted_organizations = list(
@@ -46,7 +50,7 @@ def search_organization_in_wikidata(
     )
 
     return PagedResponseSchema(
-        total=len(extracted_organizations),
+        total=total_orgs,
         offset=offset,
         limit=limit,
         results=[organization for organization in extracted_organizations],

diff --git a/pdm.lock b/pdm.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -10,7 +10,7 @@ requires-python = "<3.13,>=3.11"
 dependencies = [
     "fastapi==0.111.0",
     "httpx==0.27.0",
-    "mex-common@git+https://github.com/robert-koch-institut/mex-common.git@0.28.0",
+    "mex-common@git+https://github.com/robert-koch-institut/mex-common.git@0.30.0",
     "neo4j==5.21.0",
     "pydantic==2.7.4",
     "uvicorn[standard]==0.30.1",