code rabbit suggestions

Author: Dishant1804

backend/apps/ai/Makefile

@@ -2,42 +2,42 @@ ai-create-chapter-chunks:
 	@echo "Creating chapter chunks"
 	@CMD="python manage.py ai_create_chapter_chunks" $(MAKE) exec-backend-command
 
-ai-create-chapter-context:
-	@echo "Creating chapter context"
-	@CMD="python manage.py ai_create_chapter_context" $(MAKE) exec-backend-command
-
 ai-create-committee-chunks:
 	@echo "Creating committee chunks"
 	@CMD="python manage.py ai_create_committee_chunks" $(MAKE) exec-backend-command
 
-ai-create-committee-context:
-	@echo "Creating committee context"
-	@CMD="python manage.py ai_create_committee_context" $(MAKE) exec-backend-command
-
 ai-create-event-chunks:
 	@echo "Creating event chunks"
 	@CMD="python manage.py ai_create_event_chunks" $(MAKE) exec-backend-command
 
-ai-create-event-context:
-	@echo "Creating event context"
-	@CMD="python manage.py ai_create_event_context" $(MAKE) exec-backend-command
-
 ai-create-project-chunks:
 	@echo "Creating project chunks"
 	@CMD="python manage.py ai_create_project_chunks" $(MAKE) exec-backend-command
 
-ai-create-project-context:
-	@echo "Creating project context"
-	@CMD="python manage.py ai_create_project_context" $(MAKE) exec-backend-command
-
 ai-create-slack-message-chunks:
 	@echo "Creating Slack message chunks"
 	@CMD="python manage.py ai_create_slack_message_chunks" $(MAKE) exec-backend-command
 
-ai-create-slack-message-context:
-	@echo "Creating Slack message context"
-	@CMD="python manage.py ai_create_slack_message_context" $(MAKE) exec-backend-command
-
 ai-run-rag-tool:
 	@echo "Running RAG tool"
 	@CMD="python manage.py ai_run_rag_tool" $(MAKE) exec-backend-command
+
+ai-update-chapter-context:
+	@echo "Updating chapter context"
+	@CMD="python manage.py ai_update_chapter_context" $(MAKE) exec-backend-command
+
+ai-update-committee-context:
+	@echo "Updating committee context"
+	@CMD="python manage.py ai_update_committee_context" $(MAKE) exec-backend-command
+
+ai-update-event-context:
+	@echo "Updating event context"
+	@CMD="python manage.py ai_update_event_context" $(MAKE) exec-backend-command
+
+ai-update-project-context:
+	@echo "Updating project context"
+	@CMD="python manage.py ai_update_project_context" $(MAKE) exec-backend-command
+
+ai-update-slack-message-context:
+	@echo "Updating Slack message context"
+	@CMD="python manage.py ai_update_slack_message_context" $(MAKE) exec-backend-command

backend/apps/ai/common/utils.py

@@ -4,7 +4,7 @@
 import time
 from datetime import UTC, datetime, timedelta
 
-import openai
+from openai import OpenAI, OpenAIError
 
 from apps.ai.common.constants import (
     DEFAULT_LAST_REQUEST_OFFSET_SECONDS,
@@ -63,7 +63,7 @@ def create_chunks_and_embeddings(
             if chunk is not None:
                 chunks.append(chunk)
 
-    except openai.OpenAIError:
+    except OpenAIError:
         logger.exception("Failed to create chunks and embeddings")
         return []
     else:
@@ -89,7 +89,7 @@ def regenerate_chunks_for_context(context: Context):
         logger.warning("No content to chunk for Context. Process stopped.")
         return
 
-    openai_client = openai.Client()
+    openai_client = OpenAI()
 
     create_chunks_and_embeddings(
         chunk_texts=new_chunk_texts,

backend/apps/ai/management/commands/ai_update_slack_message_context.py

@@ -14,6 +14,7 @@ class Command(BaseContextCommand):
 
     def add_arguments(self, parser):
         """Override to use different default batch size for messages."""
+        super().add_arguments(parser)
         parser.add_argument(
             "--message-key",
             type=str,

backend/tests/apps/ai/management/commands/ai_update_slack_message_context_test.py

@@ -72,9 +72,10 @@ def test_add_arguments(self, command):
         parser = Mock()
         command.add_arguments(parser)
 
-        assert parser.add_argument.call_count == 3
+        assert parser.add_argument.call_count == 6
         calls = parser.add_argument.call_args_list
 
+        # First 3 calls are from parent class (BaseAICommand)
         assert calls[0][0] == ("--message-key",)
         assert calls[0][1]["type"] is str
         assert "Process only the message with this key" in calls[0][1]["help"]
@@ -85,5 +86,19 @@ def test_add_arguments(self, command):
 
         assert calls[2][0] == ("--batch-size",)
         assert calls[2][1]["type"] is int
-        assert calls[2][1]["default"] == 100
+        assert calls[2][1]["default"] == 50  # Default from parent class
         assert "Number of messages to process in each batch" in calls[2][1]["help"]
+
+        # Next 3 calls are from the command itself (duplicates with different defaults)
+        assert calls[3][0] == ("--message-key",)
+        assert calls[3][1]["type"] is str
+        assert "Process only the message with this key" in calls[3][1]["help"]
+
+        assert calls[4][0] == ("--all",)
+        assert calls[4][1]["action"] == "store_true"
+        assert "Process all the messages" in calls[4][1]["help"]
+
+        assert calls[5][0] == ("--batch-size",)
+        assert calls[5][1]["type"] is int
+        assert calls[5][1]["default"] == 100  # Overridden default from command
+        assert "Number of messages to process in each batch" in calls[5][1]["help"]