Made base_api_url an environment variable

README.md

@@ -18,8 +18,10 @@ The aim of query-tool-ai would be to make this search process more user-friendly
   ### Set the environment variables - 
    | Environment Variable   | Type    | Required                                 | Default Value if Not Set | Example                                                   |
    | ---------------------- | ------- | ---------------------------------------- | ------------------------ | --------------------------------------------------------- |
+   | `BASE_API_URL`                 | string  | Yes                                       | `None`                | `https://api.neurobagel.org/query/?`                                               |
    | `HOST`                 | string  | No                                       | `0.0.0.0`                | `127.0.0.1`                                               |
    | `PORT`                 | integer | No                                       | `8000`                   | `8080`                                                    |
+
 
   ### Option 1 : Docker :
   - #### First, [install Docker](https://docs.docker.com/get-docker/).

app/api/url_generator.py

@@ -1,4 +1,6 @@
 import json
+import os
+from dotenv import load_dotenv
 from app.llm_processing.extractions import extract_information
 from app.api.validators import (
     validate_age_order,
@@ -11,6 +13,8 @@
     get_image_modality_termURL,
 )
 
+load_dotenv()
+
 
 def get_api_url(user_query: str) -> str:
     """
@@ -22,7 +26,7 @@ def get_api_url(user_query: str) -> str:
     Returns:
         str: The constructed API URL.
     """
-    base_api_url = "https://api.neurobagel.org/query/?"
+    base_api_url = os.getenv("BASE_API_URL", "None")
     llm_response = extract_information(user_query)
 
     try:

entrypoint.sh

@@ -2,6 +2,7 @@
 
 # Start the Ollama services first
 ollama serve &
+ollama pull mistral &
 ollama run mistral &
 
 # Start the FastAPI application