feat: agentbase-v1.1 interface update

data/{agentbase.csv → agentbase-v1.1.csv}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:da62fa2bd13b459f299df003a79edbc1f410c66d1ea06d966510d6e7e1c16563
-size 13768035
+oid sha256:4b76495040cb17bea5f45a2c051c040893bfea9f624f18fc836ddb5d75c68cb3
+size 14146922

data/embeddings/BAAI_bge-large-en-v1.5_18d58ad4370dc14d.npz

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1ac8dde554afdb1f6b51d5ef2838d48d09f718be1c2da3bbb20d0c707f927d59
-size 37193368
+oid sha256:871437bcb334f3cd37d0863d10a98b895a24017fbc31e718f90d048fda796108
+size 37193182

data/embeddings/BAAI_bge-large-en-v1.5_35225dd43945b254.npz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dba2da69db6c7e31a1a6a11630d7c07ceb799a62d804de5494ee030099e76761
+size 37193033

data/embeddings/BAAI_bge-large-en-v1.5_63af9b74ada2385d.npz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:153580c91d9a8f52aedc70c95f2bff83ac1d0052d68e9f53fd8a216867fad38f
+size 37193486

data/embeddings/BAAI_bge-large-en-v1.5_74e561a3a431c1aa.npz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:76b07cf52ca92c1d4754fa5be65dbd5e61c8f169cde7d78df4017bfc2e68d08f
+size 37192831

data/embeddings/BAAI_bge-large-en-v1.5_8c6dc4a78211c0a6.npz

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:826b33a8f494b959e50164b59ae65bd550258ab0b6218b18533f9098b4f7f478
-size 37192686
+oid sha256:ed776a604d0ad19da7fcf8ccd22db8d2a89edb01529c2739d2431ed41b449335
+size 37191805

data/embeddings/BAAI_bge-large-en-v1.5_cf2365f1cf47d288.npz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:aebdb5484fb2ded2131984ad6e88ea4c45820fea4052ced8a9b241c88beb8d1b
+size 37192370

interface.py

@@ -6,6 +6,7 @@ Date: [Current Date]
 """
 
 from typing import List, Tuple
+from pathlib import Path
 import streamlit as st
 import pandas as pd
 import numpy as np
@@ -64,7 +65,7 @@ class AgentBaseUI:
         if "query" not in st.session_state:
             st.session_state.query = ""
 
-        query_suggestions = list(load_queries("data/queries.json").values())
+        query_suggestions = list(load_queries("data/samples.json").values())
         suggestion_cols = st.columns(len(query_suggestions))
         for i, suggestion in enumerate(query_suggestions):
             if suggestion_cols[i].button(suggestion):
@@ -105,8 +106,10 @@ class AgentBaseUI:
                 "agent_url": st.column_config.LinkColumn("agent_url", display_text="Visit →"),
                 "agent_description": st.column_config.TextColumn("agent_description", width="large"),
                 "agent_accessibility": st.column_config.TextColumn("agent_accessibility", width="small"),
+                "agent_pricing": st.column_config.TextColumn("agent_pricing", width="medium"),
+                "base_model": st.column_config.TextColumn("base_model", width="medium"),
             }
-            key_columns = ['agent_name', 'platform_name', 'agent_description', 'agent_url', 'scores']
+            key_columns = ['agent_name', 'platform_name', 'agent_description', 'agent_pricing', 'base_model', 'agent_url', 'scores']
             if (self.filtered_df['scores'] == 0).all(): key_columns.remove("scores")
             st.dataframe(
                 self.filtered_df[key_columns].head(top_k),
@@ -137,10 +140,11 @@ class AgentBaseUI:
         for index, row in filtered_df.iterrows():
             score = dict(res).get(row['agent_id'], 0)
             filtered_df.at[index, 'scores'] = score
+    
         return filtered_df.sort_values(by="scores", ascending=False)
     
     def info_panel(self):
-        with st.expander(f"View AgentBase-v1.0"):
+        with st.expander(f"View AgentBase-v1.1"):
             st.dataframe(
                 self.agents_df,
                 self.agents_df.columns,
@@ -155,7 +159,11 @@ class AgentBaseUI:
             )
 
 if __name__ == "__main__":
-    agentbaseui = AgentBaseUI(agentbase_path="data/agentbase.csv", platforms_path="data/platforms.csv")
+    BASE_DIR = Path(__file__).resolve().parent
+    agentbase_path = BASE_DIR / "data/agentbase-v1.1.csv"
+    platforms_path = BASE_DIR / "data/platforms.csv"
+
+    agentbaseui = AgentBaseUI(agentbase_path, platforms_path)
     agentbaseui.header_panel()
     agentbaseui.retrieval_panel()
     agentbaseui.info_panel()