move chat backend source files to app folder

Author: fraboniface

rag_system/backend/app/__init__.py

@@ -2,8 +2,9 @@
 from datetime import datetime
 from sqlmodel import Field, Relationship, SQLModel, create_engine, Session
 
-from dependencies import get_logger
-from config import settings
+from .config import settings
+from .dependencies import get_logger
+
 
 logger = get_logger(__name__)
 

rag_system/backend/config.py rag_system/backend/app/config.pyrag_system/backend/config.py renamed to rag_system/backend/app/config.py

@@ -1,6 +1,6 @@
 import logging
 from openai import AsyncOpenAI
-from config import settings
+from .config import settings
 
 
 def get_logger(name: str) -> logging.Logger:

rag_system/backend/database.py rag_system/backend/app/database.pyrag_system/backend/database.py renamed to rag_system/backend/app/database.py

@@ -3,13 +3,13 @@
 
 from pydantic import BaseModel, ValidationError
 
-from config import settings
-from dependencies import create_openai_client, escape_newlines
-from models import ChatMessage
+from .config import settings
+from .dependencies import create_openai_client, escape_newlines, get_logger
+from .models import ChatMessage
 
 
 generation_client = create_openai_client()
-
+logger = get_logger(__name__)
 
 async def generate_response(
     messages: list[ChatMessage],
@@ -44,7 +44,7 @@ async def generate_response(
             try:
                 return response_format.model_validate_json(response.choices[0].message.content)
             except ValidationError as e:
-                print("Validation error in query rewrite, retrying...:")
+                logger.warning("Validation error in query rewrite, retrying...")
                 kwargs["temperature"] = 0  # Retry with deterministic output
                 response = await generation_client.chat.completions.create(**kwargs)
                 return response_format.model_validate_json(response.choices[0].message.content)
@@ -53,7 +53,7 @@ async def generate_response(
     except asyncio.TimeoutError:
         return "\n\n[Generation timed out]\n\n"
     except ValidationError as e:
-        print(response.choices[0].message.content)
+        logger.error(response.choices[0].message.content)
         raise e
 
 

rag_system/backend/dependencies.py rag_system/backend/app/dependencies.pyrag_system/backend/dependencies.py renamed to rag_system/backend/app/dependencies.py

@@ -5,11 +5,11 @@
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import StreamingResponse
 
-from config import settings
-from database import create_db_and_tables, get_or_create_session, save_feedback
-from dependencies import get_logger
-from models import ChatRequest, FeedbackRequest
-from rag import generate_dummy_response, simple_rag_pipeline
+from .config import settings
+from .database import create_db_and_tables, get_or_create_session, save_feedback
+from .dependencies import get_logger
+from .models import ChatRequest, FeedbackRequest
+from .rag import generate_dummy_response, simple_rag_pipeline
 
 
 # TODO: move ml models init/clean here

rag_system/backend/generation.py rag_system/backend/app/generation.pyrag_system/backend/generation.py renamed to rag_system/backend/app/generation.py

@@ -2,17 +2,17 @@
 import json
 import time
 
-from config import settings
-from database import ChatSession, ChatTurn, save_chat_turn
-from dependencies import escape_newlines, get_logger
-from generation import (
+from .config import settings
+from .database import ChatSession, ChatTurn, save_chat_turn
+from .dependencies import escape_newlines, get_logger
+from .generation import (
     generate_response,
     simulate_stream,
     stream_response,
 )
-from models import ChatMessage, DocumentChunk, Publication, QueryRewriteResponse
-from prompts import BASE_SYSTEM_PROMPT, RAG_PROMPT, QUERY_REWRITE_PROMPT
-from retrieval import get_publications_from_chunks, retrieve_chunks
+from .models import ChatMessage, DocumentChunk, Publication, QueryRewriteResponse
+from .prompts import BASE_SYSTEM_PROMPT, RAG_PROMPT, QUERY_REWRITE_PROMPT
+from .retrieval import get_publications_from_chunks, retrieve_chunks
 
 logger = get_logger(__name__)
 

rag_system/backend/app.py rag_system/backend/app/main.pyrag_system/backend/app.py renamed to rag_system/backend/app/main.py

@@ -2,9 +2,9 @@
 
 from flashrank import Ranker, RerankRequest
 
-from config import settings
-from dependencies import create_openai_client
-from models import DocumentChunk
+from .config import settings
+from .dependencies import create_openai_client
+from .models import DocumentChunk
 
 
 ranker = Ranker(max_length=settings.max_length_reranker)