Merge pull request #150 from badaitech/feat/okx-llm-nodes

Fix stream port UI

Author: yakud

apps/chaingraph-backend/CHANGELOG.md

@@ -1,5 +1,15 @@
 # @badaitech/chaingraph-backend
 
+## 0.1.25
+
+### Patch Changes
+
+- Fix stream port UI
+- Updated dependencies
+  - @badaitech/chaingraph-nodes@0.1.25
+  - @badaitech/chaingraph-types@0.1.25
+  - @badaitech/chaingraph-trpc@0.1.25
+
 ## 0.1.24
 
 ### Patch Changes

apps/chaingraph-backend/Dockerfile

@@ -13,6 +13,8 @@ RUN turbo prune "@badaitech/chaingraph-backend" --docker
 # Second stage: Install dependencies and build
 FROM base AS builder
 
+ENV NODE_OPTIONS="--max-old-space-size=8192"
+
 COPY --from=pruner /app/out/json/ .
 COPY --from=pruner /app/out/pnpm-lock.yaml ./pnpm-lock.yaml
 

apps/chaingraph-backend/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@badaitech/chaingraph-backend",
   "type": "module",
-  "version": "0.1.24",
+  "version": "0.1.25",
   "private": false,
   "description": "Backend server for the Chaingraph project",
   "license": "BUSL-1.1",

apps/chaingraph-frontend/CHANGELOG.md

@@ -1,5 +1,16 @@
 # @badaitech/chaingraph-frontend
 
+## 0.1.25
+
+### Patch Changes
+
+- Fix stream port UI
+- Updated dependencies
+  - @badaitech/chaingraph-nodes@0.1.25
+  - @badaitech/chaingraph-types@0.1.25
+  - @badaitech/chaingraph-trpc@0.1.25
+  - @badaitech/badai-api@0.1.25
+
 ## 0.1.24
 
 ### Patch Changes

apps/chaingraph-frontend/Dockerfile

@@ -14,6 +14,7 @@ RUN turbo prune "@badaitech/chaingraph-frontend" --docker
 FROM base AS builder
 
 ARG NODE_ENV=production
+ENV NODE_OPTIONS="--max-old-space-size=8192"
 
 COPY --from=pruner /app/out/json/ .
 COPY --from=pruner /app/out/pnpm-lock.yaml ./pnpm-lock.yaml

apps/chaingraph-frontend/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@badaitech/chaingraph-frontend",
   "type": "module",
-  "version": "0.1.24",
+  "version": "0.1.25",
   "private": false,
   "description": "Frontend application for the Chaingraph project",
   "license": "BUSL-1.1",

apps/chaingraph-frontend/src/components/flow/nodes/ChaingraphNode/PortComponent.tsx

@@ -18,18 +18,19 @@ import type {
   IPort,
   NumberPortConfig,
   ObjectPortConfig,
+  StreamPortConfig,
   StringPortConfig,
 } from '@badaitech/chaingraph-types'
 import { BooleanPort } from '@/components/flow/nodes/ChaingraphNode/ports/BooleanPort/BooleanPort'
 import { NumberPort } from '@/components/flow/nodes/ChaingraphNode/ports/NumberPort/NumberPort'
 import { ObjectPort } from '@/components/flow/nodes/ChaingraphNode/ports/ObjectPort/ObjectPort'
 
+import { StreamPort } from '@/components/flow/nodes/ChaingraphNode/ports/StreamPort/StreamPort'
 import { AnyPort } from 'components/flow/nodes/ChaingraphNode/ports/AnyPort/AnyPort'
 import { ArrayPort } from './ports/ArrayPort/ArrayPort'
 import { PortContext } from './ports/context/PortContext'
 import { EnumPort } from './ports/EnumPort/EnumPort'
 import { StringPort } from './ports/StringPort/StringPort'
-import { StubPort } from './ports/StubPort/StubPort'
 
 /**
  * PortProps interface for all components rendered through PortComponent
@@ -103,7 +104,7 @@ export function PortComponent(props: PortProps) {
     }
 
     case 'stream': {
-      return <StubPort port={port} />
+      return <StreamPort node={node} port={port as IPort<StreamPortConfig>} context={context} />
     }
 
     default: {

apps/chaingraph-frontend/src/components/flow/nodes/ChaingraphNode/ports/StreamPort/StreamPort.tsx

@@ -0,0 +1,68 @@
+/*
+ * Copyright (c) 2025 BadLabs
+ *
+ * Use of this software is governed by the Business Source License 1.1 included in the file LICENSE.txt.
+ *
+ * As of the Change Date specified in that file, in accordance with the Business Source License, use of this software will be governed by the Apache License, version 2.0.
+ */
+
+import type {
+  PortContextValue,
+} from '@/components/flow/nodes/ChaingraphNode/ports/context/PortContext'
+/*
+ * Copyright (c) 2025 BadLabs
+ *
+ * Use of this software is governed by the Business Source License 1.1 included in the file LICENSE.txt.
+ *
+ * As of the Change Date specified in that file, in accordance with the Business Source License, use of this software will be governed by the Apache License, version 2.0.
+ */
+import type { INode, IPort, StreamPortConfig } from '@badaitech/chaingraph-types'
+import { cn } from '@/lib/utils'
+import { memo } from 'react'
+import { PortHandle } from '../ui/PortHandle'
+import { PortTitle } from '../ui/PortTitle'
+
+export interface StreamPortProps {
+  node: INode
+  port: IPort<StreamPortConfig>
+  context: PortContextValue
+}
+
+function StreamPortComponent(props: StreamPortProps) {
+  const { port, context } = props
+
+  const config = port.getConfig()
+  const ui = config.ui
+  const title = config.title || config.key
+
+  if (ui?.hidden)
+    return null
+
+  return (
+    <div
+      key={config.id}
+      className={cn(
+        'relative flex gap-2 group/port',
+        config.direction === 'output' ? 'justify-end' : 'justify-start',
+      )}
+    >
+      {config.direction === 'input' && <PortHandle port={port} />}
+
+      <div className={cn(
+        'flex flex-col',
+        config.direction === 'output' ? 'items-end' : 'items-start',
+        'truncate',
+      )}
+      >
+        <PortTitle>
+          {title}
+        </PortTitle>
+      </div>
+
+      {config.direction === 'output' && <PortHandle port={port} />}
+    </div>
+  )
+}
+
+// Export a memoized version of the component to prevent unnecessary re-renders
+export const StreamPort = memo(StreamPortComponent)

apps/chaingraph-frontend/src/components/flow/nodes/ChaingraphNode/ports/StubPort/StubPort.tsx

@@ -26,10 +26,13 @@ export function StubPort<C extends IPortConfig>(props: { port: IPort<C> }) {
         'relative flex items-center gap-2 group/port',
         config.direction === 'output' ? 'justify-end' : 'justify-start',
         'truncate',
+        'h-20',
       )}
     >
       {config.direction === 'output' && (
-        <PortTitle>{title}</PortTitle>
+        <PortTitle>
+          {title}
+        </PortTitle>
       )}
 
       <PortHandle port={port} />

packages/badai-api/CHANGELOG.md

@@ -1,5 +1,11 @@
 # @badaitech/badai-api
 
+## 0.1.25
+
+### Patch Changes
+
+- Fix stream port UI
+
 ## 0.1.24
 
 ### Patch Changes