itzam 1.0.0__py3-none-any.whl → 1.1.0__py3-none-any.whl

itzam/__init__.py

@@ -14,10 +14,11 @@ response = client.text.generate(
 print(response.text)
 ```
 """
-from .text.client import TextClient
+from .text import TextClient
 from .threads import ThreadsClient
 from .runs import RunsClient
 from .models import ModelsClient
+from .objects import ObjectsClient
 import dotenv, os
 
 class Itzam:
@@ -37,4 +38,5 @@ class Itzam:
     self.text = TextClient(base_url=base_url, api_key=api_key)
     self.threads = ThreadsClient(base_url=base_url, api_key=api_key)
     self.runs = RunsClient(base_url=base_url, api_key=api_key)
-    self.models = ModelsClient(base_url=base_url, api_key=api_key)
+    self.models = ModelsClient(base_url=base_url, api_key=api_key)
+    self.objects = ObjectsClient(base_url=base_url, api_key=api_key)

itzam/objects/__init__.py

@@ -0,0 +1,4 @@
+from .client import ObjectsClient
+from .models import Response
+
+__all__ = ["ObjectsClient", "Response"]

itzam/objects/client.py

@@ -0,0 +1,70 @@
+from ..base.client import BaseClient
+from ..text import Attachment
+from .models import Response
+import json
+
+class ObjectsClient(BaseClient):
+    """
+    Client for interacting with the Itzam API's objects endpoint.
+    This client provides methods to manage and interact with objects in the Itzam system.
+    """
+
+    def __init__(self, base_url: str, api_key: str):
+        """
+        Initialize the ObjectsClient with the base URL and API key.
+
+        :param base_url: The base URL for the Itzam API.
+        :param api_key: The API key for authentication.
+        """
+        super().__init__(base_url=base_url, api_key=api_key)
+
+    def generate(
+        self,
+        input: str,
+        schema: dict,
+        workflow_slug: str | None = None,
+        thread_id: str | None = None,
+        attachments: list[Attachment] = None,
+        stream: bool = False
+    ):
+        """
+        Generate text using the specified model and prompt.
+        If stream=True, returns a generator yielding text deltas.
+        """
+        if not workflow_slug and not thread_id:
+            raise ValueError("Either 'thread_id' or 'worflow_slug' must be provided.")
+        endpoint = "/api/v1/generate/object"
+
+        data = {
+            "input": input,
+            "schema": schema,
+        }
+        if workflow_slug:
+            data["workflowSlug"] = workflow_slug
+        if thread_id:
+            data["threadId"] = thread_id
+        if attachments:
+            data["attachments"] = [attachment.model_dump() for attachment in attachments]
+
+        if stream == True:
+            return self._stream_object("/api/v1/stream/object", data)
+        else:
+            response = self.request(method="POST", endpoint=endpoint, data=data)
+            return Response.model_validate(response.json())
+
+    def _stream_object(self, endpoint, data):
+        """
+        Internal method to handle streaming text responses.
+        Yields text deltas as they arrive.
+        """
+        response = self.request(method="POST", endpoint=endpoint, data=data, stream=True)
+        for line in response.iter_lines():
+            if line:
+                try:
+                    event = json.loads(line.decode("utf-8").removeprefix("data: "))
+                    if event.get("type") == "text-delta":
+                        yield event.get("textDelta")
+                    elif event.get("type") == "object":
+                        yield str(event.get("object"))
+                except Exception as e:
+                    continue

itzam/objects/models.py

@@ -0,0 +1,9 @@
+from pydantic import BaseModel, Field
+from ..text import ResponseMetadata
+
+class Response(BaseModel):
+    """
+    Represents the response from the object generation API.
+    """
+    metadata: ResponseMetadata = Field(description="Metadata about the generation process")
+    object: dict = Field(description="The generated object based on the schema provided")

{itzam-1.0.0.dist-info → itzam-1.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: itzam
-Version: 1.0.0
+Version: 1.1.0
 Summary: Python SDK to interact with Itzam's API
 Author-email: Joaquim Cassano <joaquim@cassano.com.br>
 Requires-Python: <4.0,>=3.10
@@ -9,6 +9,8 @@ Requires-Dist: pydantic
 Requires-Dist: requests
 Requires-Dist: rich
 
+![PyPI Version](https://img.shields.io/pypi/v/itzam?label=pypi%20package)
+![PyPI Downloads](https://img.shields.io/pypi/dm/itzam)
 # Itzam python sdk
 ![itzam logo](https://pbs.twimg.com/profile_banners/1930643525021937664/1749136962/600x200)
 
@@ -59,6 +61,10 @@ client = Itzam()
 - **Runs**: Inspect previous runs and their metadata.
 
 ## Usage Examples
+### See available models
+```shell
+python3 -m itzam.models
+```
 
 ### Generate Text
 

{itzam-1.0.0.dist-info → itzam-1.1.0.dist-info}/RECORD

@@ -1,10 +1,13 @@
-itzam/__init__.py,sha256=bENDRFgjtSUMWgRRjdRH_EdCm6uABG1K9O16eDKc50M,1286
+itzam/__init__.py,sha256=EF_ElsUjkMlHLTFz5AqLrtBfWMTntXQ6xUNGlVD-sfI,1383
 itzam/base/__init__.py,sha256=DalvyyupMefIabAm2Zr08bJEgMe_4YW18wZAghz9ng4,98
 itzam/base/client.py,sha256=PoZ8qfSVOSKFEqQmspr0EcOdFHWHUkaAZryxJ7xrhYg,1395
 itzam/models/__init__.py,sha256=zQnpPpnTxv1UN1nryKH0IbUPMRJj_-Z6GhIgT0Z0Dk8,69
 itzam/models/__main__.py,sha256=A5m-oX_MYktOObynqzFNpzwzhFM-s7d3Dn0q_OP0l4I,868
 itzam/models/client.py,sha256=oyzZbsbNTtQaEapiP-RPBgevBWeYDPwZ_hMmrjpGsoE,568
 itzam/models/models.py,sha256=exr_ZgRSkERB_0-gJaWfYYcGxwa3dnVx69TkEbGrDzo,329
+itzam/objects/__init__.py,sha256=mtOAakOo6jOcmRgEC1CY5whHB5H40XQwetKFSbbJRUY,103
+itzam/objects/client.py,sha256=UOwohJtffDEb96V4MPELu95SzASJ1TceTqfKi9-cRGA,2515
+itzam/objects/models.py,sha256=RPBq5geEUiuZbzHmL7HGbUWuBw-Kjjgjk3cjIkB-bys,359
 itzam/runs/__init__.py,sha256=krE8QnjXNZ-lhVAVnHh7kTSMiXJdCEJc1Jm7iu57A4k,63
 itzam/runs/client.py,sha256=zgjCUghl4-a_FwYWOFe6n1hbX0kv66G2IS0XHxcz8w4,527
 itzam/runs/models.py,sha256=feYO8lljmwoOk3IY1zXlZ_rZRYev890V1iQwbrVL6rs,828
@@ -15,7 +18,7 @@ itzam/threads/__init__.py,sha256=-JGYGwOaiiozh3f-PWKZKIKWMN8WY4Wj1Y1SWbV74CA,109
 itzam/threads/client.py,sha256=jXALIcUhiNi4kdmQNf3BGD_WjDGeXHSVknZ41tNwAi8,1816
 itzam/threads/models.py,sha256=RYgiDmm9hsgJbuI88Zu3MGfNhmsE7-UMokrs7P9-Y1Y,829
 itzam/utils/exceptions.py,sha256=Zmxn-ZQjNFwRqHFKTg2uxQf53f7cKi2sgbj65XXIHuE,5749
-itzam-1.0.0.dist-info/METADATA,sha256=L8sBw3OyBohjzrW2zhX0O6S80y7rYWWNVH92jYbyjxc,2422
-itzam-1.0.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-itzam-1.0.0.dist-info/top_level.txt,sha256=krzAg5bkN3qcDdhHvmUgpOOWct4KcdpwbfeR-p-QHJE,6
-itzam-1.0.0.dist-info/RECORD,,
+itzam-1.1.0.dist-info/METADATA,sha256=avVuH3ndReAlSYNDLWtP2AsqDp1OG7zNcniUjH6JwKo,2614
+itzam-1.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+itzam-1.1.0.dist-info/top_level.txt,sha256=krzAg5bkN3qcDdhHvmUgpOOWct4KcdpwbfeR-p-QHJE,6
+itzam-1.1.0.dist-info/RECORD,,