Merge PR #547 from soniyaLama0/feature/top-k-parameter

Author: param20h

backend/app/config.py

@@ -54,6 +54,7 @@ class Settings(BaseSettings):
     GOOGLE_SERVICE_ACCOUNT_FILE: str = ""
 
     # Celery / Redis background processing
+    CELERY_ENABLED: bool = True
     CELERY_BROKER_URL: str = "redis://localhost:6379/0"
     CELERY_RESULT_BACKEND: str = "redis://localhost:6379/1"
     CELERY_TASK_TRACK_STARTED: bool = True

backend/app/rag/agent.py

@@ -82,6 +82,7 @@ def get_agent_executor(
 
     llm = HuggingFaceEndpoint(
         repo_id=settings.LLM_MODEL,
+        model=settings.LLM_MODEL,
         huggingfacehub_api_token=token,
         max_new_tokens=settings.LLM_MAX_NEW_TOKENS,
         temperature=settings.LLM_TEMPERATURE,
@@ -156,7 +157,8 @@ def generate_answer(
                 model=settings.LLM_MODEL,
                 max_tokens=256,
             )
-            answer = response.choices[0].message.content.strip() if response.choices else "Hello! How can I help you today?"
+            content = response.choices[0].message.content if response.choices else None
+            answer = content.strip() if content else "Hello! How can I help you today?"
         except Exception:
             answer = "Hello! I'm Document AI Analyst. How can I help you with your documents?"
         return {"answer": answer, "sources": []}

backend/app/rag/summarizer.py

@@ -66,15 +66,8 @@ def generate_document_summary(
             max_tokens=settings.SUMMARY_MAX_TOKENS,
             temperature=settings.LLM_TEMPERATURE,
         )
-        
-        # Defensive check for malformed or empty response structures
-        summary = None
-        if response and getattr(response, "choices", None):
-            first_choice = response.choices[0]
-            message = getattr(first_choice, "message", None)
-            content = getattr(message, "content", None)
-            if content:
-                summary = content.strip()
+        content = response.choices[0].message.content if response.choices else None
+        summary = content.strip() if content else None
 
         return summary or None