feat(server): embedding search tool (#12810)

packages/backend/server/src/plugins/copilot/embedding/client.ts

@@ -1,4 +1,5 @@
 import { Logger } from '@nestjs/common';
+import type { ModuleRef } from '@nestjs/core';
 
 import {
   CopilotPromptNotFound,
@@ -193,12 +194,16 @@ class ProductionEmbeddingClient extends EmbeddingClient {
 
 let EMBEDDING_CLIENT: EmbeddingClient | undefined;
 export async function getEmbeddingClient(
-  providerFactory: CopilotProviderFactory,
-  prompt: PromptService
+  moduleRef: ModuleRef
 ): Promise<EmbeddingClient | undefined> {
   if (EMBEDDING_CLIENT) {
     return EMBEDDING_CLIENT;
   }
+  const providerFactory = moduleRef.get(CopilotProviderFactory, {
+    strict: false,
+  });
+  const prompt = moduleRef.get(PromptService, { strict: false });
+
   const client = new ProductionEmbeddingClient(providerFactory, prompt);
   if (await client.configured()) {
     EMBEDDING_CLIENT = client;

packages/backend/server/src/plugins/copilot/embedding/job.ts

@@ -1,4 +1,5 @@
 import { Injectable } from '@nestjs/common';
+import { ModuleRef } from '@nestjs/core';
 
 import {
   AFFiNELogger,
@@ -14,8 +15,6 @@ import {
 } from '../../../base';
 import { DocReader } from '../../../core/doc';
 import { Models } from '../../../models';
-import { PromptService } from '../prompt';
-import { CopilotProviderFactory } from '../providers';
 import { CopilotStorage } from '../storage';
 import { readStream } from '../utils';
 import { getEmbeddingClient } from './client';
@@ -31,12 +30,11 @@ export class CopilotEmbeddingJob {
   private client: EmbeddingClient | undefined;
 
   constructor(
+    private readonly moduleRef: ModuleRef,
     private readonly doc: DocReader,
     private readonly event: EventBus,
     private readonly logger: AFFiNELogger,
     private readonly models: Models,
-    private readonly providerFactory: CopilotProviderFactory,
-    private readonly prompt: PromptService,
     private readonly queue: JobQueue,
     private readonly storage: CopilotStorage
   ) {
@@ -57,7 +55,7 @@ export class CopilotEmbeddingJob {
     this.supportEmbedding =
       await this.models.copilotContext.checkEmbeddingAvailable();
     if (this.supportEmbedding) {
-      this.client = await getEmbeddingClient(this.providerFactory, this.prompt);
+      this.client = await getEmbeddingClient(this.moduleRef);
     }
   }