Skip to content

Commit 28ecba5

Browse files
NyakultNyakult
committed
format code
1 parent 4af98e6 commit 28ecba5

File tree

5 files changed

+46
-46
lines changed

5 files changed

+46
-46
lines changed

‎evaluation/scripts/locomo/locomo_ingestion.py‎

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -89,34 +89,34 @@ def process_user(conv_idx, frame, locomo_df, version):
8989

9090
client = None
9191
if frame == "mem0" or frame == "mem0_graph":
92-
from utils.client import mem0Client
92+
from utils.client import Mem0Client
9393

94-
client = mem0Client(enable_graph="graph" in frame)
94+
client = Mem0Client(enable_graph="graph" in frame)
9595
client.client.update_project(custom_instructions=custom_instructions)
9696
client.client.delete_all(user_id=speaker_a_user_id)
9797
client.client.delete_all(user_id=speaker_b_user_id)
9898
elif frame == "memos-api":
99-
from utils.client import memosApiClient
99+
from utils.client import MemosApiClient
100100

101-
client = memosApiClient()
101+
client = MemosApiClient()
102102
elif frame == "memobase":
103-
from utils.client import memobaseClient
103+
from utils.client import MemobaseClient
104104

105-
client = memobaseClient()
105+
client = MemobaseClient()
106106
all_users = client.client.get_all_users(limit=5000)
107107
for user in all_users:
108108
if user["additional_fields"]["user_id"] in [speaker_a_user_id, speaker_b_user_id]:
109109
client.client.delete_user(user["id"])
110110
speaker_a_user_id = client.client.add_user({"user_id": speaker_a_user_id})
111111
speaker_b_user_id = client.client.add_user({"user_id": speaker_b_user_id})
112112
elif frame == "memu":
113-
from utils.client import memuClient
113+
from utils.client import MemuClient
114114

115-
client = memuClient()
115+
client = MemuClient()
116116
elif frame == "supermemory":
117-
from utils.client import supermemoryClient
117+
from utils.client import SupermemoryClient
118118

119-
client = supermemoryClient()
119+
client = SupermemoryClient()
120120
sessions_to_process = []
121121
for session_idx in range(max_session_count):
122122
session_key = f"session_{session_idx}"

‎evaluation/scripts/locomo/locomo_search.py‎

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -233,31 +233,31 @@ def process_user(conv_idx, locomo_df, frame, version, top_k=20, num_workers=1):
233233

234234
client = None
235235
if frame == "mem0" or frame == "mem0_graph":
236-
from utils.client import mem0Client
236+
from utils.client import Mem0Client
237237

238-
client = mem0Client(enable_graph="graph" in frame)
238+
client = Mem0Client(enable_graph="graph" in frame)
239239
elif frame == "memos-api":
240-
from utils.client import memosApiClient
240+
from utils.client import MemosApiClient
241241

242-
client = memosApiClient()
242+
client = MemosApiClient()
243243
elif frame == "memobase":
244-
from utils.client import memobaseClient
244+
from utils.client import MemobaseClient
245245

246-
client = memobaseClient()
246+
client = MemobaseClient()
247247
users = client.client.get_all_users(limit=5000)
248248
for u in users:
249249
if u["additional_fields"]["user_id"] == speaker_a_user_id:
250250
speaker_a_user_id = u["id"]
251251
if u["additional_fields"]["user_id"] == speaker_b_user_id:
252252
speaker_b_user_id = u["id"]
253253
elif frame == "memu":
254-
from utils.client import memuClient
254+
from utils.client import MemuClient
255255

256-
client = memuClient()
256+
client = MemuClient()
257257
elif frame == "supermemory":
258-
from utils.client import supermemoryClient
258+
from utils.client import SupermemoryClient
259259

260-
client = supermemoryClient()
260+
client = SupermemoryClient()
261261

262262
metadata = {
263263
"speaker_a": speaker_a,

‎evaluation/scripts/longmemeval/lme_ingestion.py‎

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -68,31 +68,31 @@ def ingest_conv(lme_df, version, conv_idx, frame, success_records, f):
6868
print("=" * 80)
6969

7070
if frame == "mem0" or frame == "mem0_graph":
71-
from utils.client import mem0Client
71+
from utils.client import Mem0Client
7272

73-
client = mem0Client(enable_graph="graph" in frame)
73+
client = Mem0Client(enable_graph="graph" in frame)
7474
client.client.delete_all(user_id=user_id)
7575
elif frame == "memos-api":
76-
from utils.client import memosApiClient
76+
from utils.client import MemosApiClient
7777

78-
client = memosApiClient()
78+
client = MemosApiClient()
7979
elif frame == "memobase":
80-
from utils.client import memobaseClient
80+
from utils.client import MemobaseClient
8181

82-
client = memobaseClient()
82+
client = MemobaseClient()
8383
all_users = client.client.get_all_users(limit=5000)
8484
for user in all_users:
8585
if user["additional_fields"]["user_id"] == user_id:
8686
client.client.delete_user(user["id"])
8787
user_id = client.client.add_user({"user_id": user_id})
8888
elif frame == "memu":
89-
from utils.client import memuClient
89+
from utils.client import MemuClient
9090

91-
client = memuClient()
91+
client = MemuClient()
9292
elif frame == "supermemory":
93-
from utils.client import supermemoryClient
93+
from utils.client import SupermemoryClient
9494

95-
client = supermemoryClient()
95+
client = SupermemoryClient()
9696

9797
for idx, session in enumerate(sessions):
9898
if f"{conv_idx}_{idx}" not in success_records:

‎evaluation/scripts/longmemeval/lme_search.py‎

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -106,33 +106,33 @@ def process_user(lme_df, conv_idx, frame, version, top_k=20):
106106
return existing_results
107107

108108
if "mem0" in frame:
109-
from utils.client import mem0Client
109+
from utils.client import Mem0Client
110110

111-
client = mem0Client(enable_graph="graph" in frame)
111+
client = Mem0Client(enable_graph="graph" in frame)
112112
context, duration_ms = mem0_search(client, question, user_id, top_k)
113113
elif frame == "memobase":
114-
from utils.client import memobaseClient
114+
from utils.client import MemobaseClient
115115

116-
client = memobaseClient()
116+
client = MemobaseClient()
117117
users = client.client.get_all_users(limit=5000)
118118
for u in users:
119119
if u["additional_fields"]["user_id"] == user_id:
120120
user_id = u["id"]
121121
context, duration_ms = memobase_search(client, question, user_id, top_k)
122122
elif frame == "memos-api":
123-
from utils.client import memosApiClient
123+
from utils.client import MemosApiClient
124124

125-
client = memosApiClient()
125+
client = MemosApiClient()
126126
context, duration_ms = memos_search(client, question, user_id, top_k)
127127
elif frame == "memu":
128-
from utils.client import memuClient
128+
from utils.client import MemuClient
129129

130-
client = memuClient()
130+
client = MemuClient()
131131
context, duration_ms = memu_search(client, question, user_id, top_k)
132132
elif frame == "supermemory":
133-
from utils.client import supermemoryClient
133+
from utils.client import SupermemoryClient
134134

135-
client = supermemoryClient()
135+
client = SupermemoryClient()
136136
context, duration_ms = supermemory_search(client, question, user_id, top_k)
137137

138138
search_results[user_id].append(

‎evaluation/scripts/utils/client.py‎

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
load_dotenv()
1111

1212

13-
class zepClient:
13+
class ZepClient:
1414
def __init__(self):
1515
from zep_cloud.client import Zep
1616

@@ -42,7 +42,7 @@ def search(self, query, user_id, top_k):
4242
return nodes, edges
4343

4444

45-
class mem0Client:
45+
class Mem0Client:
4646
def __init__(self, enable_graph=False):
4747
from mem0 import MemoryClient
4848

@@ -85,7 +85,7 @@ def search(self, query, user_id, top_k):
8585
return res
8686

8787

88-
class memobaseClient:
88+
class MemobaseClient:
8989
def __init__(self):
9090
from memobase import MemoBaseClient
9191

@@ -114,7 +114,7 @@ def search(self, query, user_id, top_k):
114114
return memories
115115

116116

117-
class memosApiClient:
117+
class MemosApiClient:
118118
def __init__(self):
119119
self.memos_url = os.getenv("MEMOS_URL")
120120
self.headers = {"Content-Type": "application/json", "Authorization": os.getenv("MEMOS_KEY")}
@@ -155,7 +155,7 @@ def search(self, query, user_id, top_k):
155155
return json.loads(response.text)["data"]
156156

157157

158-
class supermemoryClient:
158+
class SupermemoryClient:
159159
def __init__(self):
160160
from supermemory import Supermemory
161161

@@ -197,7 +197,7 @@ def search(self, query, user_id, top_k):
197197
raise e
198198

199199

200-
class memuClient:
200+
class MemuClient:
201201
def __init__(self):
202202
from memu import MemuClient
203203

0 commit comments

Comments
 (0)