lionagi 0.9.8__tar.gz → 0.9.9__tar.gz

{lionagi-0.9.8 → lionagi-0.9.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lionagi
-Version: 0.9.8
+Version: 0.9.9
 Summary: An Intelligence Operating System.
 Author-email: HaiyangLi <quantocean.li@gmail.com>
 License:            Apache License

{lionagi-0.9.8 → lionagi-0.9.9}/lionagi/service/providers/anthropic_/messages.py

@@ -29,6 +29,7 @@ CHAT_COMPLETION_CONFIG = {
         "tools",
         "top_p",
         "top_k",
+        "cache_control",
     },
     "allowed_roles": ["user", "assistant"],
 }
@@ -49,6 +50,21 @@ class AnthropicChatCompletionEndPoint(ChatCompletionEndPoint):
         for k, v in kwargs.items():
             if k in self.acceptable_kwargs:
                 payload[k] = v
+        if "cache_control" in payload:
+            cache_control = payload.pop("cache_control")
+            if cache_control:
+                cache_control = "ephemeral"
+                last_message = payload["messages"][-1]["content"]
+                if isinstance(last_message, str):
+                    last_message = {
+                        "type": "text",
+                        "text": last_message,
+                        "cache_control": cache_control,
+                    }
+                elif isinstance(last_message, list) and isinstance(
+                    last_message[-1], dict
+                ):
+                    last_message[-1]["cache_control"] = cache_control
         if "api_key" in kwargs:
             headers["x-api-key"] = kwargs["api_key"]
         headers["anthropic-version"] = kwargs.pop(

{lionagi-0.9.8 → lionagi-0.9.9}/lionagi/service/providers/openai_/chat_completions.py

@@ -4,6 +4,8 @@
 
 from lionagi.service.endpoints.chat_completion import ChatCompletionEndPoint
 
+from .spec import reasoning_models, reasoning_not_support_params
+
 CHAT_COMPLETION_CONFIG = {
     "provider": "openai",
     "base_url": "https://api.openai.com/v1",
@@ -84,9 +86,9 @@ class OpenAIChatCompletionEndPoint(ChatCompletionEndPoint):
         if "api_key" in kwargs:
             headers["Authorization"] = f"Bearer {kwargs['api_key']}"
 
-        if payload.get("model") in ["o1", "o1-2024-12-17"]:
-            payload.pop("temperature", None)
-            payload.pop("top_p", None)
+        if payload.get("model") in reasoning_models:
+            for param in reasoning_not_support_params:
+                payload.pop(param, None)
             if payload["messages"][0].get("role") == "system":
                 payload["messages"][0]["role"] = "developer"
         else:

lionagi-0.9.9/lionagi/service/providers/openai_/spec.py

@@ -0,0 +1,14 @@
+reasoning_models = (
+    "o3-mini-2025-01-31",
+    "o3-mini",
+    "o1",
+    "o1-2024-12-17",
+)
+
+reasoning_not_support_params = (
+    "temperature",
+    "top_p",
+    "logit_bias",
+    "logprobs",
+    "top_logprobs",
+)

lionagi-0.9.9/lionagi/version.py

@@ -0,0 +1 @@
+__version__ = "0.9.9"

{lionagi-0.9.8 → lionagi-0.9.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "lionagi"
-version = "0.9.8"
+version = "0.9.9"
 description = "An Intelligence Operating System."
 authors = [
     { name = "HaiyangLi", email = "quantocean.li@gmail.com" },

{lionagi-0.9.8 → lionagi-0.9.9}/uv.lock

@@ -1208,7 +1208,7 @@ wheels = [
 
 [[package]]
 name = "lionagi"
-version = "0.9.8"
+version = "0.9.9"
 source = { editable = "." }
 dependencies = [
     { name = "aiocache" },

lionagi-0.9.8/lionagi/version.py

@@ -1 +0,0 @@
-__version__ = "0.9.8"