vrr/eigent
2
0
Fork 0
mirror of https://github.com/eigent-ai/eigent.git synced 2026-05-05 15:30:48 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

chore: fix pre commit format and pipeline issue (#1144)

Browse source
This commit is contained in:
Wendong-Fan 2026-02-04 00:06:29 +00:00 • committed by GitHub
parent 2256497dff
commit 893f51fc82
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
970 changed files with 11235 additions and 9533 deletions
Download patch file Download diff file Expand all files Collapse all files

39
server/app/controller/chat/share_controller.py
View file

@ -12,18 +12,24 @@
# limitations under the License.
# ========= Copyright 2025-2026 @ Eigent.ai All Rights Reserved. =========
from fastapi import APIRouter, Depends, HTTPException, Response
from sqlmodel import Session, asc, select
from app.component.database import session
import json
import asyncio
from itsdangerous import SignatureExpired, BadTimeSignature
from starlette.responses import StreamingResponse
from app.model.chat.chat_share import ChatHistoryShareOut, ChatShare, ChatShareIn
from app.model.chat.chat_step import ChatStep
from app.model.chat.chat_history import ChatHistory
import json
import logging
from fastapi import APIRouter, Depends, HTTPException
from itsdangerous import BadTimeSignature, SignatureExpired
from sqlmodel import Session, asc, select
from starlette.responses import StreamingResponse
from app.component.database import session
from app.model.chat.chat_history import ChatHistory
from app.model.chat.chat_share import (
ChatHistoryShareOut,
ChatShare,
ChatShareIn,
)
from app.model.chat.chat_step import ChatStep
logger = logging.getLogger("server_chat_share")
router = APIRouter(prefix="/chat", tags=["Chat Share"])
@ -63,7 +69,7 @@ async def share_playback(token: str, session: Session = Depends(session), delay_
if delay_time > 5:
logger.debug("Delay time capped", extra={"requested": delay_time, "capped": 5})
delay_time = 5
try:
task_id = ChatShare.verify_token(token, False)
except SignatureExpired:
@ -83,8 +89,11 @@ async def share_playback(token: str, session: Session = Depends(session), delay_
yield f"data: {json.dumps({'error': 'No steps found for this task.'})}\n\n"
return
logger.info("Shared chat playback started", extra={"task_id": task_id, "step_count": len(steps), "delay_time": delay_time})
logger.info(
"Shared chat playback started",
extra={"task_id": task_id, "step_count": len(steps), "delay_time": delay_time},
)
for idx, step in enumerate(steps, start=1):
step_data = {
"id": step.id,
@ -94,10 +103,10 @@ async def share_playback(token: str, session: Session = Depends(session), delay_
"created_at": step.created_at.isoformat() if step.created_at else None,
}
yield f"data: {json.dumps(step_data)}\n\n"
if delay_time > 0 and step.step != "create_agent":
await asyncio.sleep(delay_time)
logger.info("Shared chat playback completed", extra={"task_id": task_id, "step_count": len(steps)})
except Exception as e:
logger.error("Shared chat playback error", extra={"task_id": task_id, "error": str(e)}, exc_info=True)
@ -115,4 +124,4 @@ def create_share_link(data: ChatShareIn):
return {"share_token": share_token}
except Exception as e:
logger.error("Share link creation failed", extra={"task_id": data.task_id, "error": str(e)}, exc_info=True)
raise HTTPException(status_code=500, detail="Internal server error")
raise HTTPException(status_code=500, detail="Internal server error")