updateSegment method

Future<SegmentUpsertResultDto> updateSegment(
  1. UpdateSegmentDto updateSegmentDto
)

Implementation

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