Add a method to access typed metadata keys

intitni · intitni · commit 4860fc83f92f · 2023-09-21T16:11:56.000+08:00
diff --git a/Tool/Sources/LangChain/DocumentLoader/DocumentLoader.swift b/Tool/Sources/LangChain/DocumentLoader/DocumentLoader.swift
@@ -9,6 +9,11 @@ public struct Document: Codable {
         self.pageContent = pageContent
         self.metadata = metadata
     }
+    
+    public func metadata<Key>(_ keyPath: KeyPath<Key.Type, String>) -> JSONValue? {
+        let key = Key.self[keyPath: keyPath]
+        return metadata[key]
+    }
 }
 
 public protocol DocumentLoader {

Original file line number	Diff line number	Diff line change
`@@ -9,6 +9,11 @@ public struct Document: Codable {`
`9`	`9`	`self.pageContent = pageContent`
`10`	`10`	`self.metadata = metadata`
`11`	`11`	`}`
	`12`	`+`
	`13`	`+ public func metadata<Key>(_ keyPath: KeyPath<Key.Type, String>) -> JSONValue? {`
	`14`	`+ let key = Key.self[keyPath: keyPath]`
	`15`	`+ return metadata[key]`
	`16`	`+ }`
`12`	`17`	`}`
`13`	`18`
`14`	`19`	`public protocol DocumentLoader {`