diff --git a/pyproject.toml b/pyproject.toml index df838cd..e878b97 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "tiny-ai-client" -version = "0.0.4" +version = "0.0.5" description = "Tiny AI client for LLMs. As simple as it gets." authors = ["piEsposito "] license = "Apache 2.0" diff --git a/tiny_ai_client/models.py b/tiny_ai_client/models.py index 0629b6f..f92748d 100644 --- a/tiny_ai_client/models.py +++ b/tiny_ai_client/models.py @@ -20,7 +20,8 @@ def __init__( self.temperature: int = temperature self.max_new_tokens: int | None = max_new_tokens self.timeout: int = timeout - self.tools = tools or [] + tools = tools or [] + self.tools = tools self.tools_dict = {tool.__name__: tool for tool in tools} self.model_name: str = model_name