@axiom-lattice/gateway 2.1.45 → 2.1.47

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@axiom-lattice/gateway",
-  "version": "2.1.45",
+  "version": "2.1.47",
   "main": "dist/index.js",
   "module": "dist/index.mjs",
   "types": "dist/index.d.ts",
@@ -38,10 +38,10 @@
     "pg": "^8.11.0",
     "redis": "^5.0.1",
     "uuid": "^9.0.1",
-    "@axiom-lattice/core": "2.1.39",
-    "@axiom-lattice/pg-stores": "1.0.29",
-    "@axiom-lattice/protocols": "2.1.21",
-    "@axiom-lattice/queue-redis": "1.0.20"
+    "@axiom-lattice/core": "2.1.41",
+    "@axiom-lattice/pg-stores": "1.0.31",
+    "@axiom-lattice/protocols": "2.1.22",
+    "@axiom-lattice/queue-redis": "1.0.21"
   },
   "devDependencies": {
     "@types/jest": "^29.5.14",

package/src/controllers/models.ts

@@ -43,6 +43,7 @@ function extractModelKey(tenantId: string, tenantModelKey: string): string {
 interface ModelConfig {
   key: string;
   model: string;
+  displayName?: string;
   provider: "azure" | "openai" | "deepseek" | "siliconcloud" | "volcengine";
   streaming?: boolean;
   apiKey?: string;
@@ -75,10 +76,16 @@ export async function getModels(request: FastifyRequest, reply: FastifyReply) {
       // Extract config from the lattice client
       // Note: This is a simplified approach - you may need to adjust based on actual implementation
       const config = (lattice.client as any).config || {};
+      // Use provided displayName or auto-generate from key
+      const displayName = config.displayName || lattice.key
+        .split('-')
+        .map(word => word.charAt(0).toUpperCase() + word.slice(1))
+        .join(' ');
       return {
         key: lattice.key,
         model: config.model || "",
         provider: config.provider || "openai",
+        displayName: displayName,
         streaming: config.streaming || false,
         apiKey: config.apiKey || "",
         baseURL: config.baseURL || "",
@@ -144,6 +151,7 @@ export async function updateModels(
         const llmConfig: LLMConfig = {
           provider: modelConfig.provider,
           model: modelConfig.model,
+          displayName: modelConfig.displayName,
           streaming: modelConfig.streaming ?? false,
           apiKey: modelConfig.apiKey,
           baseURL: modelConfig.baseURL,

package/src/controllers/run.ts

@@ -10,6 +10,7 @@ import {
 
 interface ResumeStreamRequest {
   thread_id: string;
+  assistant_id: string
   message_id: string;
   known_content: string;
   poll_interval?: number;
@@ -73,6 +74,7 @@ export const createRun = async (
         const result = await agent.addMessage({
           input,
           command,
+          custom_run_config,
         });
 
         // const agentStatus = await agent.getRunStatus()
@@ -114,6 +116,7 @@ export const createRun = async (
       const result = await agent.invoke({
         input: { message: msg, ...restInputNonStream },
         command,
+        custom_run_config,
       });
       reply.status(200).send({
         success: true,
@@ -134,14 +137,21 @@ export const resumeStream = async (
   reply: FastifyReply
 ): Promise<void> => {
   try {
-    const { thread_id, message_id, known_content, poll_interval } =
+
+
+
+
+    const { thread_id, message_id, assistant_id, known_content, poll_interval } =
       request.body as ResumeStreamRequest;
+    const tenant_id = request.headers["x-tenant-id"] as string;
+    const workspace_id = request.headers["x-workspace-id"] as string;
+    const project_id = request.headers["x-project-id"] as string;
 
     // Validate request data
-    if (!thread_id || !message_id || known_content === undefined) {
+    if (!thread_id || !message_id || !assistant_id || known_content === undefined) {
       reply.status(400).send({
         success: false,
-        error: "thread_id, message_id, and known_content are required",
+        error: "thread_id, message_id, assistant_id, and known_content are required",
       });
       return;
     }
@@ -160,10 +170,13 @@ export const resumeStream = async (
     try {
       // Create Agent instance for accessing chunk buffer
       // Note: assistant_id, tenant_id are not required for chunkStream
-      const agent = new Agent({
-        assistant_id: '',
+      // Get or create Agent instance (ensures single instance per thread)
+      const agent = agentInstanceManager.getAgent({
+        assistant_id,
         thread_id,
-        tenant_id: '',
+        tenant_id,
+        workspace_id,
+        project_id,
       });
 
       // Get the stream from agent's chunk buffer