Merge remote-tracking branch 'origin/cz/bug/state/zustand' into update_lc

This commit is contained in:
anovazzi1 2024-01-09 10:52:34 -03:00
commit db13e00258
3 changed files with 172 additions and 107 deletions

View file

@ -1,3 +1,4 @@
import { useSignal, useSignalEffect } from "@preact/signals-react";
import { cloneDeep } from "lodash";
import { forwardRef, useEffect, useState } from "react";
import ShadTooltip from "../../components/ShadTooltipComponent";
@ -54,32 +55,31 @@ const EditNodeModal = forwardRef(
},
ref
) => {
const [myData, setMyData] = useState(data);
const setPending = useFlowStore((state) => state.setPending);
const edges = useFlowStore((state) => state.edges);
const setNode = useFlowStore((state) => state.setNode);
const myData = useSignal(data);
const [render, setRender] = useState(false);
useSignalEffect(() => {
setRender(!render);
});
function changeAdvanced(n) {
setMyData((old) => {
let newData = cloneDeep(old);
newData.node!.template[n].advanced =
!newData.node!.template[n].advanced;
return newData;
});
const newValue = cloneDeep(myData.value);
newValue.node!.template[n].advanced =
!newValue.node!.template[n].advanced;
myData.value = newValue;
}
const handleOnNewValue = (newValue: any, name) => {
setMyData((old) => {
let newData = cloneDeep(old);
newData.node!.template[name].value = newValue;
return newData;
});
const newSignalValue = cloneDeep(myData.value);
newSignalValue.node!.template[name].value = newValue;
myData.value = newSignalValue;
};
useEffect(() => {
if (open) {
setMyData(data); // reset data to what it is on node when opening modal
myData.value = cloneDeep(data); // clone data to avoid changing data on node when opening modal
}
}, [open]);
@ -92,15 +92,15 @@ const EditNodeModal = forwardRef(
open={open}
setOpen={setOpen}
onChangeOpenModal={(open) => {
setMyData(data);
myData.value = cloneDeep(data);
}}
>
<BaseModal.Trigger>
<></>
</BaseModal.Trigger>
<BaseModal.Header description={myData.node?.description!}>
<span className="pr-2">{myData.type}</span>
<Badge variant="secondary">ID: {myData.id}</Badge>
<BaseModal.Header description={myData.value.node?.description!}>
<span className="pr-2">{myData.value.type}</span>
<Badge variant="secondary">ID: {myData.value.id}</Badge>
</BaseModal.Header>
<BaseModal.Content>
<div className="flex pb-2">
@ -133,21 +133,23 @@ const EditNodeModal = forwardRef(
</TableRow>
</TableHeader>
<TableBody className="p-0">
{Object.keys(myData.node!.template)
{Object.keys(myData.value.node!.template)
.filter(
(templateParam) =>
templateParam.charAt(0) !== "_" &&
myData.node?.template[templateParam].show &&
myData.value.node?.template[templateParam].show &&
LANGFLOW_SUPPORTED_TYPES.has(
myData.node.template[templateParam].type
myData.value?.node?.template[templateParam].type
)
)
.map((templateParam, index) => {
let id = {
inputTypes:
myData.node!.template[templateParam].input_types,
type: myData.node!.template[templateParam].type,
id: myData.id,
myData.value.node!.template[templateParam]
.input_types,
type: myData.value.node!.template[templateParam]
.type,
id: myData.value.id,
fieldName: templateParam,
};
let disabled =
@ -155,12 +157,14 @@ const EditNodeModal = forwardRef(
(edge) =>
edge.targetHandle ===
scapedJSONStringfy(
myData.node!.template[templateParam].proxy
myData.value.node!.template[templateParam]
.proxy
? {
...id,
proxy:
myData.node?.template[templateParam]
.proxy,
myData.value.node?.template[
templateParam
].proxy,
}
: id
)
@ -170,39 +174,46 @@ const EditNodeModal = forwardRef(
<TableCell className="truncate p-0 text-center text-sm text-foreground sm:px-3">
<ShadTooltip
content={
myData.node?.template[templateParam].proxy
? myData.node?.template[templateParam]
.proxy?.id
myData.value.node?.template[templateParam]
.proxy
? myData.value.node?.template[
templateParam
].proxy?.id
: null
}
>
<span>
{myData.node?.template[templateParam]
{myData.value.node?.template[templateParam]
.display_name
? myData.node.template[templateParam]
.display_name
: myData.node?.template[templateParam]
.name}
? myData.value?.node?.template[
templateParam
].display_name
: myData.value.node?.template[
templateParam
].name}
</span>
</ShadTooltip>
</TableCell>
<TableCell className="w-[300px] p-0 text-center text-xs text-foreground ">
{myData.node?.template[templateParam].type ===
"str" &&
!myData.node.template[templateParam].options ? (
{myData.value.node?.template[templateParam]
.type === "str" &&
!myData.value?.node?.template[templateParam]
.options ? (
<div className="mx-auto">
{myData.node.template[templateParam]
{myData.value?.node?.template[templateParam]
.list ? (
<InputListComponent
editNode={true}
disabled={disabled}
value={
!myData.node.template[templateParam]
.value ||
myData.node.template[templateParam]
.value === ""
!myData.value?.node?.template[
templateParam
].value ||
myData.value?.node?.template[
templateParam
].value === ""
? [""]
: myData.node.template[
: myData.value?.node?.template[
templateParam
].value
}
@ -213,16 +224,18 @@ const EditNodeModal = forwardRef(
);
}}
/>
) : myData.node.template[templateParam]
.multiline ? (
) : myData.value?.node?.template[
templateParam
].multiline ? (
<TextAreaComponent
id={"textarea-edit-" + index}
data-testid={"textarea-edit-" + index}
disabled={disabled}
editNode={true}
value={
myData.node.template[templateParam]
.value ?? ""
myData.value?.node?.template[
templateParam
].value ?? ""
}
onChange={(
value: string | string[]
@ -239,12 +252,14 @@ const EditNodeModal = forwardRef(
editNode={true}
disabled={disabled}
password={
myData.node.template[templateParam]
.password ?? false
myData.value?.node?.template[
templateParam
].password ?? false
}
value={
myData.node.template[templateParam]
.value ?? ""
myData.value?.node?.template[
templateParam
].value ?? ""
}
onChange={(value) => {
handleOnNewValue(
@ -255,24 +270,25 @@ const EditNodeModal = forwardRef(
/>
)}
</div>
) : myData.node?.template[templateParam]
) : myData.value.node?.template[templateParam]
.type === "NestedDict" ? (
<div className=" w-full">
<DictComponent
disabled={disabled}
editNode={true}
value={
myData.node!.template[
myData.value.node!.template[
templateParam
]?.value?.toString() === "{}"
? {
yourkey: "value",
}
: myData.node!.template[templateParam]
.value
: myData.value.node!.template[
templateParam
].value
}
onChange={(newValue) => {
myData.node!.template[
myData.value.node!.template[
templateParam
].value = newValue;
handleOnNewValue(
@ -283,13 +299,13 @@ const EditNodeModal = forwardRef(
id="editnode-div-dict-input"
/>
</div>
) : myData.node?.template[templateParam]
) : myData.value.node?.template[templateParam]
.type === "dict" ? (
<div
className={classNames(
"max-h-48 w-full overflow-auto custom-scroll",
myData.node!.template[templateParam].value
?.length > 1
myData.value.node!.template[templateParam]
.value?.length > 1
? "my-3"
: ""
)}
@ -298,13 +314,15 @@ const EditNodeModal = forwardRef(
disabled={disabled}
editNode={true}
value={
myData.node!.template[templateParam]
.value?.length === 0 ||
!myData.node!.template[templateParam]
.value
myData.value.node!.template[
templateParam
].value?.length === 0 ||
!myData.value.node!.template[
templateParam
].value
? [{ "": "" }]
: convertObjToArray(
myData.node!.template[
myData.value.node!.template[
templateParam
].value
)
@ -313,7 +331,7 @@ const EditNodeModal = forwardRef(
onChange={(newValue) => {
const valueToNumbers =
convertValuesToNumbers(newValue);
myData.node!.template[
myData.value.node!.template[
templateParam
].value = valueToNumbers;
setErrorDuplicateKey(
@ -326,7 +344,7 @@ const EditNodeModal = forwardRef(
}}
/>
</div>
) : myData.node?.template[templateParam]
) : myData.value.node?.template[templateParam]
.type === "bool" ? (
<div className="ml-auto">
{" "}
@ -334,8 +352,9 @@ const EditNodeModal = forwardRef(
id={"toggle-edit-" + index}
disabled={disabled}
enabled={
myData.node.template[templateParam]
.value
myData.value?.node?.template[
templateParam
].value
}
setEnabled={(isEnabled) => {
handleOnNewValue(
@ -346,48 +365,52 @@ const EditNodeModal = forwardRef(
size="small"
/>
</div>
) : myData.node?.template[templateParam]
) : myData.value.node?.template[templateParam]
.type === "float" ? (
<div className="mx-auto">
<FloatComponent
disabled={disabled}
editNode={true}
rangeSpec={
myData.node!.template[templateParam]
.rangeSpec
myData.value.node!.template[
templateParam
].rangeSpec
}
value={
myData.node.template[templateParam]
.value ?? ""
myData.value?.node?.template[
templateParam
].value ?? ""
}
onChange={(value) => {
handleOnNewValue(value, templateParam);
}}
/>
</div>
) : myData.node?.template[templateParam]
) : myData.value.node?.template[templateParam]
.type === "str" &&
myData.node.template[templateParam]
myData.value?.node?.template[templateParam]
.options ? (
<div className="mx-auto">
<Dropdown
numberOfOptions={nodeLength}
editNode={true}
options={
myData.node.template[templateParam]
.options
myData.value?.node?.template[
templateParam
].options
}
onSelect={(value) =>
handleOnNewValue(value, templateParam)
}
value={
myData.node.template[templateParam]
.value ?? "Choose an option"
myData.value?.node?.template[
templateParam
].value ?? "Choose an option"
}
id={"dropdown-edit-" + index}
></Dropdown>
</div>
) : myData.node?.template[templateParam]
) : myData.value.node?.template[templateParam]
.type === "int" ? (
<div className="mx-auto">
<IntComponent
@ -395,30 +418,33 @@ const EditNodeModal = forwardRef(
disabled={disabled}
editNode={true}
value={
myData.node.template[templateParam]
.value ?? ""
myData.value?.node?.template[
templateParam
].value ?? ""
}
onChange={(value) => {
handleOnNewValue(value, templateParam);
}}
/>
</div>
) : myData.node?.template[templateParam]
) : myData.value.node?.template[templateParam]
.type === "file" ? (
<div className="mx-auto">
<InputFileComponent
editNode={true}
disabled={disabled}
value={
myData.node.template[templateParam]
.value ?? ""
myData.value?.node?.template[
templateParam
].value ?? ""
}
onChange={(value: string | string[]) => {
handleOnNewValue(value, templateParam);
}}
fileTypes={
myData.node.template[templateParam]
.fileTypes
myData.value?.node?.template[
templateParam
].fileTypes
}
onFileChange={(filePath: string) => {
data.node!.template[
@ -427,23 +453,24 @@ const EditNodeModal = forwardRef(
}}
></InputFileComponent>
</div>
) : myData.node?.template[templateParam]
) : myData.value.node?.template[templateParam]
.type === "prompt" ? (
<div className="mx-auto">
<PromptAreaComponent
readonly={
myData.node?.flow ? true : false
myData.value.node?.flow ? true : false
}
field_name={templateParam}
editNode={true}
disabled={disabled}
nodeClass={myData.node}
nodeClass={myData.value.node}
setNodeClass={(nodeClass) => {
myData.node = nodeClass;
myData.value.node = nodeClass;
}}
value={
myData.node.template[templateParam]
.value ?? ""
myData.value?.node?.template[
templateParam
].value ?? ""
}
onChange={(value: string | string[]) => {
handleOnNewValue(value, templateParam);
@ -454,14 +481,15 @@ const EditNodeModal = forwardRef(
}
/>
</div>
) : myData.node?.template[templateParam]
) : myData.value.node?.template[templateParam]
.type === "code" ? (
<div className="mx-auto">
<CodeAreaComponent
readonly={
myData.node?.flow &&
myData.node.template[templateParam]
.dynamic
myData.value.node?.flow &&
myData.value?.node?.template[
templateParam
].dynamic
? true
: false
}
@ -476,8 +504,9 @@ const EditNodeModal = forwardRef(
disabled={disabled}
editNode={true}
value={
myData.node.template[templateParam]
.value ?? ""
myData.value?.node?.template[
templateParam
].value ?? ""
}
onChange={(value: string | string[]) => {
handleOnNewValue(value, templateParam);
@ -485,7 +514,7 @@ const EditNodeModal = forwardRef(
id={"code-area-edit" + index}
/>
</div>
) : myData.node?.template[templateParam]
) : myData.value.node?.template[templateParam]
.type === "Any" ? (
"-"
) : (
@ -497,11 +526,13 @@ const EditNodeModal = forwardRef(
<ToggleShadComponent
id={
"show" +
myData.node?.template[templateParam].name
myData.value.node?.template[templateParam]
.name
}
enabled={
!myData.node?.template[templateParam]
.advanced
!myData.value.node?.template[
templateParam
].advanced
}
setEnabled={(e) => {
changeAdvanced(templateParam);
@ -531,10 +562,9 @@ const EditNodeModal = forwardRef(
...old,
data: {
...old.data,
node: myData.node,
node: myData.value.node,
},
}));
setPending(true);
setOpen(false);
}}
type="submit"