diff --git a/src/services/langflow_history_service.py b/src/services/langflow_history_service.py index 613c2113..ee3366c1 100644 --- a/src/services/langflow_history_service.py +++ b/src/services/langflow_history_service.py @@ -6,7 +6,9 @@ Simplified service that retrieves message history from Langflow using shared cli from typing import List, Dict, Optional, Any from config.settings import clients +from utils.logging_config import get_logger +logger = get_logger(__name__) class LangflowHistoryService: """Simplified service to retrieve message history from Langflow""" @@ -29,14 +31,14 @@ class LangflowHistoryService: if response.status_code == 200: session_ids = response.json() - print(f"Found {len(session_ids)} total sessions from Langflow") + logger.debug(f"Found {len(session_ids)} total sessions from Langflow") return session_ids else: - print(f"Failed to get sessions: {response.status_code} - {response.text}") + logger.error(f"Failed to get sessions: {response.status_code} - {response.text}") return [] except Exception as e: - print(f"Error getting user sessions: {e}") + logger.error(f"Error getting user sessions: {e}") return [] async def get_session_messages(self, user_id: str, session_id: str) -> List[Dict[str, Any]]: @@ -56,11 +58,11 @@ class LangflowHistoryService: # Convert to OpenRAG format return self._convert_langflow_messages(messages) else: - print(f"Failed to get messages for session {session_id}: {response.status_code}") + logger.error(f"Failed to get messages for session {session_id}: {response.status_code}") return [] except Exception as e: - print(f"Error getting session messages: {e}") + logger.error(f"Error getting session messages: {e}") return [] def _convert_langflow_messages(self, langflow_messages: List[Dict[str, Any]]) -> List[Dict[str, Any]]: @@ -114,7 +116,7 @@ class LangflowHistoryService: converted_messages.append(converted_msg) except Exception as e: - print(f"Error converting message: {e}") + logger.error(f"Error converting message: {e}") continue return converted_messages @@ -159,7 +161,7 @@ class LangflowHistoryService: } except Exception as e: - print(f"Error getting user conversation history: {e}") + logger.error(f"Error getting user conversation history: {e}") return { "error": str(e), "conversations": []