Added tool to get article link from Redis cache.

assets/text.py

@@ -28,5 +28,5 @@ DESCRIPTION = ('''
     3. `find_article()`: Uses vector search on article content to find title of article
         that user is referring to.
     4. `get_summary()`: Gets article summary from Redis cache using article title.
-
+    5. `get_link()`: Gets article link from Redis cache using article title.
 ''')

functions/tools.py

@@ -197,3 +197,32 @@ def get_summary(title: str) -> str:
 
     logger.info('Could not find summary for: "%s"', title)
     return 'No summary found'
+
+
+def get_link(title: str) -> str:
+    '''Uses article title to get link to content webpage.
+    
+    Args:
+        title: title of article to get link for
+
+    Returns:
+        Article webpage URL. Returns "No link found" if link does not exist.
+    '''
+
+    logger = logging.getLogger(__name__ + '.get_link()')
+
+    redis = Redis(
+        url='https://sensible-midge-19304.upstash.io',
+        token=os.environ['UPSTASH_REDIS_KEY']
+    )
+
+    cache_key = f'{title} link'
+    link = redis.get(cache_key)
+
+    if link:
+
+        logger.info('Got link for "%s": %s', title, link)
+        return link
+
+    logger.info('Could not find link for: "%s"', title)
+    return 'No summary found'

rss_server.py

@@ -41,6 +41,7 @@ with gr.Blocks() as demo:
     gr.HTML(text.TITLE)
     gr.Markdown(text.DESCRIPTION)
 
+
     # Log output
     with gr.Row():
         dialog_output = gr.Textbox(label='Server logs', lines=7, max_lines=5)
@@ -53,6 +54,7 @@ with gr.Blocks() as demo:
         show_api=False
     )
 
+
     # Get feed tool
     gr.Markdown('### 1. `get_feed()`')
     website_url = gr.Textbox('slashdot', label='Website')
@@ -69,6 +71,7 @@ with gr.Blocks() as demo:
         api_name='Get RSS feed content'
     )
 
+
     # Vector search tool
     gr.Markdown('### 2. `context_search()`')
 
@@ -95,6 +98,7 @@ with gr.Blocks() as demo:
         api_name='Context vector search'
     )
 
+
     # Find article tool
     gr.Markdown('### 3. `find_article()`')
 
@@ -104,8 +108,8 @@ with gr.Blocks() as demo:
     )
     article_search_output = gr.Textbox(
         label='Article search results',
-        lines=7,
-        max_lines=7
+        lines=3,
+        max_lines=3
     )
 
     with gr.Row():
@@ -121,6 +125,7 @@ with gr.Blocks() as demo:
         api_name='Article vector search'
     )
 
+
     # Get summary tool
     gr.Markdown('### 4. `get_summary()`')
 
@@ -130,8 +135,8 @@ with gr.Blocks() as demo:
     )
     article_summary = gr.Textbox(
         label='Article summary',
-        lines=7,
-        max_lines=7
+        lines=3,
+        max_lines=3
     )
 
     with gr.Row():
@@ -148,6 +153,33 @@ with gr.Blocks() as demo:
     )
 
 
+    # Get link tool
+    gr.Markdown('### 5. `get_link()`')
+
+    article_title_link = gr.Textbox(
+        'FAA To Eliminate Floppy Disks Used In Air Traffic Control Systems',
+        label='Article title'
+    )
+    article_link = gr.Textbox(
+        label='Article link',
+        lines=3,
+        max_lines=3
+    )
+
+    with gr.Row():
+        article_link_submit_button = gr.Button('Submit title')
+        article_link_clear_button = gr.ClearButton(
+            components=[article_title_link, article_link]
+        )
+
+    article_link_submit_button.click( # pylint: disable=no-member
+        fn=tool_funcs.get_link,
+        inputs=article_title_link,
+        outputs=article_link,
+        api_name='Article link search'
+    )
+
+
 if __name__ == '__main__':
 
     demo.launch(mcp_server=True)