speedy-utils 1.1.8__tar.gz → 1.1.9__tar.gz

{speedy_utils-1.1.8 → speedy_utils-1.1.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: speedy-utils
-Version: 1.1.8
+Version: 1.1.9
 Summary: Fast and easy-to-use package for data science
 Author: AnhVTH
 Author-email: anhvth.226@gmail.com

{speedy_utils-1.1.8 → speedy_utils-1.1.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "speedy-utils"
-version = "1.1.8"
+version = "1.1.9"
 description = "Fast and easy-to-use package for data science"
 authors = ["AnhVTH <anhvth.226@gmail.com>"]
 readme = "README.md"

{speedy_utils-1.1.8 → speedy_utils-1.1.9}/src/llm_utils/lm/async_lm/async_llm_task.py

@@ -7,6 +7,7 @@ import pathlib
 from abc import ABC
 from dataclasses import dataclass
 from typing import Any, Dict, Generic, List, Literal, Optional, TypeVar, Union, cast
+from venv import logger
 
 from openai.types.chat import ChatCompletionMessageParam
 from pydantic import BaseModel
@@ -512,5 +513,8 @@ class AsyncLLMTask(ABC, Generic[InputModelType, OutputModelType]):
         return self
 
     async def __aexit__(self, exc_type, exc_val, exc_tb):
-        if self._lm and hasattr(self._lm, "aclose"):  # Or self._lm.client
-            await self._lm._last_client._client.aclose()
+        if hasattr(self._lm, "_last_client"):
+            last_client = self._lm._last_client   # type: ignore
+            await last_client._client.aclose()
+        else:
+            logger.warning("No last client to close")

{speedy_utils-1.1.8 → speedy_utils-1.1.9}/src/llm_utils/lm/async_lm/async_lm_base.py

@@ -54,7 +54,6 @@ class AsyncLMBase:
         self._cache = cache
         self.ports = ports
         self._init_port = port  # <-- store the port provided at init
-        self._last_client:AsyncOpenAI = None
 
     @property
     def client(self) -> AsyncOpenAI: