Skip to content

Commit 60737fc

Browse files
committed
Updating embedding service to load in openai key
1 parent 4b7d240 commit 60737fc

File tree

3 files changed

+9
-11
lines changed

3 files changed

+9
-11
lines changed

backend/docker-compose.yml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,8 @@ services:
1414
environment:
1515
- ENVIRONMENT=dev
1616
- TESTING=0
17+
env_file:
18+
- .env
1719
# Add healthcheck for better orchestration
1820
healthcheck:
1921
test: ["CMD", "curl", "-f", "http://localhost:8000/health"]

backend/pyproject.toml

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -65,8 +65,6 @@ dependencies = [
6565
"pydantic-settings>=2.5.2",
6666
"pydantic_core>=2.23.3",
6767
"pymilvus>=2.4.6",
68-
"PyMuPDF>=1.24.10",
69-
"PyMuPDFb>=1.24.10",
7068
"pypdf>=5.0.0",
7169
"PyPDF2>=3.0.1",
7270
"python-dateutil>=2.9.0",

backend/src/app/services/embedding/openai_embedding_service.py

Lines changed: 7 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -16,15 +16,13 @@ class OpenAIEmbeddingService(EmbeddingService):
1616

1717
def __init__(self, settings: Settings) -> None:
1818
self.settings = settings
19-
if settings.openai_api_key:
20-
self.embeddings = OpenAIEmbeddings(
21-
model=self.settings.embedding_model
22-
)
23-
else:
24-
self.embeddings = None # type: ignore
25-
logger.warning(
26-
"OpenAI API key is not set. LLM features will be disabled."
27-
)
19+
if not settings.openai_api_key:
20+
raise ValueError("OpenAI API key is required but not set")
21+
22+
self.embeddings = OpenAIEmbeddings(
23+
api_key=settings.openai_api_key,
24+
model=settings.embedding_model
25+
)
2826

2927
async def get_embeddings(self, texts: List[str]) -> List[List[float]]:
3028
"""Get embeddings for text."""

0 commit comments

Comments
 (0)