"Replace HTTP with HTTPS in BilibiliLoader.swift and print the update…

…d thumbnail URL in langchain_swiftTests.swift."
buhe · Aug 1, 2023 · b5adb39 · b5adb39
1 parent 79a4469
commit b5adb39
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Sources/LangChain/document_loaders/BilibiliLoader.swift b/Sources/LangChain/document_loaders/BilibiliLoader.swift
@@ -20,7 +20,7 @@ public struct BilibiliLoader: BaseLoader {
         return [Document(page_content: info!.subtitle, metadata: [
             "title": info!.title,
             "desc": info!.desc,
-            "thumbnail": info!.thumbnail
+            "thumbnail": info!.thumbnail.replacingOccurrences(of: "http", with: "https")
         ])]
     }
 

diff --git a/Tests/LangChainTests/langchain_swiftTests.swift b/Tests/LangChainTests/langchain_swiftTests.swift
@@ -804,7 +804,7 @@ May God bless you all. May God protect our troops.
     func testBilibiliLoader() async throws {
         let loader = BilibiliLoader(videoId: "BV1fu411G7e3")
         let doc = await loader.load()
-
+        print(doc.first!.metadata["thumbnail"]!)
         XCTAssertFalse(doc.isEmpty)
         XCTAssertNotEqual("", doc.first!.page_content)
     }