From 1be12a3c3286c857191bd9726ab5ffca3f8b697c Mon Sep 17 00:00:00 2001 From: cristhianzl Date: Fri, 7 Jun 2024 15:06:11 -0300 Subject: [PATCH] undefined fix on types --- src/backend/base/langflow/api/v1/schemas.py | 6 -- .../components/parameterComponent/index.tsx | 18 ++--- src/frontend/src/modals/IOModal/index.tsx | 67 ++++++++----------- 3 files changed, 38 insertions(+), 53 deletions(-) diff --git a/src/backend/base/langflow/api/v1/schemas.py b/src/backend/base/langflow/api/v1/schemas.py index c1cc041ae..cb230c9f5 100644 --- a/src/backend/base/langflow/api/v1/schemas.py +++ b/src/backend/base/langflow/api/v1/schemas.py @@ -244,12 +244,6 @@ class VerticesOrderResponse(BaseModel): run_id: UUID vertices_to_run: List[str] - -class Log(TypedDict): - message: Union[dict, str] - type: str - - class ResultDataResponse(BaseModel): results: Optional[Any] = Field(default_factory=dict) logs: List[Log | None] = Field(default_factory=list) diff --git a/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx b/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx index 437e5a1b1..145eaaf4b 100644 --- a/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx +++ b/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx @@ -44,9 +44,9 @@ import useFetchDataOnMount from "../../../hooks/use-fetch-data-on-mount"; import useHandleOnNewValue from "../../../hooks/use-handle-new-value"; import useHandleNodeClass from "../../../hooks/use-handle-node-class"; import useHandleRefreshButtonPress from "../../../hooks/use-handle-refresh-buttons"; +import OutputModal from "../outputModal"; import TooltipRenderComponent from "../tooltipRenderComponent"; import { TEXT_FIELD_TYPES } from "./constants"; -import OutputModal from "../outputModal"; export default function ParameterComponent({ left, @@ -98,7 +98,7 @@ export default function ParameterComponent({ debouncedHandleUpdateValues, setNode, renderTooltips, - setIsLoading, + setIsLoading ); const { handleNodeClass: handleNodeClassHook } = useHandleNodeClass( @@ -107,7 +107,7 @@ export default function ParameterComponent({ takeSnapshot, setNode, updateNodeInternals, - renderTooltips, + renderTooltips ); const { handleRefreshButtonPress: handleRefreshButtonPressHook } = @@ -116,7 +116,7 @@ export default function ParameterComponent({ let disabled = edges.some( (edge) => - edge.targetHandle === scapedJSONStringfy(proxy ? { ...id, proxy } : id), + edge.targetHandle === scapedJSONStringfy(proxy ? { ...id, proxy } : id) ) ?? false; const handleRefreshButtonPress = async (name, data) => { @@ -129,12 +129,12 @@ export default function ParameterComponent({ handleUpdateValues, setNode, renderTooltips, - setIsLoading, + setIsLoading ); const handleOnNewValue = async ( newValue: string | string[] | boolean | Object[], - skipSnapshot: boolean | undefined = false, + skipSnapshot: boolean | undefined = false ): Promise => { handleOnNewValueHook(newValue, skipSnapshot); }; @@ -216,7 +216,7 @@ export default function ParameterComponent({ className={classNames( left ? "my-12 -ml-0.5 " : " my-12 -mr-0.5 ", "h-3 w-3 rounded-full border-2 bg-background", - !showNode ? "mt-0" : "", + !showNode ? "mt-0" : "" )} style={{ borderColor: color ?? nodeColors.unknown, @@ -284,7 +284,7 @@ export default function ParameterComponent({ "h-5 w-5 rounded-md", displayOutputPreview && !unknownOutput ? " hover:bg-secondary-foreground/5 hover:text-medium-indigo" - : " cursor-not-allowed text-muted-foreground", + : " cursor-not-allowed text-muted-foreground" )} name={"ScanEye"} /> @@ -334,7 +334,7 @@ export default function ParameterComponent({ } className={classNames( left ? "-ml-0.5" : "-mr-0.5", - "h-3 w-3 rounded-full border-2 bg-background", + "h-3 w-3 rounded-full border-2 bg-background" )} style={{ borderColor: color ?? nodeColors.unknown }} onClick={() => setFilterEdge(groupedEdge.current)} diff --git a/src/frontend/src/modals/IOModal/index.tsx b/src/frontend/src/modals/IOModal/index.tsx index 8eab95a30..e93bfab7a 100644 --- a/src/frontend/src/modals/IOModal/index.tsx +++ b/src/frontend/src/modals/IOModal/index.tsx @@ -3,33 +3,29 @@ import AccordionComponent from "../../components/accordionComponent"; import IconComponent from "../../components/genericIconComponent"; import ShadTooltip from "../../components/shadTooltipComponent"; import { Badge } from "../../components/ui/badge"; +import { Button } from "../../components/ui/button"; import { Tabs, TabsContent, TabsList, TabsTrigger, } from "../../components/ui/tabs"; -import { - CHAT_FORM_DIALOG_SUBTITLE, - OUTPUTS_MODAL_TITLE, - TEXT_INPUT_MODAL_TITLE, -} from "../../constants/constants"; +import { CHAT_FORM_DIALOG_SUBTITLE } from "../../constants/constants"; import { InputOutput } from "../../constants/enums"; +import { getMessagesTable } from "../../controllers/API"; +import useAlertStore from "../../stores/alertStore"; import useFlowStore from "../../stores/flowStore"; import useFlowsManagerStore from "../../stores/flowsManagerStore"; +import { useMessagesStore } from "../../stores/messagesStore"; import { IOModalPropsType } from "../../types/components"; import { NodeDataType, NodeType } from "../../types/flow"; import { updateVerticesOrder } from "../../utils/buildUtils"; import { cn } from "../../utils/utils"; import BaseModal from "../baseModal"; import IOFieldView from "./components/IOFieldView"; -import ChatView from "./components/chatView"; -import { getMessagesTable } from "../../controllers/API"; -import { useMessagesStore } from "../../stores/messagesStore"; import SessionView from "./components/SessionView"; import useRemoveSession from "./components/SessionView/hooks"; -import useAlertStore from "../../stores/alertStore"; -import { Button } from "../../components/ui/button"; +import ChatView from "./components/chatView"; export default function IOModal({ children, @@ -40,25 +36,25 @@ export default function IOModal({ const allNodes = useFlowStore((state) => state.nodes); const setMessages = useMessagesStore((state) => state.setMessages); const inputs = useFlowStore((state) => state.inputs).filter( - (input) => input.type !== "ChatInput", + (input) => input.type !== "ChatInput" ); const chatInput = useFlowStore((state) => state.inputs).find( - (input) => input.type === "ChatInput", + (input) => input.type === "ChatInput" ); const outputs = useFlowStore((state) => state.outputs).filter( - (output) => output.type !== "ChatOutput", + (output) => output.type !== "ChatOutput" ); const chatOutput = useFlowStore((state) => state.outputs).find( - (output) => output.type === "ChatOutput", + (output) => output.type === "ChatOutput" ); const nodes = useFlowStore((state) => state.nodes).filter( (node) => inputs.some((input) => input.id === node.id) || - outputs.some((output) => output.id === node.id), + outputs.some((output) => output.id === node.id) ); const haveChat = chatInput || chatOutput; const [selectedTab, setSelectedTab] = useState( - inputs.length > 0 ? 1 : outputs.length > 0 ? 2 : 0, + inputs.length > 0 ? 1 : outputs.length > 0 ? 2 : 0 ); const setErrorData = useAlertStore((state) => state.setErrorData); const setSuccessData = useAlertStore((state) => state.setSuccessData); @@ -93,12 +89,6 @@ export default function IOModal({ return updateVerticesOrder(currentFlow!.id, null); } - // useEffect(() => { - // if (open) { - // updateVertices(); - // } - // }, [open, currentFlow]); - async function sendMessage({ repeat = 1, files, @@ -125,6 +115,7 @@ export default function IOModal({ if (chatInput) { setNode(chatInput.id, (node: NodeType) => { const newNode = { ...node }; + newNode.data.node!.template["input_value"].value = chatValue; return newNode; }); @@ -133,7 +124,7 @@ export default function IOModal({ const { handleRemoveSession } = useRemoveSession( setSuccessData, - setErrorData, + setErrorData ); useEffect(() => { @@ -194,7 +185,7 @@ export default function IOModal({
{nodes .filter((node) => - inputs.some((input) => input.id === node.id), + inputs.some((input) => input.id === node.id) ) .map((node, index) => { const input = inputs.find( - (input) => input.id === node.id, + (input) => input.id === node.id )!; return (
{nodes .filter((node) => - outputs.some((output) => output.id === node.id), + outputs.some((output) => output.id === node.id) ) .map((node, index) => { const output = outputs.find( - (output) => output.id === node.id, + (output) => output.id === node.id )!; return (
- f_session?.session_id === session, + f_session?.session_id === session ) ? "Active Session" : "Inactive Session" @@ -401,10 +392,10 @@ export default function IOModal({ "h-2 w-2 rounded-full", flow_sessions.some( (f_session) => - f_session?.session_id === session, + f_session?.session_id === session ) ? "bg-status-green" - : "bg-slate-500", + : "bg-slate-500" )} >
@@ -422,7 +413,7 @@ export default function IOModal({
@@ -441,7 +432,7 @@ export default function IOModal({
{inputs.some( - (input) => input.id === selectedViewField.id, + (input) => input.id === selectedViewField.id ) && ( )} {outputs.some( - (output) => output.id === selectedViewField.id, + (output) => output.id === selectedViewField.id ) && ( )} {sessions.some( - (session) => session === selectedViewField.id, + (session) => session === selectedViewField.id ) && ( - message.session_id === selectedViewField.id, + message.session_id === selectedViewField.id )} /> )} @@ -476,7 +467,7 @@ export default function IOModal({
{haveChat ? ( @@ -508,7 +499,7 @@ export default function IOModal({ "h-4 w-4", isBuilding ? "animate-spin" - : "fill-current text-medium-indigo", + : "fill-current text-medium-indigo" )} /> ),