lionagi 0.2.3__py3-none-any.whl → 0.2.4__py3-none-any.whl
Sign up to get free protection for your applications and to get access to all the features.
- lionagi/integrations/provider/litellm.py +7 -5
- lionagi/version.py +1 -1
- {lionagi-0.2.3.dist-info → lionagi-0.2.4.dist-info}/METADATA +2 -2
- {lionagi-0.2.3.dist-info → lionagi-0.2.4.dist-info}/RECORD +7 -7
- {lionagi-0.2.3.dist-info → lionagi-0.2.4.dist-info}/WHEEL +1 -1
- {lionagi-0.2.3.dist-info → lionagi-0.2.4.dist-info}/LICENSE +0 -0
- {lionagi-0.2.3.dist-info → lionagi-0.2.4.dist-info}/top_level.txt +0 -0
|
@@ -26,9 +26,10 @@ class LiteLLMService(BaseService):
|
|
|
26
26
|
|
|
27
27
|
SysUtil.check_import("litellm")
|
|
28
28
|
|
|
29
|
-
|
|
30
|
-
|
|
31
|
-
|
|
29
|
+
import litellm
|
|
30
|
+
litellm.drop_params = True
|
|
31
|
+
|
|
32
|
+
self.acompletion = litellm.acompletion
|
|
32
33
|
self.model = model
|
|
33
34
|
self.kwargs = kwargs
|
|
34
35
|
self.allowed_kwargs = allowed_kwargs
|
|
@@ -41,12 +42,13 @@ class LiteLLMService(BaseService):
|
|
|
41
42
|
config[k] = v
|
|
42
43
|
|
|
43
44
|
kwargs = {**self.kwargs, **config}
|
|
45
|
+
kwargs["model"] = self.model or kwargs.get("model")
|
|
44
46
|
|
|
45
47
|
try:
|
|
46
48
|
completion = await self.acompletion(
|
|
47
|
-
|
|
49
|
+
messages=messages, **kwargs
|
|
48
50
|
)
|
|
49
|
-
return payload, completion
|
|
51
|
+
return payload, completion.model_dump()
|
|
50
52
|
except Exception as e:
|
|
51
53
|
self.status_tracker.num_tasks_failed += 1
|
|
52
54
|
raise e
|
lionagi/version.py
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
__version__ = "0.2.
|
|
1
|
+
__version__ = "0.2.4"
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.1
|
|
2
2
|
Name: lionagi
|
|
3
|
-
Version: 0.2.
|
|
3
|
+
Version: 0.2.4
|
|
4
4
|
Summary: Towards automated general intelligence.
|
|
5
5
|
Author: HaiyangLi
|
|
6
6
|
Author-email: Haiyang Li <ocean@lionagi.ai>
|
|
@@ -237,7 +237,7 @@ Requires-Dist: boto3 >=1.34.131
|
|
|
237
237
|
### an AGentic Intelligence Operating System
|
|
238
238
|
|
|
239
239
|
```
|
|
240
|
-
pip install lionagi==0.2.
|
|
240
|
+
pip install lionagi==0.2.3
|
|
241
241
|
```
|
|
242
242
|
|
|
243
243
|
**Powerful Intelligent Workflow Automation**
|
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
lionagi/__init__.py,sha256=amQal6CUIv4QDI4Qmb0M5qwTbn3_F430Wl5vaSNG6-U,1952
|
|
2
|
-
lionagi/version.py,sha256=
|
|
2
|
+
lionagi/version.py,sha256=SBl2EPFW-ltPvQ7vbVWItyAsz3aKYIpjO7vcfr84GkU,22
|
|
3
3
|
lionagi/core/__init__.py,sha256=Il5Q9ATdX8yXqVxtP_nYqUhExzxPC_qk_WXQ_4h0exg,16
|
|
4
4
|
lionagi/core/_setting/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
5
5
|
lionagi/core/_setting/_setting.py,sha256=p23fHtrzIZlQ8s8CDZICn8C6k7mdB_088nIDx19JaqM,1907
|
|
@@ -167,7 +167,7 @@ lionagi/integrations/loader/load.py,sha256=TCKXePhEdNDeBH7iv__7lG0GwhPU4j3qyjhTY
|
|
|
167
167
|
lionagi/integrations/loader/load_util.py,sha256=F0VAGS16sl6P4aolP0Us0E641RGv8zpTefdHgRDL-p4,6438
|
|
168
168
|
lionagi/integrations/provider/__init__.py,sha256=MJhnq2tkBRcMH-3utc0G-Co20MmsxLBbp3fUwHrJGQ8,198
|
|
169
169
|
lionagi/integrations/provider/_mapping.py,sha256=1IcE4IE7L1zhTIOE1W4LslGcanomdAzJYiYyGMPDhN8,1346
|
|
170
|
-
lionagi/integrations/provider/litellm.py,sha256=
|
|
170
|
+
lionagi/integrations/provider/litellm.py,sha256=RTCqe7JiSOWRQXj6Ic4nL-m1jxGGJG1lw8HIl_qppLQ,1320
|
|
171
171
|
lionagi/integrations/provider/mistralai.py,sha256=G-StbfrnUcWZvl0eRby6CZYXxmJf6BRMFzDaix-brmU,7
|
|
172
172
|
lionagi/integrations/provider/mlx_service.py,sha256=m9c9793xlNVNkNBXlNcavczueF5bdWD1urrIw6rDf5E,1726
|
|
173
173
|
lionagi/integrations/provider/oai.py,sha256=Bkh0VJPJwbQGYOFfBdSR5JG3sBkCfHHZm3dx3gmkIms,7054
|
|
@@ -251,8 +251,8 @@ lionagi/tests/test_core/test_structure/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JC
|
|
|
251
251
|
lionagi/tests/test_core/test_structure/test_base_structure.py,sha256=Ik_G7lI8dV3Q1jw1f6mynoPQVYnvVTkOQrgAf7fpIxY,9387
|
|
252
252
|
lionagi/tests/test_core/test_structure/test_graph.py,sha256=uC68zubRVbwSZ5JGzld7LnlpPqfGMNMZ9FDu9mcHp60,2161
|
|
253
253
|
lionagi/tests/test_core/test_structure/test_tree.py,sha256=0vJMvG0YChCl7W2NwF0yl9ukSftlkuoMOZFrHwwMgL8,1943
|
|
254
|
-
lionagi-0.2.
|
|
255
|
-
lionagi-0.2.
|
|
256
|
-
lionagi-0.2.
|
|
257
|
-
lionagi-0.2.
|
|
258
|
-
lionagi-0.2.
|
|
254
|
+
lionagi-0.2.4.dist-info/LICENSE,sha256=VXFWsdoN5AAknBCgFqQNgPWYx7OPp-PFEP961zGdOjc,11288
|
|
255
|
+
lionagi-0.2.4.dist-info/METADATA,sha256=laDb2JoSzm-iwVq0_qrZVtaxcHMXxWjF6zbia1iP5HE,16276
|
|
256
|
+
lionagi-0.2.4.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
|
|
257
|
+
lionagi-0.2.4.dist-info/top_level.txt,sha256=szvch_d2jE1Lu9ZIKsl26Ll6BGfYfbOgt5lm-UpFSo4,8
|
|
258
|
+
lionagi-0.2.4.dist-info/RECORD,,
|
|
File without changes
|
|
File without changes
|