llmir 0.0.3__tar.gz → 0.0.5__tar.gz

{llmir-0.0.3 → llmir-0.0.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: llmir
-Version: 0.0.3
+Version: 0.0.5
 Summary: Core message and tool IR for LLM pipelines
 Author: Mathis Siebert
 Requires-Python: >=3.11

{llmir-0.0.3 → llmir-0.0.5}/llmir/adapter/__init__.py

@@ -1,4 +1,4 @@
-from .openai import to_openai, OpenAIMessage, OpenAIMessageToolResponse, OpenAIContent, OpenAITextContent, OpenAIImageURLContent
+from .openai import to_openai, OpenAIMessage, OpenAIMessageToolResponse, OpenAIContent, OpenAITextContent, OpenAIImageURLContent, OpenAIImageURLURL, OpenAIToolCallContent, OpenAIToolCallFunction
 
 __all__ = [
     "to_openai",
@@ -7,4 +7,7 @@ __all__ = [
     "OpenAIContent",
     "OpenAITextContent",
     "OpenAIImageURLContent",
+    "OpenAIImageURLURL",
+    "OpenAIToolCallContent",
+    "OpenAIToolCallFunction",
 ]

{llmir-0.0.3 → llmir-0.0.5}/llmir/adapter/openai.py

@@ -1,20 +1,35 @@
 from typing import TypedDict, Literal, Union
 
-from ..messages import AIMessage, AIMessageToolResponse
-from ..chunks import AIChunk, AIChunkText, AIChunkImageURL, AIChunkFile
+from ..messages import AIMessages, AIMessageToolResponse
+from ..chunks import AIChunk, AIChunkText, AIChunkImageURL, AIChunkFile, AIChunkToolCall
 import base64
+import json
 
 
 class OpenAITextContent(TypedDict):
     type: Literal["text"]
     text: str
 
+class OpenAIImageURLURL(TypedDict):
+    url: str
 
 class OpenAIImageURLContent(TypedDict):
     type: Literal["image_url"]
-    image_url: dict[str, str]
+    image_url: OpenAIImageURLURL
 
-OpenAIContent = Union[OpenAITextContent, OpenAIImageURLContent]
+
+class OpenAIToolCallFunction(TypedDict):
+    name: str
+    arguments: str
+
+class OpenAIToolCallContent(TypedDict):
+    id: str
+    type: Literal["function"]
+    function: OpenAIToolCallFunction
+
+    
+
+OpenAIContent = Union[OpenAITextContent, OpenAIImageURLContent, OpenAIToolCallContent]
 
 
 class OpenAIMessage(TypedDict):
@@ -26,7 +41,7 @@ class OpenAIMessageToolResponse(OpenAIMessage):
     name: str
 
 
-def to_openai(messages: list[AIMessage]) -> list[OpenAIMessage]:
+def to_openai(messages: list[AIMessages]) -> list[OpenAIMessage]:
     
 
     result: list[OpenAIMessage] = []
@@ -77,7 +92,7 @@ def chunk_to_openai(chunk: AIChunk) -> OpenAIContent:
                         "url": f"data:{chunk.mimetype};base64,{base64_data}",
                     }
                 )
-            elif chunk.mimetype == ("text/plain"):
+            elif chunk.mimetype == "text/plain":
                 text = chunk.bytes.decode(encoding="utf-8")
                 return OpenAITextContent(
                     type="text",
@@ -85,5 +100,14 @@ def chunk_to_openai(chunk: AIChunk) -> OpenAIContent:
                 )
             else:
                 raise ValueError(f"Unsupported file type for OpenAI: {chunk.mimetype}")
+        case AIChunkToolCall():
+            return OpenAIToolCallContent(
+                id=chunk.id,
+                type="function",
+                function=OpenAIToolCallFunction(
+                    name=chunk.name,
+                    arguments=json.dumps(chunk.arguments)
+                )
+            )
         case _:
             raise ValueError(f"Unsupported chunk type: {type(chunk)}")

{llmir-0.0.3 → llmir-0.0.5}/llmir.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: llmir
-Version: 0.0.3
+Version: 0.0.5
 Summary: Core message and tool IR for LLM pipelines
 Author: Mathis Siebert
 Requires-Python: >=3.11

{llmir-0.0.3 → llmir-0.0.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "llmir"
-version = "0.0.3"
+version = "0.0.5"
 description = "Core message and tool IR for LLM pipelines"
 authors = [{ name="Mathis Siebert" }]
 requires-python = ">=3.11"