This commit is contained in:
Cristhian Zanforlin Lousa 2023-06-15 19:38:16 -03:00
commit 1b0c173f39
7 changed files with 86 additions and 12 deletions

View file

@ -68,6 +68,8 @@ BASE_API_URL = "${window.location.protocol}//${
window.location.host
}/ap1/v1/predict"
FLOW_ID = "${flowId}"
# You can tweak the flow by adding a tweaks dictionary
# e.g {"OpenAI-XXXXX": {"model_name": "gpt-4"}}
TWEAKS = ${JSON.stringify(tweaks, null, 2)}
def run_flow(message: str, flow_id: str, tweaks: dict = None) -> dict:

View file

@ -1,4 +1,4 @@
import _ from "lodash";
import _, { set } from "lodash";
import { useContext, useRef, useState, useEffect, useCallback } from "react";
import ReactFlow, {
OnSelectionChangeParams,
@ -15,6 +15,7 @@ import ReactFlow, {
updateEdge,
Background,
Controls,
NodeChange,
} from "reactflow";
import GenericNode from "../../../../CustomNodes/GenericNode";
import Chat from "../../../../components/chatComponent";
@ -43,7 +44,9 @@ export default function Page({ flow }: { flow: FlowType }) {
lastCopiedSelection,
setLastCopiedSelection,
tabsState,
saveFlow
saveFlow,
setTabsState,
tabId
} = useContext(TabsContext);
const { types, reactFlowInstance, setReactFlowInstance, templates } =
useContext(typesContext);
@ -145,8 +148,31 @@ export default function Page({ flow }: { flow: FlowType }) {
let newX = _.cloneDeep(x);
return newX;
});
setTabsState((prev) => {
return {
...prev,
[tabId]: {
isPending: true,
},
};
});
},
[onEdgesChange, setNodes]
[onEdgesChange, setNodes,setTabsState,tabId]
);
const onNodesChangeMod = useCallback(
(s: NodeChange[]) => {
onNodesChange(s);
setTabsState((prev) => {
return {
...prev,
[tabId]: {
isPending: true,
},
};
});
},
[onNodesChange,setTabsState,tabId]
);
const onConnect = useCallback(
@ -345,7 +371,7 @@ export default function Page({ flow }: { flow: FlowType }) {
onPaneMouseLeave={() => {
setDisableCopyPaste(true);
}}
onNodesChange={onNodesChange}
onNodesChange={onNodesChangeMod}
onEdgesChange={onEdgesChangeMod}
onConnect={onConnect}
disableKeyboardA11y={true}