seekrai 0.5.8__py3-none-any.whl → 0.5.9__py3-none-any.whl

seekrai/resources/chat/completions.py

@@ -112,11 +112,10 @@ class ChatCompletions:
             n=n,
             safety_model=safety_model,
             response_format=response_format,
-            tools=tools or [],
+            tools=tools or None,
         ).model_dump()
         if tool_choice is not None:
             parameter_payload["tool_choice"] = tool_choice
-
         response, _, _ = requestor.request(
             options=SeekrFlowRequest(
                 method="POST",
@@ -158,8 +157,8 @@ class AsyncChatCompletions:
         n: int = 1,
         safety_model: str | None = None,
         response_format: Dict[str, str | Dict[str, Any]] | None = None,
-        # tools: Dict[str, str | Dict[str, Any]] | None = None,
-        # tool_choice: str | Dict[str, str | Dict[str, str]] | None = None,
+        tools: Dict[str, str | Dict[str, Any]] | None = None,
+        tool_choice: str | Dict[str, str | Dict[str, str]] | None = None,
     ) -> AsyncGenerator[ChatCompletionChunk, None] | ChatCompletionResponse:
         """
         Async method to generate completions based on a given prompt using a specified model.
@@ -235,9 +234,10 @@ class AsyncChatCompletions:
             n=n,
             safety_model=safety_model,
             response_format=response_format,
-            # tools=tools,
-            # tool_choice=tool_choice,
+            tools=tools or None,
         ).model_dump()
+        if tool_choice is not None:
+            parameter_payload["tool_choice"] = tool_choice
 
         response, _, _ = await requestor.arequest(
             options=SeekrFlowRequest(

{seekrai-0.5.8.dist-info → seekrai-0.5.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: seekrai
-Version: 0.5.8
+Version: 0.5.9
 Summary: Python client for SeekrAI
 License: Apache-2.0
 Author: SeekrFlow

{seekrai-0.5.8.dist-info → seekrai-0.5.9.dist-info}/RECORD

@@ -13,7 +13,7 @@ seekrai/resources/agents/agents.py,sha256=d3C4VaXq4q5WlUr0f8txdRD3hvEKKvz4UQA-a_
 seekrai/resources/agents/threads.py,sha256=BwZ2_6wlezsb12PQjEw1fgdJh5S83SPgD6qZQoGvyIM,14544
 seekrai/resources/alignment.py,sha256=IOKlKK2I9_NhS9pwcrsd9-5OO7lVT8Uw0y_wuGHOnyA,5839
 seekrai/resources/chat/__init__.py,sha256=KmtPupgECtEN80NyvcnSmieTAFXhwmVxhMHP0qhspA4,618
-seekrai/resources/chat/completions.py,sha256=QcNsZqpJjkbiD_dN9fGBw8hgb_hLiHUqoMzHV_vtkqQ,12258
+seekrai/resources/chat/completions.py,sha256=-nFk5aL1ejZ9WUi_ksDHqCnEnT2CjPX-RaigT3N35gA,12317
 seekrai/resources/completions.py,sha256=JhTN_lW2mblfHHONFmPC7QZei3wo5vx6GliMs9FkbOY,8452
 seekrai/resources/deployments.py,sha256=DY7IN7QgqDduCHGNuHENSVwrE5PXFL88jWgh8SES7Qk,5970
 seekrai/resources/embeddings.py,sha256=7G-VisYrT9J35-hcKB8cXhs8BSi93IfveQKfVSC7diA,2585
@@ -65,8 +65,8 @@ seekrai/utils/api_helpers.py,sha256=0Y8BblNIr9h_R12zdmhkxgTlxgoRkbq84QNi4nNWGu8,
 seekrai/utils/files.py,sha256=7ixn_hgV-6pEhYqLyOp-EN0o8c1CzUwJzX9n3PQ5oqo,7164
 seekrai/utils/tools.py,sha256=jgJTL-dOIouDbEJLdQpQfpXhqaz_poQYS52adyUtBjo,1781
 seekrai/version.py,sha256=q6iGQVFor8zXiPP5F-3vy9TndOxKv5JXbaNJ2kdOQws,125
-seekrai-0.5.8.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-seekrai-0.5.8.dist-info/METADATA,sha256=aayyfKC7AZUcN4afB112qyaccnW9yWdblbki6N3Z5IY,4780
-seekrai-0.5.8.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-seekrai-0.5.8.dist-info/entry_points.txt,sha256=N49yOEGi1sK7Xr13F_rkkcOxQ88suyiMoOmRhUHTZ_U,48
-seekrai-0.5.8.dist-info/RECORD,,
+seekrai-0.5.9.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+seekrai-0.5.9.dist-info/METADATA,sha256=ZOnhU4QoIs5dAPLVX_zic81Zc5OZE0-EpkXad0iWJLc,4780
+seekrai-0.5.9.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+seekrai-0.5.9.dist-info/entry_points.txt,sha256=N49yOEGi1sK7Xr13F_rkkcOxQ88suyiMoOmRhUHTZ_U,48
+seekrai-0.5.9.dist-info/RECORD,,