@receptron/graphai_express 0.0.30 → 0.0.31

package/lib/graph.js

@@ -37,7 +37,7 @@ const streamGraphRunner = (agentDictionary, agentFilters = [], streamChunkCallba
             };
             const filterList = [...agentFilters, streamAgentFilter];
             const dispatcher = streamGraphRunnerInternal(agentDictionary, filterList);
-            const result = await dispatcher(req, res);
+            const result = await dispatcher(req);
             const json_data = JSON.stringify(result);
             res.write("___END___");
             res.write(json_data);
@@ -53,7 +53,7 @@ const nonStreamGraphRunner = (agentDictionary, agentFilters = []) => {
     return async (req, res, next) => {
         try {
             const dispatcher = streamGraphRunnerInternal(agentDictionary, agentFilters);
-            const result = await dispatcher(req, res);
+            const result = await dispatcher(req);
             return res.json(result);
         }
         catch (e) {
@@ -64,7 +64,7 @@ const nonStreamGraphRunner = (agentDictionary, agentFilters = []) => {
 exports.nonStreamGraphRunner = nonStreamGraphRunner;
 // internal function
 const streamGraphRunnerInternal = (agentDictionary, agentFilters = []) => {
-    return async (req, res) => {
+    return async (req) => {
         const { graphData } = req.body;
         const { config } = req;
         const graphai = new graphai_1.GraphAI(graphData, agentDictionary, { agentFilters, config: config ?? {} });

package/lib/type.d.ts

@@ -13,4 +13,4 @@ export type ExpressAgentInfo = {
     output: any;
     stream: boolean;
 };
-export type StreamChunkCallback = (context: AgentFunctionContext, token: string) => void;
+export type StreamChunkCallback = (context: AgentFunctionContext, token: string | Record<string, string>) => void;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@receptron/graphai_express",
-  "version": "0.0.30",
+  "version": "0.0.31",
   "description": "GraphAI express web server middleware.",
   "main": "lib/index.js",
   "files": [
@@ -12,7 +12,7 @@
     "server": "ts-node -r tsconfig-paths/register test/express.ts",
     "test_stream": "ts-node -r tsconfig-paths/register test/stream_client.ts",
     "test_stream2": "ts-node -r tsconfig-paths/register test/stream_graph.ts",
-    "test": "ts-node -r tsconfig-paths/register test/client.ts",
+    "test": "node --test -r tsconfig-paths/register --require ts-node/register ./test/test_*.ts",
     "format": "prettier --write '{src,test}/**/*.{yaml,ts,json}'",
     "ci": "yarn run format && yarn run eslint && yarn run build"
   },
@@ -46,7 +46,7 @@
     "@graphai/agent_filters": "^0.0.6",
     "dotenv": "^16.4.5",
     "express": "^4.19.2",
-    "graphai": "^0.5.5"
+    "graphai": "^0.5.6"
   },
   "types": "./lib/index.d.ts",
   "directories": {