feat/multi llm enhancement

backend/backend/settings/base.py

@@ -167,8 +167,6 @@ def get_required_setting(
     "CELERY_BROKER_URL", f"redis://{REDIS_HOST}:{REDIS_PORT}"
 )
 
-MAX_WAIT_TIME = int(get_required_setting("MAX_WAIT_TIME"))
-POLLING_INTERVAL = int(get_required_setting("POLLING_INTERVAL"))
 INDEXING_FLAG_TTL = int(get_required_setting("INDEXING_FLAG_TTL"))
 # Flag to Enable django admin
 ADMIN_ENABLED = False

backend/prompt_studio/prompt_studio_core/constants.py

@@ -8,6 +8,8 @@ class ToolStudioKeys:
     PLATFORM_SERVICE_API_KEY = "PLATFORM_SERVICE_API_KEY"
     SUMMARIZE_LLM_PROFILE = "summarize_llm_profile"
     DEFAULT_PROFILE = "default_profile"
+    PENDING_STATUS = "pending"
+    COMPLETED_STATUS = "completed"
 
 
 class ToolStudioErrors:
@@ -16,6 +18,7 @@ class ToolStudioErrors:
     DUPLICATE_API = "It appears that a duplicate call may have been made."
     PLATFORM_ERROR = "Seems an error occured in Platform Service."
     PROMPT_NAME_EXISTS = "Prompt with the name already exists"
+    DOCUMENT_BEING_INDEXED = "Document is being indexed"
 
 
 class ToolStudioPromptKeys:

backend/prompt_studio/prompt_studio_core/document_indexing_service.py

@@ -19,7 +19,7 @@ def set_document_indexing(cls, org_id: str, user_id: str, doc_id_key: str) -> No
     def is_document_indexing(cls, org_id: str, user_id: str, doc_id_key: str) -> bool:
         return (
             CacheService.get_key(cls._cache_key(org_id, user_id, doc_id_key))
-            == b"started"
+            == "started"
         )
 
     @classmethod
@@ -37,7 +37,7 @@ def get_indexed_document_id(
         cls, org_id: str, user_id: str, doc_id_key: str
     ) -> Optional[str]:
         result = CacheService.get_key(cls._cache_key(org_id, user_id, doc_id_key))
-        if result and result != b"started":
+        if result and result != "started":
             return result
         return None