howard/news-analyze
1
0
Fork 0
mirror of https://github.com/hpware/news-analyze.git synced 2025-06-23 15:51:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

Made a basic news summarizer. This is just to test in prod.

Browse Source
This commit is contained in:
吳元皓 2025-05-24 01:04:15 +08:00
parent d21957a8f9
commit 11de0632ae
2 changed files with 30 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
server/api/ai/chat/[slug].ts
View File

@ -4,9 +4,9 @@ import sql from "~/server/components/postgres";
const groq = new Groq();
export default defineEventHandler(async (event) => {
const host = await getRequestHost(event);
const protocol = await getRequestProtocol(event);
const hears = await getRequestHeaders(event);
const host = getRequestHost(event);
const protocol = getRequestProtocol(event);
const hears = getRequestHeaders(event);
const slug = getRouterParam(event, "slug");
const body = await readBody(event);
if (!slug) {
@ -47,8 +47,8 @@ export default defineEventHandler(async (event) => {
content: body.message,
},
],
model: "llama-3.1-8b-instant",
temperature: 1,
model: "gemma2-9b-it",
temperature: 0.71,
max_completion_tokens: 1024,
top_p: 1,
stream: true,
@ -56,7 +56,6 @@ export default defineEventHandler(async (event) => {
});
/*
// Save user message
await sql`
INSERT INTO chat_history (uuid, role, content)
VALUES (${slug}, 'user', ${body.message})
@ -78,7 +77,6 @@ export default defineEventHandler(async (event) => {
}
/*
// Save complete assistant response
if (assistantResponse) {
await sql`
INSERT INTO chat_history (uuid, role, content)

30
server/api/ai/summerize/[slug].ts
View File

@ -4,23 +4,24 @@ import sql from "~/server/components/postgres";
const groq = new Groq();
export default defineEventHandler(async (event) => {
const host = getRequestHost(event);
const protocol = getRequestProtocol(event);
const slug = getRouterParam(event, "slug");
const fetchNewsArticle = await sql`
select * from newArticle
where slug = ${slug}
`;
const buildURL = protocol + "://" + host + "/api/news/get/lt/" + slug;
const data = await fetch(buildURL);
const fetchNewsArticle = await data.json();
const chatCompletion = await groq.chat.completions.create({
messages: [
{
role: "user",
content: `${fetchNewsArticle.title}\n${fetchNewsArticle.content}`,
content: `${fetchNewsArticle.title}\n${fetchNewsArticle.paragraph}`,
},
{
role: "system",
content: `You are a news summarizer. You will be given a news article and you will summarize it into a short paragraph.`,
},
],
model: "llama3-70b-8192",
model: "gemma2-9b-it",
temperature: 1,
max_completion_tokens: 1024,
top_p: 1,
@ -28,7 +29,22 @@ export default defineEventHandler(async (event) => {
stop: null,
});
const stream = new ReadableStream({
async start(controller) {
try {
for await (const chunk of chatCompletion) {
process.stdout.write(chunk.choices[0]?.delta?.content || "");
const content = chunk.choices[0]?.delta?.content || "";
if (content) {
controller.enqueue(new TextEncoder().encode(content));
}
}
controller.close();
} catch (error) {
controller.error(error);
}
},
});
return sendStream(event, stream);
});