From 7cb624c5f513b80cb6e05e27eb55ccae7f2c5424 Mon Sep 17 00:00:00 2001 From: Cyberes Date: Wed, 4 Oct 2023 12:47:59 -0600 Subject: [PATCH] f --- llm_server/custom_redis.py | 2 +- llm_server/routes/queue.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/llm_server/custom_redis.py b/llm_server/custom_redis.py index d5d278f..485cb58 100644 --- a/llm_server/custom_redis.py +++ b/llm_server/custom_redis.py @@ -112,7 +112,7 @@ class RedisCustom(Redis): def zcard(self, name: KeyT): return self.redis.zcard(self._key(name)) - def hdel(self, name: str, *keys: List): + def hdel(self, name: str, *keys: str): return self.redis.hdel(self._key(name), *keys) def hget( diff --git a/llm_server/routes/queue.py b/llm_server/routes/queue.py index e374311..89d35e6 100644 --- a/llm_server/routes/queue.py +++ b/llm_server/routes/queue.py @@ -62,7 +62,7 @@ class RedisPriorityQueue: def decrement_ip_count(self, client_ip: str, redis_key): new_count = self.redis.hincrby(redis_key, client_ip, -1) if new_count <= 0: - self.redis.hdel(redis_key, [client_ip]) + self.redis.hdel(redis_key, client_ip) def __len__(self): return self.redis.zcard('queue')