centersl/lobehub
1
0
Fork 0
mirror of https://github.com/lobehub/lobehub.git synced 2026-03-27 13:29:15 +07:00
Code Issues Packages Projects Releases Wiki Activity

🐛 fix: improve GitHub Copilot auth retry logic (#12250)

Browse Source 
* 🐛 fix: improve GitHub Copilot auth retry logic

Simplify auth refresh tracking from counter to boolean flag and clear
cached bearer token on 401 to ensure fresh token exchange.

* 🔧 fix: update package.json formatting and import statements in GitHub Copilot provider

Signed-off-by: Innei <tukon479@gmail.com>

* 🐛 fix: refine GitHub Copilot auth refresh logic to check for exchange credential

Update the 401 error handling to refresh the token only if an exchange credential is available, improving the authentication flow.

Signed-off-by: Innei <tukon479@gmail.com>

---------

Signed-off-by: Innei <tukon479@gmail.com>
This commit is contained in:
Innei
2026-02-12 13:09:38 +08:00
committed by GitHub
parent 823aa29c67
commit 0e42ca5ca2
2 changed files with 14 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
package.json
View File

@@ -132,18 +132,18 @@
"prettier --write --no-error-on-unmatched-pattern"
],
"*.{mjs,cjs}": [
"prettier --write",
"eslint --fix"
"eslint --fix",
"prettier --write"
],
"*.{js,jsx}": [
"prettier --write",
"eslint --fix",
"stylelint --fix",
"eslint --fix"
"prettier --write"
],
"*.{ts,tsx}": [
"prettier --parser=typescript --write",
"stylelint --fix",
"eslint --fix"
"eslint --fix",
"prettier --parser=typescript --write"
]
},
"overrides": {

16
packages/model-runtime/src/providers/githubCopilot/index.ts
View File

@@ -1,11 +1,11 @@
import type { ChatModelCard } from '@lobechat/types';
import { type ChatModelCard } from '@lobechat/types';
import { ModelProvider } from 'model-bank';
import OpenAI from 'openai';
import type { LobeRuntimeAI } from '../../core/BaseAI';
import { type LobeRuntimeAI } from '../../core/BaseAI';
import { pruneReasoningPayload } from '../../core/contextBuilders/openai';
import { OpenAIStream } from '../../core/streams';
import type { ChatMethodOptions, ChatStreamPayload } from '../../types';
import { type ChatMethodOptions, type ChatStreamPayload } from '../../types';
import { AgentRuntimeErrorType } from '../../types/error';
import { AgentRuntimeError } from '../../utils/createError';
import { StreamingResponse } from '../../utils/response';
@@ -14,7 +14,6 @@ const COPILOT_BASE_URL = 'https://api.githubcopilot.com';
const TOKEN_EXCHANGE_URL = 'https://api.github.com/copilot_internal/v2/token';
const MAX_TOTAL_ATTEMPTS = 5;
const MAX_AUTH_REFRESH = 1;
const MAX_RATE_LIMIT_RETRIES = 3;
interface CachedToken {
@@ -227,7 +226,7 @@ export class LobeGithubCopilotAI implements LobeRuntimeAI {
private async executeWithRetry<T>(fn: () => Promise<T>): Promise<T> {
let totalAttempts = 0;
let authRefreshAttempts = 0;
let hasRefreshedAuth = false;
let rateLimitAttempts = 0;
while (totalAttempts < MAX_TOTAL_ATTEMPTS) {
@@ -238,9 +237,10 @@ export class LobeGithubCopilotAI implements LobeRuntimeAI {
} catch (error: any) {
const status = error?.status ?? error?.error?.status ?? error?.response?.status;
// 401: Refresh token once
if (status === 401 && authRefreshAttempts < MAX_AUTH_REFRESH) {
authRefreshAttempts++;
// 401: Refresh token once (only if we have an exchange credential to fall back to)
if (status === 401 && !hasRefreshedAuth && this.githubToken) {
hasRefreshedAuth = true;
this.cachedBearerToken = undefined;
tokenManager.invalidate(this.githubToken);
continue;
}