Analytics-Bot / src /utils.py
Viraj2307
Initial Commit
360b354
raw
history blame
533 Bytes
from pymongo import MongoClient
import sys
from config import settings
import ollama
from llama_index.llms.ollama import Ollama
sys.path.append("..")
client = MongoClient(settings.MONGODB_URI)
db = client[settings.DB_NAME]
client2 = ollama.Client(host=settings.HOST_URI)
model_name = settings.MODEL_NAME
llm = Ollama(
model="llama3.1", request_timeout=120, base_url=settings.HOST_URI, temperature=0
)
dev_url = settings.DEV_URI
orders_collection = db["orders"]
stores_collection = db["stores"]
brands_collection = db["brands"]