pylint

src/dokumetry/anthropic.py

@@ -86,6 +86,7 @@ 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)
@@ -134,6 +135,5 @@ def stream_generator():
             send_data(data, doku_url, api_key)
 
             return response
-
-
-    llm.messages.create = patched_messages_create
+
+    llm.messages.create = patched_messages_create

src/dokumetry/async_anthropic.py

@@ -6,7 +6,7 @@
 import time
 from .__helpers import send_data
 
-# pylint: disable=too-many-arguments
+# pylint: disable=too-many-arguments,too-many-statements
 def init(llm, doku_url, api_key, environment, application_name, skip_resp):
     """
     Initialize Anthropic integration with Doku.
@@ -35,6 +35,7 @@ async def patched_messages_create(*args, **kwargs):
         """
         streaming = kwargs.get('stream', False)
         start_time = time.time()
+        #pylint: disable=no-else-return
         if streaming:
             async def stream_generator():
                 accumulated_content = ""
@@ -134,6 +135,5 @@ async def stream_generator():
             send_data(data, doku_url, api_key)
 
             return response
-
-
-    llm.messages.create = patched_messages_create
+
+    llm.messages.create = patched_messages_create