allow passing history to AI

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "tiny-ai-client"
-version = "0.0.6"
+version = "0.0.7"
 description = "Tiny AI client for LLMs. As simple as it gets."
 authors = ["piEsposito <[email protected]>"]
 license = "Apache 2.0"

tiny_ai_client/models.py

@@ -15,20 +15,22 @@ def __init__(
         max_new_tokens: int | None = None,
         timeout: int = 30,
         tools: List[Union[Callable, Dict]] | None = None,
+        chat: List["Message"] | None = None,
     ):
         # llm sampling parameters
         self.temperature: int = temperature
         self.max_new_tokens: int | None = max_new_tokens
         self.timeout: int = timeout
+        self.chat: List[Message] = chat or []
+        if system:
+            self.chat.append(Message(text=system, role="system"))
+
         tools = tools or []
         self.tools = tools
         self.tools_dict = {tool.__name__: tool for tool in tools}
 
         self.model_name: str = model_name
         self.system: str = system
-        self.chat: List[Message] = (
-            [Message(role="system", text=system)] if system else []
-        )
         self.client_wrapper: LLMClientWrapper = self.get_llm_client_wrapper(
             model_name=model_name, tools=self.tools
         )