github/node-DeepResearch
1
0
Fork 0
mirror of https://github.com/jina-ai/node-DeepResearch.git synced 2026-03-22 07:29:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: clean up

Browse Source
This commit is contained in:
Han Xiao
2025-02-06 15:37:53 +08:00
parent 50dff0863c
commit bd36db37ef
1 changed files with 16 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/config.ts
View File

@@ -1,7 +1,7 @@
import dotenv from 'dotenv'; import dotenv from 'dotenv';
import { ProxyAgent, setGlobalDispatcher } from 'undici'; import { ProxyAgent, setGlobalDispatcher } from 'undici';
import { createGoogleGenerativeAI } from '@ai-sdk/google'; import { createGoogleGenerativeAI } from '@ai-sdk/google';
import { createOpenAI } from '@ai-sdk/openai'; import {createOpenAI, OpenAIProviderSettings} from '@ai-sdk/openai';
export type LLMProvider = 'openai' | 'gemini'; export type LLMProvider = 'openai' | 'gemini';
export type ToolName = keyof ToolConfigs; export type ToolName = keyof ToolConfigs;
@@ -52,6 +52,7 @@ if (process.env.https_proxy) {
} }
} }
export const OPENAI_BASE_URL = process.env.OPENAI_BASE_URL;
export const GEMINI_API_KEY = process.env.GEMINI_API_KEY as string; export const GEMINI_API_KEY = process.env.GEMINI_API_KEY as string;
export const OPENAI_API_KEY = process.env.OPENAI_API_KEY as string; export const OPENAI_API_KEY = process.env.OPENAI_API_KEY as string;
export const JINA_API_KEY = process.env.JINA_API_KEY as string; export const JINA_API_KEY = process.env.JINA_API_KEY as string;
@@ -118,10 +119,15 @@ export function getModel(toolName: ToolName) {
if (!OPENAI_API_KEY) { if (!OPENAI_API_KEY) {
throw new Error('OPENAI_API_KEY not found'); throw new Error('OPENAI_API_KEY not found');
} }
return createOpenAI({ const opt: OpenAIProviderSettings = {
apiKey: OPENAI_API_KEY, apiKey: OPENAI_API_KEY,
compatibility: 'strict' compatibility: 'strict'
})(config.model); }
if (OPENAI_BASE_URL) {
opt.baseURL = OPENAI_BASE_URL
}
return createOpenAI(opt)(config.model);
} }
if (!GEMINI_API_KEY) { if (!GEMINI_API_KEY) {
@@ -137,3 +143,9 @@ if (LLM_PROVIDER === 'openai' && !OPENAI_API_KEY) throw new Error("OPENAI_API_KE
if (!JINA_API_KEY) throw new Error("JINA_API_KEY not found"); if (!JINA_API_KEY) throw new Error("JINA_API_KEY not found");
console.log('LLM Provider:', LLM_PROVIDER) console.log('LLM Provider:', LLM_PROVIDER)
if (LLM_PROVIDER === 'openai') {
console.log('OPENAI_BASE_URL', OPENAI_BASE_URL)
console.log('Model Name', DEFAULT_OPENAI_MODEL)
} else {
console.log('Model Name', DEFAULT_GEMINI_MODEL)
}