language-model-common 3.0.1__tar.gz → 3.0.2__tar.gz

{language_model_common-3.0.1/language_model_common.egg-info → language_model_common-3.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: language-model-common
-Version: 3.0.1
+Version: 3.0.2
 Summary: Provides the underlying framework to enhance langchain and add loading configurations
 Author-email: Imran Qureshi <imran.qureshi@bwell.com>
 License: Apache-2.0

language_model_common-3.0.2/VERSION

@@ -0,0 +1 @@
+3.0.2

{language_model_common-3.0.1 → language_model_common-3.0.2/language_model_common.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: language-model-common
-Version: 3.0.1
+Version: 3.0.2
 Summary: Provides the underlying framework to enhance langchain and add loading configurations
 Author-email: Imran Qureshi <imran.qureshi@bwell.com>
 License: Apache-2.0

{language_model_common-3.0.1 → language_model_common-3.0.2}/languagemodelcommon/utilities/logger/logging_response.py

@@ -1,7 +1,8 @@
 import logging
-from typing import Any, AsyncIterator, override
+from collections.abc import AsyncGenerator
+from typing import override
 
-import httpx
+import httpx2
 
 from languagemodelcommon.utilities.logger.log_levels import SRC_LOG_LEVELS
 
@@ -9,28 +10,29 @@ logger = logging.getLogger(__name__)
 logger.setLevel(SRC_LOG_LEVELS.HTTP)
 
 
-class LoggingResponse(httpx.Response):
+class LoggingResponse(httpx2.Response):
     """
     A custom HTTP response class that logs the request and response details.
-    This class extends httpx.Response to log the request method, URL, status code,
+    This class extends httpx2.Response to log the request method, URL, status code,
     and response content in bytes as they are streamed.
     """
 
     @override
-    async def aiter_bytes(self, *args: Any, **kwargs: Any) -> AsyncIterator[bytes]:
+    async def aiter_bytes(
+        self, chunk_size: int | None = None
+    ) -> AsyncGenerator[bytes, None]:
         """
         Asynchronously iterate over the response content in bytes, logging each chunk.
         This method overrides the default aiter_bytes method to include logging.
         Args:
-            *args: Positional arguments passed to the parent method.
-            **kwargs: Keyword arguments passed to the parent method.
+            chunk_size: Passed through to the parent method.
         Yields:
             bytes: The next chunk of response content in bytes.
         """
         logger.debug(
             f"====== Response: {self.request.method} {self.url} {self.status_code} ====="
         )
-        async for chunk in super().aiter_bytes(*args, **kwargs):
+        async for chunk in super().aiter_bytes(chunk_size):
             logger.debug(chunk)
             yield chunk
         logger.debug(

{language_model_common-3.0.1 → language_model_common-3.0.2}/languagemodelcommon/utilities/logger/logging_transport.py

@@ -1,7 +1,7 @@
 import logging
 from typing import override
 
-import httpx
+import httpx2
 
 from languagemodelcommon.utilities.logger.log_levels import SRC_LOG_LEVELS
 from languagemodelcommon.utilities.logger.logging_response import (
@@ -12,29 +12,29 @@ logger = logging.getLogger(__name__)
 logger.setLevel(SRC_LOG_LEVELS.HTTP)
 
 
-class LoggingTransport(httpx.AsyncBaseTransport):
+class LoggingTransport(httpx2.AsyncBaseTransport):
     """
     A custom HTTP transport that logs request and response details.
-    This class extends httpx.AsyncBaseTransport to log the request method, URL,
+    This class extends httpx2.AsyncBaseTransport to log the request method, URL,
     headers, and content before sending the request, and logs the response status code,
     headers, and content as it is streamed back.
-    It is designed to be used with httpx for asynchronous HTTP requests.
+    It is designed to be used with httpx2 for asynchronous HTTP requests.
     It logs the request method, URL, headers, and content before sending the request,
     and logs the response status code, headers, and content as it is streamed back.
     This transport can be used to monitor and debug HTTP requests and responses in an application.
     """
 
-    def __init__(self, transport: httpx.AsyncBaseTransport) -> None:
+    def __init__(self, transport: httpx2.AsyncBaseTransport) -> None:
         """
         Initialize the LoggingTransport with a given transport.
         Args:
-            transport (httpx.AsyncBaseTransport): The underlying transport to wrap.
+            transport (httpx2.AsyncBaseTransport): The underlying transport to wrap.
             This transport will handle the actual HTTP requests and responses.
         """
-        self.transport: httpx.AsyncBaseTransport = transport
+        self.transport: httpx2.AsyncBaseTransport = transport
 
     @override
-    async def handle_async_request(self, request: httpx.Request) -> LoggingResponse:
+    async def handle_async_request(self, request: httpx2.Request) -> LoggingResponse:
         """
         Handle an asynchronous HTTP request, logging the request details and returning a LoggingResponse.
         Args:
@@ -60,7 +60,7 @@ class LoggingTransport(httpx.AsyncBaseTransport):
                 stream=response.stream,
                 extensions=response.extensions,
             )
-        except httpx.HTTPError as e:
+        except httpx2.HTTPError as e:
             logger.exception(f"HTTP error occurred: {e}")
             raise
         except Exception as e:

language_model_common-3.0.1/VERSION

@@ -1 +0,0 @@
-3.0.1