lionagi 0.3.8__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.3.8.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.3.8.dist-info/METADATA +0 -241
  563. lionagi-0.3.8.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.3.8.dist-info → lionagi-0.5.0.dist-info/licenses}/LICENSE +0 -0
@@ -1,294 +0,0 @@
1
- from typing import List, Optional
2
-
3
- from lionagi.experimental.directive.tokenizer import BaseToken
4
-
5
- from ..template.schema import ForNode, IfNode, TryNode
6
-
7
-
8
- class BaseDirectiveParser:
9
- """A base parser with lookahead, error recovery, and backtracking support.
10
-
11
- Attributes:
12
- tokens (List[BaseToken]): A list of tokens to be parsed.
13
- current_token_index (int): The index of the current token in the tokens list.
14
- current_token (Optional[BaseToken]): The current token being processed.
15
-
16
- Examples:
17
- >>> tokenizer = BaseTokenizer("IF x > 10 THEN DO something ENDIF")
18
- >>> tokens = tokenizer.get_tokens()
19
- >>> parser = BaseParser(tokens)
20
- >>> print(parser.current_token)
21
- BaseToken(KEYWORD, IF)
22
- """
23
-
24
- def __init__(self, tokens: list[BaseToken]):
25
- self.tokens = tokens
26
- self.current_token_index = -1
27
- self.current_token: BaseToken | None = None
28
- self.next_token()
29
-
30
- def next_token(self) -> None:
31
- """Advances to the next token in the list."""
32
- self.current_token_index += 1
33
- if self.current_token_index < len(self.tokens):
34
- self.current_token = self.tokens[self.current_token_index]
35
- else:
36
- self.current_token = None
37
-
38
- def peek_next_token(self, offset: int = 1) -> BaseToken | None:
39
- """Peeks at the next token without consuming it.
40
-
41
- Args:
42
- offset (int): The number of tokens to look ahead.
43
-
44
- Returns:
45
- Optional[BaseToken]: The token at the specified lookahead offset, or None if end of list.
46
- """
47
- peek_index = self.current_token_index + offset
48
- if peek_index < len(self.tokens):
49
- return self.tokens[peek_index]
50
- else:
51
- return None
52
-
53
- def skip_until(self, token_types: list[str]) -> None:
54
- """Skips tokens until a token of the specified type is found.
55
-
56
- Args:
57
- token_types (List[str]): A list of token types to stop skipping.
58
- """
59
- while (
60
- self.current_token and self.current_token.type not in token_types
61
- ):
62
- self.next_token()
63
-
64
- def mark(self) -> int:
65
- """Marks the current position in the token list for potential backtracking.
66
-
67
- Returns:
68
- int: The current token index.
69
- """
70
- return self.current_token_index
71
-
72
- def reset_to_mark(self, mark: int) -> None:
73
- """Resets the parser to a previously marked position.
74
-
75
- Args:
76
- mark (int): The token index to reset to.
77
- """
78
- self.current_token_index = mark - 1
79
- self.next_token()
80
-
81
- def skip_semicolon(self):
82
- """Skips a semicolon token if it is the current token."""
83
- if self.current_token and self.current_token.value == ";":
84
- self.next_token()
85
-
86
- def parse_expression(self):
87
- """Parses an expression until a semicolon is encountered.
88
-
89
- Returns:
90
- str: The parsed expression as a string.
91
-
92
- Raises:
93
- SyntaxError: If a semicolon is not found at the end of the expression.
94
- """
95
- expr = ""
96
- while self.current_token and self.current_token.value != ";":
97
- expr += self.current_token.value + " "
98
- self.next_token()
99
- # Expecting a semicolon at the end of the condition
100
- if self.current_token.value != ";":
101
- raise SyntaxError("Expected ';' at the end of the condition")
102
- self.next_token() # Move past the semicolon to the next part of the statement
103
- return expr.strip()
104
-
105
- def parse_if_block(self):
106
- """Parses a block of statements for an IF condition.
107
-
108
- Returns:
109
- list: The parsed block of statements as a list of strings.
110
- """
111
- block = []
112
- # Parse the block until 'ELSE', 'ENDIF', ensuring not to include semicolons as part of the block
113
- while self.current_token and self.current_token.value not in (
114
- "ENDIF",
115
- "ELSE",
116
- ):
117
- if self.current_token.value == "DO":
118
- self.next_token() # Move past 'DO' to get to the action
119
- block.append(
120
- self.current_token.value
121
- ) # Add the action to the block
122
- self.next_token() # Move to the next token, which could be a semicolon or the next action
123
- if self.current_token.value == ";":
124
- self.next_token() # Move past the semicolon
125
- return block
126
-
127
- def parse_if_statement(self):
128
- """Parses an IF statement.
129
-
130
- Returns:
131
- IfNode: The parsed IF statement as an IfNode object.
132
-
133
- Raises:
134
- SyntaxError: If the IF statement is not properly formed.
135
- """
136
- if (
137
- self.current_token.type != "KEYWORD"
138
- or self.current_token.value != "IF"
139
- ):
140
- raise SyntaxError("Expected IF statement")
141
- self.next_token() # Skip 'IF'
142
-
143
- condition = (
144
- self.parse_expression()
145
- ) # Now properly ends after the semicolon
146
-
147
- true_block = []
148
- if self.current_token.value == "DO":
149
- true_block = self.parse_if_block() # Parse true block after 'DO'
150
-
151
- false_block = None
152
- if self.current_token and self.current_token.value == "ELSE":
153
- self.next_token() # Skip 'ELSE', expect 'DO' next for the false block
154
- self.skip_semicolon()
155
- if self.current_token.value != "DO":
156
- raise SyntaxError("Expected 'DO' after 'ELSE'")
157
- self.next_token() # Skip 'DO'
158
- false_block = self.parse_if_block() # Parse false block
159
-
160
- return IfNode(condition, true_block, false_block)
161
-
162
- def parse_for_statement(self):
163
- """Parses a FOR statement.
164
-
165
- Returns:
166
- ForNode: The parsed FOR statement as a ForNode object.
167
-
168
- Raises:
169
- SyntaxError: If the FOR statement is not properly formed.
170
- """
171
- if (
172
- self.current_token.type != "KEYWORD"
173
- or self.current_token.value != "FOR"
174
- ):
175
- raise SyntaxError("Expected FOR statement")
176
- self.next_token() # Skip 'FOR'
177
-
178
- # Parse the iterator variable
179
- if self.current_token.type != "IDENTIFIER":
180
- raise SyntaxError("Expected iterator variable after FOR")
181
- iterator = self.current_token.value
182
- self.next_token() # Move past the iterator variable
183
-
184
- # Expect and skip 'IN' keyword
185
- if (
186
- self.current_token.type != "KEYWORD"
187
- or self.current_token.value != "IN"
188
- ):
189
- raise SyntaxError("Expected 'IN' after iterator variable")
190
- self.next_token() # Move past 'IN'
191
-
192
- # Parse the collection
193
- if self.current_token.type not in ["IDENTIFIER", "LITERAL"]:
194
- raise SyntaxError("Expected collection after 'IN'")
195
- collection = self.current_token.value
196
- self.next_token() # Move past the collection
197
-
198
- # Now, parse the block of statements to execute
199
- true_block = self.parse_for_block()
200
-
201
- # Construct and return a ForNode
202
- return ForNode(iterator, collection, true_block)
203
-
204
- def parse_for_block(self):
205
- """Parses a block of statements for a FOR loop.
206
-
207
- Returns:
208
- list: The parsed block of statements as a list of strings.
209
- """
210
- block = []
211
- # Skip initial 'DO' if present
212
- if self.current_token and self.current_token.value == "DO":
213
- self.next_token()
214
-
215
- while self.current_token and self.current_token.value not in (
216
- "ENDFOR",
217
- ):
218
- if self.current_token.value == ";":
219
- # If a semicolon is encountered, skip it and move to the next token
220
- self.next_token()
221
- continue
222
- # Add the current token to the block unless it's a 'DO' or ';'
223
- if self.current_token.value != "DO":
224
- block.append(self.current_token.value)
225
- self.next_token()
226
-
227
- # The loop exits when 'ENDFOR' is encountered; move past it for subsequent parsing
228
- self.next_token() # Skip 'ENDFOR'
229
- return block
230
-
231
- def parse_try_statement(self):
232
- """Parses a TRY statement.
233
-
234
- Returns:
235
- TryNode: The parsed TRY statement as a TryNode object.
236
-
237
- Raises:
238
- SyntaxError: If the TRY statement is not properly formed.
239
- """
240
- if (
241
- self.current_token.type != "KEYWORD"
242
- or self.current_token.value != "TRY"
243
- ):
244
- raise SyntaxError("Expected TRY statement")
245
- self.next_token() # Skip 'TRY'
246
-
247
- try_block = self.parse_try_block(
248
- "EXCEPT"
249
- ) # Parse the try block until 'EXCEPT'
250
-
251
- # Now expecting 'EXCEPT' keyword
252
- if not (self.current_token and self.current_token.value == "EXCEPT"):
253
- raise SyntaxError("Expected 'EXCEPT' after try block")
254
- self.next_token() # Move past 'EXCEPT'
255
-
256
- except_block = self.parse_try_block(
257
- "ENDTRY"
258
- ) # Parse the except block until 'ENDTRY'
259
-
260
- # Ensure we are correctly positioned after 'ENDTRY'
261
- if self.current_token and self.current_token.value != "ENDTRY":
262
- raise SyntaxError("Expected 'ENDTRY' at the end of except block")
263
- self.next_token() # Move past 'ENDTRY' for subsequent parsing
264
-
265
- return TryNode(try_block, except_block)
266
-
267
- def parse_try_block(self, stop_keyword):
268
- """Parses a block of statements for a TRY or EXCEPT clause.
269
-
270
- Args:
271
- stop_keyword (str): The keyword that indicates the end of the block.
272
-
273
- Returns:
274
- list: The parsed block of statements as a list of strings.
275
- """
276
- block = []
277
- while self.current_token and self.current_token.value != stop_keyword:
278
- if self.current_token.value == "DO":
279
- self.next_token() # Move past 'DO' to get to the action
280
- elif self.current_token.value == ";":
281
- self.next_token() # Move past the semicolon
282
- continue # Skip adding ';' to the block
283
- else:
284
- block.append(
285
- self.current_token.value
286
- ) # Add the action to the block
287
- self.next_token()
288
-
289
- return block
290
-
291
-
292
- # "IF condition1 && condition2; DO action2; ELSE; DO action3; ENDIF;"
293
- # "FOR input_ IN collections; DO action(input_); ENDFOR;"
294
- # "TRY; DO action(); EXCEPT; DO action(input_); ENDTRY;"
@@ -1,200 +0,0 @@
1
-
2
- """
3
- # basic syntax
4
- executeAction(param1, param2) IF condition1 && condition2 || !condition3;
5
-
6
- # verb
7
- DO
8
-
9
- # logic operator
10
-
11
- AND: &&
12
- OR: ||
13
- NOT: !
14
- GREATER_THAN: >
15
- LESS_THAN: <
16
- GREATER_THAN_OR_EQUAL: >=
17
- LESS_THAN_OR_EQUAL: <=
18
- EQUAL: ==
19
- NOT_EQUAL: !=
20
-
21
- # conditions
22
- IF
23
- ELIF
24
- ELSE
25
-
26
- # loop
27
- FOR
28
- IN
29
-
30
- # EXCEPTION
31
- TRY
32
- EXCEPT
33
-
34
- """
35
-
36
- """
37
- DO: perform a single action
38
- THEN: extend directive to include more statment
39
- IF: condition
40
-
41
- FOR
42
- IN
43
- END
44
- BEGIN
45
-
46
- TRY
47
- EXCEPT
48
-
49
- ---
50
-
51
- tools 1,2,3
52
-
53
-
54
-
55
- def main(): <-> compose
56
-
57
- If condition1 && condition2 || !condition3;
58
- DO tool1(param1, param2);
59
- ENDIF;
60
-
61
- IF must follow condition, (condition parsing logic);
62
- DO ....;
63
- ENDIF;
64
-
65
-
66
-
67
-
68
- THEN;
69
- DO tool2(param1, param2);
70
-
71
- THEN;
72
-
73
- TRY; DO ACTION_C;
74
- EXCEPT; DO ACTION_D;
75
- ENDEXCEPT;
76
- ENDTRY;
77
-
78
- RETURN <-> run
79
-
80
-
81
-
82
-
83
- COMPOSE; - indicate the beginning of a script-like block to execute
84
-
85
- THEN; - indicate continuation of a script-like block, another statement to execute
86
-
87
- RUN; - indicate the end of script-like block, and let the script execute
88
-
89
-
90
- XX; ; END XX;
91
-
92
-
93
-
94
- TRY; DO ACTION_A; END TRY;
95
-
96
-
97
-
98
- COMPOSE;
99
-
100
- IF CONDITION_A; DO ACTION_B; ELIF CONDITION_C; DO ACTION_D; ELSE; DO ACTION_E; ENDIF;
101
-
102
- THEN TRY; DO ACTION_C; EXCEPT; DO ACTION_D; ENDEXCEPT; ENDTRY;
103
-
104
-
105
- THEN TRY; DO ACTION_C ENDTRY;
106
-
107
-
108
-
109
- THEN IF EXCEPT; IF CONDITION_C; DO ACTION_D; ENDEXCEPT
110
-
111
- ;
112
- THEN FOR ITEM IN COLLECTION; DO ACTION_E(input_.param1, input_.param2); ENDFOR
113
- RUN;
114
-
115
-
116
-
117
- DO ACTION B;
118
-
119
- IF
120
- THEN
121
-
122
- GROUP
123
-
124
- """
125
-
126
- """syntax:
127
-
128
-
129
- BEGIN; IF condition1 && condition2 || !condition3; THEN DO action(param1, param2); END IF; END;
130
-
131
- BEGIN; FOR input_ IN nested_structure; DO action(input_.param1, input_.param2); END FOR; END;
132
-
133
- BEGIN; DO action1(param1, param2); DO action2(param3, param4); END GROUP; END;
134
-
135
-
136
- """
137
-
138
- """
139
- FOR input_ IN nested_structure DO
140
- executeAction(input_.param1, input_.param2);
141
- END FOR;
142
- """
143
-
144
- """
145
- example 1:
146
-
147
- BEGIN; IF condition1 && condition2 || !condition3; DO action1(param1, param2); ENDIF;
148
- FOR input_ IN collections DO action2(input_.param1, input_.param2); ENDFOR; END;
149
-
150
-
151
- example 2:
152
- BEGIN; IF condition1; TRY DO action1(param1, param2); EXCEPT DO action2(param3, param4);
153
- ENDTRY; ELIF condition2; DO action2(param1, param2); ELSE DO action3(param1, param2);
154
- ENDIF; END;
155
-
156
- """
157
-
158
- """
159
- BEGIN
160
- executeAction1(param1, param2);
161
- IF condition2 THEN executeAction2(param3, param4);
162
- FOR input_ IN nested_structure DO
163
- executeAction3(input_.param1, input_.param2);
164
- END FOR;
165
- END;
166
- """
167
-
168
- """
169
- TRY
170
- executeAction(param1, param2);
171
- EXCEPT
172
- handleErrorAction(param1, param2);
173
- """
174
-
175
- """
176
-
177
- CHAT / REACT
178
-
179
-
180
- DO 1,2
181
- session = Li.Session(..., tools=tools)
182
- await alcall(session.branches, func_1)
183
-
184
-
185
- THEN DO 3,4
186
- session.chat.py(3,4)
187
-
188
-
189
- THEN 5
190
- session.chat.py(5)
191
-
192
-
193
- call (1,2,3,4,5)
194
-
195
- run 1,2,3,4,5
196
-
197
-
198
- 1->2, or 1->3, or 1->4, THEN....
199
-
200
- """
@@ -1,71 +0,0 @@
1
- import re
2
- from typing import Any, Dict
3
-
4
- from ..evaluator.base_evaluator import BaseEvaluator
5
-
6
-
7
- class DirectiveTemplate:
8
- """Enhanced base template class for processing templates with conditionals and loops."""
9
-
10
- def __init__(self, template_str: str):
11
- self.template_str = template_str
12
- self.evaluator = BaseEvaluator()
13
-
14
- def _render_conditionals(self, context: dict[str, Any]) -> str:
15
- """Processes conditional statements with improved logic and support for 'else'."""
16
- pattern = re.compile(
17
- r"\{if (.*?)\}(.*?)\{else\}(.*?)\{endif\}", re.DOTALL
18
- )
19
-
20
- def evaluate_condition(match):
21
- condition, if_text, else_text = match.groups()
22
- if self.evaluator.evaluate(condition, context):
23
- return if_text
24
- else:
25
- return else_text
26
-
27
- return pattern.sub(evaluate_condition, self.template_str)
28
-
29
- def _render_loops(self, template: str, context: dict[str, Any]) -> str:
30
- """Processes loop statements within the template."""
31
- loop_pattern = re.compile(
32
- r"\{for (\w+) in (\w+)\}(.*?)\{endfor\}", re.DOTALL
33
- )
34
-
35
- def render_loop(match):
36
- iterator_var, collection_name, loop_body = match.groups()
37
- collection = context.get(collection_name, [])
38
- if not isinstance(collection, (list, range)):
39
- raise ValueError(
40
- f"Expected list or range for '{collection_name}', got {type(collection).__name__}."
41
- )
42
-
43
- loop_result = ""
44
- for item in collection:
45
- loop_context = context.copy()
46
- loop_context[iterator_var] = item
47
- loop_result += self.fill(loop_body, loop_context)
48
-
49
- return loop_result
50
-
51
- return loop_pattern.sub(render_loop, template)
52
-
53
- def fill(
54
- self, template_str: str = "", context: dict[str, Any] = {}
55
- ) -> str:
56
- """Fills the template with values from context after processing conditionals and loops."""
57
- if not template_str: # Use the instance's template if not provided
58
- template_str = self.template_str
59
-
60
- # First, process conditionals with 'else'
61
- template_with_conditionals = self._render_conditionals(template_str)
62
- # Then, process loops
63
- template_with_loops = self._render_loops(
64
- template_with_conditionals, context
65
- )
66
- # Finally, substitute the placeholders with context values
67
- try:
68
- return template_with_loops.format(**context)
69
- except KeyError as e:
70
- print(f"Missing key in context: {e}")
71
- return template_with_loops
@@ -1,36 +0,0 @@
1
- class Node:
2
- """Base class for all nodes in the abstract syntax tree (AST)."""
3
-
4
- pass
5
-
6
-
7
- class IfNode(Node):
8
- """Represents an 'IF' statement in the AST."""
9
-
10
- def __init__(self, condition, true_block, false_block=None):
11
- self.condition = condition
12
- self.true_block = true_block
13
- self.false_block = false_block
14
-
15
-
16
- class ForNode(Node):
17
- """Represents a 'FOR' loop in the AST."""
18
-
19
- def __init__(self, iterator, collection, block):
20
- self.iterator = iterator
21
- self.collection = collection
22
- self.block = block
23
-
24
-
25
- class TryNode(Node):
26
- """Represents a 'TRY-EXCEPT' block in the AST."""
27
-
28
- def __init__(self, try_block, except_block):
29
- self.try_block = try_block
30
- self.except_block = except_block
31
-
32
-
33
- class ActionNode(Node):
34
-
35
- def __init__(self, action) -> None:
36
- self.action = action
@@ -1,59 +0,0 @@
1
- import re
2
-
3
-
4
- class BaseToken:
5
- def __init__(self, type_, value):
6
- self.type = type_
7
- self.value = value
8
-
9
- def __repr__(self):
10
- return f"BaseDirectiveToken({self.type}, {self.value})"
11
-
12
-
13
- class BaseTokenizer:
14
- TOKEN_TYPES = {
15
- "KEYWORD": r"\b(BEGIN|END|IF|ELSE|FOR|IN|TRY|EXCEPT|ENDIF|ENDFOR|ENDTRY|DO)\b",
16
- "OPERATOR": r"(==|!=|>=|<=|>|<|&&|\|\||!)",
17
- "FUNCTION_CALL": r"\b[a-zA-Z_][a-zA-Z0-9_]*\b\((.*?)\)",
18
- "LITERAL": r'(\d+|\'.*?\'|".*?")',
19
- "IDENTIFIER": r"\b[a-zA-Z_][a-zA-Z0-9_]*\b",
20
- "PUNCTUATION": r"(;|,|\(|\))",
21
- "WHITESPACE": r"\s+",
22
- }
23
-
24
- def __init__(self, script):
25
- self.script = script
26
- self.tokens = []
27
- self.tokenize()
28
-
29
- @property
30
- def is_empty(self):
31
- return self.tokens == []
32
-
33
- def tokenize(self):
34
- position = 0
35
- while position < len(self.script):
36
- match = None
37
- for type_, pattern in self.TOKEN_TYPES.items():
38
- regex = re.compile(pattern)
39
- match = regex.match(self.script, position)
40
- if match:
41
- if type_ != "WHITESPACE": # Ignore whitespace
42
- token = BaseToken(type_, match.group())
43
- self.tokens.append(token)
44
- position = match.end() # Move past the matched token
45
- break
46
- if not match: # No match found, unrecognized token
47
- raise SyntaxError(
48
- f"Unexpected character: {self.script[position]}"
49
- )
50
- # break
51
-
52
- def get_tokens(self):
53
- if self.is_empty:
54
- try:
55
- self.tokenize()
56
- except SyntaxError as e:
57
- print(e)
58
- return []
59
- return self.tokens
@@ -1 +0,0 @@
1
- TODO