chore: run formatters and linters

src/wandbot/api/app.py

@@ -33,9 +33,9 @@
 from datetime import datetime, timezone
 
 import pandas as pd
+import wandb
 from fastapi import FastAPI
 
-import wandb
 from wandbot.api.routers import chat as chat_router
 from wandbot.api.routers import database as database_router
 from wandbot.api.routers import retrieve as retrieve_router

src/wandbot/api/routers/database.py

@@ -1,8 +1,8 @@
+import wandb
 from fastapi import APIRouter
 from starlette import status
 from starlette.responses import Response
 
-import wandb
 from wandbot.database.client import DatabaseClient
 from wandbot.database.database import engine
 from wandbot.database.models import Base

src/wandbot/chat/chat.py

@@ -26,9 +26,9 @@
 """
 from typing import List
 
+import wandb
 from weave.monitoring import StreamTable
 
-import wandb
 from wandbot.chat.config import ChatConfig
 from wandbot.chat.rag import RAGPipeline, RAGPipelineOutput
 from wandbot.chat.schemas import ChatRequest, ChatResponse

src/wandbot/evaluation/eval/async_main.py

@@ -7,11 +7,11 @@
 import aiofiles
 import httpx
 import pandas as pd
+import wandb
 from llama_index.llms.openai import OpenAI
 from tenacity import retry, stop_after_attempt, wait_random_exponential
 from tqdm import tqdm
 
-import wandb
 from wandbot.evaluation.config import EvalConfig
 from wandbot.evaluation.eval.correctness import (
     CORRECTNESS_EVAL_TEMPLATE,

src/wandbot/ingestion/prepare_data.py

@@ -23,13 +23,13 @@
 
 import nbformat
 import pandas as pd
+import wandb
 from google.cloud import bigquery
 from langchain.schema import Document
 from langchain_community.document_loaders import TextLoader
 from langchain_community.document_loaders.base import BaseLoader
 from nbconvert import MarkdownExporter
 
-import wandb
 from wandbot.ingestion.config import (
     DataStoreConfig,
     DocodileEnglishStoreConfig,

src/wandbot/ingestion/preprocess_data.py

@@ -24,9 +24,9 @@
 from typing import Any, List, Sequence
 
 import tiktoken
+import wandb
 from langchain_core.documents import BaseDocumentTransformer, Document
 
-import wandb
 from wandbot.ingestion.preprocessors.markdown import MarkdownTextTransformer
 from wandbot.ingestion.preprocessors.source_code import CodeTextTransformer
 from wandbot.utils import (

src/wandbot/ingestion/report.py

@@ -19,9 +19,8 @@
 import pathlib
 from datetime import datetime
 
-import wandb.apis.reports as wr
-
 import wandb
+import wandb.apis.reports as wr
 
 
 def log_raw_counts(metadata: dict[str, dict[str, int]]) -> list[str]:

src/wandbot/ingestion/vectorstores.py

@@ -16,12 +16,12 @@
 import pathlib
 from typing import List
 
+import wandb
 from langchain_community.vectorstores.chroma import Chroma
 from langchain_core.documents import Document
 from langchain_openai import OpenAIEmbeddings
 from tqdm import trange
 
-import wandb
 from wandbot.ingestion.config import VectorStoreConfig
 from wandbot.utils import get_logger
 

src/wandbot/retriever/base.py

@@ -1,12 +1,12 @@
 from operator import itemgetter
 from typing import List
 
+import wandb
 from langchain_community.document_transformers import EmbeddingsRedundantFilter
 from langchain_community.vectorstores.chroma import Chroma
 from langchain_core.documents import Document
 from langchain_core.runnables import RunnableLambda, RunnableParallel
 
-import wandb
 from wandbot.ingestion.config import VectorStoreConfig
 from wandbot.retriever.reranking import CohereRerankChain
 from wandbot.retriever.utils import OpenAIEmbeddingsModel

src/wandbot/utils.py

@@ -36,12 +36,11 @@
 import fasttext
 import nest_asyncio
 import tiktoken
+import wandb
 from langchain_core.documents import Document
 from pydantic import Field
 from pydantic_settings import BaseSettings, SettingsConfigDict
 
-import wandb
-
 
 def get_logger(name: str) -> logging.Logger:
     """Creates and returns a logger with the specified name.