feat: Support Fireworks embedding for ingest and tool

This commit is contained in:
2026-04-16 13:06:25 +05:30
parent a46d764336
commit 385f609e7b
13 changed files with 411 additions and 41 deletions
+2 -2
View File
@@ -14,7 +14,7 @@
import * as lancedbSdk from "@lancedb/lancedb";
import { OllamaEmbeddings } from "@langchain/ollama";
import { createSearchSpecSectionsTool } from "../../agent-tools/index.js";
import { EMBEDDING_MODEL, STORAGE_DIR } from "../../constants.js";
import { OLLAMA_EMBEDDING_MODEL, STORAGE_DIR } from "../../constants.js";
async function main() {
// Get query from command line or use default
@@ -28,7 +28,7 @@ async function main() {
try {
const embeddings = new OllamaEmbeddings({
model: EMBEDDING_MODEL,
model: OLLAMA_EMBEDDING_MODEL,
});
const db = await lancedbSdk.connect(STORAGE_DIR);
+2 -2
View File
@@ -21,10 +21,10 @@ import type { Table } from "@lancedb/lancedb";
import * as lancedbSdk from "@lancedb/lancedb";
import { OllamaEmbeddings } from "@langchain/ollama";
import { Command } from "commander";
import { EMBEDDING_MODEL, STORAGE_DIR } from "../../constants.js";
import { OLLAMA_EMBEDDING_MODEL, STORAGE_DIR } from "../../constants.js";
const embeddings = new OllamaEmbeddings({
model: EMBEDDING_MODEL,
model: OLLAMA_EMBEDDING_MODEL,
});
interface DocumentRecord {