@axiom-lattice/gateway 2.1.28 → 2.1.30

package/src/services/agent_service.ts

@@ -9,7 +9,6 @@ import {
 } from "@langchain/core/messages";
 import { Command, CommandParams } from "@langchain/langgraph";
 import { v4 } from "uuid";
-// 修改导入路径，使用 lattice_core 包
 import {
   getAgentClient,
   getAgentLattice,
@@ -19,6 +18,39 @@ import {
   hasChunkBuffer,
 } from "@axiom-lattice/core";
 
+interface DatabaseConfigEntry {
+  key: string;
+  name?: string;
+  description?: string;
+}
+
+async function fetchDatabaseConfigs(baseURL: string, apiKey?: string, tenantId?: string): Promise<DatabaseConfigEntry[]> {
+  try {
+    const headers: Record<string, string> = {};
+    if (apiKey) {
+      headers["Authorization"] = `Bearer ${apiKey}`;
+    }
+    if (tenantId) {
+      headers["x-tenant-id"] = tenantId;
+    }
+
+    const response = await fetch(`${baseURL}/api/database-configs`, { headers });
+    if (response.ok) {
+      const data: any = await response.json();
+      if (data.success && data.data && Array.isArray(data.data.records)) {
+        return data.data.records.map((record: any) => ({
+          key: record.key,
+          name: record.name,
+          description: record.description,
+        }));
+      }
+    }
+  } catch (error) {
+    console.error("Failed to fetch database configs:", error);
+  }
+  return [];
+}
+
 /**
  * Get or create the global ChunkBuffer instance
  */
@@ -38,6 +70,8 @@ export async function agent_invoke({
   thread_id,
   assistant_id,
   tenant_id,
+  workspace_id,
+  project_id,
   command,
   run_id,
 }: {
@@ -45,6 +79,8 @@ export async function agent_invoke({
   input: any;
   thread_id: string;
   tenant_id: string;
+  workspace_id?: string;
+  project_id?: string;
   run_id?: string;
   command?: CommandParams<any>;
 }) {
@@ -58,11 +94,22 @@ export async function agent_invoke({
     throw new Error(`Agent ${assistant_id} not found`);
   }
 
+  // Fetch database configs and set to global for middleware to access
+  const { configService } = await import("../config.js");
+  const gatewayConfig = configService.getConfig();
+  const databaseConfigs = await fetchDatabaseConfigs(
+    gatewayConfig.baseURL || "http://localhost:4001",
+    undefined,
+    tenant_id
+  );
+  (global as any).__DATABASE_CONFIGS__ = databaseConfigs;
+
   // Get runConfig from agent config and merge into configurable
   const runConfig = {
     ...agentLattice?.config?.runConfig || {},
     assistant_id,
-    sandboxConfig: agentLattice?.config?.connectedSandbox,
+    workspaceId: workspace_id,
+    projectId: project_id,
   }
 
   const result = await runnable_agent.invoke(
@@ -74,10 +121,12 @@ export async function agent_invoke({
         thread_id: thread_id,
         run_id: run_id || v4(),
         "x-tenant-id": tenant_id,
+        "x-workspace-id": workspace_id,
+        "x-project-id": project_id,
         "x-request-id": run_id,
         "x-thread-id": thread_id,
         "x-assistant-id": assistant_id,
-        runConfig, // Inject runConfig for tools to access
+        runConfig,
       },
       recursionLimit: 200,
     }
@@ -98,6 +147,8 @@ export async function agent_stream({
   thread_id,
   command,
   tenant_id,
+  workspace_id,
+  project_id,
   assistant_id,
   run_id,
 }: {
@@ -106,9 +157,11 @@ export async function agent_stream({
   thread_id: string;
   command?: CommandParams<any>;
   tenant_id: string;
+  workspace_id?: string;
+  project_id?: string;
   run_id?: string;
 }) {
-  const runnable_agent = getAgentClient(assistant_id) as any; // TODO: fix this
+  const runnable_agent = getAgentClient(assistant_id) as any;
   const agentLattice = getAgentLattice(assistant_id);
   const { message, ...rest } = input;
   let messages: BaseMessage[] = [];
@@ -120,11 +173,22 @@ export async function agent_stream({
   // Get ChunkBuffer instance
   const chunkBuffer = getOrCreateChunkBuffer();
 
+  // Fetch database configs and set to global for middleware to access
+  const { configService } = await import("../config.js");
+  const gatewayConfig = configService.getConfig();
+  const databaseConfigs = await fetchDatabaseConfigs(
+    gatewayConfig.baseURL || "http://localhost:4001",
+    undefined,
+    tenant_id
+  );
+  (global as any).__DATABASE_CONFIGS__ = databaseConfigs;
+
   // Get runConfig from agent config and merge into configurable
   const runConfig = {
     ...agentLattice?.config?.runConfig || {},
     assistant_id,
-    sandboxConfig: agentLattice?.config?.connectedSandbox,
+    workspaceId: workspace_id,
+    projectId: project_id,
   }
 
   try {
@@ -145,6 +209,8 @@ export async function agent_stream({
           thread_id: thread_id,
           run_id: run_id || v4(),
           "x-tenant-id": tenant_id,
+          "x-workspace-id": workspace_id,
+          "x-project-id": project_id,
           "x-request-id": run_id,
           "x-thread-id": thread_id,
           "x-assistant-id": assistant_id,