khalednabawi11 commited on
Commit
0074dc2
·
verified ·
1 Parent(s): 6f7ac23

Update app/main.py

Browse files
Files changed (1) hide show
  1. app/main.py +4 -2
app/main.py CHANGED
@@ -272,6 +272,7 @@ import asyncio
272
  import logging
273
  import signal
274
  import uvicorn
 
275
 
276
  from fastapi import FastAPI, Request, HTTPException, status
277
  from pydantic import BaseModel, Field
@@ -289,9 +290,10 @@ from contextlib import asynccontextmanager
289
  MODEL_NAME = "FreedomIntelligence/Apollo-7B"
290
  MODEL_FILE = "Apollo-7B.Q4_K_S.gguf"
291
  EMBEDDING_MODEL = "Omartificial-Intelligence-Space/GATE-AraBert-v1"
292
- QDRANT_URL = "https://12efeef2-9f10-4402-9deb-f070977ddfc8.eu-central-1-0.aws.cloud.qdrant.io:6333"
293
- QDRANT_API_KEY = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJhY2Nlc3MiOiJtIn0.Jb39rYQW2rSE9RdXrjdzKY6T1RF44XjdQzCvzFkjat4"
294
  COLLECTION_NAME = "arabic_rag_collection"
 
 
 
295
 
296
  # === LOGGING === #
297
  logging.basicConfig(level=logging.DEBUG)
 
272
  import logging
273
  import signal
274
  import uvicorn
275
+ import os
276
 
277
  from fastapi import FastAPI, Request, HTTPException, status
278
  from pydantic import BaseModel, Field
 
290
  MODEL_NAME = "FreedomIntelligence/Apollo-7B"
291
  MODEL_FILE = "Apollo-7B.Q4_K_S.gguf"
292
  EMBEDDING_MODEL = "Omartificial-Intelligence-Space/GATE-AraBert-v1"
 
 
293
  COLLECTION_NAME = "arabic_rag_collection"
294
+ QDRANT_URL = os.getenv("QDRANT_URL", "https://12efeef2-9f10-4402-9deb-f070977ddfc8.eu-central-1-0.aws.cloud.qdrant.io:6333")
295
+ QDRANT_API_KEY = os.getenv("QDRANT_API_KEY", "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJhY2Nlc3MiOiJtIn0.Jb39rYQW2rSE9RdXrjdzKY6T1RF44XjdQzCvzFkjat4")
296
+
297
 
298
  # === LOGGING === #
299
  logging.basicConfig(level=logging.DEBUG)