diff --git a/cryptoai/routes/analysis.py b/cryptoai/routes/analysis.py index 4379614..5f706fc 100644 --- a/cryptoai/routes/analysis.py +++ b/cryptoai/routes/analysis.py @@ -211,13 +211,14 @@ async def chat(request: ChatRequest, session: Session = Depends(get_db)): # 检查用户是否订阅 + is_member = False user_subscription_manager = UserSubscriptionManager(session) user_subscription = user_subscription_manager.get_subscription_by_user_id(current_user["id"]) - is_member = user_subscription and user_subscription["expire_time"] > datetime.now() - user_points = current_user["points"] + if user_subscription: + is_member = user_subscription["expire_time"] > datetime.now() - if not is_member and user_points < 1: - raise HTTPException(status_code=999, detail="你的免费次数不足,你可以订阅会员。") + if not is_member: + raise HTTPException(status_code=999, detail="你的免费次数不足,你可以订阅会员。") payload = { "inputs" : {}, @@ -246,11 +247,6 @@ async def chat(request: ChatRequest, status_code=response.status_code, detail=f"Failed to get response from Dify API: {response.text}" ) - - # 扣除用户积分 - if not is_member: - manager = UserManager(session) - manager.consume_user_points(current_user["id"], 1) # 获取response的stream def stream_response(): diff --git a/docker-compose.yml b/docker-compose.yml index 45beee4..28b8820 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -29,7 +29,7 @@ services: cryptoai-api: build: . container_name: cryptoai-api - image: cryptoai-api:0.2.24 + image: cryptoai-api:0.2.25 restart: always ports: - "8000:8000"