Added tool to get article summary from Redis cache.

assets/text.py

@@ -27,5 +27,6 @@ DESCRIPTION = ('''
     2. `context_search()`: Vector search on article content for RAG context.
     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.
 
 ''')

functions/summarization.py

@@ -25,7 +25,7 @@ def summarize_content(title: str, content: str) -> str:
     logger.info('Summarizing extracted content')
 
     # Check Redis cache for summary
-    cache_key = f'{title}  summary'
+    cache_key = f'{title} summary'
     cached_summary = REDIS.get(cache_key)
 
     if cached_summary:

functions/tools.py

@@ -8,6 +8,7 @@ import logging
 import queue
 from typing import Tuple
 from upstash_vector import Index
+from upstash_redis import Redis
 
 import functions.feed_extraction as extraction_funcs
 import functions.summarization as summarization_funcs
@@ -166,3 +167,33 @@ def find_article(query: str) -> list[Tuple[float, str]]:
         contexts.append((result.score, result.metadata['namespace']))
 
     return contexts
+
+
+def get_summary(title: str) -> str:
+    '''Uses article title to get summary of article content.
+    
+    Args:
+        title: title of article to get summary for
+
+    Returns:
+        Short summary of article content. Returns "No summary found"
+        if summary does not exist.
+    '''
+
+    logger = logging.getLogger(__name__ + '.get_summary()')
+
+    redis = Redis(
+        url='https://sensible-midge-19304.upstash.io',
+        token=os.environ['UPSTASH_REDIS_KEY']
+    )
+
+    cache_key = f'{title} summary'
+    summary = redis.get(cache_key)
+
+    if summary:
+
+        logger.info('Got summary for "%s": %s', title, summary[:100])
+        return summary
+
+    logger.info('Could not find summary for: "%s"', title)
+    return 'No summary found'

rss_server.py

@@ -55,7 +55,7 @@ with gr.Blocks() as demo:
 
     # Get feed tool
     gr.Markdown('### 1. `get_feed()`')
-    website_url = gr.Textbox('hackernews.com', label='Website')
+    website_url = gr.Textbox('slashdot', label='Website')
     feed_output = gr.Textbox(label='RSS entries', lines=7, max_lines=7)
 
     with gr.Row():
@@ -73,7 +73,7 @@ with gr.Blocks() as demo:
     gr.Markdown('### 2. `context_search()`')
 
     context_search_query = gr.Textbox(
-        'Does apple offer parental controls?',
+        'How is the air traffic control system being updated?',
         label='Context search query'
     )
     context_search_output = gr.Textbox(
@@ -99,7 +99,7 @@ with gr.Blocks() as demo:
     gr.Markdown('### 3. `find_article()`')
 
     article_search_query = gr.Textbox(
-        'Does apple offer parental controls?',
+        'How is the air traffic control system being updated?',
         label='Article search query'
     )
     article_search_output = gr.Textbox(
@@ -121,6 +121,32 @@ with gr.Blocks() as demo:
         api_name='Article vector search'
     )
 
+    # Get summary tool
+    gr.Markdown('### 4. `get_summary()`')
+
+    article_title = gr.Textbox(
+        'FAA To Eliminate Floppy Disks Used In Air Traffic Control Systems',
+        label='Article title'
+    )
+    article_summary = gr.Textbox(
+        label='Article summary',
+        lines=7,
+        max_lines=7
+    )
+
+    with gr.Row():
+        article_title_submit_button = gr.Button('Submit title')
+        article_title_clear_button = gr.ClearButton(
+            components=[article_title, article_summary]
+        )
+
+    article_title_submit_button.click( # pylint: disable=no-member
+        fn=tool_funcs.get_summary,
+        inputs=article_title,
+        outputs=article_summary,
+        api_name='Article summary search'
+    )
+
 
 if __name__ == '__main__':