remove logs and add more timeout

app/llm.py

@@ -39,7 +39,7 @@ def few_shots(prompt: str):
         "k": 4,
         "prompt": prompt,
     }
-    response = requests.post(url=f"{ALBERT_BASE_URL}/search", json=data, headers={"Authorization": f"Bearer {ALBERT_API_KEY}"})
+    response = requests.post(url=f"{ALBERT_BASE_URL}/search", json=data, headers={"Authorization": f"Bearer {ALBERT_API_KEY}"}, timeout=120)
     assert response.status_code == 200
     response = response.json()
 

app/subscriptions.py

@@ -23,7 +23,6 @@ def __init__(self, r: Redis, channels):
     def run(self):
         logging.info("listener run")
         for item in self.pubsub.listen():
-            logging.info(f"listener item: {item}")
             if item["type"] == "message" and item["channel"] == self.KILL_PILL.encode():
                 break
 
@@ -42,7 +41,6 @@ def run(self):
                     logging.error(f"\nRequest prompt:\n{data["text"]}\n\nError:\n{error_traceback}")
                     answer = f"error on {data["id"]} request, please resend prompt later."
 
-                logging.info(f"send answer to {data['id']}")
                 self.redis.set(
                     name=data["id"],  # key
                     value=answer,  # model output