no ollama

src/wikibase_helpers.py

@@ -3,6 +3,7 @@ import logging
 from wikibaseintegrator import WikibaseIntegrator, datatypes,  wbi_helpers
 from wikibaseintegrator.wbi_config import config
 from wikibaseintegrator.wbi_exceptions import MWApiError
+from wikibaseintegrator.wbi_helpers import mediawiki_api_call_helper, SearchError
 
 wikibase_api_url = 'https://fashionwiki.wikibase.cloud/w/api.php'
 config = {
@@ -42,6 +43,69 @@ def get_property_id_by_label(property_label, api_url):
     else:
         logging.error(f"Failed to search for property by label in the target Wikibase. HTTP Status Code: {response.status_code}")
         return None
+    
+
+def get_entity_id_by_label(search_string,wiki, dict_result=False) -> list:
+    """
+    Performs a search for entities in the Wikibase instance using labels and aliases.
+    You can have more information on the parameters in the MediaWiki API help (https://www.wikidata.org/w/api.php?action=help&modules=wbsearchentities)
+
+    :param search_string: A string which should be searched for in the Wikibase instance (labels and aliases)
+    :param wiki: The wiki to search in. It can be "wikidata" or "wikibase"
+    :param dict_result: If True, the result will be a list of dictionaries with the keys 'id', 'label', 'match', 'description' and 'aliases'. If False, the result will be a list of strings with the entity IDs.
+    :return: A list of dictionaries or strings with the search results
+    """
+    if wiki == "wikidata":
+        login = login_wikidata
+        mediawiki_api_url = wikidata_api_url
+    elif wiki == "wikibase":
+        login = login_wikibase
+        mediawiki_api_url = wikibase_api_url
+
+    language = "en"
+    strict_language = False
+
+    params = {
+        'action': 'wbsearchentities',
+        'search': search_string,
+        'language': language,
+        'type': "item",
+        'limit': 50,
+        'format': 'json',
+    }
+
+    if strict_language:
+        params.update({'strict_language': ''})
+
+    cont_count = 0
+    results = []
+
+    while True:
+        params.update({'continue': cont_count})
+        search_results = mediawiki_api_call_helper(data=params, login = login, mediawiki_api_url=mediawiki_api_url, user_agent = config['USER_AGENT'])
+        if search_results['success'] != 1:
+            raise SearchError('Wikibase API wbsearchentities failed')
+
+        for i in search_results['search']:
+            if dict_result:
+                description = i['description'] if 'description' in i else None
+                aliases = i['aliases'] if 'aliases' in i else None
+                results.append({
+                    'id': i['id'],
+                    'label': i['label'],
+                    'match': i['match'],
+                    'description': description,
+                    'aliases': aliases
+                })
+            else:
+                results.append(i['id'])
+
+        if 'search-continue' not in search_results:
+            break
+        cont_count = search_results['search-continue']
+        if cont_count >= 50:
+            break
+    return results
 
 
 wikibase_properties_id = {"instance of": get_property_id_by_label("instance of", wikibase_api_url),
@@ -83,4 +147,25 @@ wikibase_properties_id = {"instance of": get_property_id_by_label("instance of",
                     "date of fashion collection": get_property_id_by_label("date of fashion collection", wikibase_api_url),
                     "fashion show category": get_property_id_by_label("fashion show category", wikibase_api_url),
                     "fashion house X fashion collection": get_property_id_by_label("fashion house X fashion collection", wikibase_api_url),
-                    "designer of collection": get_property_id_by_label("designer of collection", wikibase_api_url)}
+                    "designer of collection": get_property_id_by_label("designer of collection", wikibase_api_url)}
+
+
+classes_wikibase = {"fashion designer": get_entity_id_by_label("fashion designer", "wikibase")[0],
+                        "fashion house": get_entity_id_by_label("fashion house", "wikibase")[0],
+                        "business": get_entity_id_by_label("business", "wikibase")[0],
+                        "academic institution": get_entity_id_by_label("academic institution", "wikibase")[0],
+                        "geographic location": get_entity_id_by_label("geographic location", "wikibase")[0],
+                        "fashion award": get_entity_id_by_label("fashion award","wikibase")[0],
+                        "gender":get_entity_id_by_label("gender","wikibase")[0] , 
+                        "occupation": get_entity_id_by_label("occupation","wikibase")[0],
+                        "human": get_entity_id_by_label("human","wikibase")[0],
+                        "organization": get_entity_id_by_label("organization","wikibase")[0],
+                        "brand": get_entity_id_by_label("brand","wikibase")[0],
+                        "lifestyle brand": get_entity_id_by_label("lifestyle brand","wikibase")[0],
+                        "privately held company": get_entity_id_by_label("privately held company","wikibase")[0],
+                        "fashion award": get_entity_id_by_label("fashion award", "wikibase")[0],
+                        "fashion season": get_entity_id_by_label("fashion season", "wikibase")[0],
+                        "fashion show category": get_entity_id_by_label("fashion show category", "wikibase")[0],
+                        "fashion season collection": get_entity_id_by_label("fashion season collection", "wikibase")[0],
+                        "fashion journalist": get_entity_id_by_label("fashion journalist", "wikibase")[0],
+                        }