feat: add language support across configurations and prompts

surfsense_backend/app/agents/researcher/qna_agent/configuration.py

@@ -20,6 +20,7 @@ class Configuration:
     ]  # Documents provided directly to the agent for answering
     user_id: str  # User identifier
     search_space_id: int  # Search space identifier
+    language: str | None = None  # Language for responses
 
     @classmethod
     def from_runnable_config(

surfsense_backend/app/agents/researcher/qna_agent/nodes.py

@@ -102,7 +102,7 @@ async def answer_question(state: State, config: RunnableConfig) -> dict[str, Any
     user_query = configuration.user_query
     user_id = configuration.user_id
     search_space_id = configuration.search_space_id
-
+    language = configuration.language 
     # Get user's fast LLM
     llm = await get_user_fast_llm(state.db_session, user_id, search_space_id)
     if not llm:
@@ -127,7 +127,7 @@ async def answer_question(state: State, config: RunnableConfig) -> dict[str, Any
         """
 
         # Use initial system prompt for token calculation
-        initial_system_prompt = get_qna_citation_system_prompt(chat_history_str)
+        initial_system_prompt = get_qna_citation_system_prompt(chat_history_str, language)
         base_messages = [
             SystemMessage(content=initial_system_prompt),
             HumanMessage(content=base_human_message_template),
@@ -146,9 +146,9 @@ async def answer_question(state: State, config: RunnableConfig) -> dict[str, Any
 
     # Choose system prompt based on final document availability
     system_prompt = (
-        get_qna_citation_system_prompt(chat_history_str)
+        get_qna_citation_system_prompt(chat_history_str, language)
         if has_documents
-        else get_qna_no_documents_system_prompt(chat_history_str)
+        else get_qna_no_documents_system_prompt(chat_history_str, language)
     )
 
     # Generate documents section

surfsense_backend/app/agents/researcher/qna_agent/prompts.py

@@ -1,7 +1,7 @@
 import datetime
 
 
-def get_qna_citation_system_prompt(chat_history: str | None = None):
+def get_qna_citation_system_prompt(chat_history: str | None = None, language: str | None = None):
     chat_history_section = (
         f"""
 <chat_history>
@@ -15,10 +15,15 @@ NO CHAT HISTORY PROVIDED
 </chat_history>
 """
     )
+    
+    # Add language instruction if specified
+    language_instruction = ""
+    if language:
+        language_instruction = f"\n\nIMPORTANT: Please respond in {language} language. All your responses, explanations, and analysis should be written in {language}."
 
     return f"""
 Today's date: {datetime.datetime.now().strftime("%Y-%m-%d")}
-You are SurfSense, an advanced AI research assistant that provides detailed, well-researched answers to user questions by synthesizing information from multiple personal knowledge sources.
+You are SurfSense, an advanced AI research assistant that provides detailed, well-researched answers to user questions by synthesizing information from multiple personal knowledge sources.{language_instruction}
 {chat_history_section}
 <knowledge_sources>
 - EXTENSION: "Web content saved via SurfSense browser extension" (personal browsing history)
@@ -149,7 +154,7 @@ Make sure your response:
 """
 
 
-def get_qna_no_documents_system_prompt(chat_history: str | None = None):
+def get_qna_no_documents_system_prompt(chat_history: str | None = None, language: str | None = None):
     chat_history_section = (
         f"""
 <chat_history>
@@ -163,10 +168,15 @@ NO CHAT HISTORY PROVIDED
 </chat_history>
 """
     )
+    
+    # Add language instruction if specified
+    language_instruction = ""
+    if language:
+        language_instruction = f"\n\nIMPORTANT: Please respond in {language} language. All your responses, explanations, and analysis should be written in {language}."
 
     return f"""
 Today's date: {datetime.datetime.now().strftime("%Y-%m-%d")}
-You are SurfSense, an advanced AI research assistant that provides helpful, detailed answers to user questions in a conversational manner.
+You are SurfSense, an advanced AI research assistant that provides helpful, detailed answers to user questions in a conversational manner.{language_instruction}
 {chat_history_section}
 <context>
 The user has asked a question but there are no specific documents from their personal knowledge base available to answer it. You should provide a helpful response based on: