Microsoft 365 files, emails and events as first-class citations

application/single_app/collaboration_models.py

@@ -700,6 +700,7 @@
         'cited_hybrid_citations',
         'cited_web_search_citations',
         'agent_citations',
+        'm365_citations',
         'agent_display_name',
         'agent_name',
         'extracted_text',

application/single_app/functions_chat_content_checks.py

@@ -42,7 +42,7 @@
 )
 ANSWER_FIELDS = (
     "hybrid_citations", "web_search_citations", "agent_citations",
-    "cited_hybrid_citations", "cited_web_search_citations",
+    "cited_hybrid_citations", "cited_web_search_citations", "m365_citations",
     "generated_artifacts", "generated_analysis_artifacts", "generated_tabular_outputs",
 )
 
@@ -309,7 +309,7 @@
         "done": True, "blocked": True, "role": "safety", "replace_content": True,
         "reply": result.notice, "content": result.notice, "full_content": result.notice,
         "conversation_id": conversation_id, "message_id": message_id,
-        "hybrid_citations": [], "web_search_citations": [], "agent_citations": [],
+        "hybrid_citations": [], "web_search_citations": [], "agent_citations": [], "m365_citations": [],
     }
 
 

application/single_app/functions_citation_tracking.py

@@ -7,6 +7,8 @@
 from typing import Any, Dict, Iterable, List, Mapping, Optional
 from urllib.parse import urlsplit, urlunsplit
 
+from functions_m365_citations import rebuild_conversation_used_m365_items
+
 
 CITATION_TRACKING_VERSION = 1
 USED_DOCUMENTS_TRACKING_VERSION = 1
@@ -669,9 +671,10 @@
 ) -> List[Dict[str, Any]]:
     """Rebuild exact usage after a retry, switch, deletion, or fork mutation."""
     initialize_conversation_used_document_tracking(conversation)
+    messages = list(messages or [])
     cited_hybrid_citations: List[Dict[str, Any]] = []
     legacy_hybrid_citations: List[Dict[str, Any]] = []
-    for message in messages or []:
+    for message in messages:
         if not isinstance(message, Mapping) or not _message_is_active_assistant(message):
             continue
         if _message_has_citation_tracking(message):
@@ -693,4 +696,13 @@
             legacy_hybrid_citations,
             source_document_tags=source_document_tags,
         )
+    if conversation.get("conversation_kind") != "collaborative":
+        # Microsoft 365 items are listed to their owner only, so a shared thread keeps none.
+        rebuild_conversation_used_m365_items(
+            conversation,
+            [
+                message for message in messages
+                if isinstance(message, Mapping) and _message_is_active_assistant(message)
+            ],
+        )
     return deepcopy(conversation["used_documents"])

application/single_app/functions_collaboration.py

@@ -456,6 +456,7 @@
         'citation_tracking_version',
         'cited_hybrid_citations',
         'cited_web_search_citations',
+        'm365_citations',
     ):
         if field_name in message_doc:
             payload[field_name] = deepcopy(message_doc.get(field_name))
@@ -564,13 +565,16 @@
     hybrid_citations = list(message_doc.get('hybrid_citations', []) or (source_message_doc or {}).get('hybrid_citations', []) or [])
     web_search_citations = list(message_doc.get('web_search_citations', []) or (source_message_doc or {}).get('web_search_citations', []) or [])
     agent_citations = list(message_doc.get('agent_citations', []) or (source_message_doc or {}).get('agent_citations', []) or [])
+    m365_citations = list(message_doc.get('m365_citations', []) or (source_message_doc or {}).get('m365_citations', []) or [])
 
     if hybrid_citations:
         generation_details['document_citation_count'] = len(hybrid_citations)
     if web_search_citations:
         generation_details['web_citation_count'] = len(web_search_citations)
     if agent_citations:
         generation_details['agent_citation_count'] = len(agent_citations)
+    if m365_citations:
+        generation_details['m365_citation_count'] = len(m365_citations)
 
     return generation_details
 
@@ -815,6 +819,7 @@
         'citation_tracking_version',
         'cited_hybrid_citations',
         'cited_web_search_citations',
+        'm365_citations',
     ):
         if field_name in message_doc:
             payload[field_name] = deepcopy(message_doc.get(field_name))