[0.7.3] bug fixes + update copyright header

Author: yashbonde

README.md

@@ -2,3 +2,4 @@
 
 This is **not** the official SDK for Tune AI. This is an open source python package used to build GenAI apps at Tune AI.
 
+> MIT License

benchmarks/threaded_map.py

@@ -1,4 +1,4 @@
-# Copyright © 2024- Frello Technology Private Limited
+# Copyright © 2024-2025 Frello Technology Private Limited
 
 from tuneapi.types import Thread, ModelInterface, human, system
 from tuneapi.utils import from_json

cookbooks/finetuning.ipynb

@@ -230,7 +230,7 @@
             "name": "python",
             "nbconvert_exporter": "python",
             "pygments_lexer": "ipython3",
-            "version": "3.11.7"
+            "version": "3.13.0"
         }
     },
     "nbformat": 4,

docs/changelog.rst

@@ -7,6 +7,22 @@ minor versions.
 
 All relevant steps to be taken will be mentioned here.
 
+0.7.3
+-----
+
+- Fix bug in image modality for Gemini
+- Fix bug in structured generation for OpenAI by setting ``strict=True`` in the API.
+- Fix bug in structured generation for Gemini
+- Abstracted making input structure for ``Anthropic``, ``OpenAI`` and ``Gemini`` using ``_process_input`` method
+- Abstracted outputs for ``Anthropic``, ``OpenAI`` and ``Gemini`` using ``_process_output`` method
+
+
+0.7.2
+-----
+
+- Add image modality for Gemini and OpenAI to complement Anthropic.
+
+
 0.7.1
 -----
 

docs/conf.py

@@ -11,9 +11,9 @@
 # https://www.sphinx-doc.org/en/master/usage/configuration.html#project-information
 
 project = "tuneapi"
-copyright = "2024, Frello Technologies"
+copyright = "2024-2025, Frello Technologies"
 author = "Frello Technologies"
-release = "0.7.1"
+release = "0.7.3"
 
 # -- General configuration ---------------------------------------------------
 # https://www.sphinx-doc.org/en/master/usage/configuration.html#general-configuration

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "tuneapi"
-version = "0.7.1"
+version = "0.7.3"
 description = "Tune AI APIs."
 authors = ["Frello Technology Private Limited <[email protected]>"]
 license = "MIT"
@@ -27,9 +27,6 @@ boto3 = ["boto3"]
 protobuf = ["protobuf"]
 all = ["boto3", "protobuf"]
 
-# [tool.poetry.scripts]
-# tuneapi = "tuneapi.__main__:main"
-
 [tool.poetry.group.dev.dependencies]
 sphinx = "7.2.5"
 sphinx_rtd_theme = "1.3.0"

tests/test_apis.py

@@ -0,0 +1,39 @@
+# Copyright © 2023- Frello Technology Private Limited
+
+from tuneapi import ta, tt
+
+import asyncio
+from pydantic import BaseModel
+
+# fmt: off
+print("[BLOCKING] [Anthropic]", ta.Anthropic().chat("2 + 2 =", max_tokens=10))
+print("[BLOCKING] [Openai]", ta.Openai().chat("2 + 2 =", max_tokens=10))
+print("[BLOCKING] [Gemini]", ta.Gemini().chat("2 + 2 =", max_tokens=10))
+print("[BLOCKING] [Mistral]", ta.Mistral().chat("2 + 2 =", max_tokens=10))
+print("[BLOCKING] [Groq]", ta.Groq().chat("2 + 2 =", max_tokens=10))
+print("[BLOCKING] [TuneAI]", ta.TuneModel("meta/llama-3.1-8b-instruct").chat("2 + 2 =", max_tokens=10))
+# fmt: on
+
+
+class Result(BaseModel):
+    result: int
+
+
+# fmt: off
+print("[BLOCKING] [STRUCTURE] [Openai]", ta.Openai().chat(tt.Thread(tt.human("2 + 2 ="), schema=Result), max_tokens=10).result)
+print("[BLOCKING] [STRUCTURE] [Gemini]", ta.Gemini().chat(tt.Thread(tt.human("2 + 2 ="), schema=Result), max_tokens=10).result)
+# fmt: on
+
+
+async def main():
+    # fmt: off
+    print("[ASYNC] [Anthropic]", await ta.Anthropic().chat_async("2 + 2 =", max_tokens=10))
+    print("[ASYNC] [Openai]", await ta.Openai().chat_async("2 + 2 =", max_tokens=10))
+    print("[ASYNC] [Gemini]", await ta.Gemini().chat_async("2 + 2 =", max_tokens=10))
+    print("[ASYNC] [Mistral]", await ta.Mistral().chat_async("2 + 2 =", max_tokens=10))
+    print("[ASYNC] [Groq]", await ta.Groq().chat_async("2 + 2 =", max_tokens=10))
+    print("[ASYNC] [TuneAI]", await ta.TuneModel("meta/llama-3.1-8b-instruct").chat_async("2 + 2 =", max_tokens=10))
+    # fmt: on
+
+
+asyncio.run(main())

tests/test_tree.py

@@ -1,4 +1,4 @@
-# Copyright © 2024- Frello Technology Private Limited
+# Copyright © 2024-2025 Frello Technology Private Limited
 
 import tuneapi.types as tt
 
@@ -24,7 +24,9 @@ def get_tree() -> tt.ThreadsTree:
                                 "```<function ... you go.",
                                 fn_pairs=[
                                     (
-                                        tt.function_call("{list_tools: arguments"),
+                                        tt.function_call(
+                                            {"name": "list_tools", "arguments": "{}"}
+                                        ),
                                         tt.function_resp("{...}"),
                                     )
                                 ],
@@ -46,7 +48,9 @@ def get_tree() -> tt.ThreadsTree:
                                 "```<function ... @apple.com] sent!",
                                 fn_pairs=[
                                     (
-                                        tt.function_call("{send_email: arguments"),
+                                        tt.function_call(
+                                            {"name": "send_email", "arguments": "{}"}
+                                        ),
                                         tt.function_resp("{...}"),
                                     )
                                 ],

tuneapi/apis/__init__.py

@@ -1,4 +1,4 @@
-# Copyright © 2024- Frello Technology Private Limited
+# Copyright © 2024-2025 Frello Technology Private Limited
 
 # model APIs
 from tuneapi.apis.model_tune import TuneModel