mirror of
https://github.com/hwchase17/langchain.git
synced 2025-07-07 21:50:25 +00:00
Revert "[Hub|tracing] Tag hub prompts" (#14735)
Reverts langchain-ai/langchain#14720
This commit is contained in:
parent
b9975fac89
commit
9d4100f915
@ -5,7 +5,6 @@ from typing import TYPE_CHECKING, Any, Optional
|
|||||||
|
|
||||||
from langchain_core.load.dump import dumps
|
from langchain_core.load.dump import dumps
|
||||||
from langchain_core.load.load import loads
|
from langchain_core.load.load import loads
|
||||||
from langchain_core.runnables import Runnable
|
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from langchainhub import Client
|
from langchainhub import Client
|
||||||
@ -79,7 +78,4 @@ def pull(
|
|||||||
"""
|
"""
|
||||||
client = _get_client(api_url=api_url, api_key=api_key)
|
client = _get_client(api_url=api_url, api_key=api_key)
|
||||||
resp: str = client.pull(owner_repo_commit)
|
resp: str = client.pull(owner_repo_commit)
|
||||||
loaded = loads(resp)
|
return loads(resp)
|
||||||
if isinstance(loaded, Runnable):
|
|
||||||
return loaded.with_config(metadata={"hub_owner_repo_commit": owner_repo_commit})
|
|
||||||
return loaded
|
|
||||||
|
@ -1,61 +0,0 @@
|
|||||||
from typing import Any, List
|
|
||||||
from unittest.mock import MagicMock, Mock, patch
|
|
||||||
|
|
||||||
from langchain_core.load.dump import dumps
|
|
||||||
from langchain_core.prompts import ChatPromptTemplate
|
|
||||||
from langchain_core.runnables.base import RunnableBinding
|
|
||||||
from langchain_core.tracers.base import BaseTracer
|
|
||||||
from langchain_core.tracers.schemas import Run
|
|
||||||
|
|
||||||
from langchain import hub
|
|
||||||
|
|
||||||
|
|
||||||
class FakeTracer(BaseTracer):
|
|
||||||
"""Fake tracer that records LangChain execution."""
|
|
||||||
|
|
||||||
def __init__(self) -> None:
|
|
||||||
"""Initialize the tracer."""
|
|
||||||
super().__init__()
|
|
||||||
self.runs: List[Run] = []
|
|
||||||
|
|
||||||
def _persist_run(self, run: Run) -> None:
|
|
||||||
"""Persist a run."""
|
|
||||||
self.runs.append(run)
|
|
||||||
|
|
||||||
|
|
||||||
repo_dict = {
|
|
||||||
"wfh/my-prompt-1": ChatPromptTemplate.from_messages(
|
|
||||||
[("system", "a"), ("user", "1")]
|
|
||||||
),
|
|
||||||
"wfh/my-random-object": {"Hi": "there"},
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
def repo_lookup(owner_repo_commit: str, **kwargs: Any) -> str:
|
|
||||||
return dumps(repo_dict[owner_repo_commit])
|
|
||||||
|
|
||||||
|
|
||||||
@patch("langchain.hub._get_client")
|
|
||||||
def test_hub_pull_metadata(mock_get_client: Mock) -> None:
|
|
||||||
mock_client = MagicMock()
|
|
||||||
mock_client.pull = repo_lookup
|
|
||||||
mock_get_client.return_value = mock_client
|
|
||||||
res = hub.pull("wfh/my-prompt-1")
|
|
||||||
assert isinstance(res, RunnableBinding)
|
|
||||||
tracer = FakeTracer()
|
|
||||||
result = res.invoke({}, {"callbacks": [tracer]})
|
|
||||||
assert result.messages[0].content == "a"
|
|
||||||
assert result.messages[1].content == "1"
|
|
||||||
assert len(tracer.runs) == 1
|
|
||||||
run = tracer.runs[0]
|
|
||||||
assert run.extra is not None
|
|
||||||
assert run.extra["metadata"]["hub_owner_repo_commit"] == "wfh/my-prompt-1" # type: ignore
|
|
||||||
|
|
||||||
|
|
||||||
@patch("langchain.hub._get_client")
|
|
||||||
def test_hub_pull_random_object(mock_get_client: Mock) -> None:
|
|
||||||
mock_client = MagicMock()
|
|
||||||
mock_client.pull = repo_lookup
|
|
||||||
mock_get_client.return_value = mock_client
|
|
||||||
res = hub.pull("wfh/my-random-object")
|
|
||||||
assert res == {"Hi": "there"}
|
|
Loading…
Reference in New Issue
Block a user