Revert mock-llm registry to return the same instance

TrafficGuard · Dec 5, 2024 · 93ce2c2 · 93ce2c2
1 parent 1124c8c
commit 93ce2c2
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/frontend/src/app/modules/agents/services/llm.service.spec.ts b/frontend/src/app/modules/agents/services/llm.service.spec.ts
@@ -1,10 +1,9 @@
 import { TestBed } from '@angular/core/testing';
 import { HttpClientTestingModule, HttpTestingController } from '@angular/common/http/testing';
 import { LlmService, LLM } from './llm.service';
-import {environment} from "../../../../environments/environment";
 
 
-const LLM_LIST_API_URL = `${environment.apiBaseUrl}llms/list`;
+const LLM_LIST_API_URL = `/api/llms/list`;
 
 describe('LlmService', () => {
   let service: LlmService;

diff --git a/src/llm/services/mock-llm.ts b/src/llm/services/mock-llm.ts
@@ -121,14 +121,15 @@ export class MockLLM extends BaseLLM {
 	}
 }
 
+export const mockLLM = new MockLLM();
+
 export function mockLLMRegistry(): Record<string, () => LLM> {
 	return {
-		'mock:mock': () => new MockLLM(),
+		// Tests need the same instance returned
+		'mock:mock': () => mockLLM,
 	};
 }
 
-export const mockLLM = new MockLLM();
-
 export function mockLLMs(): AgentLLMs {
 	return {
 		easy: mockLLM,