lionagi 0.9.8__py3-none-any.whl → 0.9.9__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.
Files changed (8) hide show
  1. lionagi/service/providers/anthropic_/messages.py +16 -0
  2. lionagi/service/providers/openai_/chat_completions.py +5 -3
  3. lionagi/service/providers/openai_/spec.py +14 -0
  4. lionagi/version.py +1 -1
  5. {lionagi-0.9.8.dist-info → lionagi-0.9.9.dist-info}/METADATA +1 -1
  6. {lionagi-0.9.8.dist-info → lionagi-0.9.9.dist-info}/RECORD +8 -7
  7. {lionagi-0.9.8.dist-info → lionagi-0.9.9.dist-info}/WHEEL +0 -0
  8. {lionagi-0.9.8.dist-info → lionagi-0.9.9.dist-info}/licenses/LICENSE +0 -0
lionagi/service/providers/anthropic_/messages.py CHANGED
@@ -29,6 +29,7 @@ CHAT_COMPLETION_CONFIG = {
29
29
  "tools",
30
30
  "top_p",
31
31
  "top_k",
32
+ "cache_control",
32
33
  },
33
34
  "allowed_roles": ["user", "assistant"],
34
35
  }
@@ -49,6 +50,21 @@ class AnthropicChatCompletionEndPoint(ChatCompletionEndPoint):
49
50
  for k, v in kwargs.items():
50
51
  if k in self.acceptable_kwargs:
51
52
  payload[k] = v
53
+ if "cache_control" in payload:
54
+ cache_control = payload.pop("cache_control")
55
+ if cache_control:
56
+ cache_control = "ephemeral"
57
+ last_message = payload["messages"][-1]["content"]
58
+ if isinstance(last_message, str):
59
+ last_message = {
60
+ "type": "text",
61
+ "text": last_message,
62
+ "cache_control": cache_control,
63
+ }
64
+ elif isinstance(last_message, list) and isinstance(
65
+ last_message[-1], dict
66
+ ):
67
+ last_message[-1]["cache_control"] = cache_control
52
68
  if "api_key" in kwargs:
53
69
  headers["x-api-key"] = kwargs["api_key"]
54
70
  headers["anthropic-version"] = kwargs.pop(
lionagi/service/providers/openai_/chat_completions.py CHANGED
@@ -4,6 +4,8 @@
4
4
 
5
5
  from lionagi.service.endpoints.chat_completion import ChatCompletionEndPoint
6
6
 
7
+ from .spec import reasoning_models, reasoning_not_support_params
8
+
7
9
  CHAT_COMPLETION_CONFIG = {
8
10
  "provider": "openai",
9
11
  "base_url": "https://api.openai.com/v1",
@@ -84,9 +86,9 @@ class OpenAIChatCompletionEndPoint(ChatCompletionEndPoint):
84
86
  if "api_key" in kwargs:
85
87
  headers["Authorization"] = f"Bearer {kwargs['api_key']}"
86
88
 
87
- if payload.get("model") in ["o1", "o1-2024-12-17"]:
88
- payload.pop("temperature", None)
89
- payload.pop("top_p", None)
89
+ if payload.get("model") in reasoning_models:
90
+ for param in reasoning_not_support_params:
91
+ payload.pop(param, None)
90
92
  if payload["messages"][0].get("role") == "system":
91
93
  payload["messages"][0]["role"] = "developer"
92
94
  else:
lionagi/service/providers/openai_/spec.py ADDED
@@ -0,0 +1,14 @@
1
+ reasoning_models = (
2
+ "o3-mini-2025-01-31",
3
+ "o3-mini",
4
+ "o1",
5
+ "o1-2024-12-17",
6
+ )
7
+
8
+ reasoning_not_support_params = (
9
+ "temperature",
10
+ "top_p",
11
+ "logit_bias",
12
+ "logprobs",
13
+ "top_logprobs",
14
+ )
lionagi/version.py CHANGED
@@ -1 +1 @@
1
- __version__ = "0.9.8"
1
+ __version__ = "0.9.9"
{lionagi-0.9.8.dist-info → lionagi-0.9.9.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: lionagi
3
- Version: 0.9.8
3
+ Version: 0.9.9
4
4
  Summary: An Intelligence Operating System.
5
5
  Author-email: HaiyangLi <quantocean.li@gmail.com>
6
6
  License: Apache License
{lionagi-0.9.8.dist-info → lionagi-0.9.9.dist-info}/RECORD RENAMED
@@ -4,7 +4,7 @@ lionagi/_errors.py,sha256=JlBTFJnRWtVYcRxKb7fWFiJHLbykl1E19mSJ8sXYVxg,455
4
4
  lionagi/_types.py,sha256=9g7iytvSj3UjZxD-jL06_fxuNfgZyWT3Qnp0XYp1wQU,63
5
5
  lionagi/settings.py,sha256=W52mM34E6jXF3GyqCFzVREKZrmnUqtZm_BVDsUiDI_s,1627
6
6
  lionagi/utils.py,sha256=K36D9AAGiMPR4eM9tYoiVgvH-NdPPSeMQPls09s7keQ,73223
7
- lionagi/version.py,sha256=GI3Cr15wC6XhdahtMoSOf62d2gYve5G_PiR_xx_RGKc,22
7
+ lionagi/version.py,sha256=nhsA3KKA-CXSYpbzuChuLyxpDepY_-JffnUNClcYEaU,22
8
8
  lionagi/libs/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
9
9
  lionagi/libs/parse.py,sha256=JRS3bql0InHJqATnAatl-hQv4N--XXw4P77JHhTFnrc,1011
10
10
  lionagi/libs/file/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
@@ -173,7 +173,7 @@ lionagi/service/endpoints/token_calculator.py,sha256=-AKwDvV7C8k8MTmd62ymT0ETSUP
173
173
  lionagi/service/providers/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
174
174
  lionagi/service/providers/types.py,sha256=NS91ysRFwOs0cpNeQgFhmtl7JrSz2pJm-tt7sZILmQY,683
175
175
  lionagi/service/providers/anthropic_/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
176
- lionagi/service/providers/anthropic_/messages.py,sha256=elV2_HkB6dm2NtZDxpzzuhCOAL-dlhJBAdxSl9dnWA0,1760
176
+ lionagi/service/providers/anthropic_/messages.py,sha256=QIczhn5CfrufcNGF-QqGqKnd5xOpsWEDor5g07jf-2k,2472
177
177
  lionagi/service/providers/exa_/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
178
178
  lionagi/service/providers/exa_/models.py,sha256=263KP-JSxbxmomNrFeYjB_cebquoMOsCJeWsiKZ0mL4,5420
179
179
  lionagi/service/providers/exa_/search.py,sha256=Z3pyJH8KiWiquJSJw8Rd6D7x43BwTFHb2ESsgSicCk0,1932
@@ -183,7 +183,8 @@ lionagi/service/providers/groq_/chat_completions.py,sha256=q1p_1qus4vduWWBzs9V_K
183
183
  lionagi/service/providers/ollama_/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
184
184
  lionagi/service/providers/ollama_/chat_completions.py,sha256=gPemTJO9dPd68l60kDYppO29uYqmLole4JGkv9Fz1Us,3764
185
185
  lionagi/service/providers/openai_/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
186
- lionagi/service/providers/openai_/chat_completions.py,sha256=y3RAgI5WQH5EwT1wZxp5ttnkCxUJEcOM3Cta6u9cpQo,2867
186
+ lionagi/service/providers/openai_/chat_completions.py,sha256=CxALMCOP3KH16O583xngkAfFot8Hvbr_o1HeLvCh0Us,2938
187
+ lionagi/service/providers/openai_/spec.py,sha256=GJkndsSQmGT-DpSnePz077HK7YQ-xp6-SCTlws6RRf0,217
187
188
  lionagi/service/providers/openrouter_/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
188
189
  lionagi/service/providers/openrouter_/chat_completions.py,sha256=0pdXjJCXmCPPbKKVubrnqofaodTOxWTJam8fd3NgrNk,1525
189
190
  lionagi/service/providers/perplexity_/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
@@ -213,7 +214,7 @@ lionagi/tools/file/writer.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,
213
214
  lionagi/tools/file/providers/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
214
215
  lionagi/tools/file/providers/docling_.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
215
216
  lionagi/tools/query/__init__.py,sha256=5y5joOZzfFWERl75auAcNcKC3lImVJ5ZZGvvHZUFCJM,112
216
- lionagi-0.9.8.dist-info/METADATA,sha256=Ywnd678RVNXSuGJYVwjOs9LAO8Ec6PXR8zY8nM0oemc,18436
217
- lionagi-0.9.8.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
218
- lionagi-0.9.8.dist-info/licenses/LICENSE,sha256=VXFWsdoN5AAknBCgFqQNgPWYx7OPp-PFEP961zGdOjc,11288
219
- lionagi-0.9.8.dist-info/RECORD,,
217
+ lionagi-0.9.9.dist-info/METADATA,sha256=gaYeleZglGI0tRleBY61jftb61YB83alH8cwS2EuE9o,18436
218
+ lionagi-0.9.9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
219
+ lionagi-0.9.9.dist-info/licenses/LICENSE,sha256=VXFWsdoN5AAknBCgFqQNgPWYx7OPp-PFEP961zGdOjc,11288
220
+ lionagi-0.9.9.dist-info/RECORD,,