powerailabs-contextkit 0.2.0__tar.gz → 0.3.0__tar.gz

{powerailabs_contextkit-0.2.0 → powerailabs_contextkit-0.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: powerailabs-contextkit
-Version: 0.2.0
+Version: 0.3.0
 Summary: Assemble: declare prioritized, pinnable context blocks; pack them to a token budget with an inspectable receipt.
 Author: Raghav Mishra
 License-Expression: MIT

{powerailabs_contextkit-0.2.0 → powerailabs_contextkit-0.3.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "powerailabs-contextkit"
-version = "0.2.0"
+version = "0.3.0"
 description = "Assemble: declare prioritized, pinnable context blocks; pack them to a token budget with an inspectable receipt."
 requires-python = ">=3.11"
 license = "MIT"

{powerailabs_contextkit-0.2.0 → powerailabs_contextkit-0.3.0}/src/powerailabs/contextkit/__init__.py

@@ -154,6 +154,46 @@ class Context:
         rest = [m for m in self._messages if m["role"] != "system"]
         return system, rest
 
+    def for_gemini(self) -> tuple[str, list[dict]]:
+        """Gemini adapter: returns ``(system_instruction, contents)``.
+
+        ``contents`` are ``{"role": "user"|"model", "parts": [{"text": ...}]}`` (Gemini uses
+        ``model``, not ``assistant``); system blocks become the separate ``system_instruction``.
+        """
+        if not self._messages:
+            self.assemble()
+        system = "\n\n".join(m["content"] for m in self._messages if m["role"] == "system")
+        contents = [
+            {
+                "role": "model" if m["role"] == "assistant" else "user",
+                "parts": [{"text": m["content"]}],
+            }
+            for m in self._messages
+            if m["role"] != "system"
+        ]
+        return system, contents
+
+    def for_bedrock(self) -> tuple[list[dict], list[dict]]:
+        """Bedrock Converse adapter: returns ``(system, messages)``.
+
+        ``system`` is ``[{"text": ...}]`` (or empty); ``messages`` are
+        ``{"role": "user"|"assistant", "content": [{"text": ...}]}`` — Bedrock allows only those
+        two roles, so non-user blocks map to ``assistant``.
+        """
+        if not self._messages:
+            self.assemble()
+        system_text = "\n\n".join(m["content"] for m in self._messages if m["role"] == "system")
+        system = [{"text": system_text}] if system_text else []
+        messages = [
+            {
+                "role": "user" if m["role"] == "user" else "assistant",
+                "content": [{"text": m["content"]}],
+            }
+            for m in self._messages
+            if m["role"] != "system"
+        ]
+        return system, messages
+
     # ------------------------------------------------------------------ internals
 
     def _pack(self, budget_tokens: int, *, emit: bool) -> tuple[list[dict], AssemblyReport]:

{powerailabs_contextkit-0.2.0 → powerailabs_contextkit-0.3.0}/tests/test_contextkit.py

@@ -153,3 +153,25 @@ def test_for_anthropic_splits_system():
     system, messages = ctx.for_anthropic()
     assert system == "you are helpful"
     assert all(m["role"] != "system" for m in messages)
+
+
+def test_for_gemini_adapter():
+    ctx = Context(budget_tokens=1000, model="gpt-4o")
+    ctx.add(Block("be helpful", priority=10, pin=True, role="system"))
+    ctx.add(Block("prior reply", priority=5, role="assistant"))
+    ctx.add(Block("question", priority=9, pin=True, role="user"))
+    system, contents = ctx.for_gemini()
+    assert system == "be helpful"
+    roles = [c["role"] for c in contents]
+    assert "model" in roles and "user" in roles and "system" not in roles  # assistant -> model
+    assert contents[0]["parts"] == [{"text": "be helpful"}] or contents[0]["parts"][0]["text"]
+
+
+def test_for_bedrock_adapter():
+    ctx = Context(budget_tokens=1000, model="gpt-4o")
+    ctx.add(Block("be helpful", priority=10, pin=True, role="system"))
+    ctx.add(Block("question", priority=9, pin=True, role="user"))
+    system, messages = ctx.for_bedrock()
+    assert system == [{"text": "be helpful"}]
+    assert messages == [{"role": "user", "content": [{"text": "question"}]}]
+    assert all(m["role"] in ("user", "assistant") for m in messages)