jeremy
/
simplylab
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

change to get

This commit is contained in:
Jeremy Yin 2024-03-15 22:45:53 +08:00
parent cec5266655
commit 605f84688e
2 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
simplylab/main.py
View File

@ -83,15 +83,15 @@ async def get_ai_chat_response(request: Request, req: GetAiChatResponseInput) ->
return res return res
@app.post("/api/v1/get_user_chat_history") @app.get("/api/v1/get_user_chat_history")
async def get_user_chat_history(request: Request, req: GetUserChatHistoryInput) -> GetUserChatHistoryOutput: async def get_user_chat_history(request: Request, user_name: str, last_n: int) -> GetUserChatHistoryOutput:
pvd = Providers(db=request.app.db) pvd = Providers(db=request.app.db)
user = await pvd.user.get_user_by_name(req.user_name) user = await pvd.user.get_user_by_name(user_name)
if not user: if not user:
raise UserNotFoundError(req.user_name) raise UserNotFoundError(user_name)
ctx = Context(user=user) ctx = Context(user=user)
svc = Services(ctx, pvd) svc = Services(ctx, pvd)
res = await svc.chat.get_user_chat_history(req) res = await svc.chat.get_user_chat_history(last_n=last_n)
return res return res

4
simplylab/services/chat.py
View File

@ -40,8 +40,8 @@ class ChatService:
res = GetAiChatResponseOutput(response=response_content) res = GetAiChatResponseOutput(response=response_content)
return res return res
async def get_user_chat_history(self, req: GetUserChatHistoryInput) -> GetUserChatHistoryOutput: async def get_user_chat_history(self, last_n: int) -> GetUserChatHistoryOutput:
messages = await self.pvd.chat.get_user_chat_messages(user_id=self.ctx.user.id, limit=req.last_n) messages = await self.pvd.chat.get_user_chat_messages(user_id=self.ctx.user.id, limit=last_n)
res = [] res = []
for message in messages: for message in messages:
res.append(UserChatMessage(type=message.type.value, text=message.text)) res.append(UserChatMessage(type=message.type.value, text=message.text))