mirror of
https://github.com/bendtherules/ask262.git
synced 2026-08-19 00:31:06 +00:00
feat: Support Fireworks embedding for ingest and tool
This commit is contained in:
@@ -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);
|
||||
|
||||
@@ -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 {
|
||||
|
||||
Reference in New Issue
Block a user