From 28105d351726392ad6e3354371f3e7c18f393aed Mon Sep 17 00:00:00 2001 From: Yi Ding <yi.s.ding@gmail.com> Date: Fri, 23 Jun 2023 09:46:11 -0700 Subject: [PATCH] service context copy --- packages/core/src/ServiceContext.ts | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/packages/core/src/ServiceContext.ts b/packages/core/src/ServiceContext.ts index 68b7c1fb2..d521dd093 100644 --- a/packages/core/src/ServiceContext.ts +++ b/packages/core/src/ServiceContext.ts @@ -31,3 +31,20 @@ export function serviceContextFromDefaults(options: ServiceContextOptions) { return serviceContext; } + +export function serviceContextFromServiceContext( + serviceContext: ServiceContext, + options: ServiceContextOptions +) { + const newServiceContext = { ...serviceContext }; + if (options.llmPredictor) { + newServiceContext.llmPredictor = options.llmPredictor; + } + if (options.embedModel) { + newServiceContext.embedModel = options.embedModel; + } + if (options.nodeParser) { + newServiceContext.nodeParser = options.nodeParser; + } + return newServiceContext; +} -- GitLab