pylint

src/dokumetry/__init__.py

@@ -1,12 +1,12 @@
 """
 __init__ module for dokumetry package.
 """
+from anthropic import AsyncAnthropic, Anthropic
 
 from .openai import init as init_openai
 from .anthropic import init as init_anthropic
 from .async_anthropic import init as init_async_anthropic
 from .cohere import init as init_cohere
-from anthropic import AsyncAnthropic, Anthropic
 
 # pylint: disable=too-few-public-methods
 class DokuConfig:

src/dokumetry/anthropic.py

@@ -22,6 +22,7 @@ def init(llm, doku_url, api_key, environment, application_name, skip_resp):
 
     original_messages_create = llm.messages.create
 
+    #pylint: disable=too-many-locals
     def patched_messages_create(*args, **kwargs):
         """
         Patched version of Anthropic's messages.create method.
@@ -35,6 +36,8 @@ def patched_messages_create(*args, **kwargs):
         """
         streaming = kwargs.get('stream', False)
         start_time = time.time()
+
+        # pylint: disable=no-else-return
         if streaming:
             def stream_generator():
                 accumulated_content = ""
@@ -86,7 +89,6 @@ def stream_generator():
                 send_data(data, doku_url, api_key)
 
             return stream_generator()
-        # pylint disable=no-else-return
         else:
             start_time = time.time()
             response = original_messages_create(*args, **kwargs)

src/dokumetry/async_anthropic.py

@@ -22,6 +22,7 @@ def init(llm, doku_url, api_key, environment, application_name, skip_resp):
 
     original_messages_create = llm.messages.create
 
+    #pylint: disable=too-many-locals
     async def patched_messages_create(*args, **kwargs):
         """
         Patched version of Anthropic's messages.create method.