Merge pull request #85 from smokeyScraper/codebase_restructuring

[refactor]: restructure backend for improved modularity and separation of concerns

Author: chandansgowda

backend/app/agents/__init__.py

@@ -1,6 +1,6 @@
 from .devrel.agent import DevRelAgent
-from .shared.base_agent import BaseAgent, AgentState
-from .shared.classification_router import ClassificationRouter
+from .base_agent import BaseAgent, AgentState
+from .classification_router import ClassificationRouter
 
 __all__ = [
     "DevRelAgent",

backend/app/agents/shared/base_agent.py renamed to backend/app/agents/base_agent.py

@@ -0,0 +1 @@
+

backend/app/agents/shared/classification_router.py renamed to backend/app/agents/classification_router.py

@@ -4,18 +4,18 @@
 from langgraph.graph import StateGraph, END
 from langchain_google_genai import ChatGoogleGenerativeAI
 from langgraph.checkpoint.memory import InMemorySaver
-from ..shared.base_agent import BaseAgent, AgentState
-from ..shared.classification_router import MessageCategory
+from ..base_agent import BaseAgent, AgentState
+from ..classification_router import MessageCategory
 from .tools.search_tool import TavilySearchTool
 from .tools.faq_tool import FAQTool
 from app.core.config import settings
-from .nodes.gather_context_node import gather_context_node
-from .nodes.handle_faq_node import handle_faq_node
-from .nodes.handle_web_search_node import handle_web_search_node
-from .nodes.handle_technical_support_node import handle_technical_support_node
-from .nodes.handle_onboarding_node import handle_onboarding_node
-from .nodes.generate_response_node import generate_response_node
-from .nodes.summarization_node import check_summarization_needed, summarize_conversation_node, store_summary_to_database
+from .nodes.gather_context import gather_context_node
+from .nodes.handlers.faq import handle_faq_node
+from .nodes.handlers.web_search import handle_web_search_node
+from .nodes.handlers.technical_support import handle_technical_support_node
+from .nodes.handlers.onboarding import handle_onboarding_node
+from .generate_response_node import generate_response_node
+from .nodes.summarization import check_summarization_needed, summarize_conversation_node, store_summary_to_database
 
 logger = logging.getLogger(__name__)
 

backend/app/agents/devrel/__init__.py

@@ -1,28 +1,12 @@
 import logging
 from typing import Dict, Any
-from app.agents.shared.state import AgentState
+from app.agents.state import AgentState
 from langchain_core.messages import HumanMessage
-from ..prompts.base_prompt import GENERAL_LLM_RESPONSE_PROMPT
+from .prompts.base_prompt import GENERAL_LLM_RESPONSE_PROMPT
+from .nodes.handlers.web_search import create_search_response
 
 logger = logging.getLogger(__name__)
 
-async def _create_search_response(task_result: Dict[str, Any]) -> str:
-    """Create a response string from search results."""
-    query = task_result.get("query")
-    results = task_result.get("results", [])
-    if not results:
-        return f"I couldn't find any information for '{query}'. You might want to try rephrasing your search."
-
-    response_parts = [f"Here's what I found for '{query}':"]
-    for i, result in enumerate(results[:3]):
-        title = result.get('title', 'N/A')
-        snippet = result.get('snippet', 'N/A')
-        url = result.get('url', '#')
-        result_line = f"{i+1}. {title}: {snippet}"
-        response_parts.append(result_line)
-        response_parts.append(f"   (Source: {url})")
-    response_parts.append("You can ask me to search again with a different query if these aren't helpful.")
-    return "\n".join(response_parts)
 
 async def _create_llm_response(state: AgentState, task_result: Dict[str, Any], llm) -> str:
     """Generate a response using the LLM based on the current state and task result."""
@@ -89,7 +73,7 @@ async def generate_response_node(state: AgentState, llm) -> dict:
     if task_result.get("type") == "faq":
         final_response = task_result.get("response", "I don't have a specific answer for that question.")
     elif task_result.get("type") == "web_search":
-        final_response = await _create_search_response(task_result)
+        final_response = create_search_response(task_result)
     else:
         final_response = await _create_llm_response(state, task_result, llm)
 

backend/app/agents/devrel/agent.py

@@ -1,7 +1,6 @@
 import logging
 from datetime import datetime
-from app.agents.shared.state import AgentState
-from app.agents.shared.classification_router import MessageCategory
+from app.agents.state import AgentState
 
 logger = logging.getLogger(__name__)
 

backend/app/agents/devrel/nodes/generate_response_node.py renamed to backend/app/agents/devrel/generate_response_node.py

@@ -1,5 +1,5 @@
 import logging
-from app.agents.shared.state import AgentState
+from app.agents.state import AgentState
 
 logger = logging.getLogger(__name__)
 

backend/app/agents/devrel/nodes/gather_context_node.py renamed to backend/app/agents/devrel/nodes/gather_context.py

@@ -1,5 +1,5 @@
 import logging
-from app.agents.shared.state import AgentState
+from app.agents.state import AgentState
 
 logger = logging.getLogger(__name__)