Harrison/llm saving (#331)

Co-authored-by: Akash Samant <70665700+asamant21@users.noreply.github.com>
This commit is contained in:
Harrison Chase
2022-12-13 06:46:01 -08:00
committed by GitHub
parent 595cc1ae1a
commit 9bb7195085
20 changed files with 279 additions and 28 deletions

View File

@@ -0,0 +1,16 @@
"""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