From f25d692d8e8101da93dbaa44fa3d253399338537 Mon Sep 17 00:00:00 2001
From: star-nox <dabholkar.asmita@gmail.com>
Date: Tue, 12 Mar 2024 15:43:17 -0500
Subject: [PATCH] minor changes

---
 ai_ta_backend/beam/ingest.py        |  2 +-
 ai_ta_backend/beam/nomic_logging.py | 21 ++++-----------------
 2 files changed, 5 insertions(+), 18 deletions(-)

diff --git a/ai_ta_backend/beam/ingest.py b/ai_ta_backend/beam/ingest.py
index 2d0533b2..2ccb17e5 100644
--- a/ai_ta_backend/beam/ingest.py
+++ b/ai_ta_backend/beam/ingest.py
@@ -180,7 +180,7 @@ def ingest(**inputs: Dict[str, Any]):
   print("Final success_fail_dict: ", success_fail_dict)
 
   # rebuild nomic document map after all ingests are done
-  rebuild_status = rebuild_map(course_name, map_type='document', supabase_client)
+  rebuild_status = rebuild_map(course_name, map_type='document')
   return success_fail_dict
 
 
diff --git a/ai_ta_backend/beam/nomic_logging.py b/ai_ta_backend/beam/nomic_logging.py
index 5af69aa0..e8189e10 100644
--- a/ai_ta_backend/beam/nomic_logging.py
+++ b/ai_ta_backend/beam/nomic_logging.py
@@ -606,14 +606,6 @@ def log_to_document_map(data: dict):
     # append to existing map
     project_name = "Document Map for " + course_name
     result = append_to_map(embeddings, metadata, project_name)
-
-    # check if project is accepting new datums
-    if project.is_accepting_data:
-      with project.wait_for_project_lock():
-        project.rebuild_maps()
-
-    # with project.wait_for_project_lock():
-    #   project.rebuild_maps()
     return result
 
   except Exception as e:
@@ -729,7 +721,7 @@ def data_prep_for_doc_map(df: pd.DataFrame):
 
   return embeddings, metadata
 
-def rebuild_map(course_name:str, map_type:str, supabase_client):
+def rebuild_map(course_name:str, map_type:str):
   """
   This function rebuilds a given map in Nomic.
   """
@@ -741,15 +733,10 @@ def rebuild_map(course_name:str, map_type:str, supabase_client):
     NOMIC_MAP_NAME_PREFIX = 'Conversation Map for '
 
   try:
-    # check if map exists
-    response = supabase_client.table("projects").select("doc_map_id").eq("course_name", course_name).execute()
-    if response.data:
-      project_id = response.data[0]['doc_map_id']
-    else:
-      return "No map found for this course"
-    
     # fetch project from Nomic
-    project = AtlasProject(project_id=project_id, add_datums_if_exists=True)
+    project_name = NOMIC_MAP_NAME_PREFIX + course_name
+    project = AtlasProject(name=project_name, add_datums_if_exists=True)
+
     with project.wait_for_project_lock():
       project.rebuild_maps()
     return "Successfully rebuilt map"