File size: 533 Bytes
360b354
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
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"]