mirror of
https://github.com/hwchase17/langchain.git
synced 2025-09-21 18:39:57 +00:00
core[patch]: Hide aliases when serializing (#16888)
Currently, if you dump an object initialized with an alias, we'll still dump the secret values since they're retained in the kwargs
This commit is contained in:
@@ -1,5 +1,6 @@
|
||||
"""Test for Serializable base class"""
|
||||
|
||||
import json
|
||||
import os
|
||||
from typing import Any, Dict, List
|
||||
from unittest.mock import patch
|
||||
@@ -11,6 +12,7 @@ from langchain_core.load.dump import dumps
|
||||
from langchain_core.load.serializable import Serializable
|
||||
from langchain_core.prompts.chat import ChatPromptTemplate, HumanMessagePromptTemplate
|
||||
from langchain_core.prompts.prompt import PromptTemplate
|
||||
from langchain_core.pydantic_v1 import Field, root_validator
|
||||
from langchain_core.tracers.langchain import LangChainTracer
|
||||
|
||||
from langchain.chains.llm import LLMChain
|
||||
@@ -167,3 +169,85 @@ def test_person_with_invalid_kwargs() -> None:
|
||||
person = Person(secret="hello")
|
||||
with pytest.raises(TypeError):
|
||||
dumps(person, invalid_kwarg="hello")
|
||||
|
||||
|
||||
class TestClass(Serializable):
|
||||
my_favorite_secret: str = Field(alias="my_favorite_secret_alias")
|
||||
my_other_secret: str = Field()
|
||||
|
||||
class Config:
|
||||
"""Configuration for this pydantic object."""
|
||||
|
||||
allow_population_by_field_name = True
|
||||
|
||||
@root_validator(pre=True)
|
||||
def get_from_env(cls, values: Dict) -> Dict:
|
||||
"""Get the values from the environment."""
|
||||
if "my_favorite_secret" not in values:
|
||||
values["my_favorite_secret"] = os.getenv("MY_FAVORITE_SECRET")
|
||||
if "my_other_secret" not in values:
|
||||
values["my_other_secret"] = os.getenv("MY_OTHER_SECRET")
|
||||
return values
|
||||
|
||||
@classmethod
|
||||
def is_lc_serializable(cls) -> bool:
|
||||
return True
|
||||
|
||||
@classmethod
|
||||
def get_lc_namespace(cls) -> List[str]:
|
||||
return ["my", "special", "namespace"]
|
||||
|
||||
@property
|
||||
def lc_secrets(self) -> Dict[str, str]:
|
||||
return {
|
||||
"my_favorite_secret": "MY_FAVORITE_SECRET",
|
||||
"my_other_secret": "MY_OTHER_SECRET",
|
||||
}
|
||||
|
||||
|
||||
def test_aliases_hidden() -> None:
|
||||
test_class = TestClass(my_favorite_secret="hello", my_other_secret="world")
|
||||
dumped = json.loads(dumps(test_class, pretty=True))
|
||||
expected_dump = {
|
||||
"lc": 1,
|
||||
"type": "constructor",
|
||||
"id": ["my", "special", "namespace", "TestClass"],
|
||||
"kwargs": {
|
||||
"my_favorite_secret": {
|
||||
"lc": 1,
|
||||
"type": "secret",
|
||||
"id": ["MY_FAVORITE_SECRET"],
|
||||
},
|
||||
"my_other_secret": {"lc": 1, "type": "secret", "id": ["MY_OTHER_SECRET"]},
|
||||
},
|
||||
}
|
||||
assert dumped == expected_dump
|
||||
# Check while patching the os environment
|
||||
with patch.dict(
|
||||
os.environ, {"MY_FAVORITE_SECRET": "hello", "MY_OTHER_SECRET": "world"}
|
||||
):
|
||||
test_class = TestClass()
|
||||
dumped = json.loads(dumps(test_class, pretty=True))
|
||||
|
||||
# Check by alias
|
||||
test_class = TestClass(my_favorite_secret_alias="hello", my_other_secret="world")
|
||||
dumped = json.loads(dumps(test_class, pretty=True))
|
||||
expected_dump = {
|
||||
"lc": 1,
|
||||
"type": "constructor",
|
||||
"id": ["my", "special", "namespace", "TestClass"],
|
||||
"kwargs": {
|
||||
"my_favorite_secret": {
|
||||
"lc": 1,
|
||||
"type": "secret",
|
||||
"id": ["MY_FAVORITE_SECRET"],
|
||||
},
|
||||
"my_favorite_secret_alias": {
|
||||
"lc": 1,
|
||||
"type": "secret",
|
||||
"id": ["MY_FAVORITE_SECRET"],
|
||||
},
|
||||
"my_other_secret": {"lc": 1, "type": "secret", "id": ["MY_OTHER_SECRET"]},
|
||||
},
|
||||
}
|
||||
assert dumped == expected_dump
|
||||
|
Reference in New Issue
Block a user