insertSegment method

Future<SegmentUpsertResultDto> insertSegment(
  1. InsertSegmentDto insertSegmentDto
)

Implementation

Future<SegmentUpsertResultDto> insertSegment(InsertSegmentDto insertSegmentDto) async {
  Completer<TokenUsage> completer = Completer<TokenUsage>();
  LLMSettings llmSettings = LLMSettings(llmConfig: insertSegmentDto.llmConfig.toModel(), listenToken: (TokenUsage tokenUsage) => completer.complete(tokenUsage));
  String segmentId = await _vdb.insertSegment(insertSegmentDto.docsId, insertSegmentDto.segment.toModel(), insertSegmentDto.index, llmSettings);
  TokenUsage tokenUsage = await completer.future;
  SegmentUpsertResultDto segmentUpsertResultDto = SegmentUpsertResultDto(segmentId: segmentId, tokenUsage: TokenUsageDto.fromModel(tokenUsage));
  return segmentUpsertResultDto;
}