add GRApiManager. py-utils to generalresearch

1 files changed, 10 insertions, 3 deletions

diff --git a/jb/decorators.py b/jb/decorators.py
index 5c1b1f5..cafd7a9 100644
--- a/jb/decorators.py
+++ b/jb/decorators.py
@@ -1,7 +1,7 @@
 import boto3
 from botocore.config import Config
-from generalresearchutils.pg_helper import PostgresConfig
-from generalresearchutils.redis_helper import RedisConfig
+from generalresearch.pg_helper import PostgresConfig
+from generalresearch.redis_helper import RedisConfig
 from influxdb import InfluxDBClient
 from mypy_boto3_mturk import MTurkClient
 from mypy_boto3_sns import SNSClient
@@ -11,7 +11,8 @@ from jb.managers import Permission
 from jb.managers.amt import AMTManager
 from jb.managers.assignment import AssignmentManager
 from jb.managers.bonus import BonusManager
-from jb.managers.hit import HitTypeManager, HitManager, HitQuestionManager
+from jb.managers.gr_api import GRApiManager
+from jb.managers.hit import HitManager, HitQuestionManager, HitTypeManager
 
 redis_config = RedisConfig(
     dsn=settings.redis,
@@ -32,6 +33,12 @@ CLIENT_CONFIG = Config(
     read_timeout=2.5,
 )
 
+gr_api_manager = GRApiManager(
+    base_url=str(settings.gr_api_host),
+    token=settings.gr_api_token.get_secret_value(),
+    product_id=settings.product_id,
+)
+
 # We shouldn't use this directly. Use our AMTManager wrapper
 AMT_CLIENT: MTurkClient = boto3.client(
     service_name="mturk",