chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,27 @@
|
||||
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"
|
||||
|
||||
|
||||
Reference in New Issue
Block a user