Skip to content

Commit dfde11d

Browse files
cursoragentenitrat
andcommitted
Refactor: Simplify embedding model selection and docker compose commands
Co-authored-by: msaug <[email protected]>
1 parent eb4bdd4 commit dfde11d

File tree

2 files changed

+5
-17
lines changed

2 files changed

+5
-17
lines changed

CLAUDE.md

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -52,12 +52,7 @@ uv run pytest -k "test_name" # Run specific test
5252

5353
```bash
5454
docker compose up postgres backend # Start main services
55-
56-
# Recommended: run ingester so compose exits when done
57-
docker compose up --abort-on-container-exit --exit-code-from ingester ingester
58-
59-
# Alternative: run Postgres in background, then run a one-shot ingester container
60-
docker compose up -d postgres && docker compose run --rm ingester
55+
docker compose up ingester # Run documentation ingestion
6156
```
6257

6358
## Architecture Overview

ingesters/src/generateEmbeddings.ts

Lines changed: 4 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -85,24 +85,17 @@ async function setupVectorStore(): Promise<VectorStore> {
8585
// Get database configuration
8686
const dbConfig = getVectorDbConfig();
8787

88-
// Try Gemini first, then OpenAI as fallback
89-
const geminiModels = await loadGeminiEmbeddingsModels();
90-
const openaiModels = await loadOpenAIEmbeddingsModels();
91-
const embeddingModel: Embeddings | undefined =
92-
(geminiModels['Gemini embedding 001'] as unknown as Embeddings) ||
93-
(openaiModels['Text embedding 3 large'] as unknown as Embeddings) ||
94-
(openaiModels['Text embedding 3 small'] as unknown as Embeddings);
88+
const embeddingModels = await loadGeminiEmbeddingsModels();
89+
const embeddingModel = embeddingModels['Gemini embedding 001'];
9590

9691
if (!embeddingModel) {
97-
throw new Error(
98-
'No embedding model configured. Set GEMINI_API_KEY or OPENAI_API_KEY.',
99-
);
92+
throw new Error('Text embedding 3 large model not found');
10093
}
10194

10295
// Initialize vector store
10396
vectorStore = await VectorStore.getInstance(
10497
dbConfig,
105-
embeddingModel,
98+
embeddingModel as unknown as Embeddings,
10699
);
107100
logger.info('VectorStore initialized successfully');
108101
return vectorStore;

0 commit comments

Comments
 (0)