formatting

Tiendil · Jan 1, 2025 · 5756c1c · 5756c1c
1 parent d3ee768
commit 5756c1c
Show file tree

Hide file tree

Showing 4 changed files with 32 additions and 10 deletions.
diff --git a/ffun/ffun/librarian/processors/llm_general.py b/ffun/ffun/librarian/processors/llm_general.py
@@ -5,8 +5,8 @@
 from ffun.librarian.entities import TagsExtractor, TextCleaner
 from ffun.librarian.processors import base
 from ffun.library.entities import Entry
-from ffun.llms_framework.domain import call_llm, search_for_api_key
 from ffun.llms_framework import errors as llmsf_errors
+from ffun.llms_framework.domain import call_llm, search_for_api_key
 from ffun.llms_framework.entities import (
     ChatResponse,
     LLMCollectionApiKey,

diff --git a/ffun/ffun/librarian/processors/tests/test_llm_general.py b/ffun/ffun/librarian/processors/tests/test_llm_general.py
@@ -1,6 +1,5 @@
 import pytest
 
-from pytest_mock import MockerFixture
 from ffun.librarian import errors
 from ffun.librarian.processors.llm_general import Processor
 from ffun.librarian.tag_extractors import dog_tags_extractor
@@ -76,10 +75,10 @@ async def test_process__has_api_key_found(
 
     @pytest.mark.asyncio
     async def test_process__temporary_error_processing(
-            self, llm_processor: Processor, cataloged_entry: Entry, fake_llm_api_key: LLMApiKey
+        self, llm_processor: Processor, cataloged_entry: Entry, fake_llm_api_key: LLMApiKey
     ) -> None:
 
-        entry = cataloged_entry.replace(title="@tag-1 @tag-2", body='raise TemporaryError')
+        entry = cataloged_entry.replace(title="@tag-1 @tag-2", body="raise TemporaryError")
 
         llm_processor.general_api_key = LLMGeneralApiKey(fake_llm_api_key)
 

diff --git a/ffun/ffun/librarian/tests/test_domain.py b/ffun/ffun/librarian/tests/test_domain.py
@@ -13,7 +13,12 @@
     push_entries_and_move_pointer,
 )
 from ffun.librarian.entities import ProcessorPointer
-from ffun.librarian.processors.base import AlwaysConstantProcessor, AlwaysErrorProcessor, AlwaysSkipEntryProcessor, AlwaysTemporaryErrorProcessor
+from ffun.librarian.processors.base import (
+    AlwaysConstantProcessor,
+    AlwaysErrorProcessor,
+    AlwaysSkipEntryProcessor,
+    AlwaysTemporaryErrorProcessor,
+)
 from ffun.librarian.tests import helpers, make
 from ffun.library.entities import Entry
 from ffun.library.tests import make as l_make
@@ -230,7 +235,13 @@ async def test_skip_processing(
                 entry=cataloged_entry,
             )
 
-        assert_logs(logs, processor_successed=0, processor_requested_to_skip_entry=1, entry_processed=1, processor_temporary_error=0)
+        assert_logs(
+            logs,
+            processor_successed=0,
+            processor_requested_to_skip_entry=1,
+            entry_processed=1,
+            processor_temporary_error=0,
+        )
 
         tags = await o_domain.get_tags_ids_for_entries([cataloged_entry.id])
 
@@ -260,7 +271,13 @@ async def test_temporary_error_in_processor(
                 entry=cataloged_entry,
             )
 
-        assert_logs(logs, processor_successed=0, processor_requested_to_skip_entry=0, entry_processed=1, processor_temporary_error=1)
+        assert_logs(
+            logs,
+            processor_successed=0,
+            processor_requested_to_skip_entry=0,
+            entry_processed=1,
+            processor_temporary_error=1,
+        )
 
         tags = await o_domain.get_tags_ids_for_entries([cataloged_entry.id])
 
@@ -291,7 +308,13 @@ async def test_unexpected_error(
                     entry=cataloged_entry,
                 )
 
-        assert_logs(logs, processor_successed=0, processor_requested_to_skip_entry=0, entry_processed=0, processor_temporary_error=0)
+        assert_logs(
+            logs,
+            processor_successed=0,
+            processor_requested_to_skip_entry=0,
+            entry_processed=0,
+            processor_temporary_error=0,
+        )
 
         tags = await o_domain.get_tags_ids_for_entries([cataloged_entry.id])
 

diff --git a/ffun/ffun/llms_framework/provider_interface.py b/ffun/ffun/llms_framework/provider_interface.py
@@ -67,8 +67,8 @@ def estimate_tokens(self, config: LLMConfiguration, text: str) -> int:
     async def chat_request(  # type: ignore
         self, config: LLMConfiguration, api_key: str, request: ChatRequestTest
     ) -> ChatResponseTest:
-        if 'raise TemporaryError' in request.text:
-            raise errors.TemporaryError(message='error raise by a test request')
+        if "raise TemporaryError" in request.text:
+            raise errors.TemporaryError(message="error raise by a test request")
 
         return ChatResponseTest(content=request.text)