refactor: move model info to constants

- Added EMBEDDING_MODEL and RERANKER_MODEL to `constants.ts`.
- Updated imports to use these constants in `agent.ts`, `agent_tools/reranker.ts`, and `setup/ingest.ts`.
- Replaced hard‑coded model strings with the new constants for Ollama embeddings and reranker.
This commit is contained in:
2026-04-01 18:31:30 +05:30
parent 0c45062348
commit f5d3884d5a
4 changed files with 10 additions and 5 deletions
+2 -2
View File
@@ -11,10 +11,10 @@ import {
createSectionRetrieverTool,
createSpecRetrieverTool,
} from "./agent_tools";
import { GRAPH_FILE, STORAGE_DIR } from "./constants";
import { EMBEDDING_MODEL, GRAPH_FILE, STORAGE_DIR } from "./constants";
const embeddings = new OllamaEmbeddings({
model: "qwen3-embedding:0.6b",
model: EMBEDDING_MODEL,
});
const config = JSON.parse(fs.readFileSync("./config.json", "utf-8"));
+2 -1
View File
@@ -3,7 +3,8 @@
* Uses Ollama's reranker API to score documents against a query.
*/
const RERANKER_MODEL = "dengcao/Qwen3-Reranker-0.6B:Q8_0";
import { RERANKER_MODEL } from "../constants";
const OLLAMA_HOST = process.env.OLLAMA_HOST || "http://localhost:11434";
export interface RerankResult<T> {
+4
View File
@@ -2,3 +2,7 @@ export const STORAGE_DIR = "./storage";
export const SPEC_DIR = "./spec-built/multipage";
export const CODE_DIR = "./engine262/src";
export const GRAPH_FILE = "./graphology/graph.json";
// Model configurations
export const EMBEDDING_MODEL = "qwen3-embedding:0.6b";
export const RERANKER_MODEL = "dengcao/Qwen3-Reranker-0.6B:Q8_0";
+2 -2
View File
@@ -9,10 +9,10 @@ import { RecursiveCharacterTextSplitter } from "@langchain/textsplitters";
import * as cheerio from "cheerio";
import { glob } from "glob";
import ora from "ora";
import { SPEC_DIR, STORAGE_DIR } from "../constants";
import { EMBEDDING_MODEL, SPEC_DIR, STORAGE_DIR } from "../constants";
const embeddings = new OllamaEmbeddings({
model: "qwen3-embedding:0.6b",
model: EMBEDDING_MODEL,
});
const htmlSplitter = new RecursiveCharacterTextSplitter({