Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
29 changes: 29 additions & 0 deletions space_flow/package-lock.json

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

1 change: 1 addition & 0 deletions space_flow/package.json
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@
"@types/node": "^16.18.12",
"@types/react": "^18.0.27",
"@types/react-dom": "^18.0.10",
"axios": "^1.3.2",
"react": "^18.2.0",
"react-dom": "^18.2.0",
"react-icons": "^4.7.1",
Expand Down
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import { Transition } from "@headlessui/react";
import { Handle, Position } from "reactflow";

export default function ValidatorNode({ data }) {
export default function ToolsNode({ data }) {
console.log(data);
return (
<Transition
Expand All @@ -20,9 +20,9 @@ export default function ValidatorNode({ data }) {
>
<Handle type="source" position={Position.Left}></Handle>
<label className="absolute cursor-grab text-sm -top-3 left-1 bg-white w-14 text-center">
Validator
Tools
</label>
<div className="w-full h-min text-xs text-center">validator data</div>
<div className="w-full h-min text-xs text-center">Tools data</div>
<Handle type="target" position={Position.Right}></Handle>
</div>
</Transition>
Expand Down
61 changes: 61 additions & 0 deletions space_flow/src/controllers/NodesServices/index.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,61 @@
import axios from "axios";

export async function getPrompts() {
const promises = (await axios.get("http://localhost:5003/list/prompts")).data.map(async (value, index) => {
const prompt = await axios.get("http://localhost:5003/signatures/prompt", {
params: { name: value },
});
return { name: value, type: "promptNode", ...prompt.data };
});
return Promise.all(promises);
}

export async function getChains() {
const promises = (await axios.get("http://localhost:5003/list/chains")).data.map(async (value, index) => {
const chain = await axios.get("http://localhost:5003/signatures/chain", {
params: { name: value },
});
return { name: value, type: "chainNode", ...chain.data };
});
return Promise.all(promises);
}

export async function getAgents() {
const promises = (await axios.get("http://localhost:5003/list/agents")).data.map(async (value, index) => {
const chain = await axios.get("http://localhost:5003/signatures/agent", {
params: { name: value },
});
return { name: value, type: "agentNode", ...chain.data };
});
return Promise.all(promises);
}

export async function getMemories() {
const promises = (await axios.get("http://localhost:5003/list/memories")).data.map(async (value, index) => {
const chain = await axios.get("http://localhost:5003/signatures/memory", {
params: { name: value },
});
return { name: value, type: "memoryNode", ...chain.data };
});
return Promise.all(promises);
}

export async function getTools() {
const promises = (await axios.get("http://localhost:5003/list/tools")).data.map(async (value, index) => {
const prompt = await axios.get("http://localhost:5003/signatures/tool", {
params: { name: value },
});
return { name: value, type: "toolNode", ...prompt.data };
});
return Promise.all(promises);
}

export async function getModels() {
const promises = (await axios.get("http://localhost:5003/list/llms")).data.map(async (value, index) => {
const prompt = await axios.get("http://localhost:5003/signatures/llm", {
params: { name: value },
});
return { name: value, type: "modelNode", ...prompt.data };
});
return Promise.all(promises);
}
11 changes: 11 additions & 0 deletions space_flow/src/controllers/UiGenerator/index.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,11 @@
import axios from "axios";

export function generateUiNode(data: Object) {
const fields = [];
Object.keys(data).forEach((field) => {
if (data[field].required) {
fields.push(data[field])
}
});
return fields
}
4 changes: 4 additions & 0 deletions space_flow/src/entities/apiEnum.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
export enum apiEnum
{
PromptTemplate="PromptTemplate"
}
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
import { Bars2Icon, CommandLineIcon, CpuChipIcon, LightBulbIcon, LinkIcon, RocketLaunchIcon, ShieldCheckIcon, ViewColumnsIcon } from "@heroicons/react/24/outline";
import { Bars2Icon, CommandLineIcon, CpuChipIcon, LightBulbIcon, LinkIcon, RocketLaunchIcon, WrenchScrewdriverIcon, ViewColumnsIcon } from "@heroicons/react/24/outline";
import { llm_chain } from "../../../../data_assets/llm_chain";
import { prompt } from "../../../../data_assets/prompt";
import DisclosureComponent from "../DisclosureComponent";
Expand All @@ -20,7 +20,7 @@ export function ExtraSidebar() {
if (nodeType === "agentNode") {
json = JSON.stringify({ content: "" });
}
if (nodeType === "validatorNode") {
if (nodeType === "toolNode") {
json = JSON.stringify({ content: "" });
}
if (nodeType === "memoryNode") {
Expand Down Expand Up @@ -72,14 +72,14 @@ export function ExtraSidebar() {
</div>
</DisclosureComponent>
<DisclosureComponent
button={{ title: "Validators", Icon: ShieldCheckIcon }}
button={{ title: "Tools", Icon: WrenchScrewdriverIcon }}
>
<div
draggable
className="flex justify-between text-sm p-4 items-center h-12 m-2 border-dashed border-gray-400 rounded-md border-2 cursor-grab"
onDragStart={(event) => onDragStart(event, "validatorNode")}
onDragStart={(event) => onDragStart(event, "toolNode")}
>
<span className="text-black">Validator</span>
<span className="text-black">tools</span>
<Bars2Icon className="w-6 text-gray-400" />
</div>
</DisclosureComponent>
Expand Down
14 changes: 12 additions & 2 deletions space_flow/src/pages/FlowPage/index.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -12,19 +12,29 @@ import { locationContext } from "../../contexts/locationContext";
import { ExtraSidebar } from "./components/extraSidebarComponent";
import AgentNode from "../../CustomNodes/AgentNode";
import ChainNode from "../../CustomNodes/ChainNode";
import ValidatorNode from "../../CustomNodes/ValidatorNode";
import ToolsNode from "../../CustomNodes/ToolsNode";
import MemoryNode from "../../CustomNodes/MemoryNode";
import axios from "axios";
import {getPrompts, getChains,getAgents,getMemories, getModels,getTools} from "../../controllers/NodesServices";
import { generateUiNode } from "../../controllers/UiGenerator";

const nodeTypes = {
promptNode: PromptNode,
modelNode: ModelNode,
chainNode: ChainNode,
agentNode: AgentNode,
validatorNode: ValidatorNode,
toolNode: ToolsNode,
memoryNode:MemoryNode
};

export default function FlowPage() {
// getPrompts().then(result=>result.forEach(prompt=>console.log(prompt)))
// getChains().then(result=>console.log(result))
// getAgents().then(result=>console.log(result))
// getMemories().then(result=>console.log(result))
// getModels().then(result=>result.forEach(model=>console.log(model)))
getTools().then(result=>result.forEach(tool=>console.log(tool)))

// outside component to avoid render trigger

const reactFlowWrapper = useRef(null);
Expand Down