about summary refs log tree commit diff
diff options
context:
space:
mode:
authorAlexander_Kabui2024-09-03 10:41:33 +0300
committerAlexander_Kabui2024-09-03 10:41:33 +0300
commitc16c54759cfd493250424ee3f565862e5d6009b3 (patch)
tree4cf5b43d0a6a1fa4f39c0ecc7c63746f75a061ef
parent41c352926ca178e65d6c948fdf93b0f987e2878a (diff)
downloadgenenetwork3-c16c54759cfd493250424ee3f565862e5d6009b3.tar.gz
Raise KeyError for doc_id not found in doc_ids.
-rw-r--r--gn3/llms/process.py17
1 files changed, 8 insertions, 9 deletions
diff --git a/gn3/llms/process.py b/gn3/llms/process.py
index bfce9a5..b8e47e7 100644
--- a/gn3/llms/process.py
+++ b/gn3/llms/process.py
@@ -41,13 +41,10 @@ class DocIDs():
         Args:
             doc_id: str: a search key for doc_ids
         Returns:
-              an object with doc_info if doc_id in doc_ids
+              an object if doc id exists else
+              raises a KeyError
         """
-        if doc_id in self.doc_ids.keys():
-            return self.doc_ids[doc_id]
-        else:
-            return doc_id
-
+        return self.doc_ids[doc_id]
 
 def format_bibliography_info(bib_info):
     """Utility function for formatting bibliography info
@@ -74,9 +71,11 @@ def parse_context(context, get_info_func, format_bib_func):
         combo_txt = ""
         for entry in summary:
             combo_txt += "\t" + entry["text"]
-        doc_info = get_info_func(doc_ids)
-        bib_info = doc_ids if doc_ids == doc_info else format_bib_func(
-            doc_info)
+        try:
+            doc_info = get_info_func(doc_ids)
+            bib_info = format_bib_func(doc_info)
+        except KeyError:
+            bib_info = doc_ids
         pattern = r'(https?://|www\.)[\w.-]+(\.[a-zA-Z]{2,})([/\w.-]*)*'
         combo_text = re.sub(pattern,
                             lambda x: f"<a href='{x[0]}' target=_blank> {x[0]} </a>",