vrr/eigent
2
0
Fork 0
mirror of https://github.com/eigent-ai/eigent.git synced 2026-04-28 03:30:06 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

feat: add ernie in custom model (#1487)
Some checks failed
CodeQL Advanced / Analyze (actions) (push) Has been cancelled
Details
CodeQL Advanced / Analyze (javascript-typescript) (push) Has been cancelled
Details
CodeQL Advanced / Analyze (python) (push) Has been cancelled
Details
Pre-commit / pre-commit (push) Has been cancelled
Details
Test / Run Python Tests (push) Has been cancelled
Details

Browse source 
Co-authored-by: bytecii <994513625@qq.com>
This commit is contained in:
Tao Sun 2026-03-12 19:20:07 +08:00 • committed by GitHub
parent 031cb65823
commit 8e0f09ea33
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 64 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

63
backend/app/controller/electron_browser.cjs
View file

@ -9,6 +9,8 @@ const startUrl = args[2] || 'https://www.google.com';
// This must be called before app.ready
app.commandLine.appendSwitch('remote-debugging-port', cdpPort);
app.commandLine.appendSwitch('password-store', 'basic');
app.commandLine.appendSwitch('use-mock-keychain');
console.log('[ELECTRON BROWSER] Starting with:');
console.log(' Chrome version:', process.versions.chrome);
@ -36,8 +38,14 @@ app.whenReady().then(async () => {
// Check command line switches
console.log('[ELECTRON BROWSER] Command line switches:');
console.log(' user-data-dir:', app.commandLine.getSwitchValue('user-data-dir'));
console.log(' remote-debugging-port:', app.commandLine.getSwitchValue('remote-debugging-port'));
console.log(
' user-data-dir:',
app.commandLine.getSwitchValue('user-data-dir')
);
console.log(
' remote-debugging-port:',
app.commandLine.getSwitchValue('remote-debugging-port')
);
// Log partition session info
const userLoginSession = session.fromPartition('persist:user_login');
@ -46,7 +54,12 @@ app.whenReady().then(async () => {
console.log(' Session storage path:', userLoginSession.getStoragePath());
// Check if Cookies file exists
const cookiesPath = path.join(app.getPath('userData'), 'Partitions', 'user_login', 'Cookies');
const cookiesPath = path.join(
app.getPath('userData'),
'Partitions',
'user_login',
'Cookies'
);
console.log('[ELECTRON BROWSER] Cookies path:', cookiesPath);
console.log('[ELECTRON BROWSER] Cookies exists:', fs.existsSync(cookiesPath));
if (fs.existsSync(cookiesPath)) {
@ -60,8 +73,8 @@ app.whenReady().then(async () => {
webPreferences: {
nodeIntegration: true,
contextIsolation: false,
webviewTag: true
}
webviewTag: true,
},
});
// Create navigation bar and webview HTML
@ -305,9 +318,14 @@ app.whenReady().then(async () => {
setInterval(async () => {
try {
const cookies = await userLoginSession.cookies.get({});
console.log('[ELECTRON BROWSER] Current cookies count:', cookies.length);
console.log(
'[ELECTRON BROWSER] Current cookies count:',
cookies.length
);
if (cookies.length > 0) {
console.log('[ELECTRON BROWSER] Cookie domains:', [...new Set(cookies.map(c => c.domain))]);
console.log('[ELECTRON BROWSER] Cookie domains:', [
...new Set(cookies.map((c) => c.domain)),
]);
}
} catch (error) {
console.error('[ELECTRON BROWSER] Failed to get cookies:', error);
@ -327,7 +345,10 @@ app.whenReady().then(async () => {
// Log cookies before flush
const cookiesBeforeFlush = await userLoginSession.cookies.get({});
console.log('[ELECTRON BROWSER] Cookies count before flush:', cookiesBeforeFlush.length);
console.log(
'[ELECTRON BROWSER] Cookies count before flush:',
cookiesBeforeFlush.length
);
// Flush storage
console.log('[ELECTRON BROWSER] Flushing storage data...');
@ -335,12 +356,23 @@ app.whenReady().then(async () => {
console.log('[ELECTRON BROWSER] Storage data flushed successfully');
// Check cookies file after flush
const cookiesPath = path.join(app.getPath('userData'), 'Partitions', 'user_login', 'Cookies');
const cookiesPath = path.join(
app.getPath('userData'),
'Partitions',
'user_login',
'Cookies'
);
if (fs.existsSync(cookiesPath)) {
const stats = fs.statSync(cookiesPath);
console.log('[ELECTRON BROWSER] Cookies file size after flush:', stats.size, 'bytes');
console.log(
'[ELECTRON BROWSER] Cookies file size after flush:',
stats.size,
'bytes'
);
} else {
console.log('[ELECTRON BROWSER] WARNING: Cookies file does not exist after flush!');
console.log(
'[ELECTRON BROWSER] WARNING: Cookies file does not exist after flush!'
);
}
} catch (error) {
console.error('[ELECTRON BROWSER] Failed to flush storage data:', error);
@ -368,9 +400,14 @@ app.on('before-quit', async (event) => {
// Log cookies before flush
const cookiesBeforeQuit = await userLoginSession.cookies.get({});
console.log('[ELECTRON BROWSER] Cookies count before quit:', cookiesBeforeQuit.length);
console.log(
'[ELECTRON BROWSER] Cookies count before quit:',
cookiesBeforeQuit.length
);
if (cookiesBeforeQuit.length > 0) {
console.log('[ELECTRON BROWSER] Cookie domains before quit:', [...new Set(cookiesBeforeQuit.map(c => c.domain))]);
console.log('[ELECTRON BROWSER] Cookie domains before quit:', [
...new Set(cookiesBeforeQuit.map((c) => c.domain)),
]);
}
// Flush storage

1
backend/app/model/model_platform.py
View file

@ -20,6 +20,7 @@ PLATFORM_ALIAS_MAPPING: Final[dict[str, str]] = {
"z.ai": "openai-compatible-model",
"ModelArk": "openai-compatible-model",
"grok": "openai-compatible-model",
"ernie": "openai-compatible-model",
"llama.cpp": "openai-compatible-model",
}

3
backend/tests/app/model/test_model_platform.py
View file

@ -26,6 +26,7 @@ def test_normalize_model_platform_maps_known_aliases():
assert normalize_model_platform("grok") == "openai-compatible-model"
assert normalize_model_platform("z.ai") == "openai-compatible-model"
assert normalize_model_platform("ModelArk") == "openai-compatible-model"
assert normalize_model_platform("ernie") == "openai-compatible-model"
assert normalize_model_platform("llama.cpp") == "openai-compatible-model"
@ -44,7 +45,7 @@ def test_normalized_model_platform_type_applies_in_pydantic_model():
optional_model_platform: NormalizedOptionalModelPlatform = None
item = _Model(
model_platform="llama.cpp",
model_platform="ernie",
optional_model_platform="ModelArk",
)