kamp/anything-llm
1
0
Fork 0
mirror of https://github.com/Mintplex-Labs/anything-llm.git synced 2024-11-14 02:20:12 +01:00
Code Issues Packages Projects Releases Wiki Activity

Gracefully handle bad agent auth (#1115)

Browse Source 
use provider that is set
This commit is contained in:
Timothy Carambat 2024-04-16 16:42:06 -07:00 committed by GitHub
parent f9ac27e9a4
commit 8ebe1a515b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 19 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
server/utils/agents/aibitat/plugins/websocket.js
View File

@ -48,7 +48,13 @@ const websocket = {
name: this.name, name: this.name,
setup(aibitat) { setup(aibitat) {
aibitat.onError(async (error) => { aibitat.onError(async (error) => {
console.error(chalk.red(` error: ${error?.message}`)); if (!!error?.message) {
console.error(chalk.red(` error: ${error.message}`));
aibitat.introspect(
`Error encountered while running: ${error.message}`
);
}
if (error instanceof RetryError) { if (error instanceof RetryError) {
console.error(chalk.red(` retrying in 60 seconds...`)); console.error(chalk.red(` retrying in 60 seconds...`));
setTimeout(() => { setTimeout(() => {

6
server/utils/agents/aibitat/providers/anthropic.js
View File

@ -117,10 +117,14 @@ class AnthropicProvider extends Provider {
cost, cost,
}; };
} catch (error) { } catch (error) {
// If invalid Auth error we need to abort because no amount of waiting
// will make auth better.
if (error instanceof Anthropic.AuthenticationError) throw error;
if ( if (
error instanceof Anthropic.RateLimitError || error instanceof Anthropic.RateLimitError ||
error instanceof Anthropic.InternalServerError || error instanceof Anthropic.InternalServerError ||
error instanceof Anthropic.APIError error instanceof Anthropic.APIError // Also will catch AuthenticationError!!!
) { ) {
throw new RetryError(error.message); throw new RetryError(error.message);
} }

7
server/utils/agents/aibitat/providers/openai.js
View File

@ -102,11 +102,14 @@ class OpenAIProvider extends Provider {
cost, cost,
}; };
} catch (error) { } catch (error) {
console.log(error); // If invalid Auth error we need to abort because no amount of waiting
// will make auth better.
if (error instanceof OpenAI.AuthenticationError) throw error;
if ( if (
error instanceof OpenAI.RateLimitError || error instanceof OpenAI.RateLimitError ||
error instanceof OpenAI.InternalServerError || error instanceof OpenAI.InternalServerError ||
error instanceof OpenAI.APIError error instanceof OpenAI.APIError // Also will catch AuthenticationError!!!
) { ) {
throw new RetryError(error.message); throw new RetryError(error.message);
} }

4
server/utils/agents/index.js
View File

@ -157,8 +157,8 @@ class AgentHandler {
} }
) { ) {
this.aibitat = new AIbitat({ this.aibitat = new AIbitat({
provider: "openai", provider: this.provider ?? "openai",
model: "gpt-3.5-turbo", model: this.model ?? "gpt-3.5-turbo",
chats: await this.#chatHistory(20), chats: await this.#chatHistory(20),
handlerProps: { handlerProps: {
invocation: this.invocation, invocation: this.invocation,