From b4a0d206d94cdb165a675da9c722fecd506522b5 Mon Sep 17 00:00:00 2001 From: Harrison Chase Date: Tue, 26 Dec 2023 12:32:55 -0800 Subject: [PATCH] [core: minor] fix getters (#15181) --- libs/core/langchain_core/beta/runnables/context.py | 5 ++--- libs/core/tests/unit_tests/language_models/test_imports.py | 1 + 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/libs/core/langchain_core/beta/runnables/context.py b/libs/core/langchain_core/beta/runnables/context.py index 350e3a1926..f13af30b2c 100644 --- a/libs/core/langchain_core/beta/runnables/context.py +++ b/libs/core/langchain_core/beta/runnables/context.py @@ -109,8 +109,6 @@ def _config_with_context( raise ValueError( f"Deadlock detected between context keys {key} and {dep}" ) - if len(getters) < 1: - raise ValueError(f"Expected at least one getter for context key {key}") if len(setters) != 1: raise ValueError(f"Expected exactly one setter for context key {key}") setter_idx = setters[0][1] @@ -119,7 +117,8 @@ def _config_with_context( f"Context setter for key {key} must be defined after all getters." ) - context_funcs[getters[0][0].id] = partial(getter, events[key], values) + if getters: + context_funcs[getters[0][0].id] = partial(getter, events[key], values) context_funcs[setters[0][0].id] = partial(setter, events[key], values) return patch_config(config, configurable=context_funcs) diff --git a/libs/core/tests/unit_tests/language_models/test_imports.py b/libs/core/tests/unit_tests/language_models/test_imports.py index 65d627c56f..95354f08d5 100644 --- a/libs/core/tests/unit_tests/language_models/test_imports.py +++ b/libs/core/tests/unit_tests/language_models/test_imports.py @@ -8,6 +8,7 @@ EXPECTED_ALL = [ "LLM", "LanguageModelInput", "LanguageModelOutput", + "LanguageModelLike", "get_tokenizer", "LanguageModelLike", ]