23 lines
1.1 KiB
Python
23 lines
1.1 KiB
Python
import os
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
|
|
# LLM Configuration
|
|
llm_baseurl = os.getenv("LLM_BASE_URL", default="http://localhost:11434/v1" )
|
|
llm_model = os.getenv("LLM_MODEL", default="granite4.1:8b" )
|
|
llm_api_key = os.getenv("LLM_API_KEY", default="ollama" )
|
|
llm_timeout = int( os.getenv("LLM_TIMEOUT", default="600" ) )
|
|
# Agent Configuration
|
|
AGENT_MAX_ITERATIONS = int( os.getenv("AGENT_MAX_ITERATIONS", default="10" ) )
|
|
# Tool Configuration (for future use)
|
|
MAX_TOOL_OUTPUT = int( os.getenv("MAX_TOOL_OUTPUT", default="4000" ) )
|
|
# RAG Configuration
|
|
RAG_PERSIST_DIR = os.getenv("RAG_PERSIST_DIR", default="chroma_db" )
|
|
RAG_EMBEDDING_MODEL = os.getenv("RAG_EMBEDDING_MODEL", default="nomic-embed-text" )
|
|
RAG_COLLECTIONS = {
|
|
"food_recommendations": {
|
|
"description": "Menu makanan, preferensi pelanggan, data kuliner"
|
|
},
|
|
}
|