bug fix

utils/generator.py

@@ -10,7 +10,16 @@ from langchain_core.messages import SystemMessage, HumanMessage
 
 # Local imports
 from .utils import getconfig, get_auth
-from .sources import _process_context, _build_messages, _parse_citations, _extract_sources, _create_sources_list, clean_citations
+from .prompts import system_prompt
+from .sources import (
+    _process_context,
+    _build_messages,
+    _parse_citations,
+    _extract_sources,
+    _create_sources_list,
+    clean_citations
+)
+
 
 # Set up logger
 logger = logging.getLogger(__name__)
@@ -116,7 +125,7 @@ async def generate_streaming(query: str, context: Union[str, List[Dict[str, Any]
     
     try:
         formatted_context, processed_results = _process_context(context)
-        messages = _build_messages(query, formatted_context)
+        messages = _build_messages(system_prompt, query, formatted_context)
         
         # Stream the response and accumulate for citation parsing (filter out any sources that were not cited)
         accumulated_response = ""

utils/sources.py

@@ -2,7 +2,7 @@ import re
 from typing import List, Dict, Any, Union
 import ast
 from langchain_core.messages import SystemMessage, HumanMessage
-from .prompts import system_prompt
+
 
 # ---------------------------------------------------------------------
 # Core Processing Functions