diff --git a/services/rag/langchain/agent.py b/services/rag/langchain/agent.py index b8a2bdd..13ad327 100644 --- a/services/rag/langchain/agent.py +++ b/services/rag/langchain/agent.py @@ -2,6 +2,7 @@ import os from typing import List, Dict, Any, Optional, Tuple +from dotenv import load_dotenv from langchain_core.tools import BaseTool, tool from langchain_core.runnables import RunnableConfig from langchain_core.messages import HumanMessage, AIMessage, BaseMessage @@ -15,6 +16,9 @@ from loguru import logger from retrieval import create_retriever from vector_storage import initialize_vector_store +# Load environment variables +load_dotenv() + def get_llm_model_info(llm_model: str = None) -> Tuple[str, str, str, str, str]: """ diff --git a/services/rag/langchain/cli.py b/services/rag/langchain/cli.py index 9e3e147..5017ee4 100644 --- a/services/rag/langchain/cli.py +++ b/services/rag/langchain/cli.py @@ -1,9 +1,13 @@ import os from pathlib import Path +from dotenv import load_dotenv import click from loguru import logger +# Load environment variables +load_dotenv() + # Configure logging to output to both file and stdout as specified in requirements def setup_logging(): diff --git a/services/rag/langchain/enrichment.py b/services/rag/langchain/enrichment.py index 37efe7c..072f3b6 100644 --- a/services/rag/langchain/enrichment.py +++ b/services/rag/langchain/enrichment.py @@ -4,6 +4,7 @@ import os import hashlib from pathlib import Path from typing import List, Dict, Any +from dotenv import load_dotenv from langchain_core.documents import Document from langchain_text_splitters import RecursiveCharacterTextSplitter from langchain_community.document_loaders import PyPDFLoader @@ -38,6 +39,9 @@ from sqlalchemy.orm import sessionmaker from loguru import logger import sqlite3 +# Load environment variables +load_dotenv() + # Define the path to the data directory DATA_DIR = Path("../../../data").resolve() diff --git a/services/rag/langchain/retrieval.py b/services/rag/langchain/retrieval.py index 87431a6..8ef228b 100644 --- a/services/rag/langchain/retrieval.py +++ b/services/rag/langchain/retrieval.py @@ -2,6 +2,7 @@ import os from typing import List, Optional +from dotenv import load_dotenv from langchain_core.retrievers import BaseRetriever from langchain_core.callbacks import CallbackManagerForRetrieverRun from langchain_core.documents import Document @@ -9,6 +10,9 @@ from loguru import logger from vector_storage import initialize_vector_store +# Load environment variables +load_dotenv() + class VectorStoreRetriever(BaseRetriever): """ diff --git a/services/rag/langchain/server.py b/services/rag/langchain/server.py index ea58191..37cae8e 100644 --- a/services/rag/langchain/server.py +++ b/services/rag/langchain/server.py @@ -4,6 +4,7 @@ import json import os from contextlib import asynccontextmanager from typing import Any, Dict +from dotenv import load_dotenv from fastapi import FastAPI, HTTPException from fastapi.middleware.cors import CORSMiddleware @@ -12,6 +13,9 @@ from pydantic import BaseModel from agent import chat_with_agent, get_llm_model_info +# Load environment variables +load_dotenv() + class QueryRequest(BaseModel): """Request model for the query endpoint."""