mirror of
https://github.com/discourse/discourse-ai.git
synced 2025-03-02 15:29:16 +00:00
Adds a comprehensive quota management system for LLM models that allows: - Setting per-group (applied per user in the group) token and usage limits with configurable durations - Tracking and enforcing token/usage limits across user groups - Quota reset periods (hourly, daily, weekly, or custom) - Admin UI for managing quotas with real-time updates This system provides granular control over LLM API usage by allowing admins to define limits on both total tokens and number of requests per group. Supports multiple concurrent quotas per model and automatically handles quota resets. Co-authored-by: Keegan George <kgeorge13@gmail.com>
61 lines
1.5 KiB
JavaScript
61 lines
1.5 KiB
JavaScript
import { ajax } from "discourse/lib/ajax";
|
|
import { clipboardCopyAsync } from "discourse/lib/utilities";
|
|
import { i18n } from "discourse-i18n";
|
|
|
|
export default async function (topic, fromPostNumber, toPostNumber) {
|
|
await clipboardCopyAsync(async () => {
|
|
const text = await generateClipboard(topic, fromPostNumber, toPostNumber);
|
|
return new Blob([text], {
|
|
type: "text/plain",
|
|
});
|
|
});
|
|
}
|
|
|
|
async function generateClipboard(topic, fromPostNumber, toPostNumber) {
|
|
const stream = topic.get("postStream");
|
|
|
|
let postNumbers = [];
|
|
// simpler to understand than Array.from
|
|
for (let i = fromPostNumber; i <= toPostNumber; i++) {
|
|
postNumbers.push(i);
|
|
}
|
|
|
|
const postIds = postNumbers.map((postNumber) => {
|
|
return stream.findPostIdForPostNumber(postNumber);
|
|
});
|
|
|
|
// we need raw to construct so post stream will not help
|
|
|
|
const url = `/t/${topic.id}/posts.json`;
|
|
const data = {
|
|
post_ids: postIds,
|
|
include_raw: true,
|
|
};
|
|
|
|
const response = await ajax(url, { data });
|
|
|
|
let buffer = [];
|
|
buffer.push("<details class='ai-quote'>");
|
|
buffer.push("<summary>");
|
|
buffer.push(`<span>${topic.title}</span>`);
|
|
buffer.push(
|
|
`<span title='${i18n("discourse_ai.ai_bot.ai_title")}'>${i18n(
|
|
"discourse_ai.ai_bot.ai_label"
|
|
)}</span>`
|
|
);
|
|
buffer.push("</summary>");
|
|
|
|
response.post_stream.posts.forEach((post) => {
|
|
buffer.push("");
|
|
buffer.push(`**${post.username}:**`);
|
|
buffer.push("");
|
|
buffer.push(post.raw);
|
|
});
|
|
|
|
buffer.push("</details>");
|
|
|
|
const text = buffer.join("\n");
|
|
|
|
return text;
|
|
}
|