vrr/SurfSense
2
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2025-09-01 18:19:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor: coderabbit review

Browse source
This commit is contained in:
DESKTOP-RTLN3BA\$punk 2025-04-19 23:44:42 -07:00
parent 34300ead02
commit 154c5748fd
4 changed files with 12 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
surfsense_backend/app/agents/researcher/nodes.py
View file

@ -3,7 +3,7 @@ from langchain_core.runnables import RunnableConfig
from .state import State
from typing import Any, Dict, List
from app.config import config as app_config
from .prompts import answer_outline_system_prompt
from .prompts import get_answer_outline_system_prompt
from langchain_core.messages import HumanMessage, SystemMessage
from pydantic import BaseModel, Field
import json
@ -70,7 +70,7 @@ async def write_answer_outline(state: State, config: RunnableConfig) -> Dict[str
# Create messages for the LLM
messages = [
SystemMessage(content=answer_outline_system_prompt),
SystemMessage(content=get_answer_outline_system_prompt()),
HumanMessage(content=human_message_content)
]
@ -259,7 +259,6 @@ async def process_section(
async with session_maker() as db_session:
# Fetch relevant documents using all research questions for this section
relevant_documents = await fetch_relevant_documents(
section_title=section_title,
research_questions=research_questions,
user_id=user_id,
search_space_id=search_space_id,
@ -271,10 +270,8 @@ async def process_section(
if not relevant_documents:
print(f"No relevant documents found for section: {section_title}")
relevant_documents = [
{
"content": f"No specific information was found for: {question}"
for question in research_questions
}
{"content": f"No specific information was found for: {question}"}
for question in research_questions
]
# Call the sub_section_writer graph with the appropriate config
@ -443,10 +440,8 @@ async def process_section_with_documents(
if not documents_to_use:
print(f"No relevant documents found for section: {section_title}")
documents_to_use = [
{
"content": f"No specific information was found for: {question}"
for question in section_questions
}
{"content": f"No specific information was found for: {question}"}
for question in section_questions
]
# Call the sub_section_writer graph with the appropriate config

3
surfsense_backend/app/agents/researcher/prompts.py
View file

@ -1,7 +1,8 @@
import datetime
answer_outline_system_prompt = f"""
def get_answer_outline_system_prompt():
return f"""
Today's date: {datetime.datetime.now().strftime("%Y-%m-%d")}
<answer_outline_system>
You are an expert research assistant specializing in structuring information. Your task is to create a detailed and logical research outline based on the user's query. This outline will serve as the blueprint for generating a comprehensive research report.

4
surfsense_backend/app/agents/researcher/sub_section_writer/nodes.py
View file

@ -3,7 +3,7 @@ from langchain_core.runnables import RunnableConfig
from .state import State
from typing import Any, Dict, List
from app.config import config as app_config
from .prompts import citation_system_prompt
from .prompts import get_citation_system_prompt
from langchain_core.messages import HumanMessage, SystemMessage
async def rerank_documents(state: State, config: RunnableConfig) -> Dict[str, Any]:
@ -145,7 +145,7 @@ async def write_sub_section(state: State, config: RunnableConfig) -> Dict[str, A
# Create messages for the LLM
messages = [
SystemMessage(content=citation_system_prompt),
SystemMessage(content=get_citation_system_prompt()),
HumanMessage(content=human_message_content)
]

3
surfsense_backend/app/agents/researcher/sub_section_writer/prompts.py
View file

@ -1,7 +1,8 @@
import datetime
citation_system_prompt = f"""
def get_citation_system_prompt():
return f"""
Today's date: {datetime.datetime.now().strftime("%Y-%m-%d")}
You are a research assistant tasked with analyzing documents and providing comprehensive answers with proper citations in IEEE format.