mirror of
https://github.com/bendtherules/ask262.git
synced 2026-08-18 13:21:55 +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,
|
createSectionRetrieverTool,
|
||||||
createSpecRetrieverTool,
|
createSpecRetrieverTool,
|
||||||
} from "./agent_tools";
|
} from "./agent_tools";
|
||||||
import { GRAPH_FILE, STORAGE_DIR } from "./constants";
|
import { EMBEDDING_MODEL, GRAPH_FILE, STORAGE_DIR } from "./constants";
|
||||||
|
|
||||||
const embeddings = new OllamaEmbeddings({
|
const embeddings = new OllamaEmbeddings({
|
||||||
model: "qwen3-embedding:0.6b",
|
model: EMBEDDING_MODEL,
|
||||||
});
|
});
|
||||||
|
|
||||||
const config = JSON.parse(fs.readFileSync("./config.json", "utf-8"));
|
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.
|
* 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";
|
const OLLAMA_HOST = process.env.OLLAMA_HOST || "http://localhost:11434";
|
||||||
|
|
||||||
export interface RerankResult<T> {
|
export interface RerankResult<T> {
|
||||||
|
|||||||
@@ -2,3 +2,7 @@ export const STORAGE_DIR = "./storage";
|
|||||||
export const SPEC_DIR = "./spec-built/multipage";
|
export const SPEC_DIR = "./spec-built/multipage";
|
||||||
export const CODE_DIR = "./engine262/src";
|
export const CODE_DIR = "./engine262/src";
|
||||||
export const GRAPH_FILE = "./graphology/graph.json";
|
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 * as cheerio from "cheerio";
|
||||||
import { glob } from "glob";
|
import { glob } from "glob";
|
||||||
import ora from "ora";
|
import ora from "ora";
|
||||||
import { SPEC_DIR, STORAGE_DIR } from "../constants";
|
import { EMBEDDING_MODEL, SPEC_DIR, STORAGE_DIR } from "../constants";
|
||||||
|
|
||||||
const embeddings = new OllamaEmbeddings({
|
const embeddings = new OllamaEmbeddings({
|
||||||
model: "qwen3-embedding:0.6b",
|
model: EMBEDDING_MODEL,
|
||||||
});
|
});
|
||||||
|
|
||||||
const htmlSplitter = new RecursiveCharacterTextSplitter({
|
const htmlSplitter = new RecursiveCharacterTextSplitter({
|
||||||
|
|||||||
Reference in New Issue
Block a user