From 9d9eab5db0a563c419136b4ce60fc8f8065206ca Mon Sep 17 00:00:00 2001 From: anovazzi1 Date: Mon, 13 Feb 2023 21:02:20 -0300 Subject: [PATCH] getting tools and model --- space_flow/src/controllers/NodesServices/index.ts | 10 ++++++++++ space_flow/src/pages/FlowPage/index.tsx | 6 ++++-- 2 files changed, 14 insertions(+), 2 deletions(-) diff --git a/space_flow/src/controllers/NodesServices/index.ts b/space_flow/src/controllers/NodesServices/index.ts index 169c1d2ae..883415d74 100644 --- a/space_flow/src/controllers/NodesServices/index.ts +++ b/space_flow/src/controllers/NodesServices/index.ts @@ -49,3 +49,13 @@ export async function getTools() { }); 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/templates/llm", { + params: { name: value }, + }); + return { name: value, type: "modelNode", ...prompt.data }; + }); + return Promise.all(promises); +} diff --git a/space_flow/src/pages/FlowPage/index.tsx b/space_flow/src/pages/FlowPage/index.tsx index 1bcae460a..6a3938075 100644 --- a/space_flow/src/pages/FlowPage/index.tsx +++ b/space_flow/src/pages/FlowPage/index.tsx @@ -15,7 +15,7 @@ import ChainNode from "../../CustomNodes/ChainNode"; import ToolsNode from "../../CustomNodes/ToolsNode"; import MemoryNode from "../../CustomNodes/MemoryNode"; import axios from "axios"; -import {getPrompts, getChains,getAgents,getMemories} from "../../controllers/NodesServices"; +import {getPrompts, getChains,getAgents,getMemories, getModels,getTools} from "../../controllers/NodesServices"; import { generateUiNode } from "../../controllers/UiGenerator"; const nodeTypes = { @@ -28,10 +28,12 @@ const nodeTypes = { }; export default function FlowPage() { - getPrompts().then(result=>result.forEach(prompt=>console.log(generateUiNode(prompt)))) + // 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