From 5f7520e1215803c9716da41064267523a3429504 Mon Sep 17 00:00:00 2001
From: Will Daly <will@edx.org>
Date: Wed, 28 May 2014 07:50:41 -0400
Subject: [PATCH] Add AI algorithm configuration to LMS settings

Added ORA2_FILE_PREFIX setting
---
 lms/envs/aws.py    |  6 ++++++
 lms/envs/common.py | 12 ++++++++++++
 2 files changed, 18 insertions(+)

diff --git a/lms/envs/aws.py b/lms/envs/aws.py
index 1cd3b3c972..f29c4fc19c 100644
--- a/lms/envs/aws.py
+++ b/lms/envs/aws.py
@@ -374,6 +374,12 @@ GRADES_DOWNLOAD_ROUTING_KEY = HIGH_MEM_QUEUE
 
 GRADES_DOWNLOAD = ENV_TOKENS.get("GRADES_DOWNLOAD", GRADES_DOWNLOAD)
 
+##### ORA2 ######
+# Prefix for uploads of example-based assessment AI classifiers
+# This can be used to separate uploads for different environments
+# within the same S3 bucket.
+ORA2_FILE_PREFIX = ENV_TOKENS.get("ORA2_FILE_PREFIX", ORA2_FILE_PREFIX)
+
 ##### ACCOUNT LOCKOUT DEFAULT PARAMETERS #####
 MAX_FAILED_LOGIN_ATTEMPTS_ALLOWED = ENV_TOKENS.get("MAX_FAILED_LOGIN_ATTEMPTS_ALLOWED", 5)
 MAX_FAILED_LOGIN_ATTEMPTS_LOCKOUT_PERIOD_SECS = ENV_TOKENS.get("MAX_FAILED_LOGIN_ATTEMPTS_LOCKOUT_PERIOD_SECS", 15 * 60)
diff --git a/lms/envs/common.py b/lms/envs/common.py
index afc9d39ca0..ed9f2f686c 100644
--- a/lms/envs/common.py
+++ b/lms/envs/common.py
@@ -1356,6 +1356,18 @@ LINKEDIN_API = {
 }
 
 
+############################ ORA 2 ############################################
+
+# By default, don't use a file prefix
+ORA2_FILE_PREFIX = None
+
+# AI algorithm configuration
+ORA2_AI_ALGORITHMS = {
+    'fake': 'openassessment.assessment.worker.algorithm.FakeAIAlgorithm',
+    'ease': 'openassessment.assessment.worker.algorithm.EaseAIAlgorithm'
+}
+
+
 ##### ACCOUNT LOCKOUT DEFAULT PARAMETERS #####
 MAX_FAILED_LOGIN_ATTEMPTS_ALLOWED = 5
 MAX_FAILED_LOGIN_ATTEMPTS_LOCKOUT_PERIOD_SECS = 15 * 60