Jayden/AFFiNE-Mirror
1
0
Fork 0
mirror of https://github.com/toeverything/AFFiNE.git synced 2026-02-23 09:17:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: improve context api (#10812)

Browse Source
This commit is contained in:
darkskygit
2025-03-14 05:57:40 +00:00
parent b8452f56a8
commit 92effd9b51
10 changed files with 46 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
packages/backend/server/src/plugins/copilot/context/session.ts
View File

@@ -6,8 +6,8 @@ import {
ChunkSimilarity,
ContextConfig,
ContextDoc,
ContextEmbedStatus,
ContextFile,
ContextFileStatus,
ContextList,
DocChunkSimilarity,
EmbeddingClient,
@@ -54,7 +54,7 @@ export class ContextSession implements AsyncDisposable {
if (doc) {
return doc;
}
const record = { id: docId, createdAt: Date.now() };
const record = { id: docId, createdAt: Date.now(), status: null };
this.config.docs.push(record);
await this.save();
return record;
@@ -76,7 +76,7 @@ export class ContextSession implements AsyncDisposable {
if (existsBlob) {
// use exists file id if the blob exists
// we assume that the file content pointed to by the same blobId is consistent.
if (existsBlob.status === ContextFileStatus.finished) {
if (existsBlob.status === ContextEmbedStatus.finished) {
return existsBlob;
}
fileId = existsBlob.id;
@@ -179,7 +179,7 @@ export class ContextSession implements AsyncDisposable {
if (file) {
Object.assign(file, cb({ ...file }));
} else {
const file = { id: fileId, status: ContextFileStatus.processing };
const file = { id: fileId, status: ContextEmbedStatus.processing };
files.push(cb(file));
}
await this.save(tx);