@goscribe/server 1.6.0 → 1.7.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/generated/prisma/client.d.ts +224 -0
- package/dist/generated/prisma/client.js +34 -0
- package/dist/generated/prisma/commonInputTypes.d.ts +941 -0
- package/dist/generated/prisma/commonInputTypes.js +10 -0
- package/dist/generated/prisma/enums.d.ts +67 -0
- package/dist/generated/prisma/enums.js +66 -0
- package/dist/generated/prisma/internal/class.d.ts +539 -0
- package/dist/generated/prisma/internal/class.js +49 -0
- package/dist/generated/prisma/internal/prismaNamespace.d.ts +3924 -0
- package/dist/generated/prisma/internal/prismaNamespace.js +557 -0
- package/dist/generated/prisma/models/ActivityLog.d.ts +1847 -0
- package/dist/generated/prisma/models/ActivityLog.js +1 -0
- package/dist/generated/prisma/models/Artifact.d.ts +2345 -0
- package/dist/generated/prisma/models/Artifact.js +1 -0
- package/dist/generated/prisma/models/ArtifactVersion.d.ts +1550 -0
- package/dist/generated/prisma/models/ArtifactVersion.js +1 -0
- package/dist/generated/prisma/models/Channel.d.ts +1257 -0
- package/dist/generated/prisma/models/Channel.js +1 -0
- package/dist/generated/prisma/models/Chat.d.ts +1339 -0
- package/dist/generated/prisma/models/Chat.js +1 -0
- package/dist/generated/prisma/models/CopilotConversation.d.ts +1450 -0
- package/dist/generated/prisma/models/CopilotConversation.js +1 -0
- package/dist/generated/prisma/models/CopilotMessage.d.ts +1179 -0
- package/dist/generated/prisma/models/CopilotMessage.js +1 -0
- package/dist/generated/prisma/models/FileAsset.d.ts +1832 -0
- package/dist/generated/prisma/models/FileAsset.js +1 -0
- package/dist/generated/prisma/models/Flashcard.d.ts +1460 -0
- package/dist/generated/prisma/models/Flashcard.js +1 -0
- package/dist/generated/prisma/models/FlashcardProgress.d.ts +1782 -0
- package/dist/generated/prisma/models/FlashcardProgress.js +1 -0
- package/dist/generated/prisma/models/Folder.d.ts +1685 -0
- package/dist/generated/prisma/models/Folder.js +1 -0
- package/dist/generated/prisma/models/IdempotencyRecord.d.ts +1319 -0
- package/dist/generated/prisma/models/IdempotencyRecord.js +1 -0
- package/dist/generated/prisma/models/Invoice.d.ts +1586 -0
- package/dist/generated/prisma/models/Invoice.js +1 -0
- package/dist/generated/prisma/models/KnowledgeBase.d.ts +1721 -0
- package/dist/generated/prisma/models/KnowledgeBase.js +1 -0
- package/dist/generated/prisma/models/KnowledgeBaseChunk.d.ts +1333 -0
- package/dist/generated/prisma/models/KnowledgeBaseChunk.js +1 -0
- package/dist/generated/prisma/models/KnowledgeBaseDocument.d.ts +1695 -0
- package/dist/generated/prisma/models/KnowledgeBaseDocument.js +1 -0
- package/dist/generated/prisma/models/Notification.d.ts +1992 -0
- package/dist/generated/prisma/models/Notification.js +1 -0
- package/dist/generated/prisma/models/PasswordResetToken.d.ts +1210 -0
- package/dist/generated/prisma/models/PasswordResetToken.js +1 -0
- package/dist/generated/prisma/models/Plan.d.ts +1431 -0
- package/dist/generated/prisma/models/Plan.js +1 -0
- package/dist/generated/prisma/models/PlanLimit.d.ts +1328 -0
- package/dist/generated/prisma/models/PlanLimit.js +1 -0
- package/dist/generated/prisma/models/PodcastSegment.d.ts +1564 -0
- package/dist/generated/prisma/models/PodcastSegment.js +1 -0
- package/dist/generated/prisma/models/ResourcePrice.d.ts +1008 -0
- package/dist/generated/prisma/models/ResourcePrice.js +1 -0
- package/dist/generated/prisma/models/Role.d.ts +1065 -0
- package/dist/generated/prisma/models/Role.js +1 -0
- package/dist/generated/prisma/models/Session.d.ts +1105 -0
- package/dist/generated/prisma/models/Session.js +1 -0
- package/dist/generated/prisma/models/StripeEvent.d.ts +1081 -0
- package/dist/generated/prisma/models/StripeEvent.js +1 -0
- package/dist/generated/prisma/models/StudyGuideComment.d.ts +1321 -0
- package/dist/generated/prisma/models/StudyGuideComment.js +1 -0
- package/dist/generated/prisma/models/StudyGuideHighlight.d.ts +1629 -0
- package/dist/generated/prisma/models/StudyGuideHighlight.js +1 -0
- package/dist/generated/prisma/models/Subscription.d.ts +1677 -0
- package/dist/generated/prisma/models/Subscription.js +1 -0
- package/dist/generated/prisma/models/User.d.ts +7559 -0
- package/dist/generated/prisma/models/User.js +1 -0
- package/dist/generated/prisma/models/UserCredit.d.ts +1249 -0
- package/dist/generated/prisma/models/UserCredit.js +1 -0
- package/dist/generated/prisma/models/VerificationToken.d.ts +946 -0
- package/dist/generated/prisma/models/VerificationToken.js +1 -0
- package/dist/generated/prisma/models/WorksheetPreset.d.ts +1433 -0
- package/dist/generated/prisma/models/WorksheetPreset.js +1 -0
- package/dist/generated/prisma/models/WorksheetQuestion.d.ts +1491 -0
- package/dist/generated/prisma/models/WorksheetQuestion.js +1 -0
- package/dist/generated/prisma/models/WorksheetQuestionProgress.d.ts +1620 -0
- package/dist/generated/prisma/models/WorksheetQuestionProgress.js +1 -0
- package/dist/generated/prisma/models/Workspace.d.ts +3620 -0
- package/dist/generated/prisma/models/Workspace.js +1 -0
- package/dist/generated/prisma/models/WorkspaceInvitation.d.ts +1490 -0
- package/dist/generated/prisma/models/WorkspaceInvitation.js +1 -0
- package/dist/generated/prisma/models/WorkspaceKnowledgeBase.d.ts +1410 -0
- package/dist/generated/prisma/models/WorkspaceKnowledgeBase.js +1 -0
- package/dist/generated/prisma/models/WorkspaceMember.d.ts +1326 -0
- package/dist/generated/prisma/models/WorkspaceMember.js +1 -0
- package/dist/generated/prisma/models.d.ts +39 -0
- package/dist/generated/prisma/models.js +1 -0
- package/dist/src/context.d.ts +27 -0
- package/dist/src/context.js +33 -0
- package/dist/src/index.d.ts +3 -0
- package/dist/src/index.js +1 -0
- package/dist/src/lib/ai/config.d.ts +20 -0
- package/dist/src/lib/ai/config.js +31 -0
- package/dist/src/lib/ai/embedding-client.d.ts +8 -0
- package/dist/src/lib/ai/embedding-client.js +30 -0
- package/dist/src/lib/ai/index.d.ts +48 -0
- package/dist/src/lib/ai/index.js +29 -0
- package/dist/src/lib/ai/inference-backend/client.d.ts +28 -0
- package/dist/src/lib/ai/inference-backend/client.js +301 -0
- package/dist/src/lib/ai/inference-backend/mocks.d.ts +12 -0
- package/dist/src/lib/ai/inference-backend/mocks.js +133 -0
- package/dist/src/lib/ai/inference-backend/types.d.ts +44 -0
- package/dist/src/lib/ai/inference-backend/types.js +1 -0
- package/dist/src/lib/ai/json-parse.d.ts +2 -0
- package/dist/src/lib/ai/json-parse.js +34 -0
- package/dist/src/lib/ai/llm-client.d.ts +7 -0
- package/dist/src/lib/ai/llm-client.js +36 -0
- package/dist/src/lib/ai/mock.d.ts +2 -0
- package/dist/src/lib/ai/mock.js +10 -0
- package/dist/src/lib/ai/types.d.ts +9 -0
- package/dist/src/lib/ai/types.js +1 -0
- package/dist/src/lib/auth.d.ts +36 -0
- package/dist/src/lib/auth.js +117 -0
- package/dist/src/lib/chunking.d.ts +19 -0
- package/dist/src/lib/chunking.js +47 -0
- package/dist/src/lib/constants.d.ts +13 -0
- package/dist/src/lib/constants.js +12 -0
- package/dist/src/lib/curated-kb-seed.d.ts +12 -0
- package/dist/src/lib/curated-kb-seed.js +155 -0
- package/dist/src/lib/email.d.ts +11 -0
- package/dist/src/lib/email.js +152 -0
- package/dist/src/lib/embeddings.d.ts +2 -0
- package/dist/src/lib/embeddings.js +1 -0
- package/dist/src/lib/ensure-curated-kb-catalog.d.ts +6 -0
- package/dist/src/lib/ensure-curated-kb-catalog.js +53 -0
- package/dist/src/lib/env.d.ts +41 -0
- package/dist/src/lib/env.js +57 -0
- package/dist/src/lib/errors.d.ts +33 -0
- package/dist/src/lib/errors.js +78 -0
- package/dist/src/lib/file.d.ts +0 -0
- package/dist/src/lib/file.js +1 -0
- package/dist/src/lib/inference.d.ts +1 -0
- package/dist/src/lib/inference.js +1 -0
- package/dist/src/lib/kb-meta.d.ts +8 -0
- package/dist/src/lib/kb-meta.js +77 -0
- package/dist/src/lib/logger.d.ts +62 -0
- package/dist/src/lib/logger.js +364 -0
- package/dist/src/lib/pdf.d.ts +11 -0
- package/dist/src/lib/pdf.js +11 -0
- package/dist/src/lib/prisma.d.ts +3 -0
- package/dist/src/lib/prisma.js +15 -0
- package/dist/src/lib/pusher.d.ts +38 -0
- package/dist/src/lib/pusher.js +170 -0
- package/dist/src/lib/retry.d.ts +15 -0
- package/dist/src/lib/retry.js +37 -0
- package/dist/src/lib/storage.d.ts +11 -0
- package/dist/src/lib/storage.js +71 -0
- package/dist/src/lib/stripe.d.ts +10 -0
- package/dist/src/lib/stripe.js +36 -0
- package/dist/src/lib/validation.d.ts +51 -0
- package/dist/src/lib/validation.js +64 -0
- package/dist/src/lib/workspace-kb.d.ts +5 -0
- package/dist/src/lib/workspace-kb.js +7 -0
- package/dist/src/repositories/artifact.repository.d.ts +64 -0
- package/dist/src/repositories/artifact.repository.js +40 -0
- package/dist/src/repositories/base.repository.d.ts +14 -0
- package/dist/src/repositories/base.repository.js +14 -0
- package/dist/src/repositories/invitation.repository.d.ts +104 -0
- package/dist/src/repositories/invitation.repository.js +44 -0
- package/dist/src/repositories/notification.repository.d.ts +76 -0
- package/dist/src/repositories/notification.repository.js +44 -0
- package/dist/src/repositories/user.repository.d.ts +84 -0
- package/dist/src/repositories/user.repository.js +37 -0
- package/dist/src/repositories/workspace-member.repository.d.ts +35 -0
- package/dist/src/repositories/workspace-member.repository.js +31 -0
- package/dist/src/repositories/workspace.repository.d.ts +101 -0
- package/dist/src/repositories/workspace.repository.js +79 -0
- package/dist/src/routers/_app.d.ts +3464 -0
- package/dist/src/routers/_app.js +36 -0
- package/dist/src/routers/admin.d.ts +358 -0
- package/dist/src/routers/admin.js +105 -0
- package/dist/src/routers/annotations.d.ts +219 -0
- package/dist/src/routers/annotations.js +29 -0
- package/dist/src/routers/artifactVersions.d.ts +65 -0
- package/dist/src/routers/artifactVersions.js +14 -0
- package/dist/src/routers/auth.d.ts +161 -0
- package/dist/src/routers/auth.js +97 -0
- package/dist/src/routers/chat.d.ts +170 -0
- package/dist/src/routers/chat.js +32 -0
- package/dist/src/routers/copilot.d.ts +200 -0
- package/dist/src/routers/copilot.js +52 -0
- package/dist/src/routers/flashcards.d.ts +336 -0
- package/dist/src/routers/flashcards.js +93 -0
- package/dist/src/routers/knowledgeBase.d.ts +421 -0
- package/dist/src/routers/knowledgeBase.js +118 -0
- package/dist/src/routers/members.d.ts +169 -0
- package/dist/src/routers/members.js +47 -0
- package/dist/src/routers/notifications.d.ts +99 -0
- package/dist/src/routers/notifications.js +25 -0
- package/dist/src/routers/payment.d.ts +80 -0
- package/dist/src/routers/payment.js +21 -0
- package/dist/src/routers/podcast.d.ts +287 -0
- package/dist/src/routers/podcast.js +34 -0
- package/dist/src/routers/studyguide.d.ts +36 -0
- package/dist/src/routers/studyguide.js +8 -0
- package/dist/src/routers/worksheets.d.ts +429 -0
- package/dist/src/routers/worksheets.js +139 -0
- package/dist/src/routers/workspace.d.ts +563 -0
- package/dist/src/routers/workspace.js +104 -0
- package/dist/src/scripts/purge-deleted-users.d.ts +1 -0
- package/dist/src/scripts/purge-deleted-users.js +148 -0
- package/dist/src/server.d.ts +1 -0
- package/dist/src/server.js +190 -0
- package/dist/src/services/activity/activity-human-description.service.d.ts +13 -0
- package/dist/src/services/activity/activity-human-description.service.js +221 -0
- package/dist/src/services/activity/activity-human-description.service.test.d.ts +1 -0
- package/dist/src/services/activity/activity-human-description.service.test.js +16 -0
- package/dist/src/services/activity/activity-log.service.d.ts +87 -0
- package/dist/src/services/activity/activity-log.service.js +276 -0
- package/dist/src/services/activity/activity-log.service.test.d.ts +1 -0
- package/dist/src/services/activity/activity-log.service.test.js +27 -0
- package/dist/src/services/admin/admin.service.d.ts +270 -0
- package/dist/src/services/admin/admin.service.js +476 -0
- package/dist/src/services/ai/ai-session.service.d.ts +5 -0
- package/dist/src/services/ai/ai-session.service.js +4 -0
- package/dist/src/services/artifacts/annotation.service.d.ts +177 -0
- package/dist/src/services/artifacts/annotation.service.js +154 -0
- package/dist/src/services/artifacts/artifact-version.service.d.ts +38 -0
- package/dist/src/services/artifacts/artifact-version.service.js +129 -0
- package/dist/src/services/artifacts/chat.service.d.ts +127 -0
- package/dist/src/services/artifacts/chat.service.js +182 -0
- package/dist/src/services/artifacts/study-guide.service.d.ts +18 -0
- package/dist/src/services/artifacts/study-guide.service.js +65 -0
- package/dist/src/services/auth/auth.service.d.ts +94 -0
- package/dist/src/services/auth/auth.service.js +368 -0
- package/dist/src/services/base.service.d.ts +14 -0
- package/dist/src/services/base.service.js +14 -0
- package/dist/src/services/billing/payment.service.d.ts +44 -0
- package/dist/src/services/billing/payment.service.js +365 -0
- package/dist/src/services/billing/subscription.service.d.ts +37 -0
- package/dist/src/services/billing/subscription.service.js +654 -0
- package/dist/src/services/billing/usage.service.d.ts +47 -0
- package/dist/src/services/billing/usage.service.js +149 -0
- package/dist/src/services/content/copilot.service.d.ts +113 -0
- package/dist/src/services/content/copilot.service.js +439 -0
- package/dist/src/services/content/flashcard-progress.service.d.ts +159 -0
- package/dist/src/services/content/flashcard-progress.service.js +432 -0
- package/dist/src/services/content/flashcard.service.d.ts +184 -0
- package/dist/src/services/content/flashcard.service.js +339 -0
- package/dist/src/services/content/media-analysis.service.d.ts +23 -0
- package/dist/src/services/content/media-analysis.service.js +404 -0
- package/dist/src/services/content/podcast.service.d.ts +267 -0
- package/dist/src/services/content/podcast.service.js +653 -0
- package/dist/src/services/content/worksheet-content.service.d.ts +37 -0
- package/dist/src/services/content/worksheet-content.service.js +84 -0
- package/dist/src/services/content/worksheet-content.service.test.d.ts +1 -0
- package/dist/src/services/content/worksheet-content.service.test.js +69 -0
- package/dist/src/services/content/worksheet-generation.service.d.ts +91 -0
- package/dist/src/services/content/worksheet-generation.service.js +95 -0
- package/dist/src/services/content/worksheet-generation.service.test.d.ts +1 -0
- package/dist/src/services/content/worksheet-generation.service.test.js +20 -0
- package/dist/src/services/content/worksheet.service.d.ts +347 -0
- package/dist/src/services/content/worksheet.service.js +599 -0
- package/dist/src/services/knowledge/knowledge-base.service.d.ts +316 -0
- package/dist/src/services/knowledge/knowledge-base.service.js +544 -0
- package/dist/src/services/members/invitation.service.d.ts +66 -0
- package/dist/src/services/members/invitation.service.js +348 -0
- package/dist/src/services/members/member.service.d.ts +36 -0
- package/dist/src/services/members/member.service.js +193 -0
- package/dist/src/services/notifications/notification.service.d.ts +214 -0
- package/dist/src/services/notifications/notification.service.js +550 -0
- package/dist/src/services/notifications/notification.service.test.d.ts +1 -0
- package/dist/src/services/notifications/notification.service.test.js +87 -0
- package/dist/src/services/workspace/workspace-analytics.service.d.ts +24 -0
- package/dist/src/services/workspace/workspace-analytics.service.js +95 -0
- package/dist/src/services/workspace/workspace-kb.service.d.ts +40 -0
- package/dist/src/services/workspace/workspace-kb.service.js +184 -0
- package/dist/src/services/workspace/workspace.service.d.ts +263 -0
- package/dist/src/services/workspace/workspace.service.js +401 -0
- package/dist/src/trpc.d.ts +60 -0
- package/dist/src/trpc.js +217 -0
- package/dist/src/types/index.d.ts +126 -0
- package/dist/src/types/index.js +1 -0
- package/package.json +8 -9
- package/prisma/schema.prisma +3 -4
- package/prisma/seed.mjs +5 -2
- package/prisma.config.ts +16 -0
- package/src/lib/prisma.ts +18 -9
- package/src/scripts/purge-deleted-users.ts +1 -3
- package/tsconfig.json +3 -0
|
@@ -0,0 +1,95 @@
|
|
|
1
|
+
import { BaseService } from '../base.service.js';
|
|
2
|
+
/**
|
|
3
|
+
* Study analytics: streaks, weekly activity, flashcard mastery, worksheet
|
|
4
|
+
* accuracy. Behavior-preserving extraction from `workspace.getStudyAnalytics`.
|
|
5
|
+
*/
|
|
6
|
+
export class WorkspaceAnalyticsService extends BaseService {
|
|
7
|
+
constructor(db) {
|
|
8
|
+
super(db);
|
|
9
|
+
}
|
|
10
|
+
async getStudyAnalytics(userId) {
|
|
11
|
+
const flashcardProgress = await this.db.flashcardProgress.findMany({
|
|
12
|
+
where: { userId },
|
|
13
|
+
select: { lastStudiedAt: true },
|
|
14
|
+
});
|
|
15
|
+
const worksheetProgress = await this.db.worksheetQuestionProgress.findMany({
|
|
16
|
+
where: { userId },
|
|
17
|
+
select: { updatedAt: true, completedAt: true },
|
|
18
|
+
});
|
|
19
|
+
const studyDays = new Set();
|
|
20
|
+
for (const fp of flashcardProgress) {
|
|
21
|
+
if (fp.lastStudiedAt) {
|
|
22
|
+
studyDays.add(fp.lastStudiedAt.toISOString().split('T')[0]);
|
|
23
|
+
}
|
|
24
|
+
}
|
|
25
|
+
for (const wp of worksheetProgress) {
|
|
26
|
+
if (wp.completedAt) {
|
|
27
|
+
studyDays.add(wp.completedAt.toISOString().split('T')[0]);
|
|
28
|
+
}
|
|
29
|
+
else {
|
|
30
|
+
studyDays.add(wp.updatedAt.toISOString().split('T')[0]);
|
|
31
|
+
}
|
|
32
|
+
}
|
|
33
|
+
const sortedDays = [...studyDays].sort().reverse();
|
|
34
|
+
let streak = 0;
|
|
35
|
+
if (sortedDays.length > 0) {
|
|
36
|
+
const today = new Date();
|
|
37
|
+
today.setHours(0, 0, 0, 0);
|
|
38
|
+
const yesterday = new Date(today);
|
|
39
|
+
yesterday.setDate(yesterday.getDate() - 1);
|
|
40
|
+
const todayStr = today.toISOString().split('T')[0];
|
|
41
|
+
const yesterdayStr = yesterday.toISOString().split('T')[0];
|
|
42
|
+
if (sortedDays[0] === todayStr || sortedDays[0] === yesterdayStr) {
|
|
43
|
+
streak = 1;
|
|
44
|
+
for (let i = 1; i < sortedDays.length; i++) {
|
|
45
|
+
const current = new Date(sortedDays[i - 1]);
|
|
46
|
+
const prev = new Date(sortedDays[i]);
|
|
47
|
+
const diffDays = (current.getTime() - prev.getTime()) / (1000 * 60 * 60 * 24);
|
|
48
|
+
if (diffDays === 1) {
|
|
49
|
+
streak++;
|
|
50
|
+
}
|
|
51
|
+
else {
|
|
52
|
+
break;
|
|
53
|
+
}
|
|
54
|
+
}
|
|
55
|
+
}
|
|
56
|
+
}
|
|
57
|
+
const weeklyActivity = [];
|
|
58
|
+
const today = new Date();
|
|
59
|
+
today.setHours(0, 0, 0, 0);
|
|
60
|
+
for (let i = 6; i >= 0; i--) {
|
|
61
|
+
const d = new Date(today);
|
|
62
|
+
d.setDate(d.getDate() - i);
|
|
63
|
+
const dayStr = d.toISOString().split('T')[0];
|
|
64
|
+
weeklyActivity.push(studyDays.has(dayStr));
|
|
65
|
+
}
|
|
66
|
+
const totalCards = await this.db.flashcardProgress.count({ where: { userId } });
|
|
67
|
+
const masteredCards = await this.db.flashcardProgress.count({
|
|
68
|
+
where: { userId, masteryLevel: { gte: 80 } },
|
|
69
|
+
});
|
|
70
|
+
const dueCards = await this.db.flashcardProgress.count({
|
|
71
|
+
where: { userId, nextReviewAt: { lte: new Date() } },
|
|
72
|
+
});
|
|
73
|
+
const completedQuestions = await this.db.worksheetQuestionProgress.count({
|
|
74
|
+
where: { userId, completedAt: { not: null } },
|
|
75
|
+
});
|
|
76
|
+
const correctQuestions = await this.db.worksheetQuestionProgress.count({
|
|
77
|
+
where: { userId, correct: true },
|
|
78
|
+
});
|
|
79
|
+
return {
|
|
80
|
+
streak,
|
|
81
|
+
totalStudyDays: studyDays.size,
|
|
82
|
+
weeklyActivity,
|
|
83
|
+
flashcards: {
|
|
84
|
+
total: totalCards,
|
|
85
|
+
mastered: masteredCards,
|
|
86
|
+
dueForReview: dueCards,
|
|
87
|
+
},
|
|
88
|
+
worksheets: {
|
|
89
|
+
completed: completedQuestions,
|
|
90
|
+
correct: correctQuestions,
|
|
91
|
+
accuracy: completedQuestions > 0 ? Math.round((correctQuestions / completedQuestions) * 100) : 0,
|
|
92
|
+
},
|
|
93
|
+
};
|
|
94
|
+
}
|
|
95
|
+
}
|
|
@@ -0,0 +1,40 @@
|
|
|
1
|
+
import type { PrismaClient } from '@prisma/client';
|
|
2
|
+
import { BaseService } from '../base.service.js';
|
|
3
|
+
export interface WorkspaceKbMatch {
|
|
4
|
+
chunkId: string;
|
|
5
|
+
documentId: string;
|
|
6
|
+
documentName: string;
|
|
7
|
+
chunkIndex: number;
|
|
8
|
+
content: string;
|
|
9
|
+
score: number;
|
|
10
|
+
}
|
|
11
|
+
export declare class WorkspaceKbService extends BaseService {
|
|
12
|
+
constructor(db: PrismaClient);
|
|
13
|
+
/** Returns the single auto-managed KB for a workspace, creating it if needed. */
|
|
14
|
+
ensureWorkspaceKb(workspaceId: string, userId: string, title?: string): Promise<{
|
|
15
|
+
id: string;
|
|
16
|
+
name: string;
|
|
17
|
+
createdAt: Date;
|
|
18
|
+
updatedAt: Date;
|
|
19
|
+
description: string | null;
|
|
20
|
+
createdById: string | null;
|
|
21
|
+
status: import("@prisma/client").KnowledgeBaseStatus;
|
|
22
|
+
meta: import("@prisma/client/runtime/client").JsonValue | null;
|
|
23
|
+
isCurated: boolean;
|
|
24
|
+
embeddingModel: string;
|
|
25
|
+
embeddingDim: number;
|
|
26
|
+
}>;
|
|
27
|
+
getWorkspaceKbId(workspaceId: string): Promise<string | null>;
|
|
28
|
+
indexText(workspaceId: string, userId: string, input: {
|
|
29
|
+
sourceType: string;
|
|
30
|
+
sourceId: string;
|
|
31
|
+
name: string;
|
|
32
|
+
text: string;
|
|
33
|
+
}): Promise<void>;
|
|
34
|
+
/** Fire-and-forget indexing — never blocks the caller. */
|
|
35
|
+
indexTextAsync(workspaceId: string, userId: string, input: Parameters<WorkspaceKbService['indexText']>[2]): void;
|
|
36
|
+
search(workspaceId: string, query: string, topK?: number): Promise<WorkspaceKbMatch[]>;
|
|
37
|
+
formatContext(matches: WorkspaceKbMatch[]): string;
|
|
38
|
+
retrieveContext(workspaceId: string, query: string, topK?: number): Promise<string>;
|
|
39
|
+
}
|
|
40
|
+
export declare const workspaceKbService: WorkspaceKbService;
|
|
@@ -0,0 +1,184 @@
|
|
|
1
|
+
import { BaseService } from '../base.service.js';
|
|
2
|
+
import { prisma } from '../../lib/prisma.js';
|
|
3
|
+
import { DEFAULT_EMBEDDING_MODEL, embedQuery, embedTexts, toVectorLiteral, } from '../../lib/embeddings.js';
|
|
4
|
+
import { chunkText } from '../../lib/chunking.js';
|
|
5
|
+
import { WORKSPACE_KB_KIND } from '../../lib/workspace-kb.js';
|
|
6
|
+
export class WorkspaceKbService extends BaseService {
|
|
7
|
+
constructor(db) {
|
|
8
|
+
super(db);
|
|
9
|
+
}
|
|
10
|
+
/** Returns the single auto-managed KB for a workspace, creating it if needed. */
|
|
11
|
+
async ensureWorkspaceKb(workspaceId, userId, title) {
|
|
12
|
+
const existingLink = await this.db.workspaceKnowledgeBase.findFirst({
|
|
13
|
+
where: {
|
|
14
|
+
workspaceId,
|
|
15
|
+
knowledgeBase: {
|
|
16
|
+
isCurated: false,
|
|
17
|
+
meta: {
|
|
18
|
+
path: ['kind'],
|
|
19
|
+
equals: WORKSPACE_KB_KIND,
|
|
20
|
+
},
|
|
21
|
+
},
|
|
22
|
+
},
|
|
23
|
+
include: { knowledgeBase: true },
|
|
24
|
+
});
|
|
25
|
+
if (existingLink) {
|
|
26
|
+
return existingLink.knowledgeBase;
|
|
27
|
+
}
|
|
28
|
+
const workspace = await this.db.workspace.findUnique({
|
|
29
|
+
where: { id: workspaceId },
|
|
30
|
+
select: { title: true },
|
|
31
|
+
});
|
|
32
|
+
const kb = await this.db.knowledgeBase.create({
|
|
33
|
+
data: {
|
|
34
|
+
createdById: userId,
|
|
35
|
+
name: `${title ?? workspace?.title ?? 'Workspace'} Knowledge Base`,
|
|
36
|
+
description: 'Private knowledge base for this workspace',
|
|
37
|
+
isCurated: false,
|
|
38
|
+
embeddingModel: DEFAULT_EMBEDDING_MODEL,
|
|
39
|
+
embeddingDim: 1536,
|
|
40
|
+
meta: { kind: WORKSPACE_KB_KIND, workspaceId },
|
|
41
|
+
},
|
|
42
|
+
});
|
|
43
|
+
await this.db.workspaceKnowledgeBase.create({
|
|
44
|
+
data: {
|
|
45
|
+
workspaceId,
|
|
46
|
+
knowledgeBaseId: kb.id,
|
|
47
|
+
addedById: userId,
|
|
48
|
+
},
|
|
49
|
+
});
|
|
50
|
+
return kb;
|
|
51
|
+
}
|
|
52
|
+
async getWorkspaceKbId(workspaceId) {
|
|
53
|
+
const link = await this.db.workspaceKnowledgeBase.findFirst({
|
|
54
|
+
where: {
|
|
55
|
+
workspaceId,
|
|
56
|
+
knowledgeBase: {
|
|
57
|
+
isCurated: false,
|
|
58
|
+
meta: { path: ['kind'], equals: WORKSPACE_KB_KIND },
|
|
59
|
+
},
|
|
60
|
+
},
|
|
61
|
+
select: { knowledgeBaseId: true },
|
|
62
|
+
});
|
|
63
|
+
return link?.knowledgeBaseId ?? null;
|
|
64
|
+
}
|
|
65
|
+
async indexText(workspaceId, userId, input) {
|
|
66
|
+
const trimmed = input.text.trim();
|
|
67
|
+
if (!trimmed)
|
|
68
|
+
return;
|
|
69
|
+
try {
|
|
70
|
+
const kb = await this.ensureWorkspaceKb(workspaceId, userId);
|
|
71
|
+
const docs = await this.db.knowledgeBaseDocument.findMany({
|
|
72
|
+
where: { knowledgeBaseId: kb.id },
|
|
73
|
+
select: { id: true, meta: true },
|
|
74
|
+
});
|
|
75
|
+
const existing = docs.find((doc) => {
|
|
76
|
+
const meta = doc.meta;
|
|
77
|
+
return meta?.sourceType === input.sourceType && meta?.sourceId === input.sourceId;
|
|
78
|
+
});
|
|
79
|
+
let documentId;
|
|
80
|
+
if (existing) {
|
|
81
|
+
documentId = existing.id;
|
|
82
|
+
await this.db.knowledgeBaseChunk.deleteMany({ where: { documentId } });
|
|
83
|
+
await this.db.knowledgeBaseDocument.update({
|
|
84
|
+
where: { id: documentId },
|
|
85
|
+
data: {
|
|
86
|
+
name: input.name,
|
|
87
|
+
status: 'PROCESSING',
|
|
88
|
+
errorMessage: null,
|
|
89
|
+
},
|
|
90
|
+
});
|
|
91
|
+
}
|
|
92
|
+
else {
|
|
93
|
+
const created = await this.db.knowledgeBaseDocument.create({
|
|
94
|
+
data: {
|
|
95
|
+
knowledgeBaseId: kb.id,
|
|
96
|
+
name: input.name,
|
|
97
|
+
mimeType: 'text/plain',
|
|
98
|
+
size: Buffer.byteLength(trimmed, 'utf8'),
|
|
99
|
+
status: 'PROCESSING',
|
|
100
|
+
meta: {
|
|
101
|
+
sourceType: input.sourceType,
|
|
102
|
+
sourceId: input.sourceId,
|
|
103
|
+
},
|
|
104
|
+
},
|
|
105
|
+
});
|
|
106
|
+
documentId = created.id;
|
|
107
|
+
}
|
|
108
|
+
await this.db.knowledgeBase.update({
|
|
109
|
+
where: { id: kb.id },
|
|
110
|
+
data: { status: 'INDEXING' },
|
|
111
|
+
});
|
|
112
|
+
const chunks = chunkText(trimmed, { targetTokens: 500, overlapTokens: 60 });
|
|
113
|
+
if (chunks.length === 0)
|
|
114
|
+
return;
|
|
115
|
+
const embeddings = await embedTexts(chunks.map((c) => c.content), { model: kb.embeddingModel });
|
|
116
|
+
for (let i = 0; i < chunks.length; i += 1) {
|
|
117
|
+
const chunk = chunks[i];
|
|
118
|
+
const vec = toVectorLiteral(embeddings[i] ?? []);
|
|
119
|
+
await this.db.$executeRawUnsafe(`INSERT INTO "KnowledgeBaseChunk"
|
|
120
|
+
("id", "documentId", "knowledgeBaseId", "chunkIndex",
|
|
121
|
+
"pageNumber", "content", "tokenCount", "embedding", "createdAt")
|
|
122
|
+
VALUES (
|
|
123
|
+
gen_random_uuid()::text,
|
|
124
|
+
$1, $2, $3, $4, $5, $6, $7::vector, NOW()
|
|
125
|
+
)`, documentId, kb.id, chunk.index, null, chunk.content, chunk.approxTokens, vec);
|
|
126
|
+
}
|
|
127
|
+
await this.db.knowledgeBaseDocument.update({
|
|
128
|
+
where: { id: documentId },
|
|
129
|
+
data: { status: 'READY', numChunks: chunks.length },
|
|
130
|
+
});
|
|
131
|
+
await this.db.knowledgeBase.update({
|
|
132
|
+
where: { id: kb.id },
|
|
133
|
+
data: { status: 'READY' },
|
|
134
|
+
});
|
|
135
|
+
}
|
|
136
|
+
catch (err) {
|
|
137
|
+
this.logger.error(`Failed to index ${input.sourceType}:${input.sourceId} into workspace KB`, err);
|
|
138
|
+
}
|
|
139
|
+
}
|
|
140
|
+
/** Fire-and-forget indexing — never blocks the caller. */
|
|
141
|
+
indexTextAsync(workspaceId, userId, input) {
|
|
142
|
+
void this.indexText(workspaceId, userId, input);
|
|
143
|
+
}
|
|
144
|
+
async search(workspaceId, query, topK = 5) {
|
|
145
|
+
const kbId = await this.getWorkspaceKbId(workspaceId);
|
|
146
|
+
if (!kbId)
|
|
147
|
+
return [];
|
|
148
|
+
const kb = await this.db.knowledgeBase.findUnique({ where: { id: kbId } });
|
|
149
|
+
if (!kb)
|
|
150
|
+
return [];
|
|
151
|
+
const k = Math.min(Math.max(topK, 1), 20);
|
|
152
|
+
const queryVec = await embedQuery(query, { model: kb.embeddingModel });
|
|
153
|
+
const vecLiteral = toVectorLiteral(queryVec);
|
|
154
|
+
const rows = await this.db.$queryRawUnsafe(`SELECT c."id", c."documentId", c."chunkIndex", c."content",
|
|
155
|
+
(c."embedding" <=> $1::vector) AS "distance",
|
|
156
|
+
d."name" AS "documentName"
|
|
157
|
+
FROM "KnowledgeBaseChunk" c
|
|
158
|
+
JOIN "KnowledgeBaseDocument" d ON d."id" = c."documentId"
|
|
159
|
+
WHERE c."knowledgeBaseId" = $2
|
|
160
|
+
AND c."embedding" IS NOT NULL
|
|
161
|
+
ORDER BY c."embedding" <=> $1::vector
|
|
162
|
+
LIMIT ${k}`, vecLiteral, kbId);
|
|
163
|
+
return rows.map((r) => ({
|
|
164
|
+
chunkId: r.id,
|
|
165
|
+
documentId: r.documentId,
|
|
166
|
+
documentName: r.documentName,
|
|
167
|
+
chunkIndex: r.chunkIndex,
|
|
168
|
+
content: r.content,
|
|
169
|
+
score: 1 - Number(r.distance),
|
|
170
|
+
}));
|
|
171
|
+
}
|
|
172
|
+
formatContext(matches) {
|
|
173
|
+
if (matches.length === 0)
|
|
174
|
+
return '';
|
|
175
|
+
return matches
|
|
176
|
+
.map((m, i) => `[${i + 1}] (${m.documentName}, chunk ${m.chunkIndex})\n${m.content}`)
|
|
177
|
+
.join('\n\n---\n\n');
|
|
178
|
+
}
|
|
179
|
+
async retrieveContext(workspaceId, query, topK = 5) {
|
|
180
|
+
const matches = await this.search(workspaceId, query, topK);
|
|
181
|
+
return this.formatContext(matches);
|
|
182
|
+
}
|
|
183
|
+
}
|
|
184
|
+
export const workspaceKbService = new WorkspaceKbService(prisma);
|
|
@@ -0,0 +1,263 @@
|
|
|
1
|
+
import type { PrismaClient } from '@prisma/client';
|
|
2
|
+
import { BaseService } from '../base.service.js';
|
|
3
|
+
export declare class WorkspaceService extends BaseService {
|
|
4
|
+
constructor(db: PrismaClient);
|
|
5
|
+
list(userId: string, parentId: string | null): Promise<{
|
|
6
|
+
workspaces: {
|
|
7
|
+
id: string;
|
|
8
|
+
createdAt: Date;
|
|
9
|
+
updatedAt: Date;
|
|
10
|
+
ownerId: string;
|
|
11
|
+
title: string;
|
|
12
|
+
description: string | null;
|
|
13
|
+
icon: string;
|
|
14
|
+
color: string;
|
|
15
|
+
markerColor: string | null;
|
|
16
|
+
folderId: string | null;
|
|
17
|
+
fileBeingAnalyzed: boolean;
|
|
18
|
+
analysisProgress: import("@prisma/client/runtime/client").JsonValue | null;
|
|
19
|
+
needsAnalysis: boolean;
|
|
20
|
+
}[];
|
|
21
|
+
folders: {
|
|
22
|
+
id: string;
|
|
23
|
+
name: string;
|
|
24
|
+
createdAt: Date;
|
|
25
|
+
updatedAt: Date;
|
|
26
|
+
ownerId: string;
|
|
27
|
+
color: string;
|
|
28
|
+
markerColor: string | null;
|
|
29
|
+
parentId: string | null;
|
|
30
|
+
}[];
|
|
31
|
+
}>;
|
|
32
|
+
getTree(userId: string): Promise<{
|
|
33
|
+
folders: {
|
|
34
|
+
id: string;
|
|
35
|
+
name: string;
|
|
36
|
+
updatedAt: Date;
|
|
37
|
+
color: string;
|
|
38
|
+
markerColor: string | null;
|
|
39
|
+
parentId: string | null;
|
|
40
|
+
}[];
|
|
41
|
+
workspaces: {
|
|
42
|
+
id: string;
|
|
43
|
+
updatedAt: Date;
|
|
44
|
+
uploads: {
|
|
45
|
+
id: string;
|
|
46
|
+
name: string;
|
|
47
|
+
createdAt: Date;
|
|
48
|
+
mimeType: string;
|
|
49
|
+
}[];
|
|
50
|
+
title: string;
|
|
51
|
+
icon: string;
|
|
52
|
+
color: string;
|
|
53
|
+
markerColor: string | null;
|
|
54
|
+
folderId: string | null;
|
|
55
|
+
}[];
|
|
56
|
+
}>;
|
|
57
|
+
create(userId: string, input: {
|
|
58
|
+
name: string;
|
|
59
|
+
description?: string;
|
|
60
|
+
parentId?: string;
|
|
61
|
+
markerColor?: string | null;
|
|
62
|
+
}): Promise<{
|
|
63
|
+
id: string;
|
|
64
|
+
createdAt: Date;
|
|
65
|
+
updatedAt: Date;
|
|
66
|
+
ownerId: string;
|
|
67
|
+
title: string;
|
|
68
|
+
description: string | null;
|
|
69
|
+
icon: string;
|
|
70
|
+
color: string;
|
|
71
|
+
markerColor: string | null;
|
|
72
|
+
folderId: string | null;
|
|
73
|
+
fileBeingAnalyzed: boolean;
|
|
74
|
+
analysisProgress: import("@prisma/client/runtime/client").JsonValue | null;
|
|
75
|
+
needsAnalysis: boolean;
|
|
76
|
+
}>;
|
|
77
|
+
createFolder(userId: string, input: {
|
|
78
|
+
name: string;
|
|
79
|
+
color?: string;
|
|
80
|
+
parentId?: string;
|
|
81
|
+
}): Promise<{
|
|
82
|
+
id: string;
|
|
83
|
+
name: string;
|
|
84
|
+
createdAt: Date;
|
|
85
|
+
updatedAt: Date;
|
|
86
|
+
ownerId: string;
|
|
87
|
+
color: string;
|
|
88
|
+
markerColor: string | null;
|
|
89
|
+
parentId: string | null;
|
|
90
|
+
}>;
|
|
91
|
+
updateFolder(userId: string, input: {
|
|
92
|
+
id: string;
|
|
93
|
+
name?: string;
|
|
94
|
+
markerColor?: string | null;
|
|
95
|
+
}): Promise<{
|
|
96
|
+
id: string;
|
|
97
|
+
name: string;
|
|
98
|
+
createdAt: Date;
|
|
99
|
+
updatedAt: Date;
|
|
100
|
+
ownerId: string;
|
|
101
|
+
color: string;
|
|
102
|
+
markerColor: string | null;
|
|
103
|
+
parentId: string | null;
|
|
104
|
+
}>;
|
|
105
|
+
deleteFolder(userId: string, id: string): Promise<{
|
|
106
|
+
id: string;
|
|
107
|
+
name: string;
|
|
108
|
+
createdAt: Date;
|
|
109
|
+
updatedAt: Date;
|
|
110
|
+
ownerId: string;
|
|
111
|
+
color: string;
|
|
112
|
+
markerColor: string | null;
|
|
113
|
+
parentId: string | null;
|
|
114
|
+
}>;
|
|
115
|
+
get(userId: string, id: string): Promise<{
|
|
116
|
+
id: string;
|
|
117
|
+
createdAt: Date;
|
|
118
|
+
updatedAt: Date;
|
|
119
|
+
uploads: {
|
|
120
|
+
id: string;
|
|
121
|
+
name: string;
|
|
122
|
+
createdAt: Date;
|
|
123
|
+
size: number;
|
|
124
|
+
objectKey: string | null;
|
|
125
|
+
mimeType: string;
|
|
126
|
+
}[];
|
|
127
|
+
ownerId: string;
|
|
128
|
+
title: string;
|
|
129
|
+
description: string | null;
|
|
130
|
+
icon: string;
|
|
131
|
+
color: string;
|
|
132
|
+
markerColor: string | null;
|
|
133
|
+
folderId: string | null;
|
|
134
|
+
fileBeingAnalyzed: boolean;
|
|
135
|
+
analysisProgress: import("@prisma/client/runtime/client").JsonValue;
|
|
136
|
+
needsAnalysis: boolean;
|
|
137
|
+
folder: {
|
|
138
|
+
id: string;
|
|
139
|
+
name: string;
|
|
140
|
+
color: string;
|
|
141
|
+
} | null;
|
|
142
|
+
}>;
|
|
143
|
+
getAccountSummary(userId: string): Promise<import("../billing/usage.service.js").AccountSummary>;
|
|
144
|
+
getStats(userId: string): Promise<import("../billing/usage.service.js").AccountStats>;
|
|
145
|
+
update(userId: string, input: {
|
|
146
|
+
id: string;
|
|
147
|
+
name?: string;
|
|
148
|
+
description?: string;
|
|
149
|
+
markerColor?: string | null;
|
|
150
|
+
icon?: string;
|
|
151
|
+
}): Promise<{
|
|
152
|
+
id: string;
|
|
153
|
+
createdAt: Date;
|
|
154
|
+
updatedAt: Date;
|
|
155
|
+
ownerId: string;
|
|
156
|
+
title: string;
|
|
157
|
+
description: string | null;
|
|
158
|
+
icon: string;
|
|
159
|
+
color: string;
|
|
160
|
+
markerColor: string | null;
|
|
161
|
+
folderId: string | null;
|
|
162
|
+
fileBeingAnalyzed: boolean;
|
|
163
|
+
analysisProgress: import("@prisma/client/runtime/client").JsonValue | null;
|
|
164
|
+
needsAnalysis: boolean;
|
|
165
|
+
}>;
|
|
166
|
+
delete(userId: string, id: string): Promise<boolean>;
|
|
167
|
+
getFolderInformation(userId: string, id: string): Promise<{
|
|
168
|
+
folder: {
|
|
169
|
+
id: string;
|
|
170
|
+
name: string;
|
|
171
|
+
createdAt: Date;
|
|
172
|
+
updatedAt: Date;
|
|
173
|
+
ownerId: string;
|
|
174
|
+
color: string;
|
|
175
|
+
markerColor: string | null;
|
|
176
|
+
parentId: string | null;
|
|
177
|
+
};
|
|
178
|
+
parents: {
|
|
179
|
+
id: string;
|
|
180
|
+
name: string;
|
|
181
|
+
createdAt: Date;
|
|
182
|
+
updatedAt: Date;
|
|
183
|
+
ownerId: string;
|
|
184
|
+
color: string;
|
|
185
|
+
markerColor: string | null;
|
|
186
|
+
parentId: string | null;
|
|
187
|
+
}[];
|
|
188
|
+
}>;
|
|
189
|
+
getSharedWith(userId: string, id: string): Promise<{
|
|
190
|
+
shared: {
|
|
191
|
+
id: string;
|
|
192
|
+
createdAt: Date;
|
|
193
|
+
updatedAt: Date;
|
|
194
|
+
ownerId: string;
|
|
195
|
+
title: string;
|
|
196
|
+
description: string | null;
|
|
197
|
+
icon: string;
|
|
198
|
+
color: string;
|
|
199
|
+
markerColor: string | null;
|
|
200
|
+
folderId: string | null;
|
|
201
|
+
fileBeingAnalyzed: boolean;
|
|
202
|
+
analysisProgress: import("@prisma/client/runtime/client").JsonValue | null;
|
|
203
|
+
needsAnalysis: boolean;
|
|
204
|
+
}[];
|
|
205
|
+
invitations: ({
|
|
206
|
+
workspace: {
|
|
207
|
+
id: string;
|
|
208
|
+
createdAt: Date;
|
|
209
|
+
updatedAt: Date;
|
|
210
|
+
ownerId: string;
|
|
211
|
+
title: string;
|
|
212
|
+
description: string | null;
|
|
213
|
+
icon: string;
|
|
214
|
+
color: string;
|
|
215
|
+
markerColor: string | null;
|
|
216
|
+
folderId: string | null;
|
|
217
|
+
fileBeingAnalyzed: boolean;
|
|
218
|
+
analysisProgress: import("@prisma/client/runtime/client").JsonValue | null;
|
|
219
|
+
needsAnalysis: boolean;
|
|
220
|
+
};
|
|
221
|
+
} & {
|
|
222
|
+
id: string;
|
|
223
|
+
email: string;
|
|
224
|
+
createdAt: Date;
|
|
225
|
+
updatedAt: Date;
|
|
226
|
+
role: string;
|
|
227
|
+
token: string;
|
|
228
|
+
workspaceId: string;
|
|
229
|
+
invitedById: string;
|
|
230
|
+
acceptedAt: Date | null;
|
|
231
|
+
expiresAt: Date;
|
|
232
|
+
})[];
|
|
233
|
+
}>;
|
|
234
|
+
uploadFiles(userId: string, input: {
|
|
235
|
+
id: string;
|
|
236
|
+
files: Array<{
|
|
237
|
+
filename: string;
|
|
238
|
+
contentType: string;
|
|
239
|
+
size: number;
|
|
240
|
+
}>;
|
|
241
|
+
}): Promise<{
|
|
242
|
+
fileId: string;
|
|
243
|
+
uploadUrl: string;
|
|
244
|
+
}[]>;
|
|
245
|
+
deleteFiles(userId: string, input: {
|
|
246
|
+
fileId: string[];
|
|
247
|
+
id: string;
|
|
248
|
+
}): Promise<boolean>;
|
|
249
|
+
getFileUploadUrl(userId: string, input: {
|
|
250
|
+
workspaceId: string;
|
|
251
|
+
filename: string;
|
|
252
|
+
contentType: string;
|
|
253
|
+
size: number;
|
|
254
|
+
}): Promise<{
|
|
255
|
+
fileId: string;
|
|
256
|
+
uploadUrl: string;
|
|
257
|
+
}>;
|
|
258
|
+
search(userId: string, input: {
|
|
259
|
+
query: string;
|
|
260
|
+
color?: string;
|
|
261
|
+
limit: number;
|
|
262
|
+
}): Promise<any[]>;
|
|
263
|
+
}
|