Spaces:
Sleeping
Sleeping
File size: 739 Bytes
f7b069f 3874ccc f7b069f f02ba16 f7b069f 3379ec0 f7b069f 3379ec0 f7b069f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 | from pymongo import MongoClient
from openai import OpenAI, AzureOpenAI
import os
# ======================================================
# CONFIGURATION
# ======================================================
MONGO_DATABASE_HOST=os.getenv('MONGO_DATABASE_HOST')
# MongoDB
client_mongo = MongoClient(MONGO_DATABASE_HOST)
db = client_mongo["douane_db"]
collection = db["douane_code"]
# Azure OpenAI
endpoint = os.getenv('endpoint')
embedding_model = "text-embedding-ada-002"
chat_model = "gpt-5-chat"
api_key = os.getenv('api_key')
api_version = "2024-12-01-preview"
client_embedding = OpenAI(base_url=f"{endpoint}openai/v1/", api_key=api_key)
client_chat = AzureOpenAI(api_version=api_version, azure_endpoint=endpoint, api_key=api_key)
|