correcoes cosmeticas, de acordo com pycharm, de alguns tipos de retorno

README.md

@@ -2,7 +2,7 @@
 
 Inspired by [tinygrad](https://github.com/tinygrad/tinygrad) and [simpleaichat](https://github.com/minimaxir/simpleaichat/tree/main/simpleaichat), `tiny-ai-client` is the easiest way to use and switch LLMs with vision and tool usage support. It works because it is `tiny`, `simple` and most importantly `fun` to develop.
 
-I want to change LLMs with ease, while knowing what is happening under the hood. Langchain is cool, but became bloated, complicated there is just too much chaos going on. I want to keep it simple, easy to understand and easy to use. If you want to use a LLM and have an API key, you should not need to read a 1000 lines of code and write `response.choices[0].message.content` to get the response.
+I want to change LLMs with ease, while knowing what is happening under the hood. Langchain is cool, but became bloated, complicated there is just too much chaos going on. I want to keep it simple, easy to understand and easy to use. If you want to use a LLM and have an API key, you should not need to read 1000 lines of code and write `response.choices[0].message.content` to get the response.
 
 Simple and tiny, that's the goal.
 

requirements.txt

@@ -1,3 +1,5 @@
 pydantic==2.7.3
 openai==1.31.0
 anthropic==0.28.0
+pillow~=10.3.0
+google-generativeai

tiny_ai_client/anthropic_.py

@@ -101,7 +101,7 @@ def call_llm_provider(
         temperature: int | None,
         max_new_tokens: int | None,
         timeout: int,
-    ) -> str:
+    ) -> Message:
         kwargs = {}
         input_messages, system = model_input
         if temperature is not None:
@@ -139,7 +139,7 @@ async def async_call_llm_provider(
         temperature: int | None,
         max_new_tokens: int | None,
         timeout: int,
-    ) -> str:
+    ) -> Message:
         kwargs = {}
         input_messages, system = model_input
         if temperature is not None:

tiny_ai_client/gemini_.py

@@ -31,7 +31,6 @@ def build_model_input(self, messages: List["Message"]) -> Any:
         history = []
         local_messages = deepcopy(messages)
         system = None
-        message = None
 
         for message in local_messages:
             if message.role == "system":
@@ -45,6 +44,7 @@ def build_model_input(self, messages: List["Message"]) -> Any:
                 if message.text is not None:
                     parts.append(message.text)
                 if message.images is not None:
+                    # noinspection PyTypeChecker
                     parts.extend(message.images)
                 history.append(
                     {
@@ -53,15 +53,15 @@ def build_model_input(self, messages: List["Message"]) -> Any:
                     }
                 )
 
-        return (system, history)
+        return system, history
 
     def call_llm_provider(
         self,
         model_input: Any,
         temperature: int | None,
         max_new_tokens: int | None,
         timeout: int,
-    ) -> str:
+    ) -> Message:
         system, history = model_input
 
         generation_config_kwargs = {}
@@ -92,7 +92,7 @@ async def async_call_llm_provider(
         temperature: int | None,
         max_new_tokens: int | None,
         timeout: int,
-    ) -> str:
+    ) -> Message:
         system, history = model_input
 
         generation_config_kwargs = {}

tiny_ai_client/models.py

@@ -87,7 +87,7 @@ def get_llm_client_wrapper(
 
     @property
     def model_name(self) -> str:
-        return self._model_name
+        return self.model_name
 
     @model_name.setter
     def model_name(self, value: str) -> None:

tiny_ai_client/openai_.py

@@ -68,7 +68,7 @@ def call_llm_provider(
         temperature: int | None,
         max_new_tokens: int | None,
         timeout: int,
-    ) -> str:
+    ) -> Message:
         kwargs = {}
         if temperature is not None:
             kwargs["temperature"] = temperature
@@ -103,7 +103,7 @@ async def async_call_llm_provider(
         temperature: int | None,
         max_new_tokens: int | None,
         timeout: int,
-    ) -> str:
+    ) -> Message:
         kwargs = {}
         if temperature is not None:
             kwargs["temperature"] = temperature