Unverified 提交 bbf4bb76 authored 作者: Will Chen's avatar Will Chen 提交者: GitHub

Allow overriding gateway URL (#169)

上级 35b459d8
...@@ -15,6 +15,7 @@ ...@@ -15,6 +15,7 @@
"clean": "rm -rf out && rm -rf scaffold/node_modules", "clean": "rm -rf out && rm -rf scaffold/node_modules",
"start": "electron-forge start", "start": "electron-forge start",
"dev:engine": "DYAD_LOCAL_ENGINE=http://localhost:8080/v1 npm start", "dev:engine": "DYAD_LOCAL_ENGINE=http://localhost:8080/v1 npm start",
"staging:gateway": "DYAD_GATEWAY_URL=https://staging---litellm-gcp-cloud-run-kq7pivehnq-uc.a.run.app/v1 npm start",
"package": "npm run clean && electron-forge package", "package": "npm run clean && electron-forge package",
"make": "npm run clean && electron-forge make", "make": "npm run clean && electron-forge make",
"publish": "npm run clean && electron-forge publish", "publish": "npm run clean && electron-forge publish",
......
...@@ -14,6 +14,7 @@ import { llmErrorStore } from "@/main/llm_error_store"; ...@@ -14,6 +14,7 @@ import { llmErrorStore } from "@/main/llm_error_store";
import { createDyadEngine } from "./llm_engine_provider"; import { createDyadEngine } from "./llm_engine_provider";
const dyadLocalEngine = process.env.DYAD_LOCAL_ENGINE; const dyadLocalEngine = process.env.DYAD_LOCAL_ENGINE;
const dyadGatewayUrl = process.env.DYAD_GATEWAY_URL;
const AUTO_MODELS = [ const AUTO_MODELS = [
{ {
...@@ -105,7 +106,7 @@ export async function getModelClient( ...@@ -105,7 +106,7 @@ export async function getModelClient(
}) })
: createOpenAI({ : createOpenAI({
apiKey: dyadApiKey, apiKey: dyadApiKey,
baseURL: "https://llm-gateway.dyad.sh/v1", baseURL: dyadGatewayUrl ?? "https://llm-gateway.dyad.sh/v1",
}); });
logger.info( logger.info(
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论