adds metadata

app.py

@@ -1,10 +1,11 @@
 import nh3
-from fastapi import FastAPI, Request
+from fastapi import FastAPI, HTTPException
 from doc_searcher import DocSearcher
 from suggestion_searcher import SuggestionSearcher
 from huggingface_hub import login
-from config import HUGGING_FACE_API_KEY, COLLECTION_NAME, API_KEY, COLLECTION_NAME_SUGGESTION
+from config import HUGGING_FACE_API_KEY, COLLECTION_NAME, API_KEY, COLLECTION_NAME_SUGGESTION, METADATA_API
 from fastapi.responses import JSONResponse
+import httpx
 
 login(HUGGING_FACE_API_KEY)
 
@@ -29,6 +30,15 @@ async def get_suggestions(q: str, type: int):
     data = await suggestion_searcher.search(text=xss,type=type)
     return data
 
+@app.get("/api/metadata")
+async def get_suggestions(id: str):
+    url = f"{METADATA_API}{id}"
+    async with httpx.AsyncClient() as client:
+        resp = await client.get(url, params=None, timeout=10.0)
+    if resp.status_code != 200:
+        raise HTTPException(status_code=resp.status_code, detail=resp.text)
+    return resp.json()
+
 
 # @app.middleware("http")
 # async def api_key_authentication(request: Request, call_next):

config.py

@@ -17,4 +17,6 @@ SPARSE_MODEL = os.getenv('SPARSE_MODEL')
  
 LATE_INTERACTION_MODEL = os.getenv('LATE_INTERACTION_MODEL')
 
-API_KEY = os.getenv('API_KEY')
+API_KEY = os.getenv('API_KEY')
+
+METADATA_API = os.getenv('METADATA_API')