From 27fdd1b6e6e50882ee9a17c85c5f6b845d4973ce Mon Sep 17 00:00:00 2001
From: Eddie Santos <9561596+eddie-santos@users.noreply.github.com>
Date: Sat, 7 Jun 2025 13:38:05 -0700
Subject: Add embedder (#818)

---
 packages/core/src/config/config.test.ts | 2 ++
 packages/core/src/config/config.ts      | 7 +++++++
 2 files changed, 9 insertions(+)

(limited to 'packages/core/src/config')

diff --git a/packages/core/src/config/config.test.ts b/packages/core/src/config/config.test.ts
index 411b124d..3800585d 100644
--- a/packages/core/src/config/config.test.ts
+++ b/packages/core/src/config/config.test.ts
@@ -48,9 +48,11 @@ describe('Server Config (config.ts)', () => {
   const USER_AGENT = 'ServerTestAgent/1.0';
   const USER_MEMORY = 'Test User Memory';
   const TELEMETRY = false;
+  const EMBEDDING_MODEL = 'gemini-embedding';
   const baseParams: ConfigParameters = {
     apiKey: API_KEY,
     model: MODEL,
+    embeddingModel: EMBEDDING_MODEL,
     sandbox: SANDBOX,
     targetDir: TARGET_DIR,
     debugMode: DEBUG_MODE,
diff --git a/packages/core/src/config/config.ts b/packages/core/src/config/config.ts
index 00b3e35d..75db970b 100644
--- a/packages/core/src/config/config.ts
+++ b/packages/core/src/config/config.ts
@@ -55,6 +55,7 @@ export class MCPServerConfig {
 export interface ConfigParameters {
   apiKey: string;
   model: string;
+  embeddingModel: string;
   sandbox: boolean | string;
   targetDir: string;
   debugMode: boolean;
@@ -84,6 +85,7 @@ export class Config {
   private toolRegistry: Promise<ToolRegistry>;
   private readonly apiKey: string;
   private readonly model: string;
+  private readonly embeddingModel: string;
   private readonly sandbox: boolean | string;
   private readonly targetDir: string;
   private readonly debugMode: boolean;
@@ -113,6 +115,7 @@ export class Config {
   constructor(params: ConfigParameters) {
     this.apiKey = params.apiKey;
     this.model = params.model;
+    this.embeddingModel = params.embeddingModel;
     this.sandbox = params.sandbox;
     this.targetDir = path.resolve(params.targetDir);
     this.debugMode = params.debugMode;
@@ -163,6 +166,10 @@ export class Config {
     return this.model;
   }
 
+  getEmbeddingModel(): string {
+    return this.embeddingModel;
+  }
+
   getSandbox(): boolean | string {
     return this.sandbox;
   }
-- 
cgit v1.2.3