From 3fe59a7cf5cea9c3c845c362b07a9cf218af0bd7 Mon Sep 17 00:00:00 2001
From: timothycarambat <rambat1010@gmail.com>
Date: Tue, 29 Oct 2024 10:02:46 -0700
Subject: [PATCH] patch bad reference for `EphemeralAgentHandler`

---
 server/utils/agents/ephemeral.js | 24 ++++++++++++++++++++++--
 1 file changed, 22 insertions(+), 2 deletions(-)

diff --git a/server/utils/agents/ephemeral.js b/server/utils/agents/ephemeral.js
index 1ab2a988a2..d8258eca80 100644
--- a/server/utils/agents/ephemeral.js
+++ b/server/utils/agents/ephemeral.js
@@ -20,19 +20,40 @@ const {
  * not persist between invocations
  */
 class EphemeralAgentHandler extends AgentHandler {
+  /** @type {string|null} the unique identifier for the agent invocation */
   #invocationUUID = null;
+  /** @type {import("@prisma/client").workspaces|null} the workspace to use for the agent */
   #workspace = null;
+  /** @type {import("@prisma/client").users|null} the user id to use for the agent */
   #userId = null;
+  /** @type {import("@prisma/client").workspace_threads|null} the workspace thread id to use for the agent */
   #threadId = null;
+  /** @type {string|null} the session id to use for the agent */
   #sessionId = null;
+  /** @type {string|null} the prompt to use for the agent */
   #prompt = null;
+  /** @type {string[]} the functions to load into the agent (Aibitat plugins) */
   #funcsToLoad = [];
 
+  /** @type {AIbitat|null} */
   aibitat = null;
+  /** @type {string|null} */
   channel = null;
+  /** @type {string|null} */
   provider = null;
+  /** @type {string|null} the model to use for the agent */
   model = null;
 
+  /**
+   * @param {{
+   * uuid: string,
+   * workspace: import("@prisma/client").workspaces,
+   * prompt: string,
+   * userId: import("@prisma/client").users["id"]|null,
+   * threadId: import("@prisma/client").workspace_threads["id"]|null,
+   * sessionId: string|null
+   * }} parameters
+   */
   constructor({
     uuid,
     workspace,
@@ -148,8 +169,7 @@ class EphemeralAgentHandler extends AgentHandler {
     }
 
     // The provider was explicitly set, so check if the workspace has an agent model set.
-    if (this.invocation.workspace.agentModel)
-      return this.invocation.workspace.agentModel;
+    if (this.#workspace.agentModel) return this.#workspace.agentModel;
 
     // Otherwise, we have no model to use - so guess a default model to use via the provider
     // and it's system ENV params and if that fails - we return either a base model or null.