kolzchut-ragbot 1.3.0__tar.gz → 1.4.1__tar.gz

{kolzchut_ragbot-1.3.0 → kolzchut_ragbot-1.4.1}/PKG-INFO

@@ -1,8 +1,8 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: kolzchut-ragbot
-Version: 1.3.0
+Version: 1.4.1
 Summary: A search engine using machine learning models and Elasticsearch for advanced document retrieval.
-Home-page: https://github.com/shmuelrob/ragbot
+Home-page: https://github.com/shmuelrob/rag-bot
 Author: Shmuel Robinov
 Author-email: shmuel_robinov@webiks.com
 Classifier: Programming Language :: Python :: 3

{kolzchut_ragbot-1.3.0 → kolzchut_ragbot-1.4.1}/kolzchut_ragbot/model.py

@@ -21,6 +21,31 @@ def index_from_page_id(page_id: int):
     index_postfix = round(page_id / 1000)
     return EMBEDDING_INDEX + "_" + str(index_postfix)
 
+def create_mapping():
+    """
+    Creates a mapping for the model in Elasticsearch.
+    """
+    vector_fields = {f"{name}_{semantic_model}_vectors": {"type": "dense_vector", "dims": 1024}
+                     for name, semantic_model in definitions_singleton.models.items()}
+
+    data_fields = {}
+    for field in definitions_singleton.saved_fields.keys():
+        field_type = definitions_singleton.saved_fields[field]
+        field_mapping = {"type": field_type}
+        if field_type == "date":
+            field_mapping["format"] = "yyyyMMddHHmmss"
+        data_fields[f"{field}"] = field_mapping
+
+    mappings = {
+        "properties": {
+            "last_update": {
+                "type": "date",
+            },
+            **vector_fields,
+            **data_fields,
+        }
+    }
+    return mappings
 
 class Model:
     """
@@ -59,23 +84,11 @@ class Model:
         """
         Creates an index for the model in Elasticsearch.
         """
+        mapping = create_mapping()
         if not self.es_client.indices.exists(index=os.getenv("ES_EMBEDDING_INDEX")):
-            vector_fields = {f"{name}_{semantic_model}_vectors": {"type": "dense_vector", "dims": 1024}
-                             for name, semantic_model in definitions_singleton.models.items()}
-            data_fields = {f"{field}": {"type": definitions_singleton.saved_fields[field]}
-                           for field in definitions_singleton.saved_fields.keys()}
-
             self.es_client.indices.create(
                 index=os.getenv("ES_EMBEDDING_INDEX"),
-                mappings={
-                    "properties": {
-                        "last_update": {
-                            "type": "date"
-                        },
-                        **vector_fields,
-                        **data_fields,
-                    }
-                })
+                mappings=mapping)
 
     def create_or_update_documents(self, paragraphs_dicts: list[dict], update=False):
         """

{kolzchut_ragbot-1.3.0 → kolzchut_ragbot-1.4.1}/kolzchut_ragbot.egg-info/PKG-INFO

@@ -1,8 +1,8 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: kolzchut-ragbot
-Version: 1.3.0
+Version: 1.4.1
 Summary: A search engine using machine learning models and Elasticsearch for advanced document retrieval.
-Home-page: https://github.com/shmuelrob/ragbot
+Home-page: https://github.com/shmuelrob/rag-bot
 Author: Shmuel Robinov
 Author-email: shmuel_robinov@webiks.com
 Classifier: Programming Language :: Python :: 3

{kolzchut_ragbot-1.3.0 → kolzchut_ragbot-1.4.1}/setup.py

@@ -1,18 +1,19 @@
 from setuptools import setup, find_packages
 
 # to deploy the package to PyPI, run the following command:
+# pip install setuptools wheel twine
 # python setup.py sdist bdist_wheel
 # twine upload dist/*
 
 setup(
     name='kolzchut-ragbot',
-    version='1.3.0',
+    version='1.4.1',
     author='Shmuel Robinov',
     author_email='shmuel_robinov@webiks.com',
     description='A search engine using machine learning models and Elasticsearch for advanced document retrieval.',
     long_description=open('README.md').read(),
     long_description_content_type='text/markdown',
-    url='https://github.com/shmuelrob/ragbot',
+    url='https://github.com/shmuelrob/rag-bot',
     packages=find_packages(),
     install_requires=[
         'elasticsearch==8.17.1',