Misc changes

api/docs.py

@@ -1,7 +1,7 @@
 import asyncio
 from pathlib import Path
 import traceback
-from typing import Dict, List, Literal, Tuple
+from typing import Literal, Tuple
 from fastapi.routing import APIRouter
 import logging
 import io
@@ -9,7 +9,6 @@ import zipfile
 import os
 from httpx import AsyncClient
 from pydantic import BaseModel
-import requests
 import subprocess
 import pandas as pd
 import re

app.py

@@ -1,21 +1,16 @@
-import asyncio
 import logging
 from dotenv import load_dotenv
-from typing import Literal
 from jinja2 import Environment, TemplateNotFound
 import warnings
 import os
-from fastapi import Depends, FastAPI, BackgroundTasks, HTTPException, Request, Response
+from fastapi import Depends, FastAPI, Response
 from fastapi.staticfiles import StaticFiles
 import api.solutions
-from dependencies import get_llm_router, get_prompt_templates, init_dependencies
+from dependencies import get_prompt_templates, init_dependencies
 import api.docs
 import api.requirements
-from api.docs import doc_to_txt
 from schemas import *
 from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import FileResponse, StreamingResponse
-from litellm.router import Router
 
 load_dotenv()
 
@@ -26,6 +21,8 @@ logging.basicConfig(
     datefmt='%Y-%m-%d %H:%M:%S'
 )
 
+logging.info("DEBUG logging is disabled. Set `DEBUG_LOG` env var to 1 to enable debug logging.")
+
 # Initialize global dependencies
 init_dependencies()
 
@@ -41,11 +38,14 @@ app.include_router(api.requirements.router, prefix="/requirements")
 app.include_router(api.solutions.router, prefix="/solutions")
 
 # INTERNAL ROUTE TO RETRIEVE PROMPT TEMPLATES FOR PRIVATE COMPUTE
+
+
 @app.get("/prompt/{task}", include_in_schema=True)
 async def retrieve_prompt(task: str, prompt_env: Environment = Depends(get_prompt_templates)):
     """Retrieves a prompt for client-side private inference"""
     try:
-        logging.debug(f"Retrieving template for on device private task {task}.")
+        logging.debug(
+            f"Retrieving template for on device private task {task}.")
         prompt, filename, _ = prompt_env.loader.get_source(
             prompt_env,  f"private/{task}.txt")
         return prompt

dependencies.py

@@ -10,7 +10,6 @@ from jinja2 import Environment, StrictUndefined, FileSystemLoader
 
 
 INSIGHT_FINDER_BASE_URL = "https://organizedprogrammers-insight-finder.hf.space/"
-DOC_FINDER_BASE_URL = "https://organizedprogrammers-docfinder.hf.space/"
 
 
 def init_dependencies():