small steps to make the plugin work

Author: VergaJU

HissMed/retrieve_articles.py

@@ -87,14 +87,21 @@ def batch_download_pdfs(cls, query, retmax=100, db="pmc"):
             cls.download_pdfs(pmc_ids)
 
     @classmethod
-    def run(cls, query, email):
+    def run(cls, query):
+        if Entrez.email is None:
+            raise ValueError("Please set your email address")
         cls.create_directory()
-        Entrez.email = email
         cls.batch_download_pdfs(query)
 
+    @classmethod
+    def set_email(cls, email):
+        if email == 'your.email@mail.com':
+            raise ValueError("Please set your email address")
+        Entrez.email = email
 
 if __name__ == "__main__":
     args = PapersDownloader.get_args()
     query = args.query or '(Multiple Myeloma[Title]) AND ("2021/01/01"[Publication Date] : "2021/01/10"[Publication Date])'
     email = args.email or "your.email@example.com"
-    PapersDownloader.run(query, email)
+    PapersDownloader.set_email(email)
+    PapersDownloader.run(query)

hissmed_literature_assistant.py

@@ -1,31 +1,32 @@
 from cat.mad_hatter.decorators import tool, hook, plugin
-from pydantic import BaseModel
 from datetime import datetime, date
+from cat.looking_glass.stray_cat import StrayCat
+from cat.log import log
+import os
+import sys
 
-class MySettings(BaseModel):
-    required_int: int
-    optional_int: int = 69
-    required_str: str
-    optional_str: str = "meow"
-    required_date: date
-    optional_date: date = 1679616000
+sys.path.append(os.path.join(os.path.dirname(__file__)))
 
-@plugin
-def settings_model():
-    return MySettings
-
-@tool
-def get_the_day(tool_input, cat):
-    """Get the day of the week. Input is always None."""
+from HissMed.retrieve_articles import PapersDownloader
 
-    dt = datetime.now()
 
-    return dt.strftime('%A')
-
-@hook
-def before_cat_sends_message(message, cat):
+def settings_model():
+    settings = cat.mad_hatter.get_plugin().load_settings()
+    return settings
 
-    prompt = f'Rephrase the following sentence in a grumpy way: {message["content"]}'
-    message["content"] = cat.llm(prompt)
 
-    return message
+@tool
+def hisscat_download_literature(user_message, cat):
+    """ run this tool whenever the message starts with HissMed search"""
+    # user_message = StrayCat.working_memory.user_message_json.text
+    log.debug(f"User message received: {user_message}")  # Log the user message
+    # Check if the input matches the 'HissMed search: <query>' format
+    if user_message.startswith("HissMed search: "):
+        settings = settings_model()
+        PapersDownloader.set_email(email=settings['email'])
+        # Extract the query
+        query = user_message[len("HissMed search: "):]
+        # Print the query
+        log.debug(f"Query received: {query}")
+        PapersDownloader.run(query)
+    return user_message

settings.py

@@ -1,4 +1,5 @@
-from pydantic import BaseModel, Field, field_validator
+from pydantic import BaseModel, field_validator
+from cat.mad_hatter.decorators import plugin
 
 class MySettings(BaseModel):
     email: str = "your.email@mail.com"
@@ -7,11 +8,11 @@ class MySettings(BaseModel):
     @field_validator("email")
     @classmethod
     def email_validator(cls, email):
-        if "@" not in value:
+        if "@" not in email:
             raise ValueError("Invalid email address")
-        elif "your.email@mail.com" == value:
+        elif "your.email@mail.com" == email:
             raise ValueError("Please set your email address")
 
-#@plugin
+@plugin
 def settings_model():
     return MySettings