Iyeoluwa Akinrinola
34 lines
944 B
Python
34 lines
944 B
Python
"""
|
|
Configuration settings for the AI service.
|
|
"""
|
|
|
|
import os
|
|
from dotenv import load_dotenv
|
|
|
|
# Load environment variables from .env file
|
|
import os.path
|
|
dotenv_path = os.path.join(os.path.dirname(__file__), '.env')
|
|
load_dotenv(dotenv_path=dotenv_path)
|
|
|
|
class Config:
|
|
"""Base configuration."""
|
|
|
|
# API configuration
|
|
API_HOST = os.environ.get('API_HOST', '0.0.0.0')
|
|
API_PORT = int(os.environ.get('API_PORT', 5252))
|
|
|
|
# OpenWebUI configuration
|
|
OPENWEBUI_URL = os.environ.get('OPENWEBUI_URL', 'http://104.225.217.215:8080')
|
|
OPENWEBUI_API_KEY = os.environ.get('OPENWEBUI_API_KEY', '')
|
|
|
|
# Ollama configuration
|
|
OLLAMA_API_URL = os.environ.get('OLLAMA_API_URL', 'http://127.0.0.1:11434')
|
|
DEFAULT_MODEL = os.environ.get('DEFAULT_MODEL', 'llama3.1')
|
|
|
|
# Document processing
|
|
CHUNK_SIZE = int(os.environ.get('CHUNK_SIZE', 1000))
|
|
CHUNK_OVERLAP = int(os.environ.get('CHUNK_OVERLAP', 200))
|
|
|
|
|
|
config = Config()
|