use MessageTextInput

This commit is contained in:
Nicolò Boschi 2024-06-24 12:28:05 +02:00 • committed by Gabriel Luiz Freitas Almeida
commit a158082175

View file

@ -1,6 +1,6 @@
from typing import Any from typing import Any
from langflow.custom import Component from langflow.custom import Component
from langflow.inputs.inputs import DictInput, SecretStrInput, StrInput from langflow.inputs.inputs import DictInput, SecretStrInput, StrInput, MessageTextInput
from langflow.template.field.base import Output from langflow.template.field.base import Output
@ -11,12 +11,12 @@ class AstraVectorize(Component):
icon = "AstraDB" icon = "AstraDB"
inputs = [ inputs = [
StrInput( MessageTextInput(
name="provider", name="provider",
display_name="Provider name", display_name="Provider name",
info='The embedding provider to use.', info='The embedding provider to use.',
), ),
StrInput( MessageTextInput(
name="model_name", name="model_name",
display_name="Model name", display_name="Model name",
info='The embedding model to use.', info='The embedding model to use.',