@axiom-lattice/gateway 2.1.45 → 2.1.47

package/dist/index.mjs

@@ -200,7 +200,6 @@ var getAgentGraph = async (request, reply) => {
 // src/controllers/run.ts
 import { v4 } from "uuid";
 import {
-  Agent,
   agentInstanceManager as agentInstanceManager2
 } from "@axiom-lattice/core";
 var createRun = async (request, reply) => {
@@ -244,7 +243,8 @@ var createRun = async (request, reply) => {
       try {
         const result = await agent.addMessage({
           input,
-          command
+          command,
+          custom_run_config
         });
         const stream = agent.chunkStream(result.messageId);
         for await (const chunk of stream) {
@@ -274,7 +274,8 @@ var createRun = async (request, reply) => {
       const { message: msg, ...restInputNonStream } = input;
       const result = await agent.invoke({
         input: { message: msg, ...restInputNonStream },
-        command
+        command,
+        custom_run_config
       });
       reply.status(200).send({
         success: true,
@@ -290,11 +291,14 @@ var createRun = async (request, reply) => {
 };
 var resumeStream = async (request, reply) => {
   try {
-    const { thread_id, message_id, known_content, poll_interval } = request.body;
-    if (!thread_id || !message_id || known_content === void 0) {
+    const { thread_id, message_id, assistant_id, known_content, poll_interval } = request.body;
+    const tenant_id = request.headers["x-tenant-id"];
+    const workspace_id = request.headers["x-workspace-id"];
+    const project_id = request.headers["x-project-id"];
+    if (!thread_id || !message_id || !assistant_id || known_content === void 0) {
       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;
     }
@@ -306,10 +310,12 @@ var resumeStream = async (request, reply) => {
       "Access-Control-Allow-Origin": "*"
     });
     try {
-      const agent = new Agent({
-        assistant_id: "",
+      const agent = agentInstanceManager2.getAgent({
+        assistant_id,
         thread_id,
-        tenant_id: ""
+        tenant_id,
+        workspace_id,
+        project_id
       });
       const stream = agent.chunkStream(message_id, known_content);
       for await (const chunk of stream) {
@@ -1055,10 +1061,12 @@ async function getModels(request, reply) {
     const allLattices = modelLatticeManager.getAllLattices();
     const models = allLattices.map((lattice) => {
       const config = lattice.client.config || {};
+      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,
         streaming: config.streaming || false,
         apiKey: config.apiKey || "",
         baseURL: config.baseURL || "",
@@ -1108,6 +1116,7 @@ async function updateModels(request, reply) {
         const llmConfig = {
           provider: modelConfig.provider,
           model: modelConfig.model,
+          displayName: modelConfig.displayName,
           streaming: modelConfig.streaming ?? false,
           apiKey: modelConfig.apiKey,
           baseURL: modelConfig.baseURL,