feat: improve context api (#10812)

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

View File

@@ -41,8 +41,8 @@ import { CopilotContextDocJob } from './job';
import { CopilotContextService } from './service';
import {
ContextDoc,
ContextEmbedStatus,
type ContextFile,
ContextFileStatus,
DocChunkSimilarity,
FileChunkSimilarity,
MAX_EMBEDDABLE_SIZE,
@@ -94,13 +94,16 @@ export class CopilotContextType {
workspaceId!: string;
}
registerEnumType(ContextFileStatus, { name: 'ContextFileStatus' });
registerEnumType(ContextEmbedStatus, { name: 'ContextEmbedStatus' });
@ObjectType()
class CopilotContextDoc implements ContextDoc {
@Field(() => ID)
id!: string;
@Field(() => ContextEmbedStatus, { nullable: true })
status!: ContextEmbedStatus | null;
@Field(() => SafeIntResolver)
createdAt!: number;
}
@@ -116,8 +119,8 @@ class CopilotContextFile implements ContextFile {
@Field(() => SafeIntResolver)
chunkSize!: number;
@Field(() => ContextFileStatus)
status!: ContextFileStatus;
@Field(() => ContextEmbedStatus)
status!: ContextEmbedStatus;
@Field(() => String, { nullable: true })
error!: string | null;