Selaa lähdekoodia

对话列表过滤name为空的,设置最多1000条返回

zhangqian 1 vuosi sitten
vanhempi
sitoutus
0e5f87417a
2 muutettua tiedostoa jossa 2 lisäystä ja 4 poistoa
  1. 1 2
      app/api/agent.py
  2. 1 2
      app/service/bisheng.py

+ 1 - 2
app/api/agent.py

@@ -29,7 +29,7 @@ async def agent_list(db: Session = Depends(get_db)):
 async def chat_list(
         agent_id: str,
         page: int = Query(1, ge=1),
-        limit: int = Query(10, ge=1, le=100),
+        limit: int = Query(1000, ge=1, le=1000),
         db: Session = Depends(get_db),
         current_user: UserModel = Depends(get_current_user)):
     agent = db.query(AgentModel).filter(AgentModel.id == agent_id).first()
@@ -49,7 +49,6 @@ async def chat_list(
         bisheng_service = BishengService(base_url=settings.sgb_base_url)
         try:
             token = get_bisheng_token(db, current_user.id)
-            print(f"Calling get_chat_sessions with token={token}, page={page}, limit={limit}")
             result = await bisheng_service.get_chat_sessions(token, page, limit)
         except Exception as e:
             raise HTTPException(status_code=500, detail=str(e))

+ 1 - 2
app/service/bisheng.py

@@ -70,8 +70,7 @@ class BishengService:
                     "updated_time": int(datetime.strptime(item["update_time"], "%Y-%m-%dT%H:%M:%S").timestamp() * 1000)
                 }
                 for item in data
-                if
-                "latest_message" in item and "message" in item["latest_message"] and item["latest_message"]["message"]
+                if "latest_message" in item and "message" in item["latest_message"] and item["latest_message"]["message"]
             ]
             return result