style: fix typing, reformat

style: black+isort

style: fix typing

src/chains/pipelines.py

@@ -1,14 +1,14 @@
-from typing import List, Dict, Any, Optional, Tuple, Union
-from pydantic import BaseModel, ConfigDict, Field
-from pathlib import Path
 import json
 import logging
-from tqdm import tqdm
 from datetime import datetime
-import fitz
+from pathlib import Path
+from typing import Any, Dict, List, Optional, Tuple, Union
 
-from langchain_openai.chat_models import ChatOpenAI
+import fitz
 from langchain.chains.base import Chain
+from langchain_openai.chat_models import ChatOpenAI
+from pydantic import BaseModel, ConfigDict, Field
+from tqdm import tqdm
 
 from src.chains.chains import (
     LoadPageChain,
@@ -16,11 +16,8 @@ from src.chains.chains import (
     ImageEncodeChain,
     VisionAnalysisChain
 )
-
 from src.chains.prompts import BasePrompt, JsonH1AndGDPrompt
 from src.config.navigator import Navigator
-from src.chains.prompts import BasePrompt
-
 
 logger = logging.getLogger(__name__)
 

src/chains/prompts.py

@@ -1,18 +1,15 @@
-from typing import Dict, Any, Optional, Type, Union, TypeVar
-from abc import ABC, abstractmethod
 import logging
+from abc import ABC, abstractmethod
+from textwrap import dedent
+from typing import Any, Dict, Optional, Type, TypeVar, Union
 
-from pydantic import BaseModel
-from langchain.prompts import ChatPromptTemplate
 from langchain.output_parsers import PydanticOutputParser
-from langchain_core.output_parsers import BaseOutputParser
-from langchain_core.output_parsers import StrOutputParser
-
-from textwrap import dedent
+from langchain.prompts import ChatPromptTemplate
+from langchain_core.output_parsers import BaseOutputParser, StrOutputParser
+from pydantic import BaseModel
 
 logger = logging.getLogger(__name__)
 
-T = TypeVar("T")
 
 
 class BasePrompt(ABC):
@@ -62,7 +59,7 @@ class BasePrompt(ABC):
         return self._template
 
     @abstractmethod
-    def parse(self, text: str) -> T:
+    def parse(self, text: str) -> object:
         """Parse LLM output
 
         Args:

src/processing/image_utlis.py

@@ -2,8 +2,8 @@ from PIL import Image
 import base64
 import io
 
-def image2base64(pil_image: Image):
+def image2base64(pil_image: Image.Image):
     buffered = io.BytesIO()
     pil_image.save(buffered, format="png")
     img_str = base64.b64encode(buffered.getvalue())
-    return img_str.decode("utf-8")
+    return img_str.decode("utf-8")