@astralyn/pi 0.81.1-2 → 0.82.1

package/dist/core/agent-session.js

@@ -174,7 +174,7 @@ export class AgentSession {
             }
             throw error;
         }
-        if (result?.auth.apiKey) {
+        if (result && (result.auth.apiKey || result.auth.headers)) {
             return {
                 apiKey: result.auth.apiKey,
                 headers: withoutDeletedHeaders(result.auth.headers),
@@ -1599,12 +1599,7 @@ export class AgentSession {
             let headers;
             let env;
             if (this.agent.streamFunction === streamSimple) {
-                const authResult = await this._modelRuntime.getAuth(this.model);
-                if (!authResult?.auth.apiKey)
-                    return false;
-                apiKey = authResult.auth.apiKey;
-                headers = withoutDeletedHeaders(authResult.auth.headers);
-                env = authResult.env;
+                ({ apiKey, headers, env } = await this._getRequiredRequestAuth(this.model));
             }
             else {
                 ({ apiKey, headers, env } = await this._getSummarizationRequestAuth(this.model));
@@ -2195,6 +2190,7 @@ export class AgentSession {
      * @param command The bash command to execute
      * @param onChunk Optional streaming callback for output
      * @param options.excludeFromContext If true, command output won't be sent to LLM (!! prefix)
+     * @param options.id Optional identifier included in bash execution update events
      * @param options.operations Custom BashOperations for remote execution
      */
     async executeBash(command, onChunk, options) {
@@ -2205,7 +2201,10 @@ export class AgentSession {
         const resolvedCommand = prefix ? `${prefix}\n${command}` : command;
         try {
             const result = await executeBashWithOperations(resolvedCommand, this.sessionManager.getCwd(), options?.operations ?? createLocalBashOperations({ shellPath }), {
-                onChunk,
+                onChunk: (delta) => {
+                    onChunk?.(delta);
+                    this._emit({ type: "bash_execution_update", id: options?.id, delta });
+                },
                 signal: this._bashAbortController.signal,
             });
             this.recordBashResult(command, result, options);