jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

♻️ (monitor.py): refactor update_message to return MessageModelResponse

Browse source 
♻️ (service.py): update SQL query to use index instead of id
♻️ (api.tsx): refactor duplicate request check logic
✨ (check-duplicate-requests.ts): add helper to check and store duplicate requests
🐛 (messagesStore.ts): fix message update logic to use index instead of id
This commit is contained in:
cristhianzl 2024-06-03 17:05:39 -03:00
commit 0e56617e26
5 changed files with 41 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/backend/base/langflow/api/v1/monitor.py
View file

@ -86,9 +86,10 @@ async def update_message(
):
try:
message_dict = message.model_dump(exclude_none=True)
df = monitor_service.update_message(message_id=message_id, **message_dict)
dicts = df.to_dict(orient="records")
return [MessageModelResponse(**d) for d in dicts]
message_dict.pop("index", None)
monitor_service.update_message(message_id=message_id, **message_dict)
return MessageModelResponse(index=message_id, **message_dict)
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))

4
src/backend/base/langflow/services/monitor/service.py
View file

@ -113,8 +113,8 @@ class MonitorService(Service):
return self.exec_query(query)
def update_message(self, message_id: int, **kwargs):
query = f"UPDATE messages SET {', '.join(f'{k} = {v}' for k, v in kwargs.items())} WHERE id = {message_id}"
query = f"""UPDATE messages SET {', '.join(f"{k} = '{v}'" for k, v in kwargs.items())} WHERE index = {message_id}"""
return self.exec_query(query)
def add_message(self, message: MessageModel):