Adding the Vendor field

app/backend/data_engine.py

@@ -20,7 +20,7 @@ COLUMNS_TYPES = ["markdown",
 
                  ]
 
-BRANCH = 'main'
+BRANCH = 'adding_vendor_field'
 GIT_URL = f"https://raw.githubusercontent.com/embedding-benchmark/rteb/refs/heads/{BRANCH}/results/"
 DATASET_URL = f"{GIT_URL}datasets.json"
 MODEL_URL = f"{GIT_URL}models.json"

app/backend/data_page.py

@@ -14,7 +14,7 @@ from app.backend.app_init_func import LEADERBOARD_MAP
 from app.backend.constant import LEADERBOARD_ICON_MAP, BASE_URL
 from app.backend.json_util import compress_msgpack, decompress_msgpack
 
-COLUMNS = ['model_name',
+COLUMNS = ['model_name', 'vendor',
            'embd_dtype', 'embd_dim', 'num_params', 'max_tokens', 'similarity',
            'query_instruct', 'corpus_instruct', 'reference'
 
@@ -261,6 +261,12 @@ def render_page(group_name):
                 'suppressSizeToFit': True
 
             },
+            {'headerName': "Vendor",
+             'field': 'vendor',
+             'headerStyle': HEADER_STYLE,
+             'cellStyle': CELL_STYLE,
+             # 'suppressSizeToFit': True
+             },
             {'headerName': "Overall Score",
              'field': avg_column,
              'headerStyle': HEADER_STYLE,