@rivus/agent 0.5.1 → 0.5.2

package/dist/acp.js

@@ -312,7 +312,11 @@ function createAcpStdioAgentLoop(options) {
 			resolveSession: async (input) => {
 				const processConnection = await resolveConnection();
 				const current = processConnection.sessions.get(input.sessionKey);
-				if (current) return current;
+				if (current?.isReusable) return current;
+				if (current) {
+					current.dispose();
+					processConnection.sessions.delete(input.sessionKey);
+				}
 				const session = new SdkAcpAgentSession(await processConnection.connection.agent.buildSession(options.workingDirectory).start(), processConnection.connection.agent);
 				sessionKeys.set(session.sessionId, input.sessionKey);
 				processConnection.sessions.set(input.sessionKey, session);
@@ -324,6 +328,7 @@ function createAcpStdioAgentLoop(options) {
 var SdkAcpAgentSession = class {
 	session;
 	agent;
+	reusable = true;
 	constructor(session, agent) {
 		this.session = session;
 		this.agent = agent;
@@ -331,10 +336,15 @@ var SdkAcpAgentSession = class {
 	get sessionId() {
 		return this.session.sessionId;
 	}
+	get isReusable() {
+		return this.reusable;
+	}
 	cancel() {
+		this.reusable = false;
 		return this.agent.notify(acp.methods.agent.session.cancel, { sessionId: this.session.sessionId });
 	}
 	dispose() {
+		this.reusable = false;
 		this.session.dispose();
 	}
 	async prompt(text, onUpdate) {

package/dist/index.js

@@ -2838,25 +2838,113 @@ function resolveRunPresentation(input) {
 	}
 }
 //#endregion
+//#region src/infrastructure/feishu/feishu-tenant-token-provider.ts
+var FeishuTenantAccessTokenError = class {
+	status;
+	code;
+	message;
+	_tag = "FeishuTenantAccessTokenError";
+	constructor(status, code, message) {
+		this.status = status;
+		this.code = code;
+		this.message = message;
+	}
+};
+const DEFAULT_REFRESH_WINDOW_MS = 1800 * 1e3;
+function createFeishuTenantAccessTokenProvider(options) {
+	const baseUrl = (options.baseUrl ?? "https://open.feishu.cn").replace(/\/$/, "");
+	const now = options.now ?? Date.now;
+	const refreshWindowMs = options.refreshWindowMs ?? DEFAULT_REFRESH_WINDOW_MS;
+	let cached;
+	return { getTenantAccessToken: () => Effect.gen(function* () {
+		if (cached && now() < cached.refreshAfterMs) return cached.token;
+		const response = yield* options.request({
+			body: {
+				app_id: options.appId,
+				app_secret: options.appSecret
+			},
+			headers: { "Content-Type": "application/json; charset=utf-8" },
+			method: "POST",
+			url: `${baseUrl}/open-apis/auth/v3/tenant_access_token/internal`
+		});
+		if (response.status < 200 || response.status >= 300 || response.body.code !== 0 || !response.body.tenant_access_token || typeof response.body.expire !== "number") return yield* Effect.fail(new FeishuTenantAccessTokenError(response.status, response.body.code, response.body.msg ?? "Failed to get tenant_access_token"));
+		cached = {
+			refreshAfterMs: now() + response.body.expire * 1e3 - refreshWindowMs,
+			token: response.body.tenant_access_token
+		};
+		return cached.token;
+	}) };
+}
+//#endregion
+//#region src/infrastructure/feishu/feishu-openapi-client.ts
+var FeishuOpenApiError = class extends Error {
+	status;
+	code;
+	_tag = "FeishuOpenApiError";
+	name = "FeishuOpenApiError";
+	constructor(status, code, message) {
+		super(message);
+		this.status = status;
+		this.code = code;
+	}
+};
+function createFeishuOpenApiClient(options) {
+	return { request: (request, errorMessage = "Feishu OpenAPI error") => Effect.gen(function* () {
+		const token = yield* options.getTenantAccessToken();
+		const response = yield* options.request({
+			...request,
+			headers: {
+				Authorization: `Bearer ${token}`,
+				"Content-Type": "application/json; charset=utf-8"
+			}
+		});
+		if (response.status < 200 || response.status >= 300 || response.body.code !== 0) return yield* Effect.fail(new FeishuOpenApiError(response.status, response.body.code, response.body.msg ?? errorMessage));
+		return response;
+	}) };
+}
+function createConfiguredFeishuOpenApiClient(options) {
+	const tokenProvider = createFeishuTenantAccessTokenProvider({
+		appId: options.config.feishu.appId,
+		appSecret: options.config.feishu.appSecret,
+		baseUrl: options.config.feishu.baseUrl,
+		request: (request) => options.request(request).pipe(Effect.map((response) => response))
+	});
+	return createFeishuOpenApiClient({
+		getTenantAccessToken: () => tokenProvider.getTenantAccessToken(),
+		request: (request) => options.request(request).pipe(Effect.map((response) => response))
+	});
+}
+//#endregion
 //#region src/infrastructure/feishu/feishu-cardkit-openapi-client.ts
 function createFeishuCardKitOpenApiClient(options) {
 	const baseUrl = (options.baseUrl ?? "https://open.feishu.cn").replace(/\/$/, "");
 	const agentName = options.agentName ?? "Rivus Agent";
+	const streamingClosedCards = /* @__PURE__ */ new Set();
 	const call = (request) => options.client.request(request).pipe(Effect.asVoid);
+	const updateTerminal = (request, status) => updateTerminalCard(baseUrl, agentName, request, status, call).pipe(Effect.tap(() => Effect.sync(() => streamingClosedCards.delete(request.cardId))));
 	return {
-		cancel: (request) => updateTerminalCard(baseUrl, agentName, request, "cancelled", call),
-		fail: (request) => updateTerminalCard(baseUrl, agentName, request, "failed", call),
-		finish: (request) => updateTerminalCard(baseUrl, agentName, request, "completed", call),
-		updateText: (request) => call({
-			body: {
-				content: request.content,
-				sequence: request.sequence
-			},
-			method: "PUT",
-			url: `${baseUrl}/open-apis/cardkit/v1/cards/${encodeURIComponent(request.cardId)}/elements/${encodeURIComponent(request.elementId)}/content`
-		})
+		cancel: (request) => updateTerminal(request, "cancelled"),
+		fail: (request) => updateTerminal(request, "failed"),
+		finish: (request) => updateTerminal(request, "completed"),
+		updateText: (request) => {
+			if (streamingClosedCards.has(request.cardId)) return Effect.void;
+			return call({
+				body: {
+					content: request.content,
+					sequence: request.sequence
+				},
+				method: "PUT",
+				url: `${baseUrl}/open-apis/cardkit/v1/cards/${encodeURIComponent(request.cardId)}/elements/${encodeURIComponent(request.elementId)}/content`
+			}).pipe(Effect.catchAll((error) => {
+				if (!isStreamingClosed(error)) return Effect.fail(error);
+				return Effect.sync(() => streamingClosedCards.add(request.cardId));
+			}));
+		}
 	};
 }
+function isStreamingClosed(error) {
+	return error instanceof FeishuOpenApiError && (error.code === 200850 || error.code === 300309);
+}
 function updateTerminalCard(baseUrl, agentName, request, status, call) {
 	const card = status === "completed" && "text" in request ? createFeishuAgentRunCard({
 		agentName,
@@ -2988,83 +3076,6 @@ function createConfiguredFeishuCardKitPublisher(options) {
 	}) });
 }
 //#endregion
-//#region src/infrastructure/feishu/feishu-tenant-token-provider.ts
-var FeishuTenantAccessTokenError = class {
-	status;
-	code;
-	message;
-	_tag = "FeishuTenantAccessTokenError";
-	constructor(status, code, message) {
-		this.status = status;
-		this.code = code;
-		this.message = message;
-	}
-};
-const DEFAULT_REFRESH_WINDOW_MS = 1800 * 1e3;
-function createFeishuTenantAccessTokenProvider(options) {
-	const baseUrl = (options.baseUrl ?? "https://open.feishu.cn").replace(/\/$/, "");
-	const now = options.now ?? Date.now;
-	const refreshWindowMs = options.refreshWindowMs ?? DEFAULT_REFRESH_WINDOW_MS;
-	let cached;
-	return { getTenantAccessToken: () => Effect.gen(function* () {
-		if (cached && now() < cached.refreshAfterMs) return cached.token;
-		const response = yield* options.request({
-			body: {
-				app_id: options.appId,
-				app_secret: options.appSecret
-			},
-			headers: { "Content-Type": "application/json; charset=utf-8" },
-			method: "POST",
-			url: `${baseUrl}/open-apis/auth/v3/tenant_access_token/internal`
-		});
-		if (response.status < 200 || response.status >= 300 || response.body.code !== 0 || !response.body.tenant_access_token || typeof response.body.expire !== "number") return yield* Effect.fail(new FeishuTenantAccessTokenError(response.status, response.body.code, response.body.msg ?? "Failed to get tenant_access_token"));
-		cached = {
-			refreshAfterMs: now() + response.body.expire * 1e3 - refreshWindowMs,
-			token: response.body.tenant_access_token
-		};
-		return cached.token;
-	}) };
-}
-//#endregion
-//#region src/infrastructure/feishu/feishu-openapi-client.ts
-var FeishuOpenApiError = class extends Error {
-	status;
-	code;
-	_tag = "FeishuOpenApiError";
-	name = "FeishuOpenApiError";
-	constructor(status, code, message) {
-		super(message);
-		this.status = status;
-		this.code = code;
-	}
-};
-function createFeishuOpenApiClient(options) {
-	return { request: (request, errorMessage = "Feishu OpenAPI error") => Effect.gen(function* () {
-		const token = yield* options.getTenantAccessToken();
-		const response = yield* options.request({
-			...request,
-			headers: {
-				Authorization: `Bearer ${token}`,
-				"Content-Type": "application/json; charset=utf-8"
-			}
-		});
-		if (response.status < 200 || response.status >= 300 || response.body.code !== 0) return yield* Effect.fail(new FeishuOpenApiError(response.status, response.body.code, response.body.msg ?? errorMessage));
-		return response;
-	}) };
-}
-function createConfiguredFeishuOpenApiClient(options) {
-	const tokenProvider = createFeishuTenantAccessTokenProvider({
-		appId: options.config.feishu.appId,
-		appSecret: options.config.feishu.appSecret,
-		baseUrl: options.config.feishu.baseUrl,
-		request: (request) => options.request(request).pipe(Effect.map((response) => response))
-	});
-	return createFeishuOpenApiClient({
-		getTenantAccessToken: () => tokenProvider.getTenantAccessToken(),
-		request: (request) => options.request(request).pipe(Effect.map((response) => response))
-	});
-}
-//#endregion
 //#region src/infrastructure/feishu/feishu-cardkit-target-preparation.ts
 const DEFAULT_ELEMENT_ID = "rivus_agent_answer";
 function createFeishuCardTargetPreparation(options) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@rivus/agent",
-  "version": "0.5.1",
+  "version": "0.5.2",
   "description": "A local agent daemon core built around a usable agent harness and domain events.",
   "type": "module",
   "license": "MIT",