forked from Archives/langchain
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
17 lines
581 B
Python
17 lines
581 B
Python
2 years ago
|
"""Utils for LLM Tests."""
|
||
|
|
||
|
from langchain.llms.base import LLM
|
||
|
|
||
|
|
||
|
def assert_llm_equality(llm: LLM, loaded_llm: LLM) -> None:
|
||
|
"""Assert LLM Equality for tests."""
|
||
|
# Check that they are the same type.
|
||
|
assert type(llm) == 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 llm.__fields__.keys():
|
||
|
if field != "client":
|
||
|
val = getattr(llm, field)
|
||
|
new_val = getattr(loaded_llm, field)
|
||
|
assert new_val == val
|