chore: format imports (#22)

Co-authored-by: leoguillaume <[email protected]>

app/endpoints/chat.py

@@ -1,17 +1,17 @@
-import httpx
 import json
 from typing import Union
 
-from fastapi import APIRouter, Security, HTTPException
+from fastapi import APIRouter, HTTPException, Security
 from fastapi.responses import StreamingResponse
+import httpx
 
-from app.schemas.chat import ChatCompletionRequest, ChatCompletion, ChatCompletionChunk
-from app.utils.security import check_api_key
-from app.utils.lifespan import clients
-from app.utils.config import LOGGER
+from app.schemas.chat import ChatCompletion, ChatCompletionChunk, ChatCompletionRequest
+from app.schemas.config import LANGUAGE_MODEL_TYPE
 from app.tools import *
 from app.tools import __all__ as tools_list
-from app.schemas.config import LANGUAGE_MODEL_TYPE
+from app.utils.config import LOGGER
+from app.utils.lifespan import clients
+from app.utils.security import check_api_key
 
 router = APIRouter()
 

app/endpoints/chunks.py

@@ -1,10 +1,10 @@
 from fastapi import APIRouter, Security
 from qdrant_client.http.models import Filter, HasIdCondition
 
-from app.schemas.chunks import Chunks, Chunk, ChunkRequest
-from app.utils.security import check_api_key
-from app.utils.lifespan import clients
 from app.helpers._vectorstore import VectorStore
+from app.schemas.chunks import Chunk, ChunkRequest, Chunks
+from app.utils.lifespan import clients
+from app.utils.security import check_api_key
 
 router = APIRouter()
 

app/endpoints/collections.py

@@ -1,13 +1,13 @@
-from typing import Union, Optional
+from typing import Optional, Union
 
-from fastapi import APIRouter, Security, Response
+from fastapi import APIRouter, Response, Security
 
-from app.schemas.collections import Collections, Collection
-from app.utils.security import check_api_key
-from app.utils.lifespan import clients
-from app.utils.data import delete_contents
-from app.utils.config import LOGGER
 from app.helpers import VectorStore
+from app.schemas.collections import Collection, Collections
+from app.utils.config import LOGGER
+from app.utils.data import delete_contents
+from app.utils.lifespan import clients
+from app.utils.security import check_api_key
 
 router = APIRouter()
 

app/endpoints/completions.py

@@ -4,7 +4,6 @@
 from app.utils.lifespan import clients
 from app.utils.security import check_api_key
 
-
 router = APIRouter()
 
 

app/endpoints/embeddings.py

@@ -1,9 +1,9 @@
-from fastapi import APIRouter, Security, HTTPException
+from fastapi import APIRouter, HTTPException, Security
 
-from app.schemas.embeddings import EmbeddingsRequest, Embeddings
+from app.schemas.config import EMBEDDINGS_MODEL_TYPE
+from app.schemas.embeddings import Embeddings, EmbeddingsRequest
 from app.utils.lifespan import clients
 from app.utils.security import check_api_key
-from app.schemas.config import EMBEDDINGS_MODEL_TYPE
 
 router = APIRouter()
 

app/endpoints/files.py

@@ -1,6 +1,6 @@
 import base64
-import uuid
 from typing import List, Optional, Union
+import uuid
 
 from botocore.exceptions import ClientError
 from fastapi import APIRouter, HTTPException, Response, Security, UploadFile

app/endpoints/models.py

@@ -1,12 +1,11 @@
-from typing import Union, Optional
+from typing import Optional, Union
 
 from fastapi import APIRouter, Security
 
 from app.schemas.models import Model, Models
 from app.utils.lifespan import clients
 from app.utils.security import check_api_key
 
-
 router = APIRouter()
 
 

app/endpoints/search.py

@@ -1,7 +1,7 @@
 from fastapi import APIRouter, Security
 
 from app.helpers import VectorStore
-from app.schemas.search import SearchRequest, Searches
+from app.schemas.search import Searches, SearchRequest
 from app.utils.lifespan import clients
 from app.utils.security import check_api_key
 

app/endpoints/tools.py

@@ -1,9 +1,9 @@
 from fastapi import APIRouter, Security
 
 from app.schemas.tools import Tools
-from app.utils.security import check_api_key
 from app.tools import *
 from app.tools import __all__ as tools_list
+from app.utils.security import check_api_key
 
 router = APIRouter()
 

app/helpers/_gristkeymanager.py

@@ -1,7 +1,7 @@
 import datetime as dt
 
-from grist_api import GristDocAPI
 from fastapi import HTTPException
+from grist_api import GristDocAPI
 from redis import Redis
 
 

app/helpers/_universalparser.py

@@ -2,14 +2,15 @@
 from typing import List, Optional
 
 from docx import Document
+from langchain.docstore.document import Document as LangchainDocument
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain_community.document_loaders import PDFMinerLoader
-from langchain.docstore.document import Document as LangchainDocument
 import magic
 
-from ._textcleaner import TextCleaner
 from app.schemas.files import JsonFile
 
+from ._textcleaner import TextCleaner
+
 
 class UniversalParser:
     DOCX_TYPE = "application/vnd.openxmlformats-officedocument.wordprocessingml.document"

app/schemas/chunks.py

@@ -1,4 +1,4 @@
-from typing import Literal, List
+from typing import List, Literal
 
 from pydantic import BaseModel
 

app/schemas/completions.py

@@ -1,7 +1,7 @@
-from typing import Optional, List, Union, Dict, Iterable
+from typing import Dict, Iterable, List, Optional, Union
 
-from pydantic import BaseModel, Field
 from openai.types import Completion
+from pydantic import BaseModel, Field
 
 
 class CompletionRequest(BaseModel):

app/schemas/embeddings.py

@@ -1,7 +1,7 @@
-from typing import Optional, Literal, List, Union
+from typing import List, Literal, Optional, Union
 
-from pydantic import BaseModel
 from openai.types import CreateEmbeddingResponse
+from pydantic import BaseModel
 
 
 class EmbeddingsRequest(BaseModel):

app/schemas/files.py

@@ -1,4 +1,4 @@
-from typing import Literal, List, Optional, Dict
+from typing import Dict, List, Literal, Optional
 from uuid import UUID
 
 from pydantic import BaseModel

app/schemas/models.py

@@ -1,9 +1,9 @@
-from typing import Literal, List
+from typing import List, Literal
 
-from pydantic import BaseModel
 from openai.types import Model
+from pydantic import BaseModel
 
-from app.schemas.config import LANGUAGE_MODEL_TYPE, EMBEDDINGS_MODEL_TYPE
+from app.schemas.config import EMBEDDINGS_MODEL_TYPE, LANGUAGE_MODEL_TYPE
 
 
 class Model(Model):

app/schemas/tools.py

@@ -1,4 +1,4 @@
-from typing import Literal, List
+from typing import List, Literal
 
 from pydantic import BaseModel
 

app/tests/conftest.py

@@ -1,6 +1,7 @@
+import logging
+
 import pytest
 import requests
-import logging
 
 
 def pytest_addoption(parser):

app/tests/test_files.py

@@ -1,12 +1,13 @@
-import os
 import logging
+import os
+
 import pytest
 import wget
 
-from app.schemas.files import Upload, Uploads, Files, File
-from app.schemas.collections import Collections, Collection
-from app.schemas.models import Models, Model
-from app.schemas.config import PRIVATE_COLLECTION_TYPE, METADATA_COLLECTION, LANGUAGE_MODEL_TYPE, EMBEDDINGS_MODEL_TYPE
+from app.schemas.collections import Collection, Collections
+from app.schemas.config import EMBEDDINGS_MODEL_TYPE, LANGUAGE_MODEL_TYPE, METADATA_COLLECTION, PRIVATE_COLLECTION_TYPE
+from app.schemas.files import File, Files, Upload, Uploads
+from app.schemas.models import Model, Models
 from app.utils.security import encode_string
 
 

app/tests/test_tools.py

@@ -1,6 +1,6 @@
 import pytest
 
-from app.schemas.tools import Tools, Tool
+from app.schemas.tools import Tool, Tools
 
 
 @pytest.mark.usefixtures("args", "session")

app/tests/tools/test_baserag.py

@@ -1,12 +1,12 @@
-import os
 import logging
+import os
 
 import pytest
 import wget
 
-from app.schemas.config import EMBEDDINGS_MODEL_TYPE, LANGUAGE_MODEL_TYPE
-from app.schemas.models import Models, Model
 from app.schemas.chat import ChatCompletion
+from app.schemas.config import EMBEDDINGS_MODEL_TYPE, LANGUAGE_MODEL_TYPE
+from app.schemas.models import Model, Models
 
 
 @pytest.mark.usefixtures("args", "session")

app/tools/__init__.py

@@ -1,4 +1,3 @@
 from ._baserag import BaseRAG
-from ._fewshots import FewShots
 
-__all__ = ["BaseRAG", "FewShots"]
+__all__ = ["BaseRAG"]

app/utils/lifespan.py

@@ -1,11 +1,11 @@
-import time
 from contextlib import asynccontextmanager
 from functools import partial
+import time
 from typing import Dict, List
 
-import requests
 from fastapi import FastAPI, HTTPException
 from openai import OpenAI
+import requests
 
 from app.schemas.config import EMBEDDINGS_MODEL_TYPE, LANGUAGE_MODEL_TYPE, METADATA_COLLECTION
 from app.schemas.models import Model, Models

app/utils/security.py

@@ -1,9 +1,9 @@
-from typing import Annotated
-import hashlib
 import base64
+import hashlib
+from typing import Annotated
 
-from fastapi import HTTPException, Depends
-from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
+from fastapi import Depends, HTTPException
+from fastapi.security import HTTPAuthorizationCredentials, HTTPBearer
 
 from app.utils.lifespan import clients