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

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

216
server/app/controller/chat/history_controller.py
View file

@ -12,18 +12,27 @@
# limitations under the License.
# ========= Copyright 2025-2026 @ Eigent.ai All Rights Reserved. =========
from fastapi import APIRouter, Depends, HTTPException, Response, Query
import logging
from collections import defaultdict
from fastapi import APIRouter, Depends, HTTPException, Query, Response
from fastapi_pagination import Page
from fastapi_pagination.ext.sqlmodel import paginate
from app.model.chat.chat_history import ChatHistoryOut, ChatHistoryIn, ChatHistory, ChatHistoryUpdate, ChatStatus
from app.model.chat.chat_history_grouped import ProjectGroup, GroupedHistoryResponse
from fastapi_babel import _
from sqlmodel import Session, select, desc, case
from sqlmodel import Session, case, desc, select
from app.component.auth import Auth, auth_must
from app.component.database import session
import logging
from typing import Optional, Dict, List
from collections import defaultdict
from app.model.chat.chat_history import (
ChatHistory,
ChatHistoryIn,
ChatHistoryOut,
ChatHistoryUpdate,
ChatStatus,
)
from app.model.chat.chat_history_grouped import (
GroupedHistoryResponse,
ProjectGroup,
)
logger = logging.getLogger("server_chat_history")
@ -34,18 +43,24 @@ router = APIRouter(prefix="/chat", tags=["Chat History"])
def create_chat_history(data: ChatHistoryIn, session: Session = Depends(session), auth: Auth = Depends(auth_must)):
"""Save new chat history."""
user_id = auth.user.id
try:
data.user_id = user_id
chat_history = ChatHistory(**data.model_dump())
session.add(chat_history)
session.commit()
session.refresh(chat_history)
logger.info("Chat history created", extra={"user_id": user_id, "history_id": chat_history.id, "task_id": data.task_id})
logger.info(
"Chat history created", extra={"user_id": user_id, "history_id": chat_history.id, "task_id": data.task_id}
)
return chat_history
except Exception as e:
session.rollback()
logger.error("Chat history creation failed", extra={"user_id": user_id, "task_id": data.task_id, "error": str(e)}, exc_info=True)
logger.error(
"Chat history creation failed",
extra={"user_id": user_id, "task_id": data.task_id, "error": str(e)},
exc_info=True,
)
raise HTTPException(status_code=500, detail="Internal server error")
@ -53,7 +68,7 @@ def create_chat_history(data: ChatHistoryIn, session: Session = Depends(session)
def list_chat_history(session: Session = Depends(session), auth: Auth = Depends(auth_must)) -> Page[ChatHistoryOut]:
"""List chat histories for current user."""
user_id = auth.user.id
# Order by created_at descending, but fallback to id descending for old records without timestamps
# This ensures newer records with timestamps come first, followed by old records ordered by id
stmt = (
@ -62,25 +77,25 @@ def list_chat_history(session: Session = Depends(session), auth: Auth = Depends(
.order_by(
desc(case((ChatHistory.created_at.is_(None), 0), else_=1)), # Non-null created_at first
desc(ChatHistory.created_at), # Then by created_at descending
desc(ChatHistory.id) # Finally by id descending for records with same/null created_at
desc(ChatHistory.id), # Finally by id descending for records with same/null created_at
)
)
result = paginate(session, stmt)
total = result.total if hasattr(result, 'total') else 0
total = result.total if hasattr(result, "total") else 0
logger.debug("Chat histories listed", extra={"user_id": user_id, "total": total})
return result
@router.get("/histories/grouped", name="get grouped chat history")
def list_grouped_chat_history(
include_tasks: Optional[bool] = Query(True, description="Whether to include individual tasks in groups"),
session: Session = Depends(session),
auth: Auth = Depends(auth_must)
include_tasks: bool | None = Query(True, description="Whether to include individual tasks in groups"),
session: Session = Depends(session),
auth: Auth = Depends(auth_must),
) -> GroupedHistoryResponse:
"""List chat histories grouped by project_id for current user."""
user_id = auth.user.id
# Get all histories for the user, ordered by creation time
stmt = (
select(ChatHistory)
@ -88,88 +103,93 @@ def list_grouped_chat_history(
.order_by(
desc(case((ChatHistory.created_at.is_(None), 0), else_=1)), # Non-null created_at first
desc(ChatHistory.created_at), # Then by created_at descending
desc(ChatHistory.id) # Finally by id descending for records with same/null created_at
desc(ChatHistory.id), # Finally by id descending for records with same/null created_at
)
)
histories = session.exec(stmt).all()
# Group histories by project_id
project_map: Dict[str, Dict] = defaultdict(lambda: {
'project_id': '',
'project_name': None,
'total_tokens': 0,
'task_count': 0,
'latest_task_date': '',
'last_prompt': None,
'tasks': [],
'total_completed_tasks': 0,
'total_ongoing_tasks': 0,
'average_tokens_per_task': 0
})
project_map: dict[str, dict] = defaultdict(
lambda: {
"project_id": "",
"project_name": None,
"total_tokens": 0,
"task_count": 0,
"latest_task_date": "",
"last_prompt": None,
"tasks": [],
"total_completed_tasks": 0,
"total_ongoing_tasks": 0,
"average_tokens_per_task": 0,
}
)
for history in histories:
# Use project_id if available, fallback to task_id
project_id = history.project_id if history.project_id else history.task_id
project_data = project_map[project_id]
# Initialize project data
if not project_data['project_id']:
project_data['project_id'] = project_id
project_data['project_name'] = history.project_name or f"Project {project_id}"
project_data['latest_task_date'] = history.created_at.isoformat() if history.created_at else ''
project_data['last_prompt'] = history.question # Set the most recent question
if not project_data["project_id"]:
project_data["project_id"] = project_id
project_data["project_name"] = history.project_name or f"Project {project_id}"
project_data["latest_task_date"] = history.created_at.isoformat() if history.created_at else ""
project_data["last_prompt"] = history.question # Set the most recent question
# Convert to ChatHistoryOut format
history_out = ChatHistoryOut(**history.model_dump())
# Add task to project if requested
if include_tasks:
project_data['tasks'].append(history_out)
project_data["tasks"].append(history_out)
# Update project statistics
project_data['task_count'] += 1
project_data['total_tokens'] += history.tokens or 0
project_data["task_count"] += 1
project_data["total_tokens"] += history.tokens or 0
# Count completed and failed tasks
# ChatStatus.ongoing = 1, ChatStatus.done = 2
if history.status == ChatStatus.done:
project_data['total_completed_tasks'] += 1
project_data["total_completed_tasks"] += 1
elif history.status == ChatStatus.ongoing:
project_data['total_ongoing_tasks'] += 1
project_data["total_ongoing_tasks"] += 1
else:
# Only count as failed if not ongoing and not done
project_data['total_failed_tasks'] += 1
project_data["total_failed_tasks"] += 1
# Update latest task date and last prompt
if history.created_at:
task_date = history.created_at.isoformat()
if not project_data['latest_task_date'] or task_date > project_data['latest_task_date']:
project_data['latest_task_date'] = task_date
project_data['last_prompt'] = history.question
if not project_data["latest_task_date"] or task_date > project_data["latest_task_date"]:
project_data["latest_task_date"] = task_date
project_data["last_prompt"] = history.question
# Convert to ProjectGroup objects and sort
projects = []
for project_data in project_map.values():
# Sort tasks within each project by creation date (oldest first)
if include_tasks:
project_data['tasks'].sort(key=lambda x: (x.created_at is None, x.created_at or ''), reverse=False)
project_data["tasks"].sort(key=lambda x: (x.created_at is None, x.created_at or ""), reverse=False)
project_group = ProjectGroup(**project_data)
projects.append(project_group)
# Sort projects by latest task date (newest first)
projects.sort(key=lambda x: x.latest_task_date, reverse=True)
response = GroupedHistoryResponse(projects=projects)
logger.debug("Grouped chat histories listed", extra={
"user_id": user_id,
"total_projects": response.total_projects,
"total_tasks": response.total_tasks,
"include_tasks": include_tasks
})
logger.debug(
"Grouped chat histories listed",
extra={
"user_id": user_id,
"total_projects": response.total_projects,
"total_tasks": response.total_tasks,
"include_tasks": include_tasks,
},
)
return response
@ -178,15 +198,18 @@ def delete_chat_history(history_id: str, session: Session = Depends(session), au
"""Delete chat history."""
user_id = auth.user.id
history = session.exec(select(ChatHistory).where(ChatHistory.id == history_id)).first()
if not history:
logger.warning("Chat history not found for deletion", extra={"user_id": user_id, "history_id": history_id})
raise HTTPException(status_code=404, detail="Chat History not found")
if history.user_id != user_id:
logger.warning("Unauthorized deletion attempt", extra={"user_id": user_id, "history_id": history_id, "owner_id": history.user_id})
logger.warning(
"Unauthorized deletion attempt",
extra={"user_id": user_id, "history_id": history_id, "owner_id": history.user_id},
)
raise HTTPException(status_code=403, detail="You are not allowed to delete this chat history")
try:
session.delete(history)
session.commit()
@ -194,7 +217,11 @@ def delete_chat_history(history_id: str, session: Session = Depends(session), au
return Response(status_code=204)
except Exception as e:
session.rollback()
logger.error("Chat history deletion failed", extra={"user_id": user_id, "history_id": history_id, "error": str(e)}, exc_info=True)
logger.error(
"Chat history deletion failed",
extra={"user_id": user_id, "history_id": history_id, "error": str(e)},
exc_info=True,
)
raise HTTPException(status_code=500, detail="Internal server error")
@ -211,7 +238,10 @@ def update_chat_history(
raise HTTPException(status_code=404, detail="Chat History not found")
if history.user_id != user_id:
logger.warning("Unauthorized update attempt", extra={"user_id": user_id, "history_id": history_id, "owner_id": history.user_id})
logger.warning(
"Unauthorized update attempt",
extra={"user_id": user_id, "history_id": history_id, "owner_id": history.user_id},
)
raise HTTPException(status_code=403, detail="You are not allowed to update this chat history")
try:
@ -219,29 +249,29 @@ def update_chat_history(
history.update_fields(update_data)
history.save(session)
session.refresh(history)
logger.info("Chat history updated", extra={"user_id": user_id, "history_id": history_id, "fields_updated": list(update_data.keys())})
logger.info(
"Chat history updated",
extra={"user_id": user_id, "history_id": history_id, "fields_updated": list(update_data.keys())},
)
return history
except Exception as e:
logger.error("Chat history update failed", extra={"user_id": user_id, "history_id": history_id, "error": str(e)}, exc_info=True)
logger.error(
"Chat history update failed",
extra={"user_id": user_id, "history_id": history_id, "error": str(e)},
exc_info=True,
)
raise HTTPException(status_code=500, detail="Internal server error")
@router.put("/project/{project_id}/name", name="update project name")
def update_project_name(
project_id: str,
new_name: str,
session: Session = Depends(session),
auth: Auth = Depends(auth_must)
project_id: str, new_name: str, session: Session = Depends(session), auth: Auth = Depends(auth_must)
):
"""Update project name for all tasks in a project."""
user_id = auth.user.id
# Get all histories for this project
stmt = (
select(ChatHistory)
.where(ChatHistory.project_id == project_id)
.where(ChatHistory.user_id == user_id)
)
stmt = select(ChatHistory).where(ChatHistory.project_id == project_id).where(ChatHistory.user_id == user_id)
histories = session.exec(stmt).all()
@ -257,15 +287,17 @@ def update_project_name(
session.commit()
logger.info("Project name updated", extra={
"user_id": user_id,
"project_id": project_id,
"new_name": new_name,
"updated_count": len(histories)
})
logger.info(
"Project name updated",
extra={"user_id": user_id, "project_id": project_id, "new_name": new_name, "updated_count": len(histories)},
)
return Response(status_code=200)
except Exception as e:
session.rollback()
logger.error("Project name update failed", extra={"user_id": user_id, "project_id": project_id, "error": str(e)}, exc_info=True)
raise HTTPException(status_code=500, detail="Internal server error")
logger.error(
"Project name update failed",
extra={"user_id": user_id, "project_id": project_id, "error": str(e)},
exc_info=True,
)
raise HTTPException(status_code=500, detail="Internal server error")