@economic/agents 1.6.1 → 1.6.3

package/dist/index.d.mts

@@ -152,7 +152,7 @@ interface MessageRating {
  * Skill loading, compaction, and LLM calls use `buildLLMParams` from
  * `@economic/agents` inside `onChatMessage`.
  */
-declare abstract class ChatAgent<Env extends Cloudflare.Env = Cloudflare.Env> extends AIChatAgent<Env & ChatAgentEnv> {
+declare abstract class ChatAgent<Env extends Cloudflare.Env = Cloudflare.Env, TSessionIdentity extends Record<string, unknown> = Record<string, unknown>> extends AIChatAgent<Env & ChatAgentEnv> {
   /**
    * The binding of the Durable Object instance for this agent.
    */
@@ -201,6 +201,17 @@ declare abstract class ChatAgent<Env extends Cloudflare.Env = Cloudflare.Env> ex
    * @returns JWT auth config or undefined to skip authentication
    */
   protected getJwtAuthConfig?(request: Request): JwtAuthConfig<Record<string, unknown>> | undefined;
+  /**
+   * The identity associated with the session.
+   * Define getIdentity to return the identity from the request.
+   */
+  protected identity: TSessionIdentity;
+  /**
+   * Returns the identity from the request.
+   * @returns The identity from the request.
+   * @param token - The token from the request.
+   */
+  protected getIdentity(_token: string): Promise<TSessionIdentity>;
   /**
    * Returns the user ID from the durable object name.
    */
@@ -243,7 +254,7 @@ declare abstract class ChatAgent<Env extends Cloudflare.Env = Cloudflare.Env> ex
 }
 //#endregion
 //#region src/server/agent-chat/ChatAgentHarness.d.ts
-declare abstract class ChatAgentHarness<Env extends Cloudflare.Env, RequestBody extends Record<string, unknown> = Record<string, unknown>> extends ChatAgent<Env> {
+declare abstract class ChatAgentHarness<Env extends Cloudflare.Env, RequestBody extends Record<string, unknown> = Record<string, unknown>, SessionIdentity extends Record<string, unknown> = Record<string, unknown>> extends ChatAgent<Env, SessionIdentity> {
   get binding(): {
     getByName(name: string): {
       destroy(): Promise<void>;

package/dist/index.mjs

@@ -375,7 +375,7 @@ function getJwksForIssuer(issuer) {
 function isIssuerAllowed(iss, allowed) {
 	return allowed.some((rule) => typeof rule === "string" ? rule === iss : rule.test(iss));
 }
-function extractTokenFromRequest(request) {
+function extractTokenFromConnectRequest(request) {
 	const authorization = request.headers.get("Authorization");
 	if (authorization) {
 		const match = authorization.match(/^Bearer\s+(.+)$/i);
@@ -412,7 +412,7 @@ function hasRequiredScopes(tokenScope, required) {
 * @throws Error for unexpected failures that should be logged
 */
 async function verifyJwt(request, config) {
-	const token = extractTokenFromRequest(request);
+	const token = extractTokenFromConnectRequest(request);
 	if (!token) return {
 		success: false,
 		status: 401,
@@ -849,6 +849,19 @@ var ChatAgent = class extends AIChatAgent {
 	*/
 	session;
 	/**
+	* The identity associated with the session.
+	* Define getIdentity to return the identity from the request.
+	*/
+	identity = {};
+	/**
+	* Returns the identity from the request.
+	* @returns The identity from the request.
+	* @param token - The token from the request.
+	*/
+	async getIdentity(_token) {
+		return Promise.resolve({});
+	}
+	/**
 	* Returns the user ID from the durable object name.
 	*/
 	getUserId() {
@@ -883,6 +896,8 @@ var ChatAgent = class extends AIChatAgent {
 				this.session = result.claims;
 			}
 		}
+		const token = extractTokenFromConnectRequest(ctx.request);
+		if (token) this.identity = await this.getIdentity(token);
 		return super.onConnect(connection, ctx);
 	}
 	/**

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@economic/agents",
-  "version": "1.6.1",
+  "version": "1.6.3",
   "description": "A starter for creating a TypeScript package.",
   "license": "MIT",
   "bin": {