Fix formatting

api/ingest.py

@@ -6,7 +6,7 @@
 
 from models.ingest import RequestPayload
 from service.embedding import EmbeddingService, get_encoder
-from service.ingest import handle_urls, handle_google_drive
+from service.ingest import handle_google_drive, handle_urls
 from utils.summarise import SUMMARY_SUFFIX
 
 router = APIRouter()

models/ingest.py

@@ -4,8 +4,8 @@
 from pydantic import BaseModel
 
 from models.file import File
-from models.vector_database import VectorDatabase
 from models.google_drive import GoogleDrive
+from models.vector_database import VectorDatabase
 
 
 class EncoderEnum(str, Enum):

service/code_interpreter.py

@@ -1,14 +1,14 @@
-import re
 import asyncio
 import logging
-import time
+import re
 import textwrap
+import time
 from typing import List
-from decouple import config
-from openai import AsyncOpenAI
-import pandas as pd
 
+import pandas as pd
+from decouple import config
 from e2b import Sandbox
+from openai import AsyncOpenAI
 
 logging.getLogger("e2b").setLevel(logging.INFO)
 

service/embedding.py

@@ -21,9 +21,9 @@
 from models.file import File
 from models.google_drive import GoogleDrive
 from models.ingest import Encoder, EncoderEnum
+from utils.file import get_file_extension_from_url
 from utils.logger import logger
 from utils.summarise import completion
-from utils.file import get_file_extension_from_url
 from vectordbs import get_vector_service
 
 

service/router.py

@@ -5,7 +5,6 @@
 
 from models.document import BaseDocumentChunk
 from models.query import RequestPayload
-
 from service.code_interpreter import CodeInterpreterService
 from service.embedding import get_encoder
 from utils.logger import logger
@@ -49,6 +48,7 @@ async def get_documents(
         async with code_interpreter as service:
             response = await service.run_python(code=code)
             output = response.stdout
+            print(output)
 
     reranked_chunks = await vector_service.rerank(query=payload.input, documents=chunks)
     return reranked_chunks

utils/file.py

@@ -1,5 +1,5 @@
-from urllib.parse import urlparse
 import os
+from urllib.parse import urlparse
 
 
 def get_file_extension_from_url(url: str) -> str: