feat: rename auth-example to full-cycle-example, enhance GraphQL client integration, and add environment configuration

Author: yakud

packages/badai-api-example/.env.example

@@ -0,0 +1,3 @@
+REST_API_URL=https://api.badai.io/graphql
+WS_API_URL=wss://api.badai.io/graphql-ws
+AGENT_ID=260ec0fe-92ae-4693-92a5-097364196a06

packages/badai-api-example/.gitignore

@@ -0,0 +1,26 @@
+# Logs
+logs
+*.log
+npm-debug.log*
+yarn-debug.log*
+yarn-error.log*
+pnpm-debug.log*
+lerna-debug.log*
+
+node_modules
+dist
+dist-ssr
+*.local
+
+# Editor directories and files
+.vscode/*
+!.vscode/extensions.json
+.idea
+.DS_Store
+*.suo
+*.ntvs*
+*.njsproj
+*.sln
+*.sw?
+
+*storybook.log

packages/badai-api-example/package.json

@@ -0,0 +1,32 @@
+{
+  "name": "@badaitech/badai-api-example",
+  "type": "module",
+  "version": "0.5.5-dev.20",
+  "private": true,
+  "description": "Bad AI GraphQL API client example",
+  "license": "BUSL-1.1",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com//badaitech/chaingraph.git",
+    "directory": "./packages/badai-api-example"
+  },
+  "scripts": {
+    "run": "bun run src/full-cycle-example.ts",
+    "typecheck": "tsc -b"
+  },
+  "publishConfig": {
+    "registry": "https://npm.pkg.github.com"
+  },
+  "dependencies": {
+    "@badaitech/badai-api": "workspace:*",
+    "@types/node": "^22.14.0",
+    "graphql": "^16.11.0",
+    "graphql-request": "^7.2.0",
+    "graphql-ws": "^6.0.6",
+    "keccak256": "^1.0.6",
+    "viem": "^2.37.7"
+  },
+  "devDependencies": {
+    "@badaitech/typescript-config": "workspace:*"
+  }
+}

packages/badai-api/src/auth-example.ts …ai-api-example/src/full-cycle-example.tspackages/badai-api/src/auth-example.ts renamed to packages/badai-api-example/src/full-cycle-example.ts packages/badai-api/src/auth-example.ts renamed to packages/badai-api-example/src/full-cycle-example.ts

@@ -6,21 +6,15 @@
  * 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 { MessageEvent } from './gql/client/graphql'
 import process from 'node:process'
+import { createGraphQLClient, GraphQL } from '@badaitech/badai-api'
 import { print } from 'graphql'
-import {
-  createClient,
-
-} from 'graphql-ws'
+import { createClient } from 'graphql-ws'
 import { generatePrivateKey, privateKeyToAccount } from 'viem/accounts'
-import { createGraphQLClient } from './gql/client'
-import { Event } from './gql/client/graphql'
-// import { SubscribeMessagesDocument } from './gql/client/graphql'
-import { GraphQL } from './index'
 
-const restApiUrl = process.env.REST_API_URL ?? 'http://localhost:9151/graphql'
+const restApiUrl = process.env.REST_API_URL ?? 'http://localhost:9151/graphql' // ?? 'http://localhost:9151/graphql'
 const wsApiUrl = process.env.WS_API_URL ?? 'ws://localhost:9151/graphql'
+const agentId = process.env.AGENT_ID ?? ''
 
 let activeSocket: WebSocket | null = null
 let pingTimeout: ReturnType<typeof setTimeout> | null = null
@@ -176,6 +170,7 @@ async function main() {
     GraphQL.CreateChatRoomDocument,
     {
       session,
+      agents: [agentId],
     },
   )
 
@@ -187,6 +182,8 @@ async function main() {
     resolveMessageFinished = resolve
   })
 
+  let agentResponse = ''
+
   const cancel = wsClient.subscribe(
     {
       query: print(GraphQL.SubscribeMessagesDocument),
@@ -197,16 +194,51 @@ async function main() {
       },
     },
     {
-      next: ({ data }) => {
-        console.log('Subscription data:', JSON.stringify(data, null, 2))
+      next: async ({ data }) => {
+        // console.log('Subscription data:', JSON.stringify(data, null, 2))
         if (!data?.subscribeMessages)
           return
 
-        const event = data.subscribeMessages as MessageEvent
+        const event = data.subscribeMessages as GraphQL.MessageEvent
 
         switch (event.event) {
-          case Event.MessageFinished:
-            resolveMessageFinished()
+          case GraphQL.Event.Subscribed: {
+            // send message to the chat room from the user we just created
+            console.log('Subscription confirmed, sending message...')
+            const { sendMessage } = await restClient.request(
+              GraphQL.SendMessageDocument,
+              {
+                session,
+                chat_id: createChatRoom.id,
+                message: {
+                  is_system: false,
+                  finished: true,
+                  need_answer: true,
+                  text: 'Find in the web the current weather in Paris.',
+                },
+              },
+            )
+
+            console.log('Sent Message:', sendMessage)
+          }
+            break
+
+          case GraphQL.Event.MessageCreated:
+            console.log(`Message Created by @${event.message?.participant?.username}: ${event.message?.text || '<no text>'} with version ${event.message?.version}`)
+            break
+
+          case GraphQL.Event.MessageDeltaAdd:
+            if (event.message?.participant?.is_agent) {
+              console.log(`Message Delta Add by @${event.message?.participant?.username} [${event.message?.version}]: ${event.message?.text}`)
+            }
+            break
+
+          case GraphQL.Event.MessageFinished:
+            if (event.message?.participant?.is_agent) {
+              agentResponse = event.message.text
+              resolveMessageFinished()
+            }
+
             break
         }
       },
@@ -221,28 +253,14 @@ async function main() {
     },
   )
 
-  // send message to the chat room
-  const { sendMessage } = await restClient.request(
-    GraphQL.SendMessageDocument,
-    {
-      session,
-      chat_id: createChatRoom.id,
-      message: {
-        is_system: false,
-        finished: true,
-        need_answer: true,
-        text: 'Hello, world!',
-      },
-    },
-  )
-
-  console.log('Sent Message:', sendMessage)
-
   // Wait for MessageFinished event
   console.log('Waiting for MessageFinished event...')
   await onMessageFinishedPromise
   console.log('MessageFinished event received, shutting down gracefully...')
 
+  console.log('\n\n--- Summary ---')
+  console.log('Agent response:', agentResponse)
+
   // Clean up
   cancel()
   wsClient.dispose()

packages/badai-api-example/tsconfig.json

@@ -0,0 +1,26 @@
+{
+  "$schema": "https://json.schemastore.org/tsconfig",
+  "extends": "@badaitech/typescript-config/base.json",
+  "compilerOptions": {
+    "jsx": "react-jsx",
+    "lib": ["dom", "ESNext", "ESNext.AsyncIterable"],
+    "emitDecoratorMetadata": true,
+    "experimentalDecorators": true,
+    "baseUrl": ".",
+    "rootDir": "./src",
+    "paths": {
+      "@badaitech/typescript-config/*": ["../typescript-config/*"],
+      "@badaitech/badai-api/*": ["../badai-api/src/*"]
+    },
+    /** Emit types for internal packages to speed up editor performance. */
+    "declaration": true,
+    "declarationMap": true,
+    "emitDeclarationOnly": false,
+    "outDir": "dist"
+  },
+  "include": ["src"],
+  "exclude": [
+    "node_modules",
+    "dist"
+  ]
+}

packages/chaingraph-executor/server/implementations/distributed/KafkaTaskQueue.ts

@@ -83,7 +83,7 @@ export class KafkaTaskQueue implements ITaskQueue {
       minBytes: 1, // fetch.min.bytes equivalent - fetch even single bytes
       maxBytes: 52428800, // fetch.max.bytes equivalent - 50MB max fetch
       maxBytesPerPartition: 10485760, // max.partition.fetch.bytes equivalent - 10MB per partition
-      readUncommitted: true, // Read only committed messages
+      readUncommitted: false, // Read only committed messages
       allowAutoTopicCreation: false,
       retry: {
         initialRetryTime: 100,