langchain[patch]: attach hub metadata (#18830)

libs/langchain/langchain/hub.py

@@ -1,10 +1,13 @@
 """Interface with the LangChain Hub."""
+
 from __future__ import annotations
 
+import json
 from typing import TYPE_CHECKING, Any, Optional
 
 from langchain_core.load.dump import dumps
 from langchain_core.load.load import loads
+from langchain_core.prompts import BasePromptTemplate
 
 if TYPE_CHECKING:
     from langchainhub import Client
@@ -77,5 +80,12 @@ def pull(
     :param api_key: The API key to use to authenticate with the LangChain Hub API.
     """
     client = _get_client(api_url=api_url, api_key=api_key)
-    resp: str = client.pull(owner_repo_commit)
-    return loads(resp)
+    res_dict = client.pull_repo(owner_repo_commit)
+    obj = loads(json.dumps(res_dict["manifest"]))
+    if isinstance(obj, BasePromptTemplate):
+        if obj.metadata is None:
+            obj.metadata = {}
+        obj.metadata["lc_hub_owner"] = res_dict["owner"]
+        obj.metadata["lc_hub_repo"] = res_dict["repo"]
+        obj.metadata["lc_hub_commit_hash"] = res_dict["commit_hash"]
+    return obj

libs/langchain/pyproject.toml

@@ -172,6 +172,7 @@ anthropic = "^0.3.11"
 langchain-core = {path = "../core", develop = true}
 langchain-community = {path = "../community", develop = true}
 langchain-text-splitters = {path = "../text-splitters", develop = true}
+langchainhub = "^0.1.15"
 
 [tool.poetry.group.lint]
 optional = true

libs/langchain/tests/integration_tests/test_hub.py

@@ -0,0 +1,15 @@
+from langchain_core.prompts import ChatPromptTemplate
+
+from langchain import hub
+
+
+def test_hub_pull_public_prompt() -> None:
+    prompt = hub.pull("efriis/my-first-prompt")
+    assert isinstance(prompt, ChatPromptTemplate)
+    assert prompt.metadata is not None
+    assert prompt.metadata["lc_hub_owner"] == "efriis"
+    assert prompt.metadata["lc_hub_repo"] == "my-first-prompt"
+    assert (
+        prompt.metadata["lc_hub_commit_hash"]
+        == "52668c2f392f8f52d2fc0d6b60cb964e3961934fdbd5dbe72b62926be6b51742"
+    )