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
617 B
Python
17 lines
617 B
Python
"""Utils for LLM Tests."""
|
|
|
|
from langchain.llms.base import BaseLLM
|
|
|
|
|
|
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) == 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" and field != "pipeline":
|
|
val = getattr(llm, field)
|
|
new_val = getattr(loaded_llm, field)
|
|
assert new_val == val
|