same fix path import

api/main.py

@@ -17,8 +17,8 @@ import os
 import asyncio
 
 # Import your existing system
-from core.system_initializer import setup_system
-from models.state_models import MultiCountryLegalState
+from MultiCountryRAG.core.system_initializer import setup_system
+from MultiCountryRAG.models.state_models import MultiCountryLegalState
 
 # Setup logging
 logging.basicConfig(level=logging.INFO)

core/assistance/email_service.py

@@ -9,7 +9,7 @@ sys.path.insert(0, str(Path(__file__).parent.parent))
 import re
 import logging
 from typing import Optional, Dict
-from core.email_tool import LegalAssistanceEmailer
+from MultiCountryRAG.core.email_tool import LegalAssistanceEmailer
 
 logger = logging.getLogger(__name__)
 

core/assistance/workflow_nodes.py

@@ -7,7 +7,7 @@ import logging
 import re
 from typing import Dict, Any, List
 from langchain_core.runnables import RunnableConfig
-from models.state_models import MultiCountryLegalState
+from MultiCountryRAG.models.state_models import MultiCountryLegalState
 
 logger = logging.getLogger(__name__)
 

core/chat_manager.py

@@ -12,9 +12,9 @@ from langchain_core.runnables import RunnableConfig
 from langchain_core.messages import BaseMessage
 from langgraph.types import Command
 
-from config.settings import settings
-from models.state_models import MultiCountryLegalState
-from utils.helpers import dict_to_message_obj
+from MultiCountryRAG.config.settings import settings
+from MultiCountryRAG.models.state_models import MultiCountryLegalState
+from MultiCountryRAG.utils.helpers import dict_to_message_obj
 
 logger = logging.getLogger(__name__)
 

core/graph_builder.py

@@ -10,19 +10,19 @@ import logging
 from typing import Dict, List, Any
 from langchain_core.runnables import RunnableConfig
 
-from models.state_models import MultiCountryLegalState
-from core.router import CountryRouter
-from core.retriever import LegalRetriever
-from core.conversation_repair import ConversationRepair 
-from core.human_approval_node import HumanApprovalNode
+from MultiCountryRAG.models.state_models import MultiCountryLegalState
+from MultiCountryRAG.core.router import CountryRouter
+from MultiCountryRAG.core.retriever import LegalRetriever
+from MultiCountryRAG.core.conversation_repair import ConversationRepair 
+from MultiCountryRAG.core.human_approval_node import HumanApprovalNode
 
 # Import modular components
-from core.nodes.routing_nodes import RoutingNodes
-from core.assistance.workflow_nodes import AssistanceWorkflowNodes
-from core.nodes.retrieval_nodes import RetrievalNodes
-from core.nodes.response_nodes import ResponseNodes
-from core.nodes.helper_nodes import HelperNodes
-from core.routing.routing_logic import RoutingLogic
+from MultiCountryRAG.core.nodes.routing_nodes import RoutingNodes
+from MultiCountryRAG.core.assistance.workflow_nodes import AssistanceWorkflowNodes
+from MultiCountryRAG.core.nodes.retrieval_nodes import RetrievalNodes
+from MultiCountryRAG.core.nodes.response_nodes import ResponseNodes
+from MultiCountryRAG.core.nodes.helper_nodes import HelperNodes
+from MultiCountryRAG.core.routing.routing_logic import RoutingLogic
 
 logger = logging.getLogger(__name__)
 

core/human_approval_node.py

@@ -8,8 +8,8 @@ import logging
 from typing import Literal
 from langchain_core.runnables import RunnableConfig
 from langgraph.types import interrupt, Command
-from models.state_models import MultiCountryLegalState
-from core.assistance.email_service import AssistanceEmailService
+from MultiCountryRAG.models.state_models import MultiCountryLegalState
+from MultiCountryRAG.core.assistance.email_service import AssistanceEmailService
 from datetime import datetime
 
 logger = logging.getLogger(__name__)

core/nodes/base_node.py

@@ -8,8 +8,8 @@ from typing import Dict, List, Optional, Any
 from datetime import datetime
 from langchain_core.runnables import RunnableConfig
 
-from models.state_models import MultiCountryLegalState
-from utils.helpers import dict_to_message_obj
+from MultiCountryRAG.models.state_models import MultiCountryLegalState
+from MultiCountryRAG.utils.helpers import dict_to_message_obj
 
 logger = logging.getLogger(__name__)
 

core/nodes/helper_nodes.py

@@ -9,9 +9,9 @@ from typing import Dict, Any, List
 from langchain_core.runnables import RunnableConfig
 from langchain_core.messages import HumanMessage
 
-from models.state_models import MultiCountryLegalState
-from .base_node import BaseNode
-from core.prompts.prompt_templates import PromptTemplates
+from MultiCountryRAG.models.state_models import MultiCountryLegalState
+from MultiCountryRAG.core.nodes.base_node import BaseNode
+from MultiCountryRAG.core.prompts.prompt_templates import PromptTemplates
 
 logger = logging.getLogger(__name__)
 

core/nodes/response_nodes.py

@@ -10,8 +10,8 @@ from datetime import datetime
 from typing import Dict, Any
 from langchain_core.runnables import RunnableConfig
 
-from models.state_models import MultiCountryLegalState
-from utils.helpers import dict_to_message_obj, message_obj_to_dict
+from MultiCountryRAG.models.state_models import MultiCountryLegalState
+from MultiCountryRAG.utils.helpers import dict_to_message_obj, message_obj_to_dict
 
 logger = logging.getLogger(__name__)
 

core/nodes/retrieval_nodes.py

@@ -8,8 +8,8 @@ import logging
 from typing import Dict, Any
 from langchain_core.runnables import RunnableConfig
 
-from models.state_models import MultiCountryLegalState
-from core.retriever import LegalRetriever
+from MultiCountryRAG.models.state_models import MultiCountryLegalState
+from MultiCountryRAG.core.retriever import LegalRetriever
 
 logger = logging.getLogger(__name__)
 

core/nodes/routing_nodes.py

@@ -8,10 +8,10 @@ import logging
 from typing import Dict, Any
 from langchain_core.runnables import RunnableConfig
 
-from models.state_models import MultiCountryLegalState
-from core.router import CountryRouter
-from .base_node import BaseNode
-from core.prompts.prompt_templates import PromptTemplates
+from MultiCountryRAG.models.state_models import MultiCountryLegalState
+from MultiCountryRAG.core.router import CountryRouter
+from MultiCountryRAG.core.nodes.base_node import BaseNode
+from MultiCountryRAG.core.prompts.prompt_templates import PromptTemplates
 
 logger = logging.getLogger(__name__)
 

core/retriever.py

@@ -10,8 +10,8 @@ from typing import List, Dict, Any, Tuple
 from langchain_core.documents import Document
 from langchain_mongodb.vectorstores import MongoDBAtlasVectorSearch
 
-from config.settings import settings
-from config.constants import ARTICLE_PATTERNS, CATEGORY_KEYWORDS, DOCUMENT_TYPE_KEYWORDS
+from MultiCountryRAG.config.settings import settings
+from MultiCountryRAG.config.constants import ARTICLE_PATTERNS, CATEGORY_KEYWORDS, DOCUMENT_TYPE_KEYWORDS
 
 logger = logging.getLogger(__name__)
 

core/router.py

@@ -11,8 +11,8 @@ from typing import Dict, List, Optional, Literal, Any
 from langchain_openai import ChatOpenAI
 from langchain_core.messages import SystemMessage, HumanMessage
 
-from config.settings import settings
-from models.state_models import RoutingResult
+from MultiCountryRAG.config.settings import settings
+from MultiCountryRAG.models.state_models import RoutingResult
 
 logger = logging.getLogger(__name__)
 

core/routing/routing_logic.py

@@ -5,7 +5,7 @@ from pathlib import Path
 sys.path.insert(0, str(Path(__file__).parent.parent))
 import logging
 from typing import Literal
-from models.state_models import MultiCountryLegalState
+from MultiCountryRAG.models.state_models import MultiCountryLegalState
 
 logger = logging.getLogger(__name__)
 

core/system_initializer.py

@@ -7,13 +7,13 @@ sys.path.insert(0, str(Path(__file__).parent.parent))
 import logging
 from langgraph.checkpoint.postgres.aio import AsyncPostgresSaver
 
-from core.graph_builder import GraphBuilder
-from core.chat_manager import LegalChatManager
-from core.router import CountryRouter
-from database.mongodb_client import MongoDBClient
-from database.postgres_checkpointer import PostgresCheckpointer
+from MultiCountryRAG.core.graph_builder import GraphBuilder
+from MultiCountryRAG.core.chat_manager import LegalChatManager
+from MultiCountryRAG.core.router import CountryRouter
+from MultiCountryRAG.database.mongodb_client import MongoDBClient
+from MultiCountryRAG.database.postgres_checkpointer import PostgresCheckpointer
 from langchain_openai import ChatOpenAI
-from config import settings  # Make sure this import is correct
+from MultiCountryRAG.config import settings  # Make sure this import is correct
 
 
 logger = logging.getLogger(__name__)

database/mongodb_client.py

@@ -9,7 +9,7 @@ from langchain_mongodb.vectorstores import MongoDBAtlasVectorSearch
 from langchain_openai import OpenAIEmbeddings
 from typing import Dict
 import logging
-from config.settings import settings
+from MultiCountryRAG.config.settings import settings
 
 logger = logging.getLogger(__name__)
 

main.py

@@ -14,14 +14,14 @@ import time
 from datetime import datetime
 from typing import List, Dict, Any, Optional
 
-from config.settings import settings
-from database.mongodb_client import MongoDBClient
-from database.postgres_checkpointer import PostgresCheckpointer
-from core.router import CountryRouter
-from core.retriever import LegalRetriever
-from core.graph_builder import GraphBuilder
-from core.chat_manager import LegalChatManager
-from utils.logger import setup_logging
+from MultiCountryRAG.config.settings import settings
+from MultiCountryRAG.database.mongodb_client import MongoDBClient
+from MultiCountryRAG.database.postgres_checkpointer import PostgresCheckpointer
+from MultiCountryRAG.core.router import CountryRouter
+from MultiCountryRAG.core.retriever import LegalRetriever
+from MultiCountryRAG.core.graph_builder import GraphBuilder
+from MultiCountryRAG.core.chat_manager import LegalChatManager
+from MultiCountryRAG.utils.logger import setup_logging
 
 import uuid