既存のknowledge baseを使う場合に、Third party data sourcesに対応する (#767)

aws-samples · Mar 7, 2025 · fd6239e · fd6239e
1 parent dbf945e
commit fd6239e
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/backend/app/vector_search.py b/backend/app/vector_search.py
@@ -105,6 +105,22 @@ def extract_source_from_retrieval_result(
                 source_name = urlparse(url=uri).path.split("/")[-1]
                 return (source_name, uri)
 
+            elif location_type == "CONFLUENCE":
+                url = location.get("confluenceLocation", {}).get("url", "")
+                return (url, url) if url else None
+
+            elif location_type == "SALESFORCE":
+                url = location.get("salesforceLocation", {}).get("url", "")
+                return (url, url) if url else None
+
+            elif location_type == "SHAREPOINT":
+                url = location.get("sharePointLocation", {}).get("url", "")
+                return (url, url) if url else None
+
+            elif location_type == "KENDRA":
+                url = location.get("kendraDocumentLocation", {}).get("uri", "")
+                return (url, url) if url else None
+
             return None
 
         search_results = []