mirror of
https://github.com/hwchase17/langchain.git
synced 2025-05-02 13:55:42 +00:00
98779797fe
Change all usages of __fields__ with get_fields adapter merged into
langchain_core.
Code mod generated using the following grit pattern:
```
engine marzano(0.1)
language python
`$X.__fields__` => `get_fields($X)` where {
add_import(source="langchain_core.utils.pydantic", name="get_fields")
}
```
18 lines
687 B
Python
18 lines
687 B
Python
"""Utils for LLM Tests."""
|
|
|
|
from langchain_core.language_models.llms import BaseLLM
|
|
from langchain_core.utils.pydantic import get_fields
|
|
|
|
|
|
def assert_llm_equality(llm: BaseLLM, loaded_llm: BaseLLM) -> None:
|
|
"""Assert LLM Equality for tests."""
|
|
# Check that they are the same type.
|
|
assert type(llm) is type(loaded_llm)
|
|
# Client field can be session based, so hash is different despite
|
|
# all other values being the same, so just assess all other fields
|
|
for field in get_fields(llm).keys():
|
|
if field != "client" and field != "pipeline":
|
|
val = getattr(llm, field)
|
|
new_val = getattr(loaded_llm, field)
|
|
assert new_val == val
|