wassname/stampy-chat
1
0
Fork 0
mirror of https://github.com/wassname/stampy-chat.git synced 2026-09-10 12:40:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix logging times

Browse Source
This commit is contained in:
Fraser
2023-07-08 15:36:14 -04:00
parent 8e99e566fe
commit b292e76fae
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
api/chat.py
+1 -1
View File
@@ -154,7 +154,7 @@ def talk_to_robot_internal(index, query: str, history: List[Dict[str, str]], k:
t2 = time.time()
print("Time to get response: ", t2 - t1, " s")
print(f'Time to get response: {t2-t1:.2f}s')
if DEBUG_PRINT:
print('\n' * 10)
api/get_blocks.py
+2 -2
View File
@@ -76,7 +76,7 @@ def get_top_k_blocks(index, user_query: str, k: int) -> List[Block]:
query_embedding = get_embedding(user_query)
t1 = time.time()
print("Time to get embedding: ", t1 - t)
print(f'Time to get embedding: {t1-t:.2f}s')
query_response = index.query(
namespace="alignment-search", # ugly, sorry
@@ -103,7 +103,7 @@ def get_top_k_blocks(index, user_query: str, k: int) -> List[Block]:
t2 = time.time()
print("Time to get top-k blocks: ", t2 - t1)
print(f'Time to get top-k blocks: {t2-t1:.2f}s')
# for all blocks that are "the same" (same title, author, date, url, tags),
# combine their text with "....." in between. Return them in order such