aisberg 0.2.2__py3-none-any.whl → 0.2.3__py3-none-any.whl

aisberg/models/chat.py

@@ -138,6 +138,8 @@ def format_messages(input: LanguageModelInput) -> list[dict]:
     else:
         messages = [parse_message(msg) for msg in input]
     return [
-        msg.model_dump(include={"role", "content", "tool_calls", "tool_call_id"})
+        msg.model_dump(
+            include={"role", "content", "tool_calls", "tool_call_id"}, exclude_none=True
+        )
         for msg in messages
     ]

aisberg/modules/s3.py

@@ -111,7 +111,7 @@ class BaseS3Module(ABC):
         return file_size, metadata
 
     @staticmethod
-    def _get_object_name(self, file, object_name):
+    def _get_object_name(file, object_name):
         return object_name or (file if isinstance(file, str) else "default_object_name")
 
     @abstractmethod

aisberg/utils.py

@@ -28,6 +28,9 @@ def parse_chat_line(
         if chunk["object"] != "chat.completion.chunk":
             raise ValueError(f"Unexpected object type: {chunk['object']}")
 
+        if not chunk.get("choices", []):
+            return
+
         if chunk["choices"][0]["finish_reason"] == "stop":
             return
 

{aisberg-0.2.2.dist-info → aisberg-0.2.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aisberg
-Version: 0.2.2
+Version: 0.2.3
 Summary: Aisberg SDK for Python - A simple and powerful SDK to interact with the Aisberg API
 Author: Free Pro
 Author-email: Mathis Lambert <mathis.lambert@freepro.com>

{aisberg-0.2.2.dist-info → aisberg-0.2.3.dist-info}/RECORD

@@ -3,14 +3,14 @@ aisberg/async_client.py,sha256=0hO2lzdP2Vcb4qV619aJLrsxeXmfYj5wo0mog1bda5Q,3816
 aisberg/client.py,sha256=W1i9mh-o7f283kz2s9B2nHBCegJGN0Lgf1q0lAMNx9I,3853
 aisberg/config.py,sha256=R1dvL14YeO_0VRr1nUiirWRFN8xY0Nkthurtxnnoc34,587
 aisberg/exceptions.py,sha256=T1ioitOczdkmW3-8XfPG0ta_O2WqzS1iQtzhDBucb5Y,515
-aisberg/utils.py,sha256=4wkrpC5gz8PJWw3kZqaRbd3KekrI37zlDC__JiAVitw,3552
+aisberg/utils.py,sha256=d-YxsRUQbt0tC3h9W8tAk3ga0Soe3SEjN6Y3IvoULtw,3613
 aisberg/abstract/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 aisberg/abstract/modules.py,sha256=aRmB21oiL9NCZsHUx8zB0Q8VvZAZBucsZxYv5l7JygY,1698
 aisberg/api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 aisberg/api/async_endpoints.py,sha256=UZhz7_JeGnkQ1AWE7gEvvOdbaBUGE1nofG0NddqRctg,10982
 aisberg/api/endpoints.py,sha256=ByVxBVxrdPo-_8G0vzb_PnrwkmOSPI09koBsHjzl3gg,10610
 aisberg/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-aisberg/models/chat.py,sha256=LcvOgQm9Aan7pS4S90h4lrdzENX-NebA9pAKYdcDLB8,3667
+aisberg/models/chat.py,sha256=HM37XWh4ep1GNuhE-poW_AnM8TIhF5zIMortHgzIHcc,3708
 aisberg/models/collections.py,sha256=Je9THFx-2qfTgmd_AktRd2oi6YeCw-j3iLLtlUdmaJ0,1103
 aisberg/models/documents.py,sha256=-9jNPRqMQEhkBMtZxDv8Fe0r8X91cpgQCv0FqPL8qH8,916
 aisberg/models/embeddings.py,sha256=gBGXR3DgyBCWB9Vky-iowWh_2Md7wa57I3fWxIrl77I,2409
@@ -26,14 +26,14 @@ aisberg/modules/documents.py,sha256=YJhGyyy4noj2Dj5XHDOy8RMlseqP8BYkcRnfaft3dJ4,
 aisberg/modules/embeddings.py,sha256=UV4lzNTIvooGNisAIS8sUmZ0rAxs3Omq-bnPhlyqQ74,10627
 aisberg/modules/me.py,sha256=VcbXBvcCNPoCKYHjDLvevuZWTJjQx3LLBzPFYkZKlb8,2144
 aisberg/modules/models.py,sha256=IWa2ezrWS4uQQPzx-YBYrBkRgC_ZnMJCnxFB2Az0DkA,3045
-aisberg/modules/s3.py,sha256=2pupGSaHxu4LzxwhMbxgYFt9-ILLcjF74iYmLtpmwY4,10170
+aisberg/modules/s3.py,sha256=jIX0ueI65MMBID5Y8hVDmBNY5drLsV67HxbedZt7k3g,10164
 aisberg/modules/tools.py,sha256=TnTIfjQuPhPImtWTjEy3UCmXeroPrYX3BRzTvT9nIkk,2424
 aisberg/modules/workflows.py,sha256=AxyBwCnZkRdgKHUOAoYXMAJH6J7K-qaL1psZ-f-MKhI,5001
 aisberg/requests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 aisberg/requests/async_requests.py,sha256=4J6Gj6vY5AD9lXhSTdYdAKQIqsqH9m1uaKc7XLDKWx8,2849
 aisberg/requests/sync_requests.py,sha256=ov7GDgx4C-oQ9xEZ7VQsGVY1fnuqRpPwjS0H_yH1yb8,2839
-aisberg-0.2.2.dist-info/licenses/LICENSE,sha256=dCGgiGa4f14s5mzTYnj1wIZNqGRVG2bUYyyk7nzKlEo,406
-aisberg-0.2.2.dist-info/METADATA,sha256=GsIDsQLP8zBATgBXSXGziMDs2ax9RTMRGEUacFxUJpg,5604
-aisberg-0.2.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aisberg-0.2.2.dist-info/top_level.txt,sha256=Q5EmjFuum10zWgRsk-c4cGKe25F8J7HYvga4AtTwYXQ,8
-aisberg-0.2.2.dist-info/RECORD,,
+aisberg-0.2.3.dist-info/licenses/LICENSE,sha256=dCGgiGa4f14s5mzTYnj1wIZNqGRVG2bUYyyk7nzKlEo,406
+aisberg-0.2.3.dist-info/METADATA,sha256=Ygs7157QFW5ab7YfgwtdrVA9If5oPKABAaKEadW6yS4,5604
+aisberg-0.2.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aisberg-0.2.3.dist-info/top_level.txt,sha256=Q5EmjFuum10zWgRsk-c4cGKe25F8J7HYvga4AtTwYXQ,8
+aisberg-0.2.3.dist-info/RECORD,,