wehub-resource-sync
28 lines
722 B
Python
28 lines
722 B
Python
from pymongo import MongoClient
|
|
from openai import OpenAI
|
|
import voyageai
|
|
from dotenv import load_dotenv
|
|
import os
|
|
|
|
# Load environment variables from .env file
|
|
load_dotenv()
|
|
|
|
# Environment variables (private)
|
|
MONGODB_URI = os.getenv("MONGODB_URI")
|
|
VOYAGE_API_KEY = os.getenv("VOYAGE_API_KEY")
|
|
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
|
|
|
|
# MongoDB cluster configuration
|
|
mongo_client = MongoClient(MONGODB_URI)
|
|
agent_db = mongo_client["ai_agent_db"]
|
|
vector_collection = agent_db["embeddings"]
|
|
memory_collection = agent_db["chat_history"]
|
|
|
|
# Model configuration
|
|
voyage_client = voyageai.Client(api_key=VOYAGE_API_KEY)
|
|
openai_client = OpenAI(api_key=OPENAI_API_KEY)
|
|
VOYAGE_MODEL = "voyage-3-large"
|
|
OPENAI_MODEL = "gpt-4o"
|
|
|
|
|