automator: show source type (FAQ/docs/repo) in FAQ assistant Slack replies

Prefix each source line with its type so readers can tell how authoritative it is. Closes #26.

Author: alexeygrigorev

automator/src/lambda_function.py

@@ -119,17 +119,27 @@ def post_faq_assistant_answer(channel, thread_ts, payload):
     slack.post_message_to_thread(channel, thread_ts, message)
 
 
+# Display labels for the FAQ assistant's structured `source` types.
+FAQ_SOURCE_LABELS = {'faq': 'FAQ', 'docs': 'docs', 'course-repo': 'repo'}
+
+
 def format_faq_sources(sources, found_answer=True):
     """Render the FAQ assistant's structured sources as Slack mrkdwn links.
 
     When the assistant found an answer these are its citations; otherwise they are
-    suggested resources, so the heading changes accordingly.
+    suggested resources, so the heading changes accordingly. Each line is prefixed
+    with the source type (FAQ / docs / repo) so the reader can tell how authoritative
+    it is.
     """
-    lines = [
-        f"• <{source['url']}|{source.get('title') or source.get('source') or 'source'}>"
-        for source in (sources or [])
-        if source.get('url')
-    ]
+    lines = []
+    for source in (sources or []):
+        if not source.get('url'):
+            continue
+        title = source.get('title') or source.get('source') or 'source'
+        kind = source.get('source')
+        label = FAQ_SOURCE_LABELS.get(kind, kind)
+        prefix = f'[{label}] ' if label else ''
+        lines.append(f"• {prefix}<{source['url']}|{title}>")
     if not lines:
         return ''
     heading = '*Sources:*' if found_answer else '*You can check these for more information:*'

automator/tests/test_automator_lambda.py

@@ -1149,5 +1149,44 @@ def test_handle_faq_reaction_posts_answer_to_original_thread(self, mock_call, mo
         )
 
 
+class TestFormatFaqSources(unittest.TestCase):
+    """format_faq_sources prefixes each line with the source type."""
+
+    def test_labels_known_source_types(self):
+        sources = [
+            {'source': 'faq', 'title': 'Can I use TypeScript?', 'url': 'https://x/faq#1'},
+            {'source': 'docs', 'title': 'Project', 'url': 'https://x/docs'},
+            {'source': 'course-repo', 'title': 'README', 'url': 'https://x/repo'},
+        ]
+        result = lambda_function.format_faq_sources(sources, found_answer=True)
+        self.assertIn('*Sources:*', result)
+        self.assertIn('• [FAQ] <https://x/faq#1|Can I use TypeScript?>', result)
+        self.assertIn('• [docs] <https://x/docs|Project>', result)
+        self.assertIn('• [repo] <https://x/repo|README>', result)
+
+    def test_unknown_source_type_used_verbatim(self):
+        sources = [{'source': 'blog', 'title': 'Post', 'url': 'https://x/p'}]
+        result = lambda_function.format_faq_sources(sources)
+        self.assertIn('• [blog] <https://x/p|Post>', result)
+
+    def test_missing_source_type_has_no_prefix(self):
+        sources = [{'title': 'Post', 'url': 'https://x/p'}]
+        result = lambda_function.format_faq_sources(sources)
+        self.assertIn('• <https://x/p|Post>', result)
+
+    def test_not_found_changes_heading(self):
+        sources = [{'source': 'faq', 'title': 'T', 'url': 'https://x/f'}]
+        result = lambda_function.format_faq_sources(sources, found_answer=False)
+        self.assertIn('*You can check these for more information:*', result)
+
+    def test_sources_without_url_are_skipped(self):
+        sources = [{'source': 'faq', 'title': 'No link'}]
+        self.assertEqual(lambda_function.format_faq_sources(sources), '')
+
+    def test_empty_sources(self):
+        self.assertEqual(lambda_function.format_faq_sources(None), '')
+        self.assertEqual(lambda_function.format_faq_sources([]), '')
+
+
 if __name__ == '__main__':
     unittest.main()