distill-pipeline / src /generator /generator_cli.js
htaf's picture
initial commit
9af7af9
raw
history blame contribute delete
748 Bytes
#!/usr/bin/env node
import dotenv from 'dotenv';
dotenv.config();
import { runGenerator } from ../generator/generator_core.mjs';
import { hybridSearch } from '../retrieval/retrieval.mjs';
import { createProvider } from '../src/providers/provider.mjs';
async function main() {
const query = process.argv.slice(2).join(" ");
if (!query) {
console.error("Usage: node generator.js \"your question\"");
process.exit(1);
}
const provider = createProvider(process.env.GENERATOR_MODEL);
const context = await hybridSearch(query, 5);
const result = await runGenerator(query, context, provider);
console.log(JSON.stringify(result, null, 2));
}
main().catch(err => {
console.error("Generator error:", err);
process.exit(1);
});