From 7ddb2f19bee679d72cfc6d6affe4ef4df1f6b3a1 Mon Sep 17 00:00:00 2001
From: KevinHuSh <kevinhu.sh@gmail.com>
Date: Wed, 8 May 2024 15:20:45 +0800
Subject: [PATCH] make sure to raise exception if redis is not there (#674)

### What problem does this PR solve?

### Type of change

- [x] Refactoring
---
 api/apps/llm_app.py             | 10 ++++++++++
 api/db/services/task_service.py |  2 +-
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/api/apps/llm_app.py b/api/apps/llm_app.py
index 473b748fb..5e7c302cb 100644
--- a/api/apps/llm_app.py
+++ b/api/apps/llm_app.py
@@ -142,6 +142,16 @@ def add_llm():
     return get_json_result(data=True)
 
 
+@manager.route('/delete_llm', methods=['POST'])
+@login_required
+@validate_request("llm_factory", "llm_name")
+def delete_llm():
+    req = request.json
+    TenantLLMService.filter_delete(
+            [TenantLLM.tenant_id == current_user.id, TenantLLM.llm_factory == req["llm_factory"], TenantLLM.llm_name == req["llm_name"]])
+    return get_json_result(data=True)
+
+
 @manager.route('/my_llms', methods=['GET'])
 @login_required
 def my_llms():
diff --git a/api/db/services/task_service.py b/api/db/services/task_service.py
index 36b4700f0..8d96fefbc 100644
--- a/api/db/services/task_service.py
+++ b/api/db/services/task_service.py
@@ -159,4 +159,4 @@ def queue_tasks(doc, bucket, name):
     DocumentService.begin2parse(doc["id"])
 
     for t in tsks:
-        REDIS_CONN.queue_product(SVR_QUEUE_NAME, message=t)
\ No newline at end of file
+        assert REDIS_CONN.queue_product(SVR_QUEUE_NAME, message=t), "Can't access Redis. Please check the Redis' status."
\ No newline at end of file