cyberes
/
local-llm-server
Archived
1
1
Fork 0
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity

change proompters 1 min to 5 min

This commit is contained in:
Cyberes 2023-09-20 21:21:22 -06:00
parent 8593198216
commit f9a80f3028
2 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
llm_server/routes/stats.py
View File

@ -4,7 +4,7 @@ from threading import Lock, Thread
from llm_server.routes.cache import redis
# proompters_1_min = 0
# proompters_5_min = 0
# concurrent_semaphore = Semaphore(concurrent_gens)
server_start_time = datetime.now()
@ -93,7 +93,7 @@ class SemaphoreCheckerThread(Thread):
while True:
current_time = time.time()
recent_prompters = redis.get_dict('recent_prompters')
new_recent_prompters = {ip: timestamp for ip, timestamp in recent_prompters.items() if current_time - timestamp <= 60}
new_recent_prompters = {ip: timestamp for ip, timestamp in recent_prompters.items() if current_time - timestamp <= 300}
redis.set_dict('recent_prompters', new_recent_prompters)
redis.set('proompters_1_min', len(new_recent_prompters))
redis.set('proompters_5_min', len(new_recent_prompters))
time.sleep(1)

6
llm_server/routes/v1/generate_stats.py
View File

@ -88,14 +88,14 @@ def generate_stats():
base_client_api = x.decode() if x else None
del x
x = redis.get('proompters_1_min')
proompters_1_min = int(x) if x else None
x = redis.get('proompters_5_min')
proompters_5_min = int(x) if x else None
del x
output = {
'stats': {
'proompters': {
'1_min': proompters_1_min,
'5_min': proompters_5_min,
'24_hrs': get_distinct_ips_24h(),
},
'proompts_total': get_total_proompts() if opts.show_num_prompts else None,