lionagi 0.5.5__py3-none-any.whl → 0.6.0__py3-none-any.whl

Sign up to get free protection for your applications and to get access to all the features.
Files changed (428) hide show
  1. lionagi/__init__.py +16 -24
  2. lionagi/{core/_class_registry.py → _class_registry.py} +51 -10
  3. lionagi/_errors.py +35 -0
  4. lionagi/libs/__init__.py +3 -0
  5. lionagi/libs/compress/__init__.py +3 -0
  6. lionagi/libs/compress/models.py +6 -2
  7. lionagi/libs/compress/utils.py +4 -16
  8. lionagi/libs/file/__init__.py +3 -0
  9. lionagi/libs/file/chunk.py +4 -0
  10. lionagi/libs/file/file_ops.py +4 -0
  11. lionagi/libs/file/params.py +4 -41
  12. lionagi/libs/file/process.py +4 -0
  13. lionagi/libs/file/save.py +5 -1
  14. lionagi/libs/{parse/flatten → nested}/flatten.py +4 -0
  15. lionagi/libs/{parse/nested → nested}/nfilter.py +4 -0
  16. lionagi/libs/{parse/nested → nested}/nget.py +6 -1
  17. lionagi/libs/{parse/nested → nested}/ninsert.py +5 -1
  18. lionagi/libs/{parse/nested → nested}/nmerge.py +4 -0
  19. lionagi/libs/{parse/nested → nested}/npop.py +5 -2
  20. lionagi/libs/{parse/nested → nested}/nset.py +6 -1
  21. lionagi/libs/{parse/flatten → nested}/unflatten.py +4 -0
  22. lionagi/libs/{parse/nested → nested}/utils.py +5 -1
  23. lionagi/libs/package/__init__.py +3 -0
  24. lionagi/libs/package/imports.py +6 -2
  25. lionagi/libs/package/management.py +7 -3
  26. lionagi/libs/package/params.py +4 -0
  27. lionagi/libs/package/system.py +4 -0
  28. lionagi/libs/parse.py +30 -0
  29. lionagi/libs/{parse/json → schema}/as_readable.py +10 -4
  30. lionagi/libs/{parse/string_parse/code_block.py → schema/extract_code_block.py} +4 -0
  31. lionagi/libs/{parse/string_parse/docstring.py → schema/extract_docstring.py} +4 -0
  32. lionagi/libs/{parse/string_parse/function_.py → schema/function_to_schema.py} +21 -9
  33. lionagi/libs/{parse/json/schema.py → schema/json_schema.py} +5 -1
  34. lionagi/libs/validate/common_field_validators.py +170 -0
  35. lionagi/libs/{parse/validate/keys.py → validate/fuzzy_match_keys.py} +42 -8
  36. lionagi/libs/{parse/validate/mapping.py → validate/fuzzy_validate_mapping.py} +41 -6
  37. lionagi/libs/{string_similarity/algorithms.py → validate/string_similarity.py} +115 -1
  38. lionagi/libs/{parse/validate/boolean.py → validate/validate_boolean.py} +42 -3
  39. lionagi/operations/__init__.py +13 -3
  40. lionagi/operations/brainstorm/__init__.py +3 -3
  41. lionagi/operations/brainstorm/brainstorm.py +33 -19
  42. lionagi/operations/brainstorm/prompt.py +4 -0
  43. lionagi/operations/plan/__init__.py +4 -0
  44. lionagi/operations/plan/plan.py +16 -13
  45. lionagi/operations/plan/prompt.py +4 -0
  46. lionagi/operations/select/__init__.py +4 -0
  47. lionagi/operations/select/prompt.py +4 -0
  48. lionagi/operations/select/select.py +1 -1
  49. lionagi/operations/select/utils.py +4 -4
  50. lionagi/{strategies → operations/strategies}/base.py +6 -2
  51. lionagi/{strategies → operations/strategies}/concurrent.py +8 -5
  52. lionagi/{strategies → operations/strategies}/concurrent_chunk.py +6 -3
  53. lionagi/{strategies → operations/strategies}/concurrent_sequential_chunk.py +8 -4
  54. lionagi/{strategies → operations/strategies}/params.py +10 -6
  55. lionagi/{strategies → operations/strategies}/sequential.py +6 -2
  56. lionagi/{strategies → operations/strategies}/sequential_chunk.py +7 -3
  57. lionagi/{strategies → operations/strategies}/sequential_concurrent_chunk.py +9 -4
  58. lionagi/{strategies → operations/strategies}/utils.py +6 -3
  59. lionagi/{core/models/__init__.py → operations/types.py} +3 -1
  60. lionagi/operations/utils.py +6 -3
  61. lionagi/operatives/action/function_calling.py +136 -0
  62. lionagi/operatives/action/manager.py +239 -0
  63. lionagi/operatives/action/request_response_model.py +90 -0
  64. lionagi/operatives/action/tool.py +141 -0
  65. lionagi/{protocols/operatives/action.py → operatives/action/utils.py} +52 -90
  66. lionagi/{core → operatives}/forms/base.py +9 -4
  67. lionagi/{core → operatives}/forms/form.py +8 -13
  68. lionagi/{core → operatives}/forms/report.py +5 -3
  69. lionagi/operatives/instruct/base.py +79 -0
  70. lionagi/operatives/instruct/instruct.py +105 -0
  71. lionagi/operatives/instruct/instruct_collection.py +52 -0
  72. lionagi/operatives/instruct/node.py +13 -0
  73. lionagi/{protocols/operatives → operatives/instruct}/prompts.py +0 -34
  74. lionagi/{protocols/operatives → operatives/instruct}/reason.py +14 -7
  75. lionagi/{integrations/anthropic_/version.py → operatives/manager.py} +5 -1
  76. lionagi/operatives/models/field_model.py +194 -0
  77. lionagi/operatives/models/model_params.py +307 -0
  78. lionagi/{core → operatives}/models/note.py +20 -28
  79. lionagi/{core → operatives}/models/operable_model.py +153 -71
  80. lionagi/{core → operatives}/models/schema_model.py +4 -3
  81. lionagi/{protocols/operatives → operatives}/operative.py +10 -7
  82. lionagi/{protocols/operatives → operatives}/step.py +67 -26
  83. lionagi/operatives/types.py +69 -0
  84. lionagi/protocols/_adapter.py +224 -0
  85. lionagi/protocols/_concepts.py +94 -0
  86. lionagi/protocols/generic/element.py +460 -0
  87. lionagi/protocols/generic/event.py +177 -0
  88. lionagi/protocols/generic/log.py +237 -0
  89. lionagi/{core → protocols}/generic/pile.py +172 -131
  90. lionagi/protocols/generic/processor.py +316 -0
  91. lionagi/protocols/generic/progression.py +500 -0
  92. lionagi/protocols/graph/edge.py +166 -0
  93. lionagi/protocols/graph/graph.py +290 -0
  94. lionagi/protocols/graph/node.py +109 -0
  95. lionagi/protocols/mail/exchange.py +116 -0
  96. lionagi/protocols/mail/mail.py +25 -0
  97. lionagi/protocols/mail/mailbox.py +47 -0
  98. lionagi/protocols/mail/manager.py +168 -0
  99. lionagi/protocols/mail/package.py +55 -0
  100. lionagi/protocols/messages/action_request.py +165 -0
  101. lionagi/protocols/messages/action_response.py +132 -0
  102. lionagi/{core/communication → protocols/messages}/assistant_response.py +55 -79
  103. lionagi/protocols/messages/base.py +73 -0
  104. lionagi/protocols/messages/instruction.py +582 -0
  105. lionagi/protocols/messages/manager.py +429 -0
  106. lionagi/protocols/messages/message.py +216 -0
  107. lionagi/protocols/messages/system.py +115 -0
  108. lionagi/protocols/messages/templates/assistant_response.jinja2 +6 -0
  109. lionagi/{core/communication → protocols/messages}/templates/instruction_message.jinja2 +2 -2
  110. lionagi/protocols/types.py +96 -0
  111. lionagi/service/__init__.py +10 -12
  112. lionagi/service/endpoints/base.py +517 -0
  113. lionagi/service/endpoints/chat_completion.py +102 -0
  114. lionagi/service/endpoints/match_endpoint.py +60 -0
  115. lionagi/service/endpoints/rate_limited_processor.py +145 -0
  116. lionagi/service/endpoints/token_calculator.py +209 -0
  117. lionagi/service/imodel.py +263 -96
  118. lionagi/service/manager.py +45 -0
  119. lionagi/service/providers/anthropic_/messages.py +64 -0
  120. lionagi/service/providers/groq_/chat_completions.py +56 -0
  121. lionagi/service/providers/openai_/chat_completions.py +62 -0
  122. lionagi/service/providers/openrouter_/chat_completions.py +62 -0
  123. lionagi/service/providers/perplexity_/__init__.py +3 -0
  124. lionagi/service/providers/perplexity_/chat_completions.py +40 -0
  125. lionagi/session/__init__.py +3 -0
  126. lionagi/session/branch.py +1287 -0
  127. lionagi/session/session.py +296 -0
  128. lionagi/settings.py +62 -118
  129. lionagi/utils.py +2386 -0
  130. lionagi/version.py +1 -1
  131. {lionagi-0.5.5.dist-info → lionagi-0.6.0.dist-info}/METADATA +7 -6
  132. lionagi-0.6.0.dist-info/RECORD +160 -0
  133. lionagi/core/action/action_manager.py +0 -289
  134. lionagi/core/action/base.py +0 -109
  135. lionagi/core/action/function_calling.py +0 -153
  136. lionagi/core/action/tool.py +0 -202
  137. lionagi/core/action/types.py +0 -16
  138. lionagi/core/communication/action_request.py +0 -163
  139. lionagi/core/communication/action_response.py +0 -149
  140. lionagi/core/communication/base_mail.py +0 -49
  141. lionagi/core/communication/instruction.py +0 -376
  142. lionagi/core/communication/message.py +0 -286
  143. lionagi/core/communication/message_manager.py +0 -543
  144. lionagi/core/communication/system.py +0 -116
  145. lionagi/core/communication/templates/assistant_response.jinja2 +0 -2
  146. lionagi/core/communication/types.py +0 -27
  147. lionagi/core/communication/utils.py +0 -256
  148. lionagi/core/forms/types.py +0 -13
  149. lionagi/core/generic/component.py +0 -422
  150. lionagi/core/generic/edge.py +0 -163
  151. lionagi/core/generic/element.py +0 -199
  152. lionagi/core/generic/graph.py +0 -377
  153. lionagi/core/generic/log.py +0 -151
  154. lionagi/core/generic/log_manager.py +0 -320
  155. lionagi/core/generic/node.py +0 -11
  156. lionagi/core/generic/progression.py +0 -395
  157. lionagi/core/generic/types.py +0 -23
  158. lionagi/core/generic/utils.py +0 -53
  159. lionagi/core/models/base.py +0 -28
  160. lionagi/core/models/field_model.py +0 -145
  161. lionagi/core/models/model_params.py +0 -194
  162. lionagi/core/models/types.py +0 -19
  163. lionagi/core/session/branch.py +0 -130
  164. lionagi/core/session/branch_mixins.py +0 -581
  165. lionagi/core/session/session.py +0 -163
  166. lionagi/core/session/types.py +0 -8
  167. lionagi/core/typing/__init__.py +0 -9
  168. lionagi/core/typing/_concepts.py +0 -173
  169. lionagi/core/typing/_id.py +0 -104
  170. lionagi/core/typing/_pydantic.py +0 -33
  171. lionagi/core/typing/_typing.py +0 -54
  172. lionagi/integrations/__init__.py +0 -0
  173. lionagi/integrations/_services.py +0 -17
  174. lionagi/integrations/anthropic_/AnthropicModel.py +0 -268
  175. lionagi/integrations/anthropic_/AnthropicService.py +0 -127
  176. lionagi/integrations/anthropic_/anthropic_max_output_token_data.yaml +0 -12
  177. lionagi/integrations/anthropic_/anthropic_price_data.yaml +0 -34
  178. lionagi/integrations/anthropic_/api_endpoints/api_request.py +0 -277
  179. lionagi/integrations/anthropic_/api_endpoints/data_models.py +0 -40
  180. lionagi/integrations/anthropic_/api_endpoints/match_response.py +0 -119
  181. lionagi/integrations/anthropic_/api_endpoints/messages/request/message_models.py +0 -14
  182. lionagi/integrations/anthropic_/api_endpoints/messages/request/request_body.py +0 -74
  183. lionagi/integrations/anthropic_/api_endpoints/messages/response/__init__.py +0 -0
  184. lionagi/integrations/anthropic_/api_endpoints/messages/response/content_models.py +0 -32
  185. lionagi/integrations/anthropic_/api_endpoints/messages/response/response_body.py +0 -101
  186. lionagi/integrations/anthropic_/api_endpoints/messages/response/usage_models.py +0 -25
  187. lionagi/integrations/groq_/GroqModel.py +0 -325
  188. lionagi/integrations/groq_/GroqService.py +0 -156
  189. lionagi/integrations/groq_/api_endpoints/__init__.py +0 -0
  190. lionagi/integrations/groq_/api_endpoints/data_models.py +0 -187
  191. lionagi/integrations/groq_/api_endpoints/groq_request.py +0 -288
  192. lionagi/integrations/groq_/api_endpoints/match_response.py +0 -106
  193. lionagi/integrations/groq_/api_endpoints/response_utils.py +0 -105
  194. lionagi/integrations/groq_/groq_max_output_token_data.yaml +0 -21
  195. lionagi/integrations/groq_/groq_price_data.yaml +0 -58
  196. lionagi/integrations/groq_/groq_rate_limits.yaml +0 -105
  197. lionagi/integrations/groq_/version.py +0 -5
  198. lionagi/integrations/litellm_/imodel.py +0 -76
  199. lionagi/integrations/ollama_/OllamaModel.py +0 -244
  200. lionagi/integrations/ollama_/OllamaService.py +0 -142
  201. lionagi/integrations/ollama_/api_endpoints/api_request.py +0 -179
  202. lionagi/integrations/ollama_/api_endpoints/chat_completion/message_models.py +0 -31
  203. lionagi/integrations/ollama_/api_endpoints/chat_completion/request_body.py +0 -46
  204. lionagi/integrations/ollama_/api_endpoints/chat_completion/response_body.py +0 -67
  205. lionagi/integrations/ollama_/api_endpoints/chat_completion/tool_models.py +0 -49
  206. lionagi/integrations/ollama_/api_endpoints/completion/__init__.py +0 -0
  207. lionagi/integrations/ollama_/api_endpoints/completion/request_body.py +0 -72
  208. lionagi/integrations/ollama_/api_endpoints/completion/response_body.py +0 -59
  209. lionagi/integrations/ollama_/api_endpoints/data_models.py +0 -15
  210. lionagi/integrations/ollama_/api_endpoints/embedding/__init__.py +0 -0
  211. lionagi/integrations/ollama_/api_endpoints/embedding/request_body.py +0 -33
  212. lionagi/integrations/ollama_/api_endpoints/embedding/response_body.py +0 -29
  213. lionagi/integrations/ollama_/api_endpoints/match_data_model.py +0 -62
  214. lionagi/integrations/ollama_/api_endpoints/match_response.py +0 -190
  215. lionagi/integrations/ollama_/api_endpoints/model/copy_model.py +0 -13
  216. lionagi/integrations/ollama_/api_endpoints/model/create_model.py +0 -28
  217. lionagi/integrations/ollama_/api_endpoints/model/delete_model.py +0 -11
  218. lionagi/integrations/ollama_/api_endpoints/model/list_model.py +0 -60
  219. lionagi/integrations/ollama_/api_endpoints/model/pull_model.py +0 -34
  220. lionagi/integrations/ollama_/api_endpoints/model/push_model.py +0 -35
  221. lionagi/integrations/ollama_/api_endpoints/model/show_model.py +0 -36
  222. lionagi/integrations/ollama_/api_endpoints/option_models.py +0 -68
  223. lionagi/integrations/openai_/OpenAIModel.py +0 -419
  224. lionagi/integrations/openai_/OpenAIService.py +0 -435
  225. lionagi/integrations/openai_/__init__.py +0 -0
  226. lionagi/integrations/openai_/api_endpoints/__init__.py +0 -3
  227. lionagi/integrations/openai_/api_endpoints/api_request.py +0 -277
  228. lionagi/integrations/openai_/api_endpoints/audio/__init__.py +0 -9
  229. lionagi/integrations/openai_/api_endpoints/audio/speech_models.py +0 -34
  230. lionagi/integrations/openai_/api_endpoints/audio/transcription_models.py +0 -136
  231. lionagi/integrations/openai_/api_endpoints/audio/translation_models.py +0 -41
  232. lionagi/integrations/openai_/api_endpoints/audio/types.py +0 -41
  233. lionagi/integrations/openai_/api_endpoints/batch/__init__.py +0 -17
  234. lionagi/integrations/openai_/api_endpoints/batch/batch_models.py +0 -146
  235. lionagi/integrations/openai_/api_endpoints/batch/cancel_batch.py +0 -7
  236. lionagi/integrations/openai_/api_endpoints/batch/create_batch.py +0 -26
  237. lionagi/integrations/openai_/api_endpoints/batch/list_batch.py +0 -37
  238. lionagi/integrations/openai_/api_endpoints/batch/request_object_models.py +0 -65
  239. lionagi/integrations/openai_/api_endpoints/batch/retrieve_batch.py +0 -7
  240. lionagi/integrations/openai_/api_endpoints/batch/types.py +0 -4
  241. lionagi/integrations/openai_/api_endpoints/chat_completions/__init__.py +0 -1
  242. lionagi/integrations/openai_/api_endpoints/chat_completions/request/__init__.py +0 -39
  243. lionagi/integrations/openai_/api_endpoints/chat_completions/request/message_models.py +0 -121
  244. lionagi/integrations/openai_/api_endpoints/chat_completions/request/request_body.py +0 -221
  245. lionagi/integrations/openai_/api_endpoints/chat_completions/request/response_format.py +0 -71
  246. lionagi/integrations/openai_/api_endpoints/chat_completions/request/stream_options.py +0 -14
  247. lionagi/integrations/openai_/api_endpoints/chat_completions/request/tool_choice_models.py +0 -17
  248. lionagi/integrations/openai_/api_endpoints/chat_completions/request/tool_models.py +0 -54
  249. lionagi/integrations/openai_/api_endpoints/chat_completions/request/types.py +0 -18
  250. lionagi/integrations/openai_/api_endpoints/chat_completions/response/__init__.py +0 -0
  251. lionagi/integrations/openai_/api_endpoints/chat_completions/response/choice_models.py +0 -62
  252. lionagi/integrations/openai_/api_endpoints/chat_completions/response/function_models.py +0 -16
  253. lionagi/integrations/openai_/api_endpoints/chat_completions/response/log_prob_models.py +0 -47
  254. lionagi/integrations/openai_/api_endpoints/chat_completions/response/message_models.py +0 -25
  255. lionagi/integrations/openai_/api_endpoints/chat_completions/response/response_body.py +0 -99
  256. lionagi/integrations/openai_/api_endpoints/chat_completions/response/types.py +0 -8
  257. lionagi/integrations/openai_/api_endpoints/chat_completions/response/usage_models.py +0 -24
  258. lionagi/integrations/openai_/api_endpoints/chat_completions/util.py +0 -46
  259. lionagi/integrations/openai_/api_endpoints/data_models.py +0 -23
  260. lionagi/integrations/openai_/api_endpoints/embeddings/__init__.py +0 -3
  261. lionagi/integrations/openai_/api_endpoints/embeddings/request_body.py +0 -79
  262. lionagi/integrations/openai_/api_endpoints/embeddings/response_body.py +0 -67
  263. lionagi/integrations/openai_/api_endpoints/files/__init__.py +0 -11
  264. lionagi/integrations/openai_/api_endpoints/files/delete_file.py +0 -20
  265. lionagi/integrations/openai_/api_endpoints/files/file_models.py +0 -56
  266. lionagi/integrations/openai_/api_endpoints/files/list_files.py +0 -27
  267. lionagi/integrations/openai_/api_endpoints/files/retrieve_file.py +0 -9
  268. lionagi/integrations/openai_/api_endpoints/files/upload_file.py +0 -38
  269. lionagi/integrations/openai_/api_endpoints/fine_tuning/__init__.py +0 -37
  270. lionagi/integrations/openai_/api_endpoints/fine_tuning/cancel_jobs.py +0 -9
  271. lionagi/integrations/openai_/api_endpoints/fine_tuning/create_jobs.py +0 -133
  272. lionagi/integrations/openai_/api_endpoints/fine_tuning/fine_tuning_job_checkpoint_models.py +0 -58
  273. lionagi/integrations/openai_/api_endpoints/fine_tuning/fine_tuning_job_event_models.py +0 -31
  274. lionagi/integrations/openai_/api_endpoints/fine_tuning/fine_tuning_job_models.py +0 -140
  275. lionagi/integrations/openai_/api_endpoints/fine_tuning/list_fine_tuning_checkpoints.py +0 -51
  276. lionagi/integrations/openai_/api_endpoints/fine_tuning/list_fine_tuning_events.py +0 -42
  277. lionagi/integrations/openai_/api_endpoints/fine_tuning/list_fine_tuning_jobs.py +0 -31
  278. lionagi/integrations/openai_/api_endpoints/fine_tuning/retrieve_jobs.py +0 -9
  279. lionagi/integrations/openai_/api_endpoints/fine_tuning/training_format.py +0 -30
  280. lionagi/integrations/openai_/api_endpoints/images/__init__.py +0 -9
  281. lionagi/integrations/openai_/api_endpoints/images/image_edit_models.py +0 -69
  282. lionagi/integrations/openai_/api_endpoints/images/image_models.py +0 -56
  283. lionagi/integrations/openai_/api_endpoints/images/image_variation_models.py +0 -56
  284. lionagi/integrations/openai_/api_endpoints/images/response_body.py +0 -30
  285. lionagi/integrations/openai_/api_endpoints/match_data_model.py +0 -197
  286. lionagi/integrations/openai_/api_endpoints/match_response.py +0 -336
  287. lionagi/integrations/openai_/api_endpoints/models/__init__.py +0 -7
  288. lionagi/integrations/openai_/api_endpoints/models/delete_fine_tuned_model.py +0 -17
  289. lionagi/integrations/openai_/api_endpoints/models/models_models.py +0 -31
  290. lionagi/integrations/openai_/api_endpoints/models/retrieve_model.py +0 -9
  291. lionagi/integrations/openai_/api_endpoints/moderations/__init__.py +0 -3
  292. lionagi/integrations/openai_/api_endpoints/moderations/request_body.py +0 -20
  293. lionagi/integrations/openai_/api_endpoints/moderations/response_body.py +0 -139
  294. lionagi/integrations/openai_/api_endpoints/uploads/__init__.py +0 -19
  295. lionagi/integrations/openai_/api_endpoints/uploads/add_upload_part.py +0 -11
  296. lionagi/integrations/openai_/api_endpoints/uploads/cancel_upload.py +0 -7
  297. lionagi/integrations/openai_/api_endpoints/uploads/complete_upload.py +0 -18
  298. lionagi/integrations/openai_/api_endpoints/uploads/create_upload.py +0 -17
  299. lionagi/integrations/openai_/api_endpoints/uploads/uploads_models.py +0 -52
  300. lionagi/integrations/openai_/image_token_calculator/__init__.py +0 -0
  301. lionagi/integrations/openai_/image_token_calculator/image_token_calculator.py +0 -98
  302. lionagi/integrations/openai_/image_token_calculator/openai_image_token_data.yaml +0 -15
  303. lionagi/integrations/openai_/openai_max_output_token_data.yaml +0 -12
  304. lionagi/integrations/openai_/openai_price_data.yaml +0 -26
  305. lionagi/integrations/openai_/version.py +0 -1
  306. lionagi/integrations/pandas_/__init__.py +0 -24
  307. lionagi/integrations/pandas_/extend_df.py +0 -61
  308. lionagi/integrations/pandas_/read.py +0 -103
  309. lionagi/integrations/pandas_/remove_rows.py +0 -61
  310. lionagi/integrations/pandas_/replace_keywords.py +0 -65
  311. lionagi/integrations/pandas_/save.py +0 -131
  312. lionagi/integrations/pandas_/search_keywords.py +0 -69
  313. lionagi/integrations/pandas_/to_df.py +0 -196
  314. lionagi/integrations/pandas_/update_cells.py +0 -54
  315. lionagi/integrations/perplexity_/PerplexityModel.py +0 -274
  316. lionagi/integrations/perplexity_/PerplexityService.py +0 -118
  317. lionagi/integrations/perplexity_/api_endpoints/__init__.py +0 -0
  318. lionagi/integrations/perplexity_/api_endpoints/api_request.py +0 -171
  319. lionagi/integrations/perplexity_/api_endpoints/chat_completions/__init__.py +0 -0
  320. lionagi/integrations/perplexity_/api_endpoints/chat_completions/request/__init__.py +0 -0
  321. lionagi/integrations/perplexity_/api_endpoints/chat_completions/request/request_body.py +0 -121
  322. lionagi/integrations/perplexity_/api_endpoints/chat_completions/response/__init__.py +0 -0
  323. lionagi/integrations/perplexity_/api_endpoints/chat_completions/response/response_body.py +0 -146
  324. lionagi/integrations/perplexity_/api_endpoints/data_models.py +0 -63
  325. lionagi/integrations/perplexity_/api_endpoints/match_response.py +0 -26
  326. lionagi/integrations/perplexity_/perplexity_max_output_token_data.yaml +0 -3
  327. lionagi/integrations/perplexity_/perplexity_price_data.yaml +0 -10
  328. lionagi/integrations/perplexity_/version.py +0 -1
  329. lionagi/integrations/pydantic_/__init__.py +0 -8
  330. lionagi/integrations/pydantic_/break_down_annotation.py +0 -81
  331. lionagi/integrations/pydantic_/new_model.py +0 -208
  332. lionagi/libs/constants.py +0 -98
  333. lionagi/libs/file/path.py +0 -301
  334. lionagi/libs/file/types.py +0 -22
  335. lionagi/libs/func/__init__.py +0 -0
  336. lionagi/libs/func/async_calls/__init__.py +0 -24
  337. lionagi/libs/func/async_calls/alcall.py +0 -210
  338. lionagi/libs/func/async_calls/bcall.py +0 -130
  339. lionagi/libs/func/async_calls/mcall.py +0 -134
  340. lionagi/libs/func/async_calls/pcall.py +0 -149
  341. lionagi/libs/func/async_calls/rcall.py +0 -217
  342. lionagi/libs/func/async_calls/tcall.py +0 -114
  343. lionagi/libs/func/async_calls/ucall.py +0 -85
  344. lionagi/libs/func/decorators.py +0 -277
  345. lionagi/libs/func/lcall.py +0 -57
  346. lionagi/libs/func/params.py +0 -64
  347. lionagi/libs/func/throttle.py +0 -119
  348. lionagi/libs/func/types.py +0 -39
  349. lionagi/libs/func/utils.py +0 -96
  350. lionagi/libs/package/types.py +0 -26
  351. lionagi/libs/parse/__init__.py +0 -1
  352. lionagi/libs/parse/flatten/__init__.py +0 -9
  353. lionagi/libs/parse/flatten/params.py +0 -52
  354. lionagi/libs/parse/json/__init__.py +0 -27
  355. lionagi/libs/parse/json/extract.py +0 -102
  356. lionagi/libs/parse/json/parse.py +0 -179
  357. lionagi/libs/parse/json/to_json.py +0 -71
  358. lionagi/libs/parse/nested/__init__.py +0 -33
  359. lionagi/libs/parse/nested/to_flat_list.py +0 -64
  360. lionagi/libs/parse/params.py +0 -0
  361. lionagi/libs/parse/string_parse/__init__.py +0 -11
  362. lionagi/libs/parse/type_convert/__init__.py +0 -19
  363. lionagi/libs/parse/type_convert/params.py +0 -145
  364. lionagi/libs/parse/type_convert/to_dict.py +0 -333
  365. lionagi/libs/parse/type_convert/to_list.py +0 -186
  366. lionagi/libs/parse/type_convert/to_num.py +0 -358
  367. lionagi/libs/parse/type_convert/to_str.py +0 -195
  368. lionagi/libs/parse/types.py +0 -9
  369. lionagi/libs/parse/validate/__init__.py +0 -14
  370. lionagi/libs/parse/validate/params.py +0 -62
  371. lionagi/libs/parse/xml/__init__.py +0 -10
  372. lionagi/libs/parse/xml/convert.py +0 -56
  373. lionagi/libs/parse/xml/parser.py +0 -93
  374. lionagi/libs/string_similarity/__init__.py +0 -32
  375. lionagi/libs/string_similarity/matcher.py +0 -102
  376. lionagi/libs/string_similarity/utils.py +0 -15
  377. lionagi/libs/utils.py +0 -266
  378. lionagi/protocols/adapters/__init__.py +0 -0
  379. lionagi/protocols/adapters/adapter.py +0 -79
  380. lionagi/protocols/adapters/json_adapter.py +0 -43
  381. lionagi/protocols/adapters/pandas_adapter.py +0 -96
  382. lionagi/protocols/configs/__init__.py +0 -0
  383. lionagi/protocols/configs/branch_config.py +0 -86
  384. lionagi/protocols/configs/id_config.py +0 -15
  385. lionagi/protocols/configs/imodel_config.py +0 -73
  386. lionagi/protocols/configs/log_config.py +0 -93
  387. lionagi/protocols/configs/retry_config.py +0 -29
  388. lionagi/protocols/configs/types.py +0 -15
  389. lionagi/protocols/operatives/instruct.py +0 -194
  390. lionagi/protocols/operatives/types.py +0 -19
  391. lionagi/protocols/registries/_component_registry.py +0 -23
  392. lionagi/protocols/registries/_pile_registry.py +0 -30
  393. lionagi/service/complete_request_info.py +0 -11
  394. lionagi/service/rate_limiter.py +0 -108
  395. lionagi/service/service.py +0 -41
  396. lionagi/service/service_match_util.py +0 -131
  397. lionagi/service/service_util.py +0 -72
  398. lionagi/service/token_calculator.py +0 -51
  399. lionagi/strategies/__init__.py +0 -0
  400. lionagi/strategies/types.py +0 -21
  401. lionagi-0.5.5.dist-info/RECORD +0 -374
  402. /lionagi/{core → libs/nested}/__init__.py +0 -0
  403. /lionagi/{core/action → libs/schema}/__init__.py +0 -0
  404. /lionagi/{core/communication → libs/validate}/__init__.py +0 -0
  405. /lionagi/{core/forms → operations/strategies}/__init__.py +0 -0
  406. /lionagi/{core/generic → operatives}/__init__.py +0 -0
  407. /lionagi/{core/session → operatives/action}/__init__.py +0 -0
  408. /lionagi/{integrations/anthropic_ → operatives/forms}/__init__.py +0 -0
  409. /lionagi/{core → operatives}/forms/utils.py +0 -0
  410. /lionagi/{integrations/anthropic_/api_endpoints → operatives/instruct}/__init__.py +0 -0
  411. /lionagi/{integrations/anthropic_/api_endpoints/messages → operatives/models}/__init__.py +0 -0
  412. /lionagi/{integrations/anthropic_/api_endpoints/messages/request → protocols/generic}/__init__.py +0 -0
  413. /lionagi/{integrations/groq_ → protocols/graph}/__init__.py +0 -0
  414. /lionagi/{integrations/litellm_ → protocols/mail}/__init__.py +0 -0
  415. /lionagi/{integrations/ollama_ → protocols/messages}/__init__.py +0 -0
  416. /lionagi/{core/communication → protocols/messages}/templates/README.md +0 -0
  417. /lionagi/{core/communication → protocols/messages}/templates/action_request.jinja2 +0 -0
  418. /lionagi/{core/communication → protocols/messages}/templates/action_response.jinja2 +0 -0
  419. /lionagi/{core/communication → protocols/messages}/templates/system_message.jinja2 +0 -0
  420. /lionagi/{core/communication → protocols/messages}/templates/tool_schemas.jinja2 +0 -0
  421. /lionagi/{integrations/ollama_/api_endpoints → service/endpoints}/__init__.py +0 -0
  422. /lionagi/{integrations/ollama_/api_endpoints/chat_completion → service/providers}/__init__.py +0 -0
  423. /lionagi/{integrations/ollama_/api_endpoints/model → service/providers/anthropic_}/__init__.py +0 -0
  424. /lionagi/{integrations/perplexity_ → service/providers/groq_}/__init__.py +0 -0
  425. /lionagi/{protocols/operatives → service/providers/openai_}/__init__.py +0 -0
  426. /lionagi/{protocols/registries → service/providers/openrouter_}/__init__.py +0 -0
  427. {lionagi-0.5.5.dist-info → lionagi-0.6.0.dist-info}/WHEEL +0 -0
  428. {lionagi-0.5.5.dist-info → lionagi-0.6.0.dist-info}/licenses/LICENSE +0 -0
@@ -1,358 +0,0 @@
1
- import re
2
- from decimal import Decimal
3
- from typing import Any
4
-
5
- from lionagi.libs.constants import NUM_TYPES, PATTERNS, TYPE_MAP
6
-
7
-
8
- def to_num(
9
- input_: Any,
10
- /,
11
- *,
12
- upper_bound: int | float | None = None,
13
- lower_bound: int | float | None = None,
14
- num_type: NUM_TYPES = float,
15
- precision: int | None = None,
16
- num_count: int = 1,
17
- ) -> int | float | complex | list[int | float | complex]:
18
- """Convert input to numeric type(s) with validation and bounds checking.
19
-
20
- Args:
21
- input_value: The input to convert to number(s).
22
- upper_bound: Maximum allowed value (inclusive).
23
- lower_bound: Minimum allowed value (inclusive).
24
- num_type: Target numeric type ('int', 'float', 'complex' or type objects).
25
- precision: Number of decimal places for rounding (float only).
26
- num_count: Number of numeric values to extract.
27
-
28
- Returns:
29
- Converted number(s). Single value if num_count=1, else list.
30
-
31
- Raises:
32
- ValueError: For invalid input or out of bounds values.
33
- TypeError: For invalid input types or invalid type conversions.
34
- """
35
- # Validate input
36
- if isinstance(input_, (list, tuple)):
37
- raise TypeError("Input cannot be a sequence")
38
-
39
- # Handle boolean input
40
- if isinstance(input_, bool):
41
- return validate_num_type(num_type)(input_)
42
-
43
- # Handle direct numeric input
44
- if isinstance(input_, (int, float, complex, Decimal)):
45
- inferred_type = type(input_)
46
- if isinstance(input_, Decimal):
47
- inferred_type = float
48
- value = float(input_) if not isinstance(input_, complex) else input_
49
- value = apply_bounds(value, upper_bound, lower_bound)
50
- value = apply_precision(value, precision)
51
- return convert_type(value, validate_num_type(num_type), inferred_type)
52
-
53
- # Convert input to string and extract numbers
54
- input_str = str(input_)
55
- number_matches = extract_numbers(input_str)
56
-
57
- if not number_matches:
58
- raise ValueError(f"No valid numbers found in: {input_str}")
59
-
60
- # Process numbers
61
- results = []
62
- target_type = validate_num_type(num_type)
63
-
64
- number_matches = (
65
- number_matches[:num_count]
66
- if num_count < len(number_matches)
67
- else number_matches
68
- )
69
-
70
- for type_and_value in number_matches:
71
- try:
72
- # Infer appropriate type
73
- inferred_type = infer_type(type_and_value)
74
-
75
- # Parse to numeric value
76
- value = parse_number(type_and_value)
77
-
78
- # Apply bounds if not complex
79
- value = apply_bounds(value, upper_bound, lower_bound)
80
-
81
- # Apply precision
82
- value = apply_precision(value, precision)
83
-
84
- # Convert to target type if different from inferred
85
- value = convert_type(value, target_type, inferred_type)
86
-
87
- results.append(value)
88
-
89
- except Exception as e:
90
- if len(type_and_value) == 2:
91
- raise type(e)(
92
- f"Error processing {type_and_value[1]}: {str(e)}"
93
- )
94
- raise type(e)(f"Error processing {type_and_value}: {str(e)}")
95
-
96
- if results and num_count == 1:
97
- return results[0]
98
- return results
99
-
100
-
101
- def extract_numbers(text: str) -> list[tuple[str, str]]:
102
- """Extract numeric values from text using ordered regex patterns.
103
-
104
- Args:
105
- text: The text to extract numbers from.
106
-
107
- Returns:
108
- List of tuples containing (pattern_type, matched_value).
109
- """
110
- combined_pattern = "|".join(PATTERNS.values())
111
- matches = re.finditer(combined_pattern, text, re.IGNORECASE)
112
- numbers = []
113
-
114
- for match in matches:
115
- value = match.group()
116
- # Check which pattern matched
117
- for pattern_name, pattern in PATTERNS.items():
118
- if re.fullmatch(pattern, value, re.IGNORECASE):
119
- numbers.append((pattern_name, value))
120
- break
121
-
122
- return numbers
123
-
124
-
125
- def validate_num_type(num_type: NUM_TYPES) -> type:
126
- """Validate and normalize numeric type specification.
127
-
128
- Args:
129
- num_type: The numeric type to validate.
130
-
131
- Returns:
132
- The normalized Python type object.
133
-
134
- Raises:
135
- ValueError: If the type specification is invalid.
136
- """
137
- if isinstance(num_type, str):
138
- if num_type not in TYPE_MAP:
139
- raise ValueError(f"Invalid number type: {num_type}")
140
- return TYPE_MAP[num_type]
141
-
142
- if num_type not in (int, float, complex):
143
- raise ValueError(f"Invalid number type: {num_type}")
144
- return num_type
145
-
146
-
147
- def infer_type(value: tuple[str, str]) -> type:
148
- """Infer appropriate numeric type from value.
149
-
150
- Args:
151
- value: Tuple of (pattern_type, matched_value).
152
-
153
- Returns:
154
- The inferred Python type.
155
- """
156
- pattern_type, _ = value
157
- if pattern_type in ("complex", "complex_sci", "pure_imaginary"):
158
- return complex
159
- return float
160
-
161
-
162
- def convert_special(value: str) -> float:
163
- """Convert special float values (inf, -inf, nan).
164
-
165
- Args:
166
- value: The string value to convert.
167
-
168
- Returns:
169
- The converted float value.
170
- """
171
- value = value.lower()
172
- if "infinity" in value or "inf" in value:
173
- return float("-inf") if value.startswith("-") else float("inf")
174
- return float("nan")
175
-
176
-
177
- def convert_percentage(value: str) -> float:
178
- """Convert percentage string to float.
179
-
180
- Args:
181
- value: The percentage string to convert.
182
-
183
- Returns:
184
- The converted float value.
185
-
186
- Raises:
187
- ValueError: If the percentage value is invalid.
188
- """
189
- try:
190
- return float(value.rstrip("%")) / 100
191
- except ValueError as e:
192
- raise ValueError(f"Invalid percentage value: {value}") from e
193
-
194
-
195
- def convert_complex(value: str) -> complex:
196
- """Convert complex number string to complex.
197
-
198
- Args:
199
- value: The complex number string to convert.
200
-
201
- Returns:
202
- The converted complex value.
203
-
204
- Raises:
205
- ValueError: If the complex number is invalid.
206
- """
207
- try:
208
- # Handle pure imaginary numbers
209
- if value.endswith("j") or value.endswith("J"):
210
- if value in ("j", "J"):
211
- return complex(0, 1)
212
- if value in ("+j", "+J"):
213
- return complex(0, 1)
214
- if value in ("-j", "-J"):
215
- return complex(0, -1)
216
- if "+" not in value and "-" not in value[1:]:
217
- # Pure imaginary number
218
- imag = float(value[:-1] or "1")
219
- return complex(0, imag)
220
-
221
- return complex(value.replace(" ", ""))
222
- except ValueError as e:
223
- raise ValueError(f"Invalid complex number: {value}") from e
224
-
225
-
226
- def convert_type(
227
- value: float | complex,
228
- target_type: type,
229
- inferred_type: type,
230
- ) -> int | float | complex:
231
- """Convert value to target type if specified, otherwise use inferred type.
232
-
233
- Args:
234
- value: The value to convert.
235
- target_type: The requested target type.
236
- inferred_type: The inferred type from the value.
237
-
238
- Returns:
239
- The converted value.
240
-
241
- Raises:
242
- TypeError: If the conversion is not possible.
243
- """
244
- try:
245
- # If no specific type requested, use inferred type
246
- if target_type is float and inferred_type is complex:
247
- return value
248
-
249
- # Handle explicit type conversions
250
- if target_type is int and isinstance(value, complex):
251
- raise TypeError("Cannot convert complex number to int")
252
- return target_type(value)
253
- except (ValueError, TypeError) as e:
254
- raise TypeError(
255
- f"Cannot convert {value} to {target_type.__name__}"
256
- ) from e
257
-
258
-
259
- def apply_bounds(
260
- value: float | complex,
261
- upper_bound: float | None = None,
262
- lower_bound: float | None = None,
263
- ) -> float | complex:
264
- """Apply bounds checking to numeric value.
265
-
266
- Args:
267
- value: The value to check.
268
- upper_bound: Maximum allowed value (inclusive).
269
- lower_bound: Minimum allowed value (inclusive).
270
-
271
- Returns:
272
- The validated value.
273
-
274
- Raises:
275
- ValueError: If the value is outside bounds.
276
- """
277
- if isinstance(value, complex):
278
- return value
279
-
280
- if upper_bound is not None and value > upper_bound:
281
- raise ValueError(f"Value {value} exceeds upper bound {upper_bound}")
282
- if lower_bound is not None and value < lower_bound:
283
- raise ValueError(f"Value {value} below lower bound {lower_bound}")
284
- return value
285
-
286
-
287
- def apply_precision(
288
- value: float | complex,
289
- precision: int | None,
290
- ) -> float | complex:
291
- """Apply precision rounding to numeric value.
292
-
293
- Args:
294
- value: The value to round.
295
- precision: Number of decimal places.
296
-
297
- Returns:
298
- The rounded value.
299
- """
300
- if precision is None or isinstance(value, complex):
301
- return value
302
- if isinstance(value, float):
303
- return round(value, precision)
304
- return value
305
-
306
-
307
- def parse_number(type_and_value: tuple[str, str]) -> float | complex:
308
- """Parse string to numeric value based on pattern type.
309
-
310
- Args:
311
- type_and_value: Tuple of (pattern_type, matched_value).
312
-
313
- Returns:
314
- The parsed numeric value.
315
-
316
- Raises:
317
- ValueError: If parsing fails.
318
- """
319
- num_type, value = type_and_value
320
- value = value.strip()
321
-
322
- try:
323
- if num_type == "special":
324
- return convert_special(value)
325
-
326
- if num_type == "percentage":
327
- return convert_percentage(value)
328
-
329
- if num_type == "fraction":
330
- if "/" not in value:
331
- raise ValueError(f"Invalid fraction: {value}")
332
- if value.count("/") > 1:
333
- raise ValueError(f"Invalid fraction: {value}")
334
- num, denom = value.split("/")
335
- if not (num.strip("-").isdigit() and denom.isdigit()):
336
- raise ValueError(f"Invalid fraction: {value}")
337
- denom_val = float(denom)
338
- if denom_val == 0:
339
- raise ValueError("Division by zero")
340
- return float(num) / denom_val
341
- if num_type in ("complex", "complex_sci", "pure_imaginary"):
342
- return convert_complex(value)
343
- if num_type == "scientific":
344
- if "e" not in value.lower():
345
- raise ValueError(f"Invalid scientific notation: {value}")
346
- parts = value.lower().split("e")
347
- if len(parts) != 2:
348
- raise ValueError(f"Invalid scientific notation: {value}")
349
- if not (parts[1].lstrip("+-").isdigit()):
350
- raise ValueError(f"Invalid scientific notation: {value}")
351
- return float(value)
352
- if num_type == "decimal":
353
- return float(value)
354
-
355
- raise ValueError(f"Unknown number type: {num_type}")
356
- except Exception as e:
357
- # Preserve the specific error type but wrap with more context
358
- raise type(e)(f"Failed to parse {value} as {num_type}: {str(e)}")
@@ -1,195 +0,0 @@
1
- import json
2
- from collections.abc import Callable, Mapping
3
- from typing import Any, Literal
4
-
5
- from pydantic_core import PydanticUndefined, PydanticUndefinedType
6
-
7
- from lionagi.libs.constants import UNDEFINED, UndefinedType
8
-
9
- from ..xml.convert import dict_to_xml
10
- from .to_dict import to_dict
11
-
12
-
13
- def _serialize_as(
14
- input_,
15
- /,
16
- *,
17
- serialize_as: Literal["json", "xml"],
18
- strip_lower: bool = False,
19
- chars: str | None = None,
20
- str_type: Literal["json", "xml"] | None = None,
21
- use_model_dump: bool = False,
22
- str_parser: Callable[[str], dict[str, Any]] | None = None,
23
- parser_kwargs: dict = {},
24
- **kwargs: Any,
25
- ) -> str:
26
- try:
27
- dict_ = to_dict(
28
- input_,
29
- use_model_dump=use_model_dump,
30
- str_type=str_type,
31
- suppress=True,
32
- parser=str_parser,
33
- **parser_kwargs,
34
- )
35
- if any((str_type, chars)):
36
- str_ = json.dumps(dict_)
37
- str_ = _process_string(str_, strip_lower=strip_lower, chars=chars)
38
- dict_ = json.loads(str_)
39
-
40
- if serialize_as == "json":
41
- return json.dumps(dict_, **kwargs)
42
-
43
- if serialize_as == "xml":
44
- return dict_to_xml(dict_, **kwargs)
45
- except Exception as e:
46
- raise ValueError(
47
- f"Failed to serialize input of {type(input_).__name__} "
48
- f"into <{str_type}>"
49
- ) from e
50
-
51
-
52
- def _to_str_type(input_: Any, /) -> str:
53
- if input_ in [set(), [], {}]:
54
- return ""
55
-
56
- if isinstance(input_, type(None) | UndefinedType | PydanticUndefinedType):
57
- return ""
58
-
59
- if isinstance(input_, bytes | bytearray):
60
- return input_.decode("utf-8", errors="replace")
61
-
62
- if isinstance(input_, str):
63
- return input_
64
-
65
- if isinstance(input_, Mapping):
66
- return json.dumps(dict(input_))
67
-
68
- try:
69
- return str(input_)
70
- except Exception as e:
71
- raise ValueError(
72
- f"Could not convert input of type <{type(input_).__name__}> "
73
- "to string"
74
- ) from e
75
-
76
-
77
- def to_str(
78
- input_: Any,
79
- /,
80
- *,
81
- strip_lower: bool = False,
82
- chars: str | None = None,
83
- str_type: Literal["json", "xml"] | None = None,
84
- serialize_as: Literal["json", "xml"] | None = None,
85
- use_model_dump: bool = False,
86
- str_parser: Callable[[str], dict[str, Any]] | None = None,
87
- parser_kwargs: dict = {},
88
- **kwargs: Any,
89
- ) -> str:
90
- """Convert any input to its string representation.
91
-
92
- Handles various input types, with options for serialization and formatting.
93
-
94
- Args:
95
- input_: The input to convert to a string.
96
- strip_lower: If True, strip whitespace and convert to lowercase.
97
- chars: Specific characters to strip from the result.
98
- str_type: Type of string input ("json" or "xml") if applicable.
99
- serialize_as: Output serialization format ("json" or "xml").
100
- use_model_dump: Use model_dump for Pydantic models if available.
101
- str_parser: Custom parser function for string inputs.
102
- parser_kwargs: Additional keyword arguments for the parser.
103
- **kwargs: Additional arguments passed to json.dumps or serialization.
104
-
105
- Returns:
106
- str: The string representation of the input.
107
-
108
- Raises:
109
- ValueError: If serialization or conversion fails.
110
-
111
- Examples:
112
- >>> to_str(123)
113
- '123'
114
- >>> to_str(" HELLO ", strip_lower=True)
115
- 'hello'
116
- >>> to_str({"a": 1}, serialize_as="json")
117
- '{"a": 1}'
118
- >>> to_str({"a": 1}, serialize_as="xml")
119
- '<root><a>1</a></root>'
120
- """
121
-
122
- if serialize_as:
123
- return _serialize_as(
124
- input_,
125
- serialize_as=serialize_as,
126
- strip_lower=strip_lower,
127
- chars=chars,
128
- str_type=str_type,
129
- use_model_dump=use_model_dump,
130
- str_parser=str_parser,
131
- parser_kwargs=parser_kwargs,
132
- **kwargs,
133
- )
134
-
135
- str_ = _to_str_type(input_, **kwargs)
136
- if any((strip_lower, chars)):
137
- str_ = _process_string(str_, strip_lower=strip_lower, chars=chars)
138
- return str_
139
-
140
-
141
- def _process_string(s: str, strip_lower: bool, chars: str | None) -> str:
142
- if s in [UNDEFINED, PydanticUndefined, None, [], {}]:
143
- return ""
144
-
145
- if strip_lower:
146
- s = s.lower()
147
- s = s.strip(chars) if chars is not None else s.strip()
148
- return s
149
-
150
-
151
- def strip_lower(
152
- input_: Any,
153
- /,
154
- *,
155
- chars: str | None = None,
156
- str_type: Literal["json", "xml"] | None = None,
157
- serialize_as: Literal["json", "xml"] | None = None,
158
- use_model_dump: bool = False,
159
- str_parser: Callable[[str], dict[str, Any]] | None = None,
160
- parser_kwargs: dict = {},
161
- **kwargs: Any,
162
- ) -> str:
163
- """
164
- Convert input to stripped and lowercase string representation.
165
-
166
- This function is a convenience wrapper around to_str that always
167
- applies stripping and lowercasing.
168
-
169
- Args:
170
- input_: The input to convert to a string.
171
- use_model_dump: If True, use model_dump for Pydantic models.
172
- chars: Characters to strip from the result.
173
- **kwargs: Additional arguments to pass to to_str.
174
-
175
- Returns:
176
- Stripped and lowercase string representation of the input.
177
-
178
- Raises:
179
- ValueError: If conversion fails.
180
-
181
- Example:
182
- >>> strip_lower(" HELLO WORLD ")
183
- 'hello world'
184
- """
185
- return to_str(
186
- input_,
187
- strip_lower=True,
188
- chars=chars,
189
- str_type=str_type,
190
- serialize_as=serialize_as,
191
- use_model_dump=use_model_dump,
192
- str_parser=str_parser,
193
- parser_kwargs=parser_kwargs,
194
- **kwargs,
195
- )
@@ -1,9 +0,0 @@
1
- """Parsing utilities for various data formats and structures."""
2
-
3
- from .flatten import *
4
- from .json import *
5
- from .nested import *
6
- from .string_parse import *
7
- from .type_convert import *
8
- from .validate import *
9
- from .xml import *
@@ -1,14 +0,0 @@
1
- """Validation utilities for various data types and structures."""
2
-
3
- from .boolean import validate_boolean
4
- from .keys import KeysDict, validate_keys
5
- from .mapping import validate_mapping
6
- from .params import ValidateKeysParams
7
-
8
- __all__ = [
9
- "validate_boolean",
10
- "validate_keys",
11
- "validate_mapping",
12
- "KeysDict",
13
- "ValidateKeysParams",
14
- ]
@@ -1,62 +0,0 @@
1
- from collections.abc import Callable
2
- from typing import Any, Literal
3
-
4
- from pydantic import BaseModel, Field, field_validator
5
-
6
- from lionagi.libs.string_similarity import SIMILARITY_TYPE
7
- from lionagi.libs.string_similarity.algorithms import SIMILARITY_ALGO_MAP
8
-
9
-
10
- class ValidateKeysParams(BaseModel):
11
- """Parameters for validate_keys function.
12
-
13
- Attributes:
14
- similarity_algo: String similarity algorithm to use or custom function
15
- similarity_threshold: Minimum similarity score for fuzzy matching
16
- fuzzy_match: If True, use fuzzy matching for key correction
17
- handle_unmatched: How to handle unmatched keys
18
- fill_value: Default value for filling unmatched keys
19
- fill_mapping: Dictionary mapping unmatched keys to default values
20
- strict: If True, raise ValueError if any expected key is missing
21
- """
22
-
23
- similarity_algo: SIMILARITY_TYPE | Callable[[str, str], float] = Field(
24
- default="jaro_winkler",
25
- description="String similarity algorithm to use or custom function",
26
- )
27
- similarity_threshold: float = Field(
28
- default=0.85,
29
- description="Minimum similarity score for fuzzy matching",
30
- ge=0.0,
31
- le=1.0,
32
- )
33
- fuzzy_match: bool = Field(
34
- default=True,
35
- description="If True, use fuzzy matching for key correction",
36
- )
37
- handle_unmatched: Literal["ignore", "raise", "remove", "fill", "force"] = (
38
- Field(
39
- default="ignore",
40
- description="How to handle unmatched keys",
41
- )
42
- )
43
- fill_value: Any = Field(
44
- default=None,
45
- description="Default value for filling unmatched keys",
46
- )
47
- fill_mapping: dict[str, Any] | None = Field(
48
- default=None,
49
- description="Dictionary mapping unmatched keys to default values",
50
- )
51
- strict: bool = Field(
52
- default=False,
53
- description="If True, raise ValueError if any expected key is missing",
54
- )
55
-
56
- @field_validator("similarity_algo")
57
- def validate_similarity_algo(
58
- cls, v: SIMILARITY_TYPE | Callable[[str, str], float]
59
- ) -> SIMILARITY_TYPE | Callable[[str, str], float]:
60
- if isinstance(v, str) and v not in SIMILARITY_ALGO_MAP:
61
- raise ValueError(f"Unknown similarity algorithm: {v}")
62
- return v
@@ -1,10 +0,0 @@
1
- """XML parsing and conversion utilities."""
2
-
3
- from .convert import dict_to_xml, xml_to_dict
4
- from .parser import XMLParser
5
-
6
- __all__ = [
7
- "dict_to_xml",
8
- "xml_to_dict",
9
- "XMLParser",
10
- ]
@@ -1,56 +0,0 @@
1
- import xml.etree.ElementTree as ET
2
- from typing import Any
3
-
4
- from .parser import XMLParser
5
-
6
-
7
- def xml_to_dict(
8
- xml_string: str,
9
- /,
10
- suppress=False,
11
- remove_root: bool = True,
12
- root_tag: str = None,
13
- ) -> dict[str, Any]:
14
- """
15
- Parse an XML string into a nested dictionary structure.
16
-
17
- This function converts an XML string into a dictionary where:
18
- - Element tags become dictionary keys
19
- - Text content is assigned directly to the tag key if there are no children
20
- - Attributes are stored in a '@attributes' key
21
- - Multiple child elements with the same tag are stored as lists
22
-
23
- Args:
24
- xml_string: The XML string to parse.
25
-
26
- Returns:
27
- A dictionary representation of the XML structure.
28
-
29
- Raises:
30
- ValueError: If the XML is malformed or parsing fails.
31
- """
32
- try:
33
- a = XMLParser(xml_string).parse()
34
- if remove_root and (root_tag or "root") in a:
35
- a = a[root_tag or "root"]
36
- return a
37
- except ValueError as e:
38
- if not suppress:
39
- raise e
40
-
41
-
42
- def dict_to_xml(data: dict, /, root_tag: str = "root") -> str:
43
-
44
- root = ET.Element(root_tag)
45
-
46
- def convert(dict_obj: dict, parent: Any) -> None:
47
- for key, val in dict_obj.items():
48
- if isinstance(val, dict):
49
- element = ET.SubElement(parent, key)
50
- convert(dict_obj=val, parent=element)
51
- else:
52
- element = ET.SubElement(parent, key)
53
- element.text = str(object=val)
54
-
55
- convert(dict_obj=data, parent=root)
56
- return ET.tostring(root, encoding="unicode")