Merge branch 'main' of https://github.com/Aws-killer/Trasnscription

App/Embedding/utils/Initialize.py

@@ -4,7 +4,11 @@ from langchain.vectorstores import MongoDBAtlasVectorSearch
 from langchain.vectorstores import FAISS
 from pymongo import MongoClient
 from motor.motor_asyncio import AsyncIOMotorClient
+import os
 
+
+completion_base=os.environ.get("completion_base")
+openai_api_key=os.environ.get("openai_api_key")
 template = """### Given the following context 
 ### Context
 {context}
@@ -14,7 +18,7 @@ template = """### Given the following context
 
 
 async def fetch_data(question, context):
-    url = "https://chimeragpt.adventblocks.cc/api/v1/chat/completions"
+    url = completion_base
 
     payload = json.dumps(
         {
@@ -39,7 +43,7 @@ async def fetch_data(question, context):
 
     headers = {
         "Content-Type": "application/json",
-        "Authorization": "Bearer 7j3MQ912kGjUx8AUKHUiFLC7MCWBPT_cmlJRVdYGUtE",
+        "Authorization": f"Bearer {openai_api_key}",
     }
 
     async with aiohttp.ClientSession() as session:
@@ -50,7 +54,7 @@ async def fetch_data(question, context):
 
 async def delete_documents(task_id):
     client = AsyncIOMotorClient(
-        "mongodb+srv://pMBONEA:[email protected]/?retryWrites=true&w=majority"
+       mongoDB
     )
     db = client["transcriptions"]
     collection = db["videos"]
@@ -60,7 +64,7 @@ async def delete_documents(task_id):
 
 
 # mongo_client = MongoClient(
-#     "mongodb+srv://pMBONEA:[email protected]/?retryWrites=true&w=majority"
+#    mongoDB
 # )
 # model_name = "BAAI/bge-base-en"
 # collection = mongo_client["transcriptions"]["videos"]
@@ -91,7 +95,7 @@ def generateChunks(chunks, task_id, n=100):
 
 def search(query: str, task_id: str):
     mongo_client = MongoClient(
-        "mongodb+srv://pMBONEA:[email protected]/?retryWrites=true&w=majority"
+       mongoDB
     )
     model_name = "BAAI/bge-base-en"
     collection = mongo_client["transcriptions"]["videos"]
@@ -124,7 +128,7 @@ def search(query: str, task_id: str):
 
 def encode(temp: list[Document]):
     mongo_client = MongoClient(
-        "mongodb+srv://pMBONEA:[email protected]/?retryWrites=true&w=majority"
+       mongoDB
     )
     model_name = "BAAI/bge-base-en"
     collection = mongo_client["transcriptions"]["videos"]