chore: better import

Author: codycjy

.pylintrc

@@ -5,3 +5,4 @@ disable=
     C0116, # missing-function-docstring
     W0511, # fixme
     W0718, # broad-exception-caught
+    W0719, # broad-excepttion-raised

api/ai_providers/base.py

@@ -1,6 +1,6 @@
 # ai_providers/base.py
 from abc import ABC, abstractmethod
-from typing import Optional, Dict, Any
+from typing import Optional
 
 class BaseSummarizer(ABC):
     def __init__(self, api_key: str, **kwargs):

api/ai_providers/claude_summarizer.py

@@ -1,7 +1,7 @@
 # ai_providers/claude_summarizer.py
 from typing import Optional
-from ai_providers.base import BaseSummarizer
 from anthropic import Anthropic
+from ai_providers.base import BaseSummarizer
 from prompt import DEFAULT_PROMPT
 DEFAULT_MODEL = "claude-3-haiku-20240307"
 
@@ -27,7 +27,7 @@ async def summarize(self, text: str, max_length: Optional[int] = 1000) -> str:
 
     async def is_available(self) -> bool:
         try:
-            response = self.client.messages.create(
+            self.client.messages.create(
                 model=self.model,
                 max_tokens=5,
                 messages=[{

api/ai_providers/openai_summarizer.py

@@ -1,5 +1,5 @@
 # ai_providers/openai_summarizer.py
-from typing import Optional, Dict, Any
+from typing import Optional
 from openai import OpenAI
 from prompt import DEFAULT_PROMPT
 from ai_providers.base import BaseSummarizer
@@ -28,7 +28,7 @@ async def summarize(self, text: str, max_length: Optional[int] = 1000) -> str:
 
     async def is_available(self) -> bool:
         try:
-            response = await self.client.chat.completions.create(
+            await self.client.chat.completions.create(
                 model=self.model,
                 messages=[{"role": "user", "content": "test"}],
                 max_tokens=5

api/api/routes/models.py

@@ -1,8 +1,7 @@
 # api/routes/models.py
-from fastapi import APIRouter, HTTPException, BackgroundTasks, Depends
-from api.schemas import ModelRequest, ModelResponse
-from services.transcription import generate_summary
+from fastapi import APIRouter, HTTPException
 from config import WHISPER_MODEL, AVAILABLE_PROVIDERS, MODEL_MAP
+from api.schemas import ModelRequest, ModelResponse
 
 
 router = APIRouter(
@@ -20,10 +19,9 @@ async def get_models():
 async def summarize_text():
     return {"provider": AVAILABLE_PROVIDERS}
 
+
 @router.post("/available_models")
 async def available_models(request: ModelRequest):
     if request.provider not in AVAILABLE_PROVIDERS:
         raise HTTPException(status_code=404, detail="Provider not found")
     return ModelResponse(available_models=MODEL_MAP[request.provider])
-
-

api/api/schemas.py

@@ -1,7 +1,7 @@
-from pydantic import BaseModel, HttpUrl
-from typing import Optional
 from enum import Enum
 from datetime import datetime
+from typing import Optional
+from pydantic import BaseModel, HttpUrl
 
 
 class TaskStatus(str, Enum):

api/config.py

@@ -1,3 +1,4 @@
+import json
 from pathlib import Path
 import os
 
@@ -21,7 +22,6 @@
 
 DEFAULT_PROVIDERS= ["openai", "anthropic"]
 env_providers= os.getenv("AVAILABLE_PROVIDERS","")
-import json
 
 try:
     providers = json.loads(env_providers)

api/database/dependencies.py

@@ -1,6 +1,5 @@
 # database/dependencies.py
-from fastapi import Depends
-from .manager import DBManager
+from database.manager import DBManager
 
 def get_db():
     return DBManager()

api/database/manager.py

@@ -2,9 +2,9 @@
 import asyncio
 from datetime import datetime
 from typing import Optional, Dict, Any
+from config import DATABASE_PATH
 from api.schemas import TaskStatus, TranscriptionResponse
 from utils.logger import logger # TODO: use logger
-from config import DATABASE_PATH
 
 
 class DBManager:

api/services/transcription.py

@@ -1,9 +1,9 @@
+import os
 from utils.logger import logger
 from database.dependencies import get_db
 from api.schemas import TaskStatus, SummaryRequest
 from services.video import VideoProcessor
 from ai_providers.factory import SummarizerFactory
-import os
 
 
 def check_model_availability(provider: str, model: str):