jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

ref: Add and use update_component_build_config utility (#5226)

Browse source 
* Add and use update_component_build_config utility

* [autofix.ci] apply automated fixes

---------

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
This commit is contained in:
Christophe Bornet 2024-12-13 00:58:19 +01:00 • committed by GitHub
commit 384ac5e80e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
21 changed files with 78 additions and 50 deletions
Download patch file Download diff file Expand all files Collapse all files

10
src/backend/tests/unit/components/models/test_chatollama_component.py
View file

@ -49,7 +49,7 @@ async def test_update_build_config_mirostat_disabled(component):
field_value = "Disabled"
field_name = "mirostat"
updated_config = await component.aupdate_build_config(build_config, field_value, field_name)
updated_config = await component.update_build_config(build_config, field_value, field_name)
assert updated_config["mirostat_eta"]["advanced"] is True
assert updated_config["mirostat_tau"]["advanced"] is True
@ -65,7 +65,7 @@ async def test_update_build_config_mirostat_enabled(component):
field_value = "Mirostat 2.0"
field_name = "mirostat"
updated_config = await component.aupdate_build_config(build_config, field_value, field_name)
updated_config = await component.update_build_config(build_config, field_value, field_name)
assert updated_config["mirostat_eta"]["advanced"] is False
assert updated_config["mirostat_tau"]["advanced"] is False
@ -88,7 +88,7 @@ async def test_update_build_config_model_name(mock_get, component):
field_value = None
field_name = "model_name"
updated_config = await component.aupdate_build_config(build_config, field_value, field_name)
updated_config = await component.update_build_config(build_config, field_value, field_name)
assert updated_config["model_name"]["options"] == ["model1", "model2"]
@ -98,12 +98,12 @@ async def test_update_build_config_keep_alive(component):
field_value = "Keep"
field_name = "keep_alive_flag"
updated_config = await component.aupdate_build_config(build_config, field_value, field_name)
updated_config = await component.update_build_config(build_config, field_value, field_name)
assert updated_config["keep_alive"]["value"] == "-1"
assert updated_config["keep_alive"]["advanced"] is True
field_value = "Immediately"
updated_config = await component.aupdate_build_config(build_config, field_value, field_name)
updated_config = await component.update_build_config(build_config, field_value, field_name)
assert updated_config["keep_alive"]["value"] == "0"
assert updated_config["keep_alive"]["advanced"] is True

23
src/backend/tests/unit/custom/custom_component/test_component.py
View file

@ -7,6 +7,7 @@ from langflow.components.custom_component import CustomComponent
from langflow.components.inputs import ChatInput
from langflow.components.models import OpenAIModelComponent
from langflow.components.outputs import ChatOutput
from langflow.custom.utils import update_component_build_config
from langflow.schema import dotdict
from langflow.template import Output
from typing_extensions import override
@ -65,7 +66,7 @@ def test_set_required_inputs_various_components():
assert _assert_all_outputs_have_different_required_inputs(agent.outputs)
async def test_update_build_config_backward_compatibility():
async def test_update_component_build_config_sync():
class TestComponent(CustomComponent):
@override
def update_build_config(
@ -79,5 +80,23 @@ async def test_update_build_config_backward_compatibility():
component = TestComponent()
build_config = dotdict()
build_config = await component.aupdate_build_config(build_config, "", "")
build_config = await update_component_build_config(component, build_config, "", "")
assert build_config["foo"] == "bar"
async def test_update_component_build_config_async():
class TestComponent(CustomComponent):
@override
async def update_build_config(
self,
build_config: dotdict,
field_value: Any,
field_name: str | None = None,
):
build_config["foo"] = "bar"
return build_config
component = TestComponent()
build_config = dotdict()
build_config = await update_component_build_config(component, build_config, "", "")
assert build_config["foo"] == "bar"