diff --git a/api/apps/llm_app.py b/api/apps/llm_app.py index 473b748fb8..5e7c302cbc 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 36b4700f00..8d96fefbcb 100644 --- a/api/db/services/task_service.py +++ b/api/db/services/task_service.py @@ -159,4 +159,4 @@ def new_task(): 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