lionagi 0.4.0__py3-none-any.whl → 0.5.0__py3-none-any.whl

Sign up to get free protection for your applications and to get access to all the features.
Files changed (584) hide show
  1. lionagi/__init__.py +14 -46
  2. lionagi/core/__init__.py +3 -1
  3. lionagi/core/_class_registry.py +69 -0
  4. lionagi/core/action/__init__.py +3 -13
  5. lionagi/core/action/action_manager.py +287 -0
  6. lionagi/core/action/base.py +109 -0
  7. lionagi/core/action/function_calling.py +127 -92
  8. lionagi/core/action/tool.py +172 -70
  9. lionagi/core/action/types.py +16 -0
  10. lionagi/core/communication/__init__.py +3 -0
  11. lionagi/core/communication/action_request.py +163 -0
  12. lionagi/core/communication/action_response.py +149 -0
  13. lionagi/core/communication/assistant_response.py +161 -0
  14. lionagi/core/communication/base_mail.py +49 -0
  15. lionagi/core/communication/instruction.py +376 -0
  16. lionagi/core/communication/message.py +286 -0
  17. lionagi/core/communication/message_manager.py +530 -0
  18. lionagi/core/communication/system.py +116 -0
  19. lionagi/core/communication/templates/README.md +28 -0
  20. lionagi/core/communication/templates/action_request.jinja2 +5 -0
  21. lionagi/core/communication/templates/action_response.jinja2 +9 -0
  22. lionagi/core/communication/templates/assistant_response.jinja2 +2 -0
  23. lionagi/core/communication/templates/instruction_message.jinja2 +61 -0
  24. lionagi/core/communication/templates/system_message.jinja2 +11 -0
  25. lionagi/core/communication/templates/tool_schemas.jinja2 +7 -0
  26. lionagi/core/communication/types.py +27 -0
  27. lionagi/core/communication/utils.py +254 -0
  28. lionagi/core/forms/__init__.py +3 -0
  29. lionagi/core/forms/base.py +232 -0
  30. lionagi/core/forms/form.py +791 -0
  31. lionagi/core/forms/report.py +321 -0
  32. lionagi/core/forms/types.py +13 -0
  33. lionagi/core/forms/utils.py +26 -0
  34. lionagi/core/generic/__init__.py +3 -6
  35. lionagi/core/generic/component.py +422 -0
  36. lionagi/core/generic/edge.py +143 -101
  37. lionagi/core/generic/element.py +195 -0
  38. lionagi/core/generic/graph.py +297 -180
  39. lionagi/core/generic/log.py +151 -0
  40. lionagi/core/generic/log_manager.py +320 -0
  41. lionagi/core/generic/node.py +7 -229
  42. lionagi/core/generic/pile.py +1017 -0
  43. lionagi/core/generic/progression.py +388 -0
  44. lionagi/core/generic/types.py +23 -0
  45. lionagi/core/generic/utils.py +50 -0
  46. lionagi/core/models/__init__.py +5 -0
  47. lionagi/core/models/base.py +85 -0
  48. lionagi/core/models/field_model.py +122 -0
  49. lionagi/core/models/new_model_params.py +195 -0
  50. lionagi/core/models/note.py +351 -0
  51. lionagi/core/models/operable_model.py +392 -0
  52. lionagi/core/models/schema_model.py +50 -0
  53. lionagi/core/models/types.py +10 -0
  54. lionagi/core/session/__init__.py +3 -0
  55. lionagi/core/session/branch.py +115 -415
  56. lionagi/core/session/branch_mixins.py +507 -0
  57. lionagi/core/session/session.py +122 -257
  58. lionagi/core/session/types.py +8 -0
  59. lionagi/core/typing/__init__.py +9 -0
  60. lionagi/core/typing/concepts.py +132 -0
  61. lionagi/core/typing/config.py +15 -0
  62. lionagi/core/typing/id.py +221 -0
  63. lionagi/core/typing/pydantic_.py +33 -0
  64. lionagi/core/typing/typing_.py +54 -0
  65. lionagi/integrations/__init__.py +0 -1
  66. lionagi/integrations/anthropic_/AnthropicModel.py +268 -0
  67. lionagi/integrations/anthropic_/AnthropicService.py +113 -0
  68. lionagi/integrations/anthropic_/__init__.py +3 -0
  69. lionagi/integrations/anthropic_/anthropic_max_output_token_data.yaml +7 -0
  70. lionagi/integrations/anthropic_/anthropic_price_data.yaml +14 -0
  71. lionagi/integrations/anthropic_/api_endpoints/__init__.py +3 -0
  72. lionagi/integrations/anthropic_/api_endpoints/api_request.py +277 -0
  73. lionagi/integrations/anthropic_/api_endpoints/data_models.py +40 -0
  74. lionagi/integrations/anthropic_/api_endpoints/match_response.py +119 -0
  75. lionagi/integrations/anthropic_/api_endpoints/messages/__init__.py +3 -0
  76. lionagi/integrations/anthropic_/api_endpoints/messages/request/__init__.py +3 -0
  77. lionagi/integrations/anthropic_/api_endpoints/messages/request/message_models.py +14 -0
  78. lionagi/integrations/anthropic_/api_endpoints/messages/request/request_body.py +74 -0
  79. lionagi/integrations/anthropic_/api_endpoints/messages/response/content_models.py +32 -0
  80. lionagi/integrations/anthropic_/api_endpoints/messages/response/response_body.py +101 -0
  81. lionagi/integrations/anthropic_/api_endpoints/messages/response/usage_models.py +25 -0
  82. lionagi/integrations/anthropic_/version.py +5 -0
  83. lionagi/integrations/groq_/GroqModel.py +318 -0
  84. lionagi/integrations/groq_/GroqService.py +147 -0
  85. lionagi/integrations/groq_/__init__.py +3 -0
  86. lionagi/integrations/groq_/api_endpoints/data_models.py +187 -0
  87. lionagi/integrations/groq_/api_endpoints/groq_request.py +288 -0
  88. lionagi/integrations/groq_/api_endpoints/match_response.py +106 -0
  89. lionagi/integrations/groq_/api_endpoints/response_utils.py +105 -0
  90. lionagi/integrations/groq_/groq_max_output_token_data.yaml +21 -0
  91. lionagi/integrations/groq_/groq_price_data.yaml +58 -0
  92. lionagi/integrations/groq_/groq_rate_limits.yaml +105 -0
  93. lionagi/integrations/groq_/version.py +5 -0
  94. lionagi/integrations/litellm_/__init__.py +3 -0
  95. lionagi/integrations/litellm_/imodel.py +69 -0
  96. lionagi/integrations/ollama_/OllamaModel.py +244 -0
  97. lionagi/integrations/ollama_/OllamaService.py +138 -0
  98. lionagi/integrations/ollama_/__init__.py +3 -0
  99. lionagi/integrations/ollama_/api_endpoints/__init__.py +3 -0
  100. lionagi/integrations/ollama_/api_endpoints/api_request.py +179 -0
  101. lionagi/integrations/ollama_/api_endpoints/chat_completion/__init__.py +3 -0
  102. lionagi/integrations/ollama_/api_endpoints/chat_completion/message_models.py +31 -0
  103. lionagi/integrations/ollama_/api_endpoints/chat_completion/request_body.py +46 -0
  104. lionagi/integrations/ollama_/api_endpoints/chat_completion/response_body.py +67 -0
  105. lionagi/integrations/ollama_/api_endpoints/chat_completion/tool_models.py +49 -0
  106. lionagi/integrations/ollama_/api_endpoints/completion/request_body.py +72 -0
  107. lionagi/integrations/ollama_/api_endpoints/completion/response_body.py +59 -0
  108. lionagi/integrations/ollama_/api_endpoints/data_models.py +15 -0
  109. lionagi/integrations/ollama_/api_endpoints/embedding/request_body.py +33 -0
  110. lionagi/integrations/ollama_/api_endpoints/embedding/response_body.py +29 -0
  111. lionagi/integrations/ollama_/api_endpoints/match_data_model.py +62 -0
  112. lionagi/integrations/ollama_/api_endpoints/match_response.py +190 -0
  113. lionagi/integrations/ollama_/api_endpoints/model/__init__.py +3 -0
  114. lionagi/integrations/ollama_/api_endpoints/model/copy_model.py +13 -0
  115. lionagi/integrations/ollama_/api_endpoints/model/create_model.py +28 -0
  116. lionagi/integrations/ollama_/api_endpoints/model/delete_model.py +11 -0
  117. lionagi/integrations/ollama_/api_endpoints/model/list_model.py +60 -0
  118. lionagi/integrations/ollama_/api_endpoints/model/pull_model.py +34 -0
  119. lionagi/integrations/ollama_/api_endpoints/model/push_model.py +35 -0
  120. lionagi/integrations/ollama_/api_endpoints/model/show_model.py +36 -0
  121. lionagi/integrations/ollama_/api_endpoints/option_models.py +68 -0
  122. lionagi/integrations/openai_/OpenAIModel.py +414 -0
  123. lionagi/integrations/openai_/OpenAIService.py +426 -0
  124. lionagi/integrations/openai_/api_endpoints/__init__.py +3 -0
  125. lionagi/integrations/openai_/api_endpoints/api_request.py +277 -0
  126. lionagi/integrations/openai_/api_endpoints/audio/__init__.py +9 -0
  127. lionagi/integrations/openai_/api_endpoints/audio/speech_models.py +34 -0
  128. lionagi/integrations/openai_/api_endpoints/audio/transcription_models.py +136 -0
  129. lionagi/integrations/openai_/api_endpoints/audio/translation_models.py +41 -0
  130. lionagi/integrations/openai_/api_endpoints/audio/types.py +41 -0
  131. lionagi/integrations/openai_/api_endpoints/batch/__init__.py +17 -0
  132. lionagi/integrations/openai_/api_endpoints/batch/batch_models.py +146 -0
  133. lionagi/integrations/openai_/api_endpoints/batch/cancel_batch.py +7 -0
  134. lionagi/integrations/openai_/api_endpoints/batch/create_batch.py +26 -0
  135. lionagi/integrations/openai_/api_endpoints/batch/list_batch.py +37 -0
  136. lionagi/integrations/openai_/api_endpoints/batch/request_object_models.py +65 -0
  137. lionagi/integrations/openai_/api_endpoints/batch/retrieve_batch.py +7 -0
  138. lionagi/integrations/openai_/api_endpoints/batch/types.py +4 -0
  139. lionagi/integrations/openai_/api_endpoints/chat_completions/__init__.py +1 -0
  140. lionagi/integrations/openai_/api_endpoints/chat_completions/request/__init__.py +39 -0
  141. lionagi/integrations/openai_/api_endpoints/chat_completions/request/message_models.py +121 -0
  142. lionagi/integrations/openai_/api_endpoints/chat_completions/request/request_body.py +221 -0
  143. lionagi/integrations/openai_/api_endpoints/chat_completions/request/response_format.py +71 -0
  144. lionagi/integrations/openai_/api_endpoints/chat_completions/request/stream_options.py +14 -0
  145. lionagi/integrations/openai_/api_endpoints/chat_completions/request/tool_choice_models.py +17 -0
  146. lionagi/integrations/openai_/api_endpoints/chat_completions/request/tool_models.py +54 -0
  147. lionagi/integrations/openai_/api_endpoints/chat_completions/request/types.py +18 -0
  148. lionagi/integrations/openai_/api_endpoints/chat_completions/response/choice_models.py +62 -0
  149. lionagi/integrations/openai_/api_endpoints/chat_completions/response/function_models.py +16 -0
  150. lionagi/integrations/openai_/api_endpoints/chat_completions/response/log_prob_models.py +47 -0
  151. lionagi/integrations/openai_/api_endpoints/chat_completions/response/message_models.py +25 -0
  152. lionagi/integrations/openai_/api_endpoints/chat_completions/response/response_body.py +99 -0
  153. lionagi/integrations/openai_/api_endpoints/chat_completions/response/types.py +8 -0
  154. lionagi/integrations/openai_/api_endpoints/chat_completions/response/usage_models.py +24 -0
  155. lionagi/integrations/openai_/api_endpoints/chat_completions/util.py +46 -0
  156. lionagi/integrations/openai_/api_endpoints/data_models.py +23 -0
  157. lionagi/integrations/openai_/api_endpoints/embeddings/__init__.py +3 -0
  158. lionagi/integrations/openai_/api_endpoints/embeddings/request_body.py +79 -0
  159. lionagi/integrations/openai_/api_endpoints/embeddings/response_body.py +67 -0
  160. lionagi/integrations/openai_/api_endpoints/files/__init__.py +11 -0
  161. lionagi/integrations/openai_/api_endpoints/files/delete_file.py +20 -0
  162. lionagi/integrations/openai_/api_endpoints/files/file_models.py +56 -0
  163. lionagi/integrations/openai_/api_endpoints/files/list_files.py +27 -0
  164. lionagi/integrations/openai_/api_endpoints/files/retrieve_file.py +9 -0
  165. lionagi/integrations/openai_/api_endpoints/files/upload_file.py +38 -0
  166. lionagi/integrations/openai_/api_endpoints/fine_tuning/__init__.py +37 -0
  167. lionagi/integrations/openai_/api_endpoints/fine_tuning/cancel_jobs.py +9 -0
  168. lionagi/integrations/openai_/api_endpoints/fine_tuning/create_jobs.py +133 -0
  169. lionagi/integrations/openai_/api_endpoints/fine_tuning/fine_tuning_job_checkpoint_models.py +58 -0
  170. lionagi/integrations/openai_/api_endpoints/fine_tuning/fine_tuning_job_event_models.py +31 -0
  171. lionagi/integrations/openai_/api_endpoints/fine_tuning/fine_tuning_job_models.py +140 -0
  172. lionagi/integrations/openai_/api_endpoints/fine_tuning/list_fine_tuning_checkpoints.py +51 -0
  173. lionagi/integrations/openai_/api_endpoints/fine_tuning/list_fine_tuning_events.py +42 -0
  174. lionagi/integrations/openai_/api_endpoints/fine_tuning/list_fine_tuning_jobs.py +31 -0
  175. lionagi/integrations/openai_/api_endpoints/fine_tuning/retrieve_jobs.py +9 -0
  176. lionagi/integrations/openai_/api_endpoints/fine_tuning/training_format.py +30 -0
  177. lionagi/integrations/openai_/api_endpoints/images/__init__.py +9 -0
  178. lionagi/integrations/openai_/api_endpoints/images/image_edit_models.py +69 -0
  179. lionagi/integrations/openai_/api_endpoints/images/image_models.py +56 -0
  180. lionagi/integrations/openai_/api_endpoints/images/image_variation_models.py +56 -0
  181. lionagi/integrations/openai_/api_endpoints/images/response_body.py +30 -0
  182. lionagi/integrations/openai_/api_endpoints/match_data_model.py +197 -0
  183. lionagi/integrations/openai_/api_endpoints/match_response.py +336 -0
  184. lionagi/integrations/openai_/api_endpoints/models/__init__.py +7 -0
  185. lionagi/integrations/openai_/api_endpoints/models/delete_fine_tuned_model.py +17 -0
  186. lionagi/integrations/openai_/api_endpoints/models/models_models.py +31 -0
  187. lionagi/integrations/openai_/api_endpoints/models/retrieve_model.py +9 -0
  188. lionagi/integrations/openai_/api_endpoints/moderations/__init__.py +3 -0
  189. lionagi/integrations/openai_/api_endpoints/moderations/request_body.py +20 -0
  190. lionagi/integrations/openai_/api_endpoints/moderations/response_body.py +139 -0
  191. lionagi/integrations/openai_/api_endpoints/uploads/__init__.py +19 -0
  192. lionagi/integrations/openai_/api_endpoints/uploads/add_upload_part.py +11 -0
  193. lionagi/integrations/openai_/api_endpoints/uploads/cancel_upload.py +7 -0
  194. lionagi/integrations/openai_/api_endpoints/uploads/complete_upload.py +18 -0
  195. lionagi/integrations/openai_/api_endpoints/uploads/create_upload.py +17 -0
  196. lionagi/integrations/openai_/api_endpoints/uploads/uploads_models.py +52 -0
  197. lionagi/integrations/openai_/image_token_calculator/image_token_calculator.py +92 -0
  198. lionagi/integrations/openai_/image_token_calculator/openai_image_token_data.yaml +15 -0
  199. lionagi/integrations/openai_/openai_max_output_token_data.yaml +12 -0
  200. lionagi/integrations/openai_/openai_price_data.yaml +26 -0
  201. lionagi/integrations/openai_/version.py +1 -0
  202. lionagi/integrations/pandas_/__init__.py +24 -0
  203. lionagi/integrations/pandas_/extend_df.py +61 -0
  204. lionagi/integrations/pandas_/read.py +103 -0
  205. lionagi/integrations/pandas_/remove_rows.py +61 -0
  206. lionagi/integrations/pandas_/replace_keywords.py +65 -0
  207. lionagi/integrations/pandas_/save.py +131 -0
  208. lionagi/integrations/pandas_/search_keywords.py +69 -0
  209. lionagi/integrations/pandas_/to_df.py +196 -0
  210. lionagi/integrations/pandas_/update_cells.py +54 -0
  211. lionagi/integrations/perplexity_/PerplexityModel.py +269 -0
  212. lionagi/integrations/perplexity_/PerplexityService.py +109 -0
  213. lionagi/integrations/perplexity_/__init__.py +3 -0
  214. lionagi/integrations/perplexity_/api_endpoints/api_request.py +171 -0
  215. lionagi/integrations/perplexity_/api_endpoints/chat_completions/request/request_body.py +121 -0
  216. lionagi/integrations/perplexity_/api_endpoints/chat_completions/response/response_body.py +146 -0
  217. lionagi/integrations/perplexity_/api_endpoints/data_models.py +63 -0
  218. lionagi/integrations/perplexity_/api_endpoints/match_response.py +26 -0
  219. lionagi/integrations/perplexity_/perplexity_max_output_token_data.yaml +3 -0
  220. lionagi/integrations/perplexity_/perplexity_price_data.yaml +10 -0
  221. lionagi/integrations/perplexity_/version.py +1 -0
  222. lionagi/integrations/pydantic_/__init__.py +8 -0
  223. lionagi/integrations/pydantic_/break_down_annotation.py +81 -0
  224. lionagi/integrations/pydantic_/new_model.py +208 -0
  225. lionagi/integrations/services.py +17 -0
  226. lionagi/libs/__init__.py +0 -55
  227. lionagi/libs/compress/models.py +62 -0
  228. lionagi/libs/compress/utils.py +81 -0
  229. lionagi/libs/constants.py +98 -0
  230. lionagi/libs/file/chunk.py +265 -0
  231. lionagi/libs/file/file_ops.py +114 -0
  232. lionagi/libs/file/params.py +212 -0
  233. lionagi/libs/file/path.py +301 -0
  234. lionagi/libs/file/process.py +139 -0
  235. lionagi/libs/file/save.py +90 -0
  236. lionagi/libs/file/types.py +22 -0
  237. lionagi/libs/func/async_calls/__init__.py +21 -0
  238. lionagi/libs/func/async_calls/alcall.py +157 -0
  239. lionagi/libs/func/async_calls/bcall.py +82 -0
  240. lionagi/libs/func/async_calls/mcall.py +134 -0
  241. lionagi/libs/func/async_calls/pcall.py +149 -0
  242. lionagi/libs/func/async_calls/rcall.py +185 -0
  243. lionagi/libs/func/async_calls/tcall.py +114 -0
  244. lionagi/libs/func/async_calls/ucall.py +85 -0
  245. lionagi/libs/func/decorators.py +277 -0
  246. lionagi/libs/func/lcall.py +57 -0
  247. lionagi/libs/func/params.py +64 -0
  248. lionagi/libs/func/throttle.py +119 -0
  249. lionagi/libs/func/types.py +39 -0
  250. lionagi/libs/func/utils.py +96 -0
  251. lionagi/libs/package/imports.py +162 -0
  252. lionagi/libs/package/management.py +58 -0
  253. lionagi/libs/package/params.py +26 -0
  254. lionagi/libs/package/system.py +18 -0
  255. lionagi/libs/package/types.py +26 -0
  256. lionagi/libs/parse/__init__.py +1 -0
  257. lionagi/libs/parse/flatten/__init__.py +9 -0
  258. lionagi/libs/parse/flatten/flatten.py +168 -0
  259. lionagi/libs/parse/flatten/params.py +52 -0
  260. lionagi/libs/parse/flatten/unflatten.py +79 -0
  261. lionagi/libs/parse/json/__init__.py +27 -0
  262. lionagi/libs/parse/json/as_readable.py +104 -0
  263. lionagi/libs/parse/json/extract.py +102 -0
  264. lionagi/libs/parse/json/parse.py +179 -0
  265. lionagi/libs/parse/json/schema.py +227 -0
  266. lionagi/libs/parse/json/to_json.py +71 -0
  267. lionagi/libs/parse/nested/__init__.py +33 -0
  268. lionagi/libs/parse/nested/nfilter.py +55 -0
  269. lionagi/libs/parse/nested/nget.py +40 -0
  270. lionagi/libs/parse/nested/ninsert.py +103 -0
  271. lionagi/libs/parse/nested/nmerge.py +155 -0
  272. lionagi/libs/parse/nested/npop.py +66 -0
  273. lionagi/libs/parse/nested/nset.py +89 -0
  274. lionagi/libs/parse/nested/to_flat_list.py +64 -0
  275. lionagi/libs/parse/nested/utils.py +185 -0
  276. lionagi/libs/parse/string_parse/__init__.py +11 -0
  277. lionagi/libs/parse/string_parse/code_block.py +73 -0
  278. lionagi/libs/parse/string_parse/docstring.py +179 -0
  279. lionagi/libs/parse/string_parse/function_.py +92 -0
  280. lionagi/libs/parse/type_convert/__init__.py +19 -0
  281. lionagi/libs/parse/type_convert/params.py +145 -0
  282. lionagi/libs/parse/type_convert/to_dict.py +333 -0
  283. lionagi/libs/parse/type_convert/to_list.py +186 -0
  284. lionagi/libs/parse/type_convert/to_num.py +358 -0
  285. lionagi/libs/parse/type_convert/to_str.py +195 -0
  286. lionagi/libs/parse/types.py +9 -0
  287. lionagi/libs/parse/validate/__init__.py +14 -0
  288. lionagi/libs/parse/validate/boolean.py +96 -0
  289. lionagi/libs/parse/validate/keys.py +150 -0
  290. lionagi/libs/parse/validate/mapping.py +109 -0
  291. lionagi/libs/parse/validate/params.py +62 -0
  292. lionagi/libs/parse/xml/__init__.py +10 -0
  293. lionagi/libs/parse/xml/convert.py +56 -0
  294. lionagi/libs/parse/xml/parser.py +93 -0
  295. lionagi/libs/string_similarity/__init__.py +32 -0
  296. lionagi/libs/string_similarity/algorithms.py +219 -0
  297. lionagi/libs/string_similarity/matcher.py +102 -0
  298. lionagi/libs/string_similarity/utils.py +15 -0
  299. lionagi/libs/utils.py +255 -0
  300. lionagi/operations/__init__.py +3 -6
  301. lionagi/operations/brainstorm/__init__.py +3 -0
  302. lionagi/operations/brainstorm/brainstorm.py +204 -0
  303. lionagi/operations/brainstorm/prompt.py +1 -0
  304. lionagi/operations/plan/__init__.py +3 -0
  305. lionagi/operations/plan/plan.py +172 -0
  306. lionagi/operations/plan/prompt.py +21 -0
  307. lionagi/operations/select/__init__.py +3 -0
  308. lionagi/operations/select/prompt.py +1 -0
  309. lionagi/operations/select/select.py +100 -0
  310. lionagi/operations/select/utils.py +107 -0
  311. lionagi/operations/utils.py +35 -0
  312. lionagi/protocols/adapters/adapter.py +79 -0
  313. lionagi/protocols/adapters/json_adapter.py +43 -0
  314. lionagi/protocols/adapters/pandas_adapter.py +96 -0
  315. lionagi/protocols/configs/__init__.py +15 -0
  316. lionagi/protocols/configs/branch_config.py +86 -0
  317. lionagi/protocols/configs/id_config.py +15 -0
  318. lionagi/protocols/configs/imodel_config.py +73 -0
  319. lionagi/protocols/configs/log_config.py +93 -0
  320. lionagi/protocols/configs/retry_config.py +29 -0
  321. lionagi/protocols/operatives/__init__.py +15 -0
  322. lionagi/protocols/operatives/action.py +181 -0
  323. lionagi/protocols/operatives/instruct.py +196 -0
  324. lionagi/protocols/operatives/operative.py +182 -0
  325. lionagi/protocols/operatives/prompts.py +232 -0
  326. lionagi/protocols/operatives/reason.py +56 -0
  327. lionagi/protocols/operatives/step.py +217 -0
  328. lionagi/protocols/registries/_component_registry.py +19 -0
  329. lionagi/protocols/registries/_pile_registry.py +26 -0
  330. lionagi/service/__init__.py +13 -0
  331. lionagi/service/complete_request_info.py +11 -0
  332. lionagi/service/imodel.py +110 -0
  333. lionagi/service/rate_limiter.py +108 -0
  334. lionagi/service/service.py +37 -0
  335. lionagi/service/service_match_util.py +131 -0
  336. lionagi/service/service_util.py +72 -0
  337. lionagi/service/token_calculator.py +51 -0
  338. lionagi/settings.py +136 -0
  339. lionagi/strategies/base.py +53 -0
  340. lionagi/strategies/concurrent.py +71 -0
  341. lionagi/strategies/concurrent_chunk.py +43 -0
  342. lionagi/strategies/concurrent_sequential_chunk.py +104 -0
  343. lionagi/strategies/params.py +128 -0
  344. lionagi/strategies/sequential.py +23 -0
  345. lionagi/strategies/sequential_chunk.py +89 -0
  346. lionagi/strategies/sequential_concurrent_chunk.py +100 -0
  347. lionagi/strategies/types.py +21 -0
  348. lionagi/strategies/utils.py +49 -0
  349. lionagi/version.py +1 -1
  350. lionagi-0.5.0.dist-info/METADATA +348 -0
  351. lionagi-0.5.0.dist-info/RECORD +373 -0
  352. {lionagi-0.4.0.dist-info → lionagi-0.5.0.dist-info}/WHEEL +1 -1
  353. lionagi/core/_setting/_setting.py +0 -59
  354. lionagi/core/action/README.md +0 -20
  355. lionagi/core/action/manual.py +0 -1
  356. lionagi/core/action/node.py +0 -94
  357. lionagi/core/action/tool_manager.py +0 -342
  358. lionagi/core/agent/README.md +0 -1
  359. lionagi/core/agent/base_agent.py +0 -82
  360. lionagi/core/agent/eval/README.md +0 -1
  361. lionagi/core/agent/eval/evaluator.py +0 -1
  362. lionagi/core/agent/eval/vote.py +0 -40
  363. lionagi/core/agent/learn/learner.py +0 -59
  364. lionagi/core/agent/plan/unit_template.py +0 -1
  365. lionagi/core/collections/README.md +0 -23
  366. lionagi/core/collections/__init__.py +0 -16
  367. lionagi/core/collections/_logger.py +0 -312
  368. lionagi/core/collections/abc/README.md +0 -63
  369. lionagi/core/collections/abc/__init__.py +0 -53
  370. lionagi/core/collections/abc/component.py +0 -620
  371. lionagi/core/collections/abc/concepts.py +0 -277
  372. lionagi/core/collections/abc/exceptions.py +0 -136
  373. lionagi/core/collections/abc/util.py +0 -45
  374. lionagi/core/collections/exchange.py +0 -146
  375. lionagi/core/collections/flow.py +0 -416
  376. lionagi/core/collections/model.py +0 -465
  377. lionagi/core/collections/pile.py +0 -1232
  378. lionagi/core/collections/progression.py +0 -221
  379. lionagi/core/collections/util.py +0 -73
  380. lionagi/core/director/README.md +0 -1
  381. lionagi/core/director/direct.py +0 -298
  382. lionagi/core/director/director.py +0 -2
  383. lionagi/core/director/operations/select.py +0 -3
  384. lionagi/core/director/operations/utils.py +0 -6
  385. lionagi/core/engine/branch_engine.py +0 -361
  386. lionagi/core/engine/instruction_map_engine.py +0 -213
  387. lionagi/core/engine/sandbox_.py +0 -16
  388. lionagi/core/engine/script_engine.py +0 -89
  389. lionagi/core/executor/base_executor.py +0 -97
  390. lionagi/core/executor/graph_executor.py +0 -335
  391. lionagi/core/executor/neo4j_executor.py +0 -394
  392. lionagi/core/generic/README.md +0 -0
  393. lionagi/core/generic/edge_condition.py +0 -17
  394. lionagi/core/generic/hyperedge.py +0 -1
  395. lionagi/core/generic/tree.py +0 -49
  396. lionagi/core/generic/tree_node.py +0 -85
  397. lionagi/core/mail/__init__.py +0 -11
  398. lionagi/core/mail/mail.py +0 -26
  399. lionagi/core/mail/mail_manager.py +0 -185
  400. lionagi/core/mail/package.py +0 -49
  401. lionagi/core/mail/start_mail.py +0 -36
  402. lionagi/core/message/__init__.py +0 -18
  403. lionagi/core/message/action_request.py +0 -114
  404. lionagi/core/message/action_response.py +0 -121
  405. lionagi/core/message/assistant_response.py +0 -80
  406. lionagi/core/message/instruction.py +0 -194
  407. lionagi/core/message/message.py +0 -86
  408. lionagi/core/message/system.py +0 -71
  409. lionagi/core/message/util.py +0 -274
  410. lionagi/core/report/__init__.py +0 -4
  411. lionagi/core/report/base.py +0 -201
  412. lionagi/core/report/form.py +0 -212
  413. lionagi/core/report/report.py +0 -150
  414. lionagi/core/report/util.py +0 -15
  415. lionagi/core/rule/_default.py +0 -17
  416. lionagi/core/rule/action.py +0 -87
  417. lionagi/core/rule/base.py +0 -234
  418. lionagi/core/rule/boolean.py +0 -56
  419. lionagi/core/rule/choice.py +0 -48
  420. lionagi/core/rule/mapping.py +0 -82
  421. lionagi/core/rule/number.py +0 -73
  422. lionagi/core/rule/rulebook.py +0 -45
  423. lionagi/core/rule/string.py +0 -43
  424. lionagi/core/rule/util.py +0 -0
  425. lionagi/core/session/directive_mixin.py +0 -307
  426. lionagi/core/structure/__init__.py +0 -1
  427. lionagi/core/structure/chain.py +0 -1
  428. lionagi/core/structure/forest.py +0 -1
  429. lionagi/core/structure/graph.py +0 -1
  430. lionagi/core/structure/tree.py +0 -1
  431. lionagi/core/unit/__init__.py +0 -4
  432. lionagi/core/unit/parallel_unit.py +0 -234
  433. lionagi/core/unit/template/action.py +0 -65
  434. lionagi/core/unit/template/base.py +0 -35
  435. lionagi/core/unit/template/plan.py +0 -69
  436. lionagi/core/unit/template/predict.py +0 -95
  437. lionagi/core/unit/template/score.py +0 -108
  438. lionagi/core/unit/template/select.py +0 -91
  439. lionagi/core/unit/unit.py +0 -452
  440. lionagi/core/unit/unit_form.py +0 -290
  441. lionagi/core/unit/unit_mixin.py +0 -1166
  442. lionagi/core/unit/util.py +0 -103
  443. lionagi/core/validator/validator.py +0 -376
  444. lionagi/core/work/work.py +0 -59
  445. lionagi/core/work/work_edge.py +0 -102
  446. lionagi/core/work/work_function.py +0 -114
  447. lionagi/core/work/work_function_node.py +0 -50
  448. lionagi/core/work/work_queue.py +0 -90
  449. lionagi/core/work/work_task.py +0 -151
  450. lionagi/core/work/worker.py +0 -410
  451. lionagi/core/work/worker_engine.py +0 -208
  452. lionagi/core/work/worklog.py +0 -108
  453. lionagi/experimental/compressor/base.py +0 -47
  454. lionagi/experimental/compressor/llm_compressor.py +0 -265
  455. lionagi/experimental/compressor/llm_summarizer.py +0 -61
  456. lionagi/experimental/compressor/util.py +0 -70
  457. lionagi/experimental/directive/README.md +0 -1
  458. lionagi/experimental/directive/__init__.py +0 -19
  459. lionagi/experimental/directive/parser/base_parser.py +0 -294
  460. lionagi/experimental/directive/parser/base_syntax.txt +0 -200
  461. lionagi/experimental/directive/template/base_template.py +0 -71
  462. lionagi/experimental/directive/template/schema.py +0 -36
  463. lionagi/experimental/directive/tokenizer.py +0 -59
  464. lionagi/experimental/evaluator/README.md +0 -1
  465. lionagi/experimental/evaluator/ast_evaluator.py +0 -119
  466. lionagi/experimental/evaluator/base_evaluator.py +0 -213
  467. lionagi/experimental/knowledge/__init__.py +0 -0
  468. lionagi/experimental/knowledge/base.py +0 -10
  469. lionagi/experimental/knowledge/graph.py +0 -0
  470. lionagi/experimental/memory/__init__.py +0 -0
  471. lionagi/experimental/strategies/__init__.py +0 -0
  472. lionagi/experimental/strategies/base.py +0 -1
  473. lionagi/integrations/bridge/__init__.py +0 -4
  474. lionagi/integrations/bridge/autogen_/__init__.py +0 -0
  475. lionagi/integrations/bridge/autogen_/autogen_.py +0 -127
  476. lionagi/integrations/bridge/langchain_/__init__.py +0 -0
  477. lionagi/integrations/bridge/langchain_/documents.py +0 -138
  478. lionagi/integrations/bridge/langchain_/langchain_bridge.py +0 -68
  479. lionagi/integrations/bridge/llamaindex_/__init__.py +0 -0
  480. lionagi/integrations/bridge/llamaindex_/index.py +0 -36
  481. lionagi/integrations/bridge/llamaindex_/llama_index_bridge.py +0 -108
  482. lionagi/integrations/bridge/llamaindex_/llama_pack.py +0 -256
  483. lionagi/integrations/bridge/llamaindex_/node_parser.py +0 -92
  484. lionagi/integrations/bridge/llamaindex_/reader.py +0 -201
  485. lionagi/integrations/bridge/llamaindex_/textnode.py +0 -59
  486. lionagi/integrations/bridge/pydantic_/__init__.py +0 -0
  487. lionagi/integrations/bridge/pydantic_/pydantic_bridge.py +0 -7
  488. lionagi/integrations/bridge/transformers_/__init__.py +0 -0
  489. lionagi/integrations/bridge/transformers_/install_.py +0 -39
  490. lionagi/integrations/chunker/__init__.py +0 -0
  491. lionagi/integrations/chunker/chunk.py +0 -314
  492. lionagi/integrations/config/__init__.py +0 -4
  493. lionagi/integrations/config/mlx_configs.py +0 -1
  494. lionagi/integrations/config/oai_configs.py +0 -154
  495. lionagi/integrations/config/ollama_configs.py +0 -1
  496. lionagi/integrations/config/openrouter_configs.py +0 -74
  497. lionagi/integrations/langchain_/__init__.py +0 -0
  498. lionagi/integrations/llamaindex_/__init__.py +0 -0
  499. lionagi/integrations/loader/__init__.py +0 -0
  500. lionagi/integrations/loader/load.py +0 -257
  501. lionagi/integrations/loader/load_util.py +0 -214
  502. lionagi/integrations/provider/__init__.py +0 -11
  503. lionagi/integrations/provider/_mapping.py +0 -47
  504. lionagi/integrations/provider/litellm.py +0 -53
  505. lionagi/integrations/provider/mistralai.py +0 -1
  506. lionagi/integrations/provider/mlx_service.py +0 -55
  507. lionagi/integrations/provider/oai.py +0 -196
  508. lionagi/integrations/provider/ollama.py +0 -55
  509. lionagi/integrations/provider/openrouter.py +0 -170
  510. lionagi/integrations/provider/services.py +0 -138
  511. lionagi/integrations/provider/transformers.py +0 -108
  512. lionagi/integrations/storage/__init__.py +0 -3
  513. lionagi/integrations/storage/neo4j.py +0 -681
  514. lionagi/integrations/storage/storage_util.py +0 -302
  515. lionagi/integrations/storage/structure_excel.py +0 -291
  516. lionagi/integrations/storage/to_csv.py +0 -70
  517. lionagi/integrations/storage/to_excel.py +0 -91
  518. lionagi/libs/ln_api.py +0 -944
  519. lionagi/libs/ln_async.py +0 -208
  520. lionagi/libs/ln_context.py +0 -37
  521. lionagi/libs/ln_convert.py +0 -671
  522. lionagi/libs/ln_dataframe.py +0 -187
  523. lionagi/libs/ln_func_call.py +0 -1328
  524. lionagi/libs/ln_image.py +0 -114
  525. lionagi/libs/ln_knowledge_graph.py +0 -422
  526. lionagi/libs/ln_nested.py +0 -822
  527. lionagi/libs/ln_parse.py +0 -750
  528. lionagi/libs/ln_queue.py +0 -107
  529. lionagi/libs/ln_tokenize.py +0 -179
  530. lionagi/libs/ln_validate.py +0 -299
  531. lionagi/libs/special_tokens.py +0 -172
  532. lionagi/libs/sys_util.py +0 -710
  533. lionagi/lions/__init__.py +0 -0
  534. lionagi/lions/coder/__init__.py +0 -0
  535. lionagi/lions/coder/add_feature.py +0 -20
  536. lionagi/lions/coder/base_prompts.py +0 -22
  537. lionagi/lions/coder/code_form.py +0 -15
  538. lionagi/lions/coder/coder.py +0 -184
  539. lionagi/lions/coder/util.py +0 -101
  540. lionagi/lions/director/__init__.py +0 -0
  541. lionagi/lions/judge/__init__.py +0 -0
  542. lionagi/lions/judge/config.py +0 -8
  543. lionagi/lions/judge/data/__init__.py +0 -0
  544. lionagi/lions/judge/data/sample_codes.py +0 -526
  545. lionagi/lions/judge/data/sample_rurbic.py +0 -48
  546. lionagi/lions/judge/forms/__init__.py +0 -0
  547. lionagi/lions/judge/forms/code_analysis_form.py +0 -126
  548. lionagi/lions/judge/rubric.py +0 -34
  549. lionagi/lions/judge/services/__init__.py +0 -0
  550. lionagi/lions/judge/services/judge_code.py +0 -49
  551. lionagi/lions/researcher/__init__.py +0 -0
  552. lionagi/lions/researcher/data_source/__init__.py +0 -0
  553. lionagi/lions/researcher/data_source/finhub_.py +0 -192
  554. lionagi/lions/researcher/data_source/google_.py +0 -207
  555. lionagi/lions/researcher/data_source/wiki_.py +0 -98
  556. lionagi/lions/researcher/data_source/yfinance_.py +0 -21
  557. lionagi/operations/brainstorm.py +0 -87
  558. lionagi/operations/config.py +0 -6
  559. lionagi/operations/rank.py +0 -102
  560. lionagi/operations/score.py +0 -144
  561. lionagi/operations/select.py +0 -141
  562. lionagi-0.4.0.dist-info/METADATA +0 -241
  563. lionagi-0.4.0.dist-info/RECORD +0 -249
  564. /lionagi/{core/_setting → integrations/anthropic_/api_endpoints/messages/response}/__init__.py +0 -0
  565. /lionagi/{core/agent → integrations/groq_/api_endpoints}/__init__.py +0 -0
  566. /lionagi/{core/agent/eval → integrations/ollama_/api_endpoints/completion}/__init__.py +0 -0
  567. /lionagi/{core/agent/learn → integrations/ollama_/api_endpoints/embedding}/__init__.py +0 -0
  568. /lionagi/{core/agent/plan → integrations/openai_}/__init__.py +0 -0
  569. /lionagi/{core/director → integrations/openai_/api_endpoints/chat_completions/response}/__init__.py +0 -0
  570. /lionagi/{core/director/operations → integrations/openai_/image_token_calculator}/__init__.py +0 -0
  571. /lionagi/{core/engine → integrations/perplexity_/api_endpoints}/__init__.py +0 -0
  572. /lionagi/{core/executor → integrations/perplexity_/api_endpoints/chat_completions}/__init__.py +0 -0
  573. /lionagi/{core/generic/registry/component_registry → integrations/perplexity_/api_endpoints/chat_completions/request}/__init__.py +0 -0
  574. /lionagi/{core/rule → integrations/perplexity_/api_endpoints/chat_completions/response}/__init__.py +0 -0
  575. /lionagi/{core/unit/template → libs/compress}/__init__.py +0 -0
  576. /lionagi/{core/validator → libs/file}/__init__.py +0 -0
  577. /lionagi/{core/work → libs/func}/__init__.py +0 -0
  578. /lionagi/{experimental → libs/package}/__init__.py +0 -0
  579. /lionagi/{core/agent/plan/plan.py → libs/parse/params.py} +0 -0
  580. /lionagi/{experimental/compressor → protocols}/__init__.py +0 -0
  581. /lionagi/{experimental/directive/parser → protocols/adapters}/__init__.py +0 -0
  582. /lionagi/{experimental/directive/template → protocols/registries}/__init__.py +0 -0
  583. /lionagi/{experimental/evaluator → strategies}/__init__.py +0 -0
  584. {lionagi-0.4.0.dist-info → lionagi-0.5.0.dist-info/licenses}/LICENSE +0 -0
@@ -0,0 +1,1017 @@
1
+ # Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
2
+ #
3
+ # SPDX-License-Identifier: Apache-2.0
4
+
5
+ import asyncio
6
+ import threading
7
+ from collections.abc import AsyncIterator, Callable, Iterator, Sequence
8
+ from functools import wraps
9
+ from pathlib import Path
10
+
11
+ import pandas as pd
12
+ from typing_extensions import override
13
+
14
+ from lionagi.core.typing import (
15
+ ID,
16
+ UNDEFINED,
17
+ Any,
18
+ ClassVar,
19
+ Field,
20
+ FieldInfo,
21
+ Generic,
22
+ ItemExistsError,
23
+ ItemNotFoundError,
24
+ Observable,
25
+ Self,
26
+ TypeVar,
27
+ field_serializer,
28
+ )
29
+ from lionagi.libs.parse import is_same_dtype, to_list
30
+ from lionagi.protocols.adapters.adapter import Adapter, AdapterRegistry
31
+ from lionagi.protocols.registries._pile_registry import PileAdapterRegistry
32
+
33
+ from .element import Element
34
+ from .progression import Progression
35
+ from .utils import to_list_type, validate_order
36
+
37
+ T = TypeVar("T", bound=Element)
38
+ D = TypeVar("D")
39
+
40
+
41
+ def synchronized(func: Callable):
42
+ @wraps(func)
43
+ def wrapper(self: "Pile", *args, **kwargs):
44
+ with self.lock:
45
+ return func(self, *args, **kwargs)
46
+
47
+ return wrapper
48
+
49
+
50
+ def async_synchronized(func: Callable):
51
+ @wraps(func)
52
+ async def wrapper(self: "Pile", *args, **kwargs):
53
+ async with self.async_lock:
54
+ return await func(self, *args, **kwargs)
55
+
56
+ return wrapper
57
+
58
+
59
+ class Pile(Element, Generic[T]):
60
+ """Thread-safe async-compatible, ordered collection of elements.
61
+
62
+ The Pile class provides a thread-safe, async-compatible collection with:
63
+ - Type validation and enforcement
64
+ - Order preservation
65
+ - Format adapters (JSON, CSV, Excel)
66
+ - Memory efficient storage
67
+
68
+ Attributes:
69
+ pile_ (dict[str, T]): Internal storage mapping IDs to elements
70
+ item_type (set[type[T]] | None): Allowed element types
71
+ progress (Progression): Order tracking
72
+ strict_type (bool): Whether to enforce strict type checking
73
+ """
74
+
75
+ pile_: dict[str, T] = Field(default_factory=dict)
76
+ item_type: set[type[T]] | None = Field(
77
+ default=None,
78
+ description="Set of allowed types for items in the pile.",
79
+ exclude=True,
80
+ )
81
+ progress: Progression = Field(
82
+ default_factory=Progression,
83
+ description="Progression specifying the order of items in the pile.",
84
+ exclude=True,
85
+ )
86
+ strict_type: bool = Field(
87
+ default=False,
88
+ description="Specify if enforce a strict type check",
89
+ frozen=True,
90
+ )
91
+
92
+ _adapter_registry: ClassVar[AdapterRegistry] = PileAdapterRegistry
93
+
94
+ def __pydantic_extra__(self) -> dict[str, FieldInfo]:
95
+ return {
96
+ "_lock": Field(default_factory=threading.Lock),
97
+ "_async": Field(default_factory=asyncio.Lock),
98
+ }
99
+
100
+ def __pydantic_private__(self) -> dict[str, FieldInfo]:
101
+ return self.__pydantic_extra__()
102
+
103
+ @override
104
+ def __init__(
105
+ self,
106
+ items: ID.ItemSeq = None,
107
+ item_type: set[type[T]] = None,
108
+ order: ID.RefSeq = None,
109
+ strict_type: bool = False,
110
+ **kwargs,
111
+ ) -> None:
112
+ """Initialize a Pile instance.
113
+
114
+ Args:
115
+ items: Initial items for the pile.
116
+ item_type: Allowed types for items in the pile.
117
+ order: Initial order of items (as Progression).
118
+ strict_type: If True, enforce strict type checking.
119
+ """
120
+ _config = {}
121
+ if "ln_id" in kwargs:
122
+ _config["ln_id"] = kwargs["ln_id"]
123
+ if "created" in kwargs:
124
+ _config["created"] = kwargs["created"]
125
+
126
+ super().__init__(strict_type=strict_type, **_config)
127
+ self.item_type = self._validate_item_type(item_type)
128
+ self.pile_ = self._validate_pile(items or kwargs.get("pile_", {}))
129
+ self.progress = self._validate_order(order)
130
+
131
+ # Sync Interface methods
132
+ @override
133
+ @classmethod
134
+ def from_dict(
135
+ cls,
136
+ data: dict[str, Any],
137
+ /,
138
+ ) -> "Pile":
139
+ """Create a Pile instance from a dictionary.
140
+
141
+ Args:
142
+ data: A dictionary containing Pile data.
143
+
144
+ Returns:
145
+ A new Pile instance created from the provided data.
146
+
147
+ Raises:
148
+ ValueError: If the dictionary format is invalid.
149
+ """
150
+ items = data.pop("pile_", [])
151
+ items = [Element.from_dict(i) for i in items]
152
+ return cls(items=items, **data)
153
+
154
+ def __setitem__(
155
+ self,
156
+ key: ID.Ref | ID.RefSeq | int | slice,
157
+ item: ID.ItemSeq | ID.Item,
158
+ ) -> None:
159
+ """Set an item or items in the Pile.
160
+
161
+ Args:
162
+ key: The key to set (index, ID, or slice).
163
+ item: The item(s) to set.
164
+
165
+ Raises:
166
+ TypeError: If item type not allowed.
167
+ KeyError: If key invalid.
168
+ """
169
+ self._setitem(key, item)
170
+
171
+ @synchronized
172
+ def pop(
173
+ self,
174
+ key: ID.Ref | ID.RefSeq | int | slice,
175
+ default: D = UNDEFINED,
176
+ /,
177
+ ) -> T | "Pile" | D:
178
+ """Remove and return item(s) from the Pile.
179
+
180
+ Args:
181
+ key: Key of item(s) to remove.
182
+ default: Value if key not found.
183
+
184
+ Returns:
185
+ Removed item(s) or default.
186
+
187
+ Raises:
188
+ KeyError: If key not found and no default.
189
+ """
190
+ return self._pop(key, default)
191
+
192
+ def remove(
193
+ self,
194
+ item: T,
195
+ /,
196
+ ) -> None:
197
+ """Remove a specific item from the Pile.
198
+
199
+ Args:
200
+ item: Item to remove.
201
+
202
+ Raises:
203
+ ValueError: If item not found.
204
+ """
205
+ self._remove(item)
206
+
207
+ def include(
208
+ self,
209
+ item: ID.ItemSeq | ID.Item,
210
+ /,
211
+ ) -> None:
212
+ """Include item(s) if not present.
213
+
214
+ Args:
215
+ item: Item(s) to include.
216
+
217
+ Raises:
218
+ TypeError: If item type not allowed.
219
+ """
220
+ self._include(item)
221
+
222
+ def exclude(
223
+ self,
224
+ item: ID.ItemSeq | ID.Item,
225
+ /,
226
+ ) -> None:
227
+ """Exclude item(s) if present.
228
+
229
+ Args:
230
+ item: Item(s) to exclude.
231
+ """
232
+ self._exclude(item)
233
+
234
+ @synchronized
235
+ def clear(self) -> None:
236
+ """Remove all items."""
237
+ self._clear()
238
+
239
+ def update(
240
+ self,
241
+ other: ID.Item | ID.ItemSeq,
242
+ /,
243
+ ) -> None:
244
+ """Update with items from another source.
245
+
246
+ Args:
247
+ other: Items to update from.
248
+
249
+ Raises:
250
+ TypeError: If item types not allowed.
251
+ """
252
+ self._update(other)
253
+
254
+ @synchronized
255
+ def insert(self, index: int, item: T, /) -> None:
256
+ """Insert item at position.
257
+
258
+ Args:
259
+ index: Position to insert at.
260
+ item: Item to insert.
261
+
262
+ Raises:
263
+ IndexError: If index out of range.
264
+ TypeError: If item type not allowed.
265
+ """
266
+ self._insert(index, item)
267
+
268
+ @synchronized
269
+ def append(self, item: T, /) -> None:
270
+ """Append item to end (alias for include).
271
+
272
+ Args:
273
+ item: Item to append.
274
+
275
+ Raises:
276
+ TypeError: If item type not allowed.
277
+ """
278
+ self.update(item)
279
+
280
+ @synchronized
281
+ def get(
282
+ self,
283
+ key: ID.Ref | ID.RefSeq | int | slice,
284
+ default: D = UNDEFINED,
285
+ /,
286
+ ) -> T | "Pile" | D:
287
+ """Get item(s) by key with default.
288
+
289
+ Args:
290
+ key: Key to get items by.
291
+ default: Value if not found.
292
+
293
+ Returns:
294
+ Item(s) or default.
295
+ """
296
+ return self._get(key, default)
297
+
298
+ def keys(self) -> Sequence[str]:
299
+ """Get all Lion IDs in order."""
300
+ return list(self.progress)
301
+
302
+ def values(self) -> Sequence[T]:
303
+ """Get all items in order."""
304
+ return [self.pile_[key] for key in self.progress]
305
+
306
+ def items(self) -> Sequence[tuple[str, T]]:
307
+ """Get all (ID, item) pairs in order."""
308
+ return [(key, self.pile_[key]) for key in self.progress]
309
+
310
+ def is_empty(self) -> bool:
311
+ """Check if empty."""
312
+ return len(self.progress) == 0
313
+
314
+ def size(self) -> int:
315
+ """Get number of items."""
316
+ return len(self.progress)
317
+
318
+ def __iter__(self) -> Iterator[T]:
319
+ """Iterate over items safely."""
320
+ with self.lock:
321
+ current_order = list(self.progress)
322
+
323
+ for key in current_order:
324
+ yield self.pile_[key]
325
+
326
+ def __next__(self) -> T:
327
+ """Get next item."""
328
+ try:
329
+ return next(iter(self))
330
+ except StopIteration:
331
+ raise StopIteration("End of pile")
332
+
333
+ def __getitem__(
334
+ self, key: ID.Ref | ID.RefSeq | int | slice
335
+ ) -> Any | list | T:
336
+ """Get item(s) by key.
337
+
338
+ Args:
339
+ key: Key to get items by.
340
+
341
+ Returns:
342
+ Item(s) or sliced Pile.
343
+
344
+ Raises:
345
+ KeyError: If key not found.
346
+ """
347
+ return self._getitem(key)
348
+
349
+ def __contains__(self, item: ID.RefSeq | ID.Ref) -> bool:
350
+ """Check if item exists."""
351
+ return item in self.progress
352
+
353
+ def __len__(self) -> int:
354
+ """Get number of items."""
355
+ return len(self.pile_)
356
+
357
+ @override
358
+ def __bool__(self) -> bool:
359
+ """Check if not empty."""
360
+ return not self.is_empty()
361
+
362
+ def __list__(self) -> list[T]:
363
+ """Convert to list."""
364
+ return self.values()
365
+
366
+ def __ior__(self, other: "Pile") -> Self:
367
+ """In-place union."""
368
+ if not isinstance(other, Pile):
369
+ raise TypeError(
370
+ "Invalid type for Pile operation.",
371
+ expected_type=Pile,
372
+ actual_type=type(other),
373
+ )
374
+ other = self._validate_pile(list(other))
375
+ self.include(other)
376
+ return self
377
+
378
+ def __or__(self, other: "Pile") -> "Pile":
379
+ """Union."""
380
+ if not isinstance(other, Pile):
381
+ raise TypeError(
382
+ "Invalid type for Pile operation.",
383
+ expected_type=Pile,
384
+ actual_type=type(other),
385
+ )
386
+
387
+ result = self.__class__(
388
+ items=self.values(),
389
+ item_type=self.item_type,
390
+ order=self.progress,
391
+ )
392
+ result.include(list(other))
393
+ return result
394
+
395
+ def __ixor__(self, other: "Pile") -> Self:
396
+ """In-place symmetric difference."""
397
+ if not isinstance(other, Pile):
398
+ raise TypeError(
399
+ "Invalid type for Pile operation.",
400
+ expected_type=Pile,
401
+ actual_type=type(other),
402
+ )
403
+
404
+ to_exclude = []
405
+ for i in other:
406
+ if i in self:
407
+ to_exclude.append(i)
408
+
409
+ other = [i for i in other if i not in to_exclude]
410
+ self.exclude(to_exclude)
411
+ self.include(other)
412
+ return self
413
+
414
+ def __xor__(self, other: "Pile") -> "Pile":
415
+ """Symmetric difference."""
416
+ if not isinstance(other, Pile):
417
+ raise TypeError(
418
+ "Invalid type for Pile operation.",
419
+ expected_type=Pile,
420
+ actual_type=type(other),
421
+ )
422
+
423
+ to_exclude = []
424
+ for i in other:
425
+ if i in self:
426
+ to_exclude.append(i)
427
+
428
+ values = [i for i in self if i not in to_exclude] + [
429
+ i for i in other if i not in to_exclude
430
+ ]
431
+
432
+ result = self.__class__(
433
+ items=values,
434
+ item_type=self.item_type,
435
+ )
436
+ return result
437
+
438
+ def __iand__(self, other: "Pile") -> Self:
439
+ """In-place intersection."""
440
+ if not isinstance(other, Pile):
441
+ raise TypeError(
442
+ "Invalid type for Pile operation.",
443
+ expected_type=Pile,
444
+ actual_type=type(other),
445
+ )
446
+
447
+ to_exclude = []
448
+ for i in self.values():
449
+ if i not in other:
450
+ to_exclude.append(i)
451
+ self.exclude(to_exclude)
452
+ return self
453
+
454
+ def __and__(self, other: "Pile") -> "Pile":
455
+ """Intersection."""
456
+ if not isinstance(other, Pile):
457
+ raise TypeError(
458
+ "Invalid type for Pile operation.",
459
+ expected_type=Pile,
460
+ actual_type=type(other),
461
+ )
462
+
463
+ values = [i for i in self if i in other]
464
+ return self.__class__(
465
+ items=values,
466
+ item_type=self.item_type,
467
+ )
468
+
469
+ @override
470
+ def __str__(self) -> str:
471
+ """Simple string representation."""
472
+ return f"Pile({len(self)})"
473
+
474
+ @override
475
+ def __repr__(self) -> str:
476
+ """Detailed string representation."""
477
+ length = len(self)
478
+ if length == 0:
479
+ return "Pile()"
480
+ elif length == 1:
481
+ return f"Pile({next(iter(self.pile_.values())).__repr__()})"
482
+ else:
483
+ return f"Pile({length})"
484
+
485
+ def __getstate__(self):
486
+ """Prepare for pickling."""
487
+ state = self.__dict__.copy()
488
+ state["_lock"] = None
489
+ state["_async_lock"] = None
490
+ return state
491
+
492
+ def __setstate__(self, state):
493
+ """Restore after unpickling."""
494
+ self.__dict__.update(state)
495
+ self._lock = threading.Lock()
496
+ self._async_lock = asyncio.Lock()
497
+
498
+ @property
499
+ def lock(self):
500
+ """Thread lock."""
501
+ if not hasattr(self, "_lock") or self._lock is None:
502
+ self._lock = threading.Lock()
503
+ return self._lock
504
+
505
+ @property
506
+ def async_lock(self):
507
+ """Async lock."""
508
+ if not hasattr(self, "_async_lock") or self._async_lock is None:
509
+ self._async_lock = asyncio.Lock()
510
+ return self._async_lock
511
+
512
+ # Async Interface methods
513
+ @async_synchronized
514
+ async def asetitem(
515
+ self,
516
+ key: ID.Ref | ID.RefSeq | int | slice,
517
+ item: ID.Item | ID.ItemSeq,
518
+ /,
519
+ ) -> None:
520
+ """Async set item(s)."""
521
+ self._setitem(key, item)
522
+
523
+ @async_synchronized
524
+ async def apop(
525
+ self,
526
+ key: ID.Ref | ID.RefSeq | int | slice,
527
+ default: Any = UNDEFINED,
528
+ /,
529
+ ):
530
+ """Async remove and return item(s)."""
531
+ return self._pop(key, default)
532
+
533
+ @async_synchronized
534
+ async def aremove(
535
+ self,
536
+ item: ID.Ref | ID.RefSeq,
537
+ /,
538
+ ) -> None:
539
+ """Async remove item."""
540
+ self._remove(item)
541
+
542
+ @async_synchronized
543
+ async def ainclude(
544
+ self,
545
+ item: ID.ItemSeq | ID.Item,
546
+ /,
547
+ ) -> None:
548
+ """Async include item(s)."""
549
+ self._include(item)
550
+ if item not in self:
551
+ raise TypeError(f"Item {item} is not of allowed types")
552
+
553
+ @async_synchronized
554
+ async def aexclude(
555
+ self,
556
+ item: ID.Ref | ID.RefSeq,
557
+ /,
558
+ ) -> None:
559
+ """Async exclude item(s)."""
560
+ self._exclude(item)
561
+
562
+ @async_synchronized
563
+ async def aclear(self) -> None:
564
+ """Async clear all items."""
565
+ self._clear()
566
+
567
+ @async_synchronized
568
+ async def aupdate(
569
+ self,
570
+ other: ID.ItemSeq | ID.Item,
571
+ /,
572
+ ) -> None:
573
+ """Async update with items."""
574
+ self._update(other)
575
+
576
+ @async_synchronized
577
+ async def aget(
578
+ self,
579
+ key: Any,
580
+ default=UNDEFINED,
581
+ /,
582
+ ) -> list | Any | T:
583
+ """Async get item(s)."""
584
+ return self._get(key, default)
585
+
586
+ async def __aiter__(self) -> AsyncIterator[T]:
587
+ """Async iterate over items."""
588
+ async with self.async_lock:
589
+ current_order = list(self.progress)
590
+
591
+ for key in current_order:
592
+ yield self.pile_[key]
593
+ await asyncio.sleep(0) # Yield control to the event loop
594
+
595
+ async def __anext__(self) -> T:
596
+ """Async get next item."""
597
+ try:
598
+ return await anext(self.AsyncPileIterator(self))
599
+ except StopAsyncIteration:
600
+ raise StopAsyncIteration("End of pile")
601
+
602
+ # private methods
603
+ def _getitem(self, key: Any) -> Any | list | T:
604
+ if key is None:
605
+ raise ValueError("getitem key not provided.")
606
+
607
+ if isinstance(key, int | slice):
608
+ try:
609
+ result_ids = self.progress[key]
610
+ result_ids = (
611
+ [result_ids]
612
+ if not isinstance(result_ids, list)
613
+ else result_ids
614
+ )
615
+ result = []
616
+ for i in result_ids:
617
+ result.append(self.pile_[i])
618
+ return result[0] if len(result) == 1 else result
619
+ except Exception as e:
620
+ raise ItemNotFoundError(f"index {key}. Error: {e}")
621
+
622
+ elif isinstance(key, str):
623
+ try:
624
+ return self.pile_[key]
625
+ except Exception as e:
626
+ raise ItemNotFoundError(f"key {key}. Error: {e}")
627
+
628
+ else:
629
+ key = to_list_type(key)
630
+ result = []
631
+ try:
632
+ for k in key:
633
+ result_id = ID.get_id(k)
634
+ result.append(self.pile_[result_id])
635
+
636
+ if len(result) == 0:
637
+ raise ItemNotFoundError(f"key {key} item not found")
638
+ if len(result) == 1:
639
+ return result[0]
640
+ return result
641
+ except Exception as e:
642
+ raise ItemNotFoundError(f"Key {key}. Error:{e}")
643
+
644
+ def _setitem(
645
+ self,
646
+ key: ID.Ref | ID.RefSeq | int | slice,
647
+ item: ID.Item | ID.ItemSeq,
648
+ ) -> None:
649
+ item_dict = self._validate_pile(item)
650
+
651
+ item_order = []
652
+ for i in item_dict.keys():
653
+ if i in self.progress:
654
+ raise ItemExistsError(f"item {i} already exists in the pile")
655
+ item_order.append(i)
656
+ if isinstance(key, int | slice):
657
+ try:
658
+ delete_order = (
659
+ list(self.progress[key])
660
+ if isinstance(self.progress[key], Progression)
661
+ else [self.progress[key]]
662
+ )
663
+ self.progress[key] = item_order
664
+ for i in to_list(delete_order, flatten=True):
665
+ self.pile_.pop(i)
666
+ self.pile_.update(item_dict)
667
+ except Exception as e:
668
+ raise ValueError(f"Failed to set pile. Error: {e}")
669
+ else:
670
+ key = to_list_type(key)
671
+ if isinstance(key[0], list):
672
+ key = to_list(key, flatten=True, dropna=True)
673
+ if len(key) != len(item_order):
674
+ raise KeyError(
675
+ f"Invalid key {key}. Key and item does not match.",
676
+ )
677
+ for k in key:
678
+ id_ = ID.get_id(k)
679
+ if id_ not in item_order:
680
+ raise KeyError(
681
+ f"Invalid key {id_}. Key and item does not match.",
682
+ )
683
+ self.progress += key
684
+ self.pile_.update(item_dict)
685
+
686
+ def _get(self, key: Any, default: D = UNDEFINED) -> T | "Pile" | D:
687
+ if isinstance(key, int | slice):
688
+ try:
689
+ return self[key]
690
+ except Exception as e:
691
+ if default is UNDEFINED:
692
+ raise ItemNotFoundError(f"Item not found. Error: {e}")
693
+ return default
694
+ else:
695
+ check = None
696
+ if isinstance(key, list):
697
+ check = True
698
+ for i in key:
699
+ if type(i) is not int:
700
+ check = False
701
+ break
702
+ try:
703
+ if not check:
704
+ key = validate_order(key)
705
+ result = []
706
+ for k in key:
707
+ result.append(self[k])
708
+ if len(result) == 0:
709
+ raise ItemNotFoundError(f"key {key} item not found")
710
+ if len(result) == 1:
711
+ return result[0]
712
+ return result
713
+
714
+ except Exception as e:
715
+ if default is UNDEFINED:
716
+ raise ItemNotFoundError(f"Item not found. Error: {e}")
717
+ return default
718
+
719
+ def _pop(
720
+ self,
721
+ key: ID.Ref | ID.RefSeq | int | slice,
722
+ default: D = UNDEFINED,
723
+ ) -> T | "Pile" | D:
724
+ if isinstance(key, int | slice):
725
+ try:
726
+ pops = self.progress[key]
727
+ pops = [pops] if isinstance(pops, str) else pops
728
+ result = []
729
+ for i in pops:
730
+ self.progress.remove(i)
731
+ result.append(self.pile_.pop(i))
732
+ result = (
733
+ self.__class__(items=result, item_type=self.item_type)
734
+ if len(result) > 1
735
+ else result[0]
736
+ )
737
+ return result
738
+ except Exception as e:
739
+ if default is UNDEFINED:
740
+ raise ItemNotFoundError(f"Item not found. Error: {e}")
741
+ return default
742
+ else:
743
+ try:
744
+ key = validate_order(key)
745
+ result = []
746
+ for k in key:
747
+ self.progress.remove(k)
748
+ result.append(self.pile_.pop(k))
749
+ if len(result) == 0:
750
+ raise ItemNotFoundError(f"key {key} item not found")
751
+ elif len(result) == 1:
752
+ return result[0]
753
+ return result
754
+ except Exception as e:
755
+ if default is UNDEFINED:
756
+ raise ItemNotFoundError(f"Item not found. Error: {e}")
757
+ return default
758
+
759
+ def _remove(self, item: ID.Ref | ID.RefSeq):
760
+ if isinstance(item, int | slice):
761
+ raise TypeError(
762
+ "Invalid item type for remove, should be ID or Item(s)"
763
+ )
764
+ if item in self:
765
+ self.pop(item)
766
+ return
767
+ raise ItemNotFoundError(f"{item}")
768
+
769
+ def _include(self, item: ID.ItemSeq | ID.Item):
770
+ item_dict = self._validate_pile(item)
771
+
772
+ item_order = []
773
+ for i in item_dict.keys():
774
+ if i not in self.progress:
775
+ item_order.append(i)
776
+
777
+ self.progress.append(item_order)
778
+ self.pile_.update(item_dict)
779
+
780
+ def _exclude(self, item: ID.Ref | ID.RefSeq):
781
+ item = to_list_type(item)
782
+ exclude_list = []
783
+ for i in item:
784
+ if i in self:
785
+ exclude_list.append(i)
786
+ if exclude_list:
787
+ self.pop(exclude_list)
788
+
789
+ def _clear(self) -> None:
790
+ self.pile_.clear()
791
+ self.progress.clear()
792
+
793
+ def _update(self, other: ID.ItemSeq | ID.Item):
794
+ others = self._validate_pile(other)
795
+ for i in others.keys():
796
+ if i in self.pile_:
797
+ self.pile_[i] = others[i]
798
+ else:
799
+ self.include(others[i])
800
+
801
+ def _validate_item_type(self, value) -> set[type[T]] | None:
802
+ if value is None:
803
+ return None
804
+
805
+ value = to_list_type(value)
806
+
807
+ for i in value:
808
+ if not issubclass(i, Observable):
809
+ raise TypeError(
810
+ message="Item type must be a subclass of T.",
811
+ expected_type=T,
812
+ actual_type=type(i),
813
+ )
814
+
815
+ if len(value) != len(set(value)):
816
+ raise ValueError(
817
+ "Detected duplicated item types in item_type.",
818
+ )
819
+
820
+ if len(value) > 0:
821
+ return set(value)
822
+
823
+ def _validate_pile(self, value: Any) -> dict[str, T]:
824
+ if not value:
825
+ return {}
826
+
827
+ value = to_list_type(value)
828
+
829
+ result = {}
830
+ for i in value:
831
+ if self.item_type:
832
+ if self.strict_type:
833
+ if type(i) not in self.item_type:
834
+ raise TypeError(
835
+ message="Invalid item type in pile."
836
+ f" Expected {self.item_type}",
837
+ )
838
+ else:
839
+ if not any(issubclass(type(i), t) for t in self.item_type):
840
+ raise TypeError(
841
+ "Invalid item type in pile. Expected "
842
+ f"{self.item_type} or the subclasses",
843
+ )
844
+ else:
845
+ if not isinstance(i, Observable):
846
+ raise ValueError(f"Invalid pile item {i}")
847
+
848
+ result[i.ln_id] = i
849
+
850
+ return result
851
+
852
+ def _validate_order(self, value: Any) -> Progression:
853
+ if not value:
854
+ return self.progress.__class__(order=list(self.pile_.keys()))
855
+
856
+ if isinstance(value, Progression):
857
+ value = list(value)
858
+ else:
859
+ value = to_list_type(value)
860
+
861
+ value_set = set(value)
862
+ if len(value_set) != len(value):
863
+ raise ValueError("There are duplicate elements in the order")
864
+ if len(value_set) != len(self.pile_.keys()):
865
+ raise ValueError(
866
+ "The length of the order does not match the length of the pile"
867
+ )
868
+
869
+ for i in value_set:
870
+ if ID.get_id(i) not in self.pile_.keys():
871
+ raise ValueError(
872
+ f"The order does not match the pile. {i} not found"
873
+ )
874
+
875
+ return self.progress.__class__(order=value)
876
+
877
+ def _insert(self, index: int, item: ID.Item):
878
+ item_dict = self._validate_pile(item)
879
+
880
+ item_order = []
881
+ for i in item_dict.keys():
882
+ if i in self.progress:
883
+ raise ItemExistsError(f"item {i} already exists in the pile")
884
+ item_order.append(i)
885
+ self.progress.insert(index, item_order)
886
+ self.pile_.update(item_dict)
887
+
888
+ @field_serializer("pile_")
889
+ def _(self, value: dict[str, T]):
890
+ return [i.to_dict() for i in value.values()]
891
+
892
+ class AsyncPileIterator:
893
+ def __init__(self, pile: "Pile"):
894
+ self.pile = pile
895
+ self.index = 0
896
+
897
+ def __aiter__(self) -> AsyncIterator[T]:
898
+ return self
899
+
900
+ async def __anext__(self) -> T:
901
+ if self.index >= len(self.pile):
902
+ raise StopAsyncIteration
903
+ item = self.pile[self.pile.progress[self.index]]
904
+ self.index += 1
905
+ await asyncio.sleep(0) # Yield control to the event loop
906
+ return item
907
+
908
+ async def __aenter__(self) -> Self:
909
+ """Enter async context."""
910
+ await self.async_lock.acquire()
911
+ return self
912
+
913
+ async def __aexit__(
914
+ self,
915
+ exc_type: type[BaseException] | None,
916
+ exc_val: BaseException | None,
917
+ exc_tb: Any,
918
+ ) -> None:
919
+ """Exit async context."""
920
+ self.async_lock.release()
921
+
922
+ def is_homogenous(self) -> bool:
923
+ """Check if all items are same type."""
924
+ return len(self.pile_) < 2 or all(is_same_dtype(self.pile_.values()))
925
+
926
+ def adapt_to(self, obj_key: str, /, **kwargs: Any) -> Any:
927
+ """Convert to another format."""
928
+ return self._get_adapter_registry().adapt_to(self, obj_key, **kwargs)
929
+
930
+ @classmethod
931
+ def list_adapters(cls):
932
+ """List available adapters."""
933
+ return cls._get_adapter_registry().list_adapters()
934
+
935
+ @classmethod
936
+ def register_adapter(cls, adapter: type[Adapter]):
937
+ """Register new adapter."""
938
+ cls._get_adapter_registry().register(adapter)
939
+
940
+ @classmethod
941
+ def _get_adapter_registry(cls) -> AdapterRegistry:
942
+ if isinstance(cls._adapter_registry, type):
943
+ cls._adapter_registry = cls._adapter_registry()
944
+ return cls._adapter_registry
945
+
946
+ @classmethod
947
+ def adapt_from(cls, obj: Any, obj_key: str, /, **kwargs: Any):
948
+ """Create from another format."""
949
+ dict_ = cls._get_adapter_registry().adapt_from(
950
+ cls, obj, obj_key, **kwargs
951
+ )
952
+ if isinstance(dict_, list):
953
+ dict_ = {"pile_": dict_}
954
+ return cls.from_dict(dict_)
955
+
956
+ def to_df(
957
+ self,
958
+ columns: list[str] | None = None,
959
+ **kwargs: Any,
960
+ ):
961
+ """Convert to DataFrame."""
962
+ return self.adapt_to("pd_dataframe", columns=columns, **kwargs)
963
+
964
+ def to_csv(self, fp: str | Path, **kwargs: Any) -> None:
965
+ """Save to CSV file."""
966
+ self.adapt_to(".csv", fp=fp, **kwargs)
967
+
968
+ def to_excel(self, fp: str | Path, **kwargs: Any) -> None:
969
+ """Save to Excel file."""
970
+ self.adapt_to(".xlsx", fp=fp, **kwargs)
971
+
972
+
973
+ def pile(
974
+ items: Any = None,
975
+ /,
976
+ item_type: type[T] | set[type[T]] | None = None,
977
+ order: list[str] | None = None,
978
+ strict_type: bool = False,
979
+ df: pd.DataFrame | None = None, # priority 1
980
+ fp: str | Path | None = None, # priority 2
981
+ **kwargs,
982
+ ) -> Pile:
983
+ """Create a new Pile instance.
984
+
985
+ Args:
986
+ items: Initial items for the pile.
987
+ item_type: Allowed types for items in the pile.
988
+ order: Initial order of items.
989
+ strict: If True, enforce strict type checking.
990
+
991
+ Returns:
992
+ Pile: A new Pile instance.
993
+ """
994
+
995
+ if df:
996
+ return Pile.adapt_from(df, "pd_dataframe", **kwargs)
997
+
998
+ if fp:
999
+ fp = Path(fp)
1000
+ if fp.suffix == ".csv":
1001
+ return Pile.adapt_from(fp, ".csv", **kwargs)
1002
+ if fp.suffix == ".xlsx":
1003
+ return Pile.adapt_from(fp, ".xlsx", **kwargs)
1004
+ if fp.suffix == ".json":
1005
+ return Pile.adapt_from(fp, ".json", **kwargs)
1006
+
1007
+ return Pile(
1008
+ items,
1009
+ item_type=item_type,
1010
+ order=order,
1011
+ strict=strict_type,
1012
+ **kwargs,
1013
+ )
1014
+
1015
+
1016
+ __all__ = [Pile, pile]
1017
+ # File: autoos/generic/pile.py