This commit is contained in:
Samuel Jędrzejewski
2025-02-07 18:53:09 +01:00
parent d533f63cb3
commit 0d71aa9f07
2 changed files with 22 additions and 8 deletions

View File

@@ -14,15 +14,19 @@ const openai = createOpenAI({
} as CustomOpenAIProviderSettings);
const isCustomEndpoint =
process.env.OPENAI_ENDPOINT && process.env.OPENAI_ENDPOINT !== 'https://api.openai.com/v1';
process.env.OPENAI_ENDPOINT &&
process.env.OPENAI_ENDPOINT !== 'https://api.openai.com/v1';
const customModel = process.env.OPENAI_MODEL;
// Models
export const o3MiniModel = openai(isCustomEndpoint && customModel ? customModel : 'o3-mini', {
reasoningEffort: 'medium',
structuredOutputs: true,
});
export const o3MiniModel = openai(
isCustomEndpoint && customModel ? customModel : 'o3-mini',
{
reasoningEffort: 'medium',
structuredOutputs: true,
},
);
const MinChunkSize = 140;
const encoder = getEncoding('o200k_base');

View File

@@ -25,9 +25,17 @@ async function run() {
// Get breath and depth parameters
const breadth =
parseInt(await askQuestion('Enter research breadth (recommended 2-10, default 4): '), 10) || 4;
parseInt(
await askQuestion(
'Enter research breadth (recommended 2-10, default 4): ',
),
10,
) || 4;
const depth =
parseInt(await askQuestion('Enter research depth (recommended 1-5, default 2): '), 10) || 2;
parseInt(
await askQuestion('Enter research depth (recommended 1-5, default 2): '),
10,
) || 2;
console.log(`Creating research plan...`);
@@ -63,7 +71,9 @@ ${followUpQuestions.map((q, i) => `Q: ${q}\nA: ${answers[i]}`).join('\n')}
});
console.log(`\n\nLearnings:\n\n${learnings.join('\n')}`);
console.log(`\n\nVisited URLs (${visitedUrls.length}):\n\n${visitedUrls.join('\n')}`);
console.log(
`\n\nVisited URLs (${visitedUrls.length}):\n\n${visitedUrls.join('\n')}`,
);
console.log('Writing final report...');
const report = await writeFinalReport({