llmir 0.0.1__py3-none-any.whl → 0.0.3__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
llmir/__init__.py CHANGED
@@ -1,6 +1,6 @@
1
1
  from .chunks import AIChunk, AIChunkText, AIChunkFile, AIChunkImageURL, AIChunkToolCall
2
2
  from .roles import AIRoles
3
- from .messages import AIMessage, AIMessageToolResponse
3
+ from .messages import AIMessages, AIMessage, AIMessageToolResponse
4
4
  from .tools import Tool
5
5
 
6
6
  __all__ = [
@@ -10,6 +10,7 @@ __all__ = [
10
10
  "AIChunkImageURL",
11
11
  "AIChunkToolCall",
12
12
  "AIRoles",
13
+ "AIMessages",
13
14
  "AIMessage",
14
15
  "AIMessageToolResponse",
15
16
 
llmir/messages.py CHANGED
@@ -1,18 +1,18 @@
1
1
  from pydantic import BaseModel, Field
2
-
2
+ from typing import Literal
3
3
  from .chunks import AIChunk
4
4
  from .roles import AIRoles
5
5
 
6
-
7
6
  class AIMessage(BaseModel):
8
7
 
9
- role: AIRoles
8
+ role: Literal[AIRoles.USER, AIRoles.MODEL, AIRoles.SYSTEM]
10
9
  chunks: list[AIChunk] = Field(default_factory=list[AIChunk])
11
10
 
12
- class AIMessageToolResponse(AIMessage):
11
+ class AIMessageToolResponse(BaseModel):
13
12
 
13
+ role: Literal[AIRoles.TOOL] = AIRoles.TOOL
14
+ chunks: list[AIChunk] = Field(default_factory=list[AIChunk])
14
15
  id: str
15
16
  name: str
16
- role: AIRoles = AIRoles.TOOL
17
-
18
17
 
18
+ AIMessages = AIMessage | AIMessageToolResponse
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: llmir
3
- Version: 0.0.1
3
+ Version: 0.0.3
4
4
  Summary: Core message and tool IR for LLM pipelines
5
5
  Author: Mathis Siebert
6
6
  Requires-Python: >=3.11
@@ -1,13 +1,13 @@
1
- llmir/__init__.py,sha256=mxV4TWB8xrR8qa3y5gfQZn-YfKRQTYWsPVvyeIjz7js,381
1
+ llmir/__init__.py,sha256=CxdIqErvDMHNP7n7VO4PAsm-WzytCgYKyGEGHEd5B_s,411
2
2
  llmir/chunks.py,sha256=lnN7-3kjYalM6YO0v4ZM2-SYDNZDFfrergj3KEzcv0o,658
3
- llmir/messages.py,sha256=G_F_gLGvBqzPZbni0iaWAMYG3-cDPJCeLJL2-9ZcAqI,311
3
+ llmir/messages.py,sha256=uABQ3xs_DJ0IWWoePl7M5YqAcpUWyT2vtokz_OuqkJQ,505
4
4
  llmir/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
5
5
  llmir/rich_repr.py,sha256=ZGAfpcPkYJSUpFdir8B1MoORVFBSzVckCrRHVVkJIj4,559
6
6
  llmir/roles.py,sha256=B9PmxTHAb8XoiA9en_J1yIP19woddzx4SRUz5XkUxFc,133
7
7
  llmir/tools.py,sha256=fpkC3IafHqZJnwo7imtHdkTdGLBHQuR6DtdztXZrlWg,147
8
8
  llmir/adapter/__init__.py,sha256=2DKTM3v35_UPVTKiseo6rtIHj_P2OdUgxhNY_PKs5K8,289
9
9
  llmir/adapter/openai.py,sha256=_DRbBWhTYUdSEsdCTyn-FraqCQ5Ke2XRnHotJHAWyGM,2644
10
- llmir-0.0.1.dist-info/METADATA,sha256=Oh-jOxYnQTw17bXZWhkJ3T8fENnEhUIUPCG-UiT_EZg,177
11
- llmir-0.0.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
12
- llmir-0.0.1.dist-info/top_level.txt,sha256=fm538OsNUSYa_71IPztSSN2PBCvcbaD5PmUIzaO5hQs,6
13
- llmir-0.0.1.dist-info/RECORD,,
10
+ llmir-0.0.3.dist-info/METADATA,sha256=oplLtL3stW_GU_HcoBQzDSXbH-dEjnhVIhTZPzp78js,177
11
+ llmir-0.0.3.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
12
+ llmir-0.0.3.dist-info/top_level.txt,sha256=fm538OsNUSYa_71IPztSSN2PBCvcbaD5PmUIzaO5hQs,6
13
+ llmir-0.0.3.dist-info/RECORD,,
File without changes