Apply to metadata

langchain-ai · Jan 10, 2025 · ade4dcb · ade4dcb
1 parent e03d734
commit ade4dcb
Showing 1 changed file with 14 additions and 8 deletions.
diff --git a/langchain/src/hub.ts b/langchain/src/hub.ts
@@ -55,12 +55,18 @@ export async function pull<T extends Runnable>(
     includeModel: options?.includeModel,
   });
 
-  const prompt = await load<T>(JSON.stringify(promptObject.manifest));
-  return prompt.withConfig({
-    metadata: {
-      lc_hub_owner: promptObject.owner,
-      lc_hub_repo: promptObject.repo,
-      lc_hub_commit_hash: promptObject.commit_hash,
-    },
-  });
+  if (promptObject.manifest.kwargs?.metadata === undefined) {
+    promptObject.manifest.kwargs = {
+      ...promptObject.manifest.kwargs,
+      metadata: {},
+    };
+  }
+
+  promptObject.manifest.kwargs.metadata = {
+    ...promptObject.manifest.kwargs.metadata,
+    lc_hub_owner: promptObject.owner,
+    lc_hub_repo: promptObject.repo,
+    lc_hub_commit_hash: promptObject.commit_hash,
+  };
+  return load<T>(JSON.stringify(promptObject.manifest));
 }