diff --git a/src/libs/agent-runtime/github/index.ts b/src/libs/agent-runtime/github/index.ts index d730059228b8..fd7fa6f280bc 100644 --- a/src/libs/agent-runtime/github/index.ts +++ b/src/libs/agent-runtime/github/index.ts @@ -1,27 +1,8 @@ import { AgentRuntimeErrorType } from '../error'; -import { ChatStreamPayload, ModelProvider, OpenAIChatMessage } from '../types'; +import { o1Models, pruneO1Payload } from '../openai'; +import { ModelProvider } from '../types'; import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory'; -// TODO: 临时写法,后续要重构成 model card 展示配置 -export const o1Models = new Set([ - 'o1-preview', - 'o1-mini', -]); - -export const pruneO1Payload = (payload: ChatStreamPayload) => ({ - ...payload, - frequency_penalty: 0, - messages: payload.messages.map((message: OpenAIChatMessage) => ({ - ...message, - role: message.role === 'system' ? 'user' : message.role, - })), - presence_penalty: 0, - stream: false, - temperature: 1, - top_p: 1, -}); - - export const LobeGithubAI = LobeOpenAICompatibleFactory({ baseURL: 'https://models.inference.ai.azure.com', chatCompletion: {