Skip to content

Commit d76a260

Browse files
committed
fix(ENV): use MODEL_ENV in test-generator
1 parent 21c1b98 commit d76a260

File tree

2 files changed

+3
-19
lines changed

2 files changed

+3
-19
lines changed

backend/docker-compose.yml

Lines changed: 0 additions & 17 deletions
This file was deleted.

backend/src/services/ai-test-generator.service.ts

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@ import type { OpenApiOperation, TestCase } from "../types/ai.types.js";
22
import { jsonrepair } from "jsonrepair";
33
import logger from "../lib/logger.js";
44
import axios from "axios";
5+
import { MODEL_ENV } from "../config/dotenv.js";
56
export async function generateTestCasesForEndpoint(
67
method: string,
78
path: string,
@@ -43,8 +44,8 @@ JSON Response:
4344

4445
try {
4546
logger.info(`Sending prompt to AI for ${method.toUpperCase()} ${path}`);
46-
const response = await axios.post("http://127.0.0.1:11434/api/generate", {
47-
model: "gemma2:2b",
47+
const response = await axios.post(`${MODEL_ENV.OLLAMA_HOST}/api/generate`, {
48+
model: MODEL_ENV.MODEL_NAME,
4849
prompt: prompt,
4950
stream: false,
5051
});

0 commit comments

Comments
 (0)