mirror of
https://github.com/bendtherules/ask262.git
synced 2026-08-18 21:31:46 +00:00
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:
@@ -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"));
|
||||
|
||||
@@ -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> {
|
||||
|
||||
@@ -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
@@ -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({
|
||||
|
||||
Reference in New Issue
Block a user