update

receptron · Jun 16, 2024 · cee4745 · cee4745
1 parent 06ed833
commit cee4745
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 1 deletion.
diff --git a/src/views/Home.vue b/src/views/Home.vue
@@ -48,6 +48,20 @@ export default defineComponent({
       node_watch.pos = [700, 200];
       graph.add(node_watch);
 
+      const openai_node = LiteGraph.createNode("graphai/OpenAIAgent");
+      graph.add(openai_node);
+      openai_node.pos = [700, 400];
+
+      const string_node = LiteGraph.createNode("basic/stringInput");
+      graph.add(string_node);
+      string_node.pos = [200, 600];
+      string_node.connect(0, openai_node, 2)
+
+      const string_node2 = LiteGraph.createNode("graphai/StringTemplateAgent");
+      graph.add(string_node2);
+      string_node2.pos = [200, 400];
+      string_node2.connect(0, openai_node, 0)
+
       node_const.connect(0, node_watch, 0);
 
       graph.start();

diff --git a/src/views/nodes.ts b/src/views/nodes.ts
@@ -47,8 +47,13 @@ export class TextInputAgentNode extends LGraphNode {
 export class OpenAIAgentNode extends LGraphNode {
   constructor() {
     super("OpenAI");
-    this.addInput("In", "string");
+    this.addInput("prompt", "string");
+    this.addInput("model", "string");
+    this.addInput("system", "string");
+    this.addInput("baseUrl", "string");
+
     this.addOutput(".choices.$0.message.content", "string");
+    this.addOutput(".", "object");
   }
 }