Skip to content

Commit 90bb9e2

Browse files
committed
clean message in list
1 parent e966389 commit 90bb9e2

File tree

2 files changed

+7
-4
lines changed

2 files changed

+7
-4
lines changed

src/codegate/api/v1.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -510,9 +510,10 @@ async def get_workspace_messages(
510510
logger.warning(f"Skipping prompt {prompt.id}. No messages found")
511511
continue
512512

513-
# message is just the first entry in the request
513+
# message is just the first entry in the request, cleaned properly
514+
message = v1_processing.parse_question_answer(messages[0])
514515
message_obj = v1_models.ChatMessage(
515-
message=messages[0], timestamp=prompt.timestamp, message_id=prompt.id
516+
message=message, timestamp=prompt.timestamp, message_id=prompt.id
516517
)
517518

518519
# count total alerts for the prompt

src/codegate/db/connection.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -746,13 +746,15 @@ def _build_prompt_query(
746746

747747
if filter_by_alert_trigger_categories:
748748
filter_conditions.append(
749-
"AND (a.trigger_category IN :filter_by_alert_trigger_categories OR a.trigger_category IS NULL)"
749+
"""AND (a.trigger_category IN :filter_by_alert_trigger_categories
750+
OR a.trigger_category IS NULL)"""
750751
)
751752
conditions["filter_by_alert_trigger_categories"] = filter_by_alert_trigger_categories
752753

753754
if filter_by_alert_trigger_types:
754755
filter_conditions.append(
755-
"AND EXISTS (SELECT 1 FROM alerts a2 WHERE a2.prompt_id = p.id AND a2.trigger_type IN :filter_by_alert_trigger_types)"
756+
"""AND EXISTS (SELECT 1 FROM alerts a2 WHERE
757+
a2.prompt_id = p.id AND a2.trigger_type IN :filter_by_alert_trigger_types)"""
756758
)
757759
conditions["filter_by_alert_trigger_types"] = filter_by_alert_trigger_types
758760

0 commit comments

Comments
 (0)