lionagi 0.4.0__py3-none-any.whl → 0.5.1__py3-none-any.whl
Sign up to get free protection for your applications and to get access to all the features.
- lionagi/__init__.py +14 -46
- lionagi/core/__init__.py +3 -1
- lionagi/core/_class_registry.py +69 -0
- lionagi/core/action/__init__.py +3 -13
- lionagi/core/action/action_manager.py +287 -0
- lionagi/core/action/base.py +109 -0
- lionagi/core/action/function_calling.py +127 -92
- lionagi/core/action/tool.py +172 -70
- lionagi/core/action/types.py +16 -0
- lionagi/core/communication/__init__.py +3 -0
- lionagi/core/communication/action_request.py +163 -0
- lionagi/core/communication/action_response.py +149 -0
- lionagi/core/communication/assistant_response.py +161 -0
- lionagi/core/communication/base_mail.py +49 -0
- lionagi/core/communication/instruction.py +376 -0
- lionagi/core/communication/message.py +286 -0
- lionagi/core/communication/message_manager.py +530 -0
- lionagi/core/communication/system.py +116 -0
- lionagi/core/communication/templates/README.md +28 -0
- lionagi/core/communication/templates/action_request.jinja2 +5 -0
- lionagi/core/communication/templates/action_response.jinja2 +9 -0
- lionagi/core/communication/templates/assistant_response.jinja2 +2 -0
- lionagi/core/communication/templates/instruction_message.jinja2 +61 -0
- lionagi/core/communication/templates/system_message.jinja2 +11 -0
- lionagi/core/communication/templates/tool_schemas.jinja2 +7 -0
- lionagi/core/communication/types.py +27 -0
- lionagi/core/communication/utils.py +254 -0
- lionagi/core/forms/__init__.py +3 -0
- lionagi/core/forms/base.py +232 -0
- lionagi/core/forms/form.py +791 -0
- lionagi/core/forms/report.py +321 -0
- lionagi/core/forms/types.py +13 -0
- lionagi/core/forms/utils.py +26 -0
- lionagi/core/generic/__init__.py +3 -6
- lionagi/core/generic/component.py +422 -0
- lionagi/core/generic/edge.py +143 -101
- lionagi/core/generic/element.py +195 -0
- lionagi/core/generic/graph.py +297 -180
- lionagi/core/generic/log.py +151 -0
- lionagi/core/generic/log_manager.py +320 -0
- lionagi/core/generic/node.py +7 -229
- lionagi/core/generic/pile.py +1017 -0
- lionagi/core/generic/progression.py +388 -0
- lionagi/core/generic/types.py +23 -0
- lionagi/core/generic/utils.py +50 -0
- lionagi/core/models/__init__.py +5 -0
- lionagi/core/models/base.py +85 -0
- lionagi/core/models/field_model.py +122 -0
- lionagi/core/models/new_model_params.py +195 -0
- lionagi/core/models/note.py +351 -0
- lionagi/core/models/operable_model.py +392 -0
- lionagi/core/models/schema_model.py +50 -0
- lionagi/core/models/types.py +10 -0
- lionagi/core/session/__init__.py +3 -0
- lionagi/core/session/branch.py +115 -415
- lionagi/core/session/branch_mixins.py +545 -0
- lionagi/core/session/session.py +122 -257
- lionagi/core/session/types.py +8 -0
- lionagi/core/typing/__init__.py +9 -0
- lionagi/core/typing/concepts.py +132 -0
- lionagi/core/typing/config.py +15 -0
- lionagi/core/typing/id.py +221 -0
- lionagi/core/typing/pydantic_.py +33 -0
- lionagi/core/typing/typing_.py +54 -0
- lionagi/integrations/__init__.py +0 -1
- lionagi/integrations/anthropic_/AnthropicModel.py +268 -0
- lionagi/integrations/anthropic_/AnthropicService.py +117 -0
- lionagi/integrations/anthropic_/__init__.py +3 -0
- lionagi/integrations/anthropic_/anthropic_max_output_token_data.yaml +7 -0
- lionagi/integrations/anthropic_/anthropic_price_data.yaml +14 -0
- lionagi/integrations/anthropic_/api_endpoints/__init__.py +3 -0
- lionagi/integrations/anthropic_/api_endpoints/api_request.py +277 -0
- lionagi/integrations/anthropic_/api_endpoints/data_models.py +40 -0
- lionagi/integrations/anthropic_/api_endpoints/match_response.py +119 -0
- lionagi/integrations/anthropic_/api_endpoints/messages/__init__.py +3 -0
- lionagi/integrations/anthropic_/api_endpoints/messages/request/__init__.py +3 -0
- lionagi/integrations/anthropic_/api_endpoints/messages/request/message_models.py +14 -0
- lionagi/integrations/anthropic_/api_endpoints/messages/request/request_body.py +74 -0
- lionagi/integrations/anthropic_/api_endpoints/messages/response/content_models.py +32 -0
- lionagi/integrations/anthropic_/api_endpoints/messages/response/response_body.py +101 -0
- lionagi/integrations/anthropic_/api_endpoints/messages/response/usage_models.py +25 -0
- lionagi/integrations/anthropic_/version.py +5 -0
- lionagi/integrations/groq_/GroqModel.py +318 -0
- lionagi/integrations/groq_/GroqService.py +151 -0
- lionagi/integrations/groq_/__init__.py +3 -0
- lionagi/integrations/groq_/api_endpoints/data_models.py +187 -0
- lionagi/integrations/groq_/api_endpoints/groq_request.py +288 -0
- lionagi/integrations/groq_/api_endpoints/match_response.py +106 -0
- lionagi/integrations/groq_/api_endpoints/response_utils.py +105 -0
- lionagi/integrations/groq_/groq_max_output_token_data.yaml +21 -0
- lionagi/integrations/groq_/groq_price_data.yaml +58 -0
- lionagi/integrations/groq_/groq_rate_limits.yaml +105 -0
- lionagi/integrations/groq_/version.py +5 -0
- lionagi/integrations/litellm_/__init__.py +3 -0
- lionagi/integrations/litellm_/imodel.py +73 -0
- lionagi/integrations/ollama_/OllamaModel.py +244 -0
- lionagi/integrations/ollama_/OllamaService.py +142 -0
- lionagi/integrations/ollama_/__init__.py +3 -0
- lionagi/integrations/ollama_/api_endpoints/__init__.py +3 -0
- lionagi/integrations/ollama_/api_endpoints/api_request.py +179 -0
- lionagi/integrations/ollama_/api_endpoints/chat_completion/__init__.py +3 -0
- lionagi/integrations/ollama_/api_endpoints/chat_completion/message_models.py +31 -0
- lionagi/integrations/ollama_/api_endpoints/chat_completion/request_body.py +46 -0
- lionagi/integrations/ollama_/api_endpoints/chat_completion/response_body.py +67 -0
- lionagi/integrations/ollama_/api_endpoints/chat_completion/tool_models.py +49 -0
- lionagi/integrations/ollama_/api_endpoints/completion/request_body.py +72 -0
- lionagi/integrations/ollama_/api_endpoints/completion/response_body.py +59 -0
- lionagi/integrations/ollama_/api_endpoints/data_models.py +15 -0
- lionagi/integrations/ollama_/api_endpoints/embedding/request_body.py +33 -0
- lionagi/integrations/ollama_/api_endpoints/embedding/response_body.py +29 -0
- lionagi/integrations/ollama_/api_endpoints/match_data_model.py +62 -0
- lionagi/integrations/ollama_/api_endpoints/match_response.py +190 -0
- lionagi/integrations/ollama_/api_endpoints/model/__init__.py +3 -0
- lionagi/integrations/ollama_/api_endpoints/model/copy_model.py +13 -0
- lionagi/integrations/ollama_/api_endpoints/model/create_model.py +28 -0
- lionagi/integrations/ollama_/api_endpoints/model/delete_model.py +11 -0
- lionagi/integrations/ollama_/api_endpoints/model/list_model.py +60 -0
- lionagi/integrations/ollama_/api_endpoints/model/pull_model.py +34 -0
- lionagi/integrations/ollama_/api_endpoints/model/push_model.py +35 -0
- lionagi/integrations/ollama_/api_endpoints/model/show_model.py +36 -0
- lionagi/integrations/ollama_/api_endpoints/option_models.py +68 -0
- lionagi/integrations/openai_/OpenAIModel.py +418 -0
- lionagi/integrations/openai_/OpenAIService.py +426 -0
- lionagi/integrations/openai_/api_endpoints/__init__.py +3 -0
- lionagi/integrations/openai_/api_endpoints/api_request.py +277 -0
- lionagi/integrations/openai_/api_endpoints/audio/__init__.py +9 -0
- lionagi/integrations/openai_/api_endpoints/audio/speech_models.py +34 -0
- lionagi/integrations/openai_/api_endpoints/audio/transcription_models.py +136 -0
- lionagi/integrations/openai_/api_endpoints/audio/translation_models.py +41 -0
- lionagi/integrations/openai_/api_endpoints/audio/types.py +41 -0
- lionagi/integrations/openai_/api_endpoints/batch/__init__.py +17 -0
- lionagi/integrations/openai_/api_endpoints/batch/batch_models.py +146 -0
- lionagi/integrations/openai_/api_endpoints/batch/cancel_batch.py +7 -0
- lionagi/integrations/openai_/api_endpoints/batch/create_batch.py +26 -0
- lionagi/integrations/openai_/api_endpoints/batch/list_batch.py +37 -0
- lionagi/integrations/openai_/api_endpoints/batch/request_object_models.py +65 -0
- lionagi/integrations/openai_/api_endpoints/batch/retrieve_batch.py +7 -0
- lionagi/integrations/openai_/api_endpoints/batch/types.py +4 -0
- lionagi/integrations/openai_/api_endpoints/chat_completions/__init__.py +1 -0
- lionagi/integrations/openai_/api_endpoints/chat_completions/request/__init__.py +39 -0
- lionagi/integrations/openai_/api_endpoints/chat_completions/request/message_models.py +121 -0
- lionagi/integrations/openai_/api_endpoints/chat_completions/request/request_body.py +221 -0
- lionagi/integrations/openai_/api_endpoints/chat_completions/request/response_format.py +71 -0
- lionagi/integrations/openai_/api_endpoints/chat_completions/request/stream_options.py +14 -0
- lionagi/integrations/openai_/api_endpoints/chat_completions/request/tool_choice_models.py +17 -0
- lionagi/integrations/openai_/api_endpoints/chat_completions/request/tool_models.py +54 -0
- lionagi/integrations/openai_/api_endpoints/chat_completions/request/types.py +18 -0
- lionagi/integrations/openai_/api_endpoints/chat_completions/response/choice_models.py +62 -0
- lionagi/integrations/openai_/api_endpoints/chat_completions/response/function_models.py +16 -0
- lionagi/integrations/openai_/api_endpoints/chat_completions/response/log_prob_models.py +47 -0
- lionagi/integrations/openai_/api_endpoints/chat_completions/response/message_models.py +25 -0
- lionagi/integrations/openai_/api_endpoints/chat_completions/response/response_body.py +99 -0
- lionagi/integrations/openai_/api_endpoints/chat_completions/response/types.py +8 -0
- lionagi/integrations/openai_/api_endpoints/chat_completions/response/usage_models.py +24 -0
- lionagi/integrations/openai_/api_endpoints/chat_completions/util.py +46 -0
- lionagi/integrations/openai_/api_endpoints/data_models.py +23 -0
- lionagi/integrations/openai_/api_endpoints/embeddings/__init__.py +3 -0
- lionagi/integrations/openai_/api_endpoints/embeddings/request_body.py +79 -0
- lionagi/integrations/openai_/api_endpoints/embeddings/response_body.py +67 -0
- lionagi/integrations/openai_/api_endpoints/files/__init__.py +11 -0
- lionagi/integrations/openai_/api_endpoints/files/delete_file.py +20 -0
- lionagi/integrations/openai_/api_endpoints/files/file_models.py +56 -0
- lionagi/integrations/openai_/api_endpoints/files/list_files.py +27 -0
- lionagi/integrations/openai_/api_endpoints/files/retrieve_file.py +9 -0
- lionagi/integrations/openai_/api_endpoints/files/upload_file.py +38 -0
- lionagi/integrations/openai_/api_endpoints/fine_tuning/__init__.py +37 -0
- lionagi/integrations/openai_/api_endpoints/fine_tuning/cancel_jobs.py +9 -0
- lionagi/integrations/openai_/api_endpoints/fine_tuning/create_jobs.py +133 -0
- lionagi/integrations/openai_/api_endpoints/fine_tuning/fine_tuning_job_checkpoint_models.py +58 -0
- lionagi/integrations/openai_/api_endpoints/fine_tuning/fine_tuning_job_event_models.py +31 -0
- lionagi/integrations/openai_/api_endpoints/fine_tuning/fine_tuning_job_models.py +140 -0
- lionagi/integrations/openai_/api_endpoints/fine_tuning/list_fine_tuning_checkpoints.py +51 -0
- lionagi/integrations/openai_/api_endpoints/fine_tuning/list_fine_tuning_events.py +42 -0
- lionagi/integrations/openai_/api_endpoints/fine_tuning/list_fine_tuning_jobs.py +31 -0
- lionagi/integrations/openai_/api_endpoints/fine_tuning/retrieve_jobs.py +9 -0
- lionagi/integrations/openai_/api_endpoints/fine_tuning/training_format.py +30 -0
- lionagi/integrations/openai_/api_endpoints/images/__init__.py +9 -0
- lionagi/integrations/openai_/api_endpoints/images/image_edit_models.py +69 -0
- lionagi/integrations/openai_/api_endpoints/images/image_models.py +56 -0
- lionagi/integrations/openai_/api_endpoints/images/image_variation_models.py +56 -0
- lionagi/integrations/openai_/api_endpoints/images/response_body.py +30 -0
- lionagi/integrations/openai_/api_endpoints/match_data_model.py +197 -0
- lionagi/integrations/openai_/api_endpoints/match_response.py +336 -0
- lionagi/integrations/openai_/api_endpoints/models/__init__.py +7 -0
- lionagi/integrations/openai_/api_endpoints/models/delete_fine_tuned_model.py +17 -0
- lionagi/integrations/openai_/api_endpoints/models/models_models.py +31 -0
- lionagi/integrations/openai_/api_endpoints/models/retrieve_model.py +9 -0
- lionagi/integrations/openai_/api_endpoints/moderations/__init__.py +3 -0
- lionagi/integrations/openai_/api_endpoints/moderations/request_body.py +20 -0
- lionagi/integrations/openai_/api_endpoints/moderations/response_body.py +139 -0
- lionagi/integrations/openai_/api_endpoints/uploads/__init__.py +19 -0
- lionagi/integrations/openai_/api_endpoints/uploads/add_upload_part.py +11 -0
- lionagi/integrations/openai_/api_endpoints/uploads/cancel_upload.py +7 -0
- lionagi/integrations/openai_/api_endpoints/uploads/complete_upload.py +18 -0
- lionagi/integrations/openai_/api_endpoints/uploads/create_upload.py +17 -0
- lionagi/integrations/openai_/api_endpoints/uploads/uploads_models.py +52 -0
- lionagi/integrations/openai_/image_token_calculator/image_token_calculator.py +92 -0
- lionagi/integrations/openai_/image_token_calculator/openai_image_token_data.yaml +15 -0
- lionagi/integrations/openai_/openai_max_output_token_data.yaml +12 -0
- lionagi/integrations/openai_/openai_price_data.yaml +26 -0
- lionagi/integrations/openai_/version.py +1 -0
- lionagi/integrations/pandas_/__init__.py +24 -0
- lionagi/integrations/pandas_/extend_df.py +61 -0
- lionagi/integrations/pandas_/read.py +103 -0
- lionagi/integrations/pandas_/remove_rows.py +61 -0
- lionagi/integrations/pandas_/replace_keywords.py +65 -0
- lionagi/integrations/pandas_/save.py +131 -0
- lionagi/integrations/pandas_/search_keywords.py +69 -0
- lionagi/integrations/pandas_/to_df.py +196 -0
- lionagi/integrations/pandas_/update_cells.py +54 -0
- lionagi/integrations/perplexity_/PerplexityModel.py +269 -0
- lionagi/integrations/perplexity_/PerplexityService.py +113 -0
- lionagi/integrations/perplexity_/__init__.py +3 -0
- lionagi/integrations/perplexity_/api_endpoints/api_request.py +171 -0
- lionagi/integrations/perplexity_/api_endpoints/chat_completions/request/request_body.py +121 -0
- lionagi/integrations/perplexity_/api_endpoints/chat_completions/response/response_body.py +146 -0
- lionagi/integrations/perplexity_/api_endpoints/data_models.py +63 -0
- lionagi/integrations/perplexity_/api_endpoints/match_response.py +26 -0
- lionagi/integrations/perplexity_/perplexity_max_output_token_data.yaml +3 -0
- lionagi/integrations/perplexity_/perplexity_price_data.yaml +10 -0
- lionagi/integrations/perplexity_/version.py +1 -0
- lionagi/integrations/pydantic_/__init__.py +8 -0
- lionagi/integrations/pydantic_/break_down_annotation.py +81 -0
- lionagi/integrations/pydantic_/new_model.py +208 -0
- lionagi/integrations/services.py +17 -0
- lionagi/libs/__init__.py +0 -55
- lionagi/libs/compress/models.py +62 -0
- lionagi/libs/compress/utils.py +81 -0
- lionagi/libs/constants.py +98 -0
- lionagi/libs/file/chunk.py +265 -0
- lionagi/libs/file/file_ops.py +114 -0
- lionagi/libs/file/params.py +212 -0
- lionagi/libs/file/path.py +301 -0
- lionagi/libs/file/process.py +139 -0
- lionagi/libs/file/save.py +90 -0
- lionagi/libs/file/types.py +22 -0
- lionagi/libs/func/async_calls/__init__.py +21 -0
- lionagi/libs/func/async_calls/alcall.py +157 -0
- lionagi/libs/func/async_calls/bcall.py +82 -0
- lionagi/libs/func/async_calls/mcall.py +134 -0
- lionagi/libs/func/async_calls/pcall.py +149 -0
- lionagi/libs/func/async_calls/rcall.py +185 -0
- lionagi/libs/func/async_calls/tcall.py +114 -0
- lionagi/libs/func/async_calls/ucall.py +85 -0
- lionagi/libs/func/decorators.py +277 -0
- lionagi/libs/func/lcall.py +57 -0
- lionagi/libs/func/params.py +64 -0
- lionagi/libs/func/throttle.py +119 -0
- lionagi/libs/func/types.py +39 -0
- lionagi/libs/func/utils.py +96 -0
- lionagi/libs/package/imports.py +162 -0
- lionagi/libs/package/management.py +58 -0
- lionagi/libs/package/params.py +26 -0
- lionagi/libs/package/system.py +18 -0
- lionagi/libs/package/types.py +26 -0
- lionagi/libs/parse/__init__.py +1 -0
- lionagi/libs/parse/flatten/__init__.py +9 -0
- lionagi/libs/parse/flatten/flatten.py +168 -0
- lionagi/libs/parse/flatten/params.py +52 -0
- lionagi/libs/parse/flatten/unflatten.py +79 -0
- lionagi/libs/parse/json/__init__.py +27 -0
- lionagi/libs/parse/json/as_readable.py +104 -0
- lionagi/libs/parse/json/extract.py +102 -0
- lionagi/libs/parse/json/parse.py +179 -0
- lionagi/libs/parse/json/schema.py +227 -0
- lionagi/libs/parse/json/to_json.py +71 -0
- lionagi/libs/parse/nested/__init__.py +33 -0
- lionagi/libs/parse/nested/nfilter.py +55 -0
- lionagi/libs/parse/nested/nget.py +40 -0
- lionagi/libs/parse/nested/ninsert.py +103 -0
- lionagi/libs/parse/nested/nmerge.py +155 -0
- lionagi/libs/parse/nested/npop.py +66 -0
- lionagi/libs/parse/nested/nset.py +89 -0
- lionagi/libs/parse/nested/to_flat_list.py +64 -0
- lionagi/libs/parse/nested/utils.py +185 -0
- lionagi/libs/parse/string_parse/__init__.py +11 -0
- lionagi/libs/parse/string_parse/code_block.py +73 -0
- lionagi/libs/parse/string_parse/docstring.py +179 -0
- lionagi/libs/parse/string_parse/function_.py +92 -0
- lionagi/libs/parse/type_convert/__init__.py +19 -0
- lionagi/libs/parse/type_convert/params.py +145 -0
- lionagi/libs/parse/type_convert/to_dict.py +333 -0
- lionagi/libs/parse/type_convert/to_list.py +186 -0
- lionagi/libs/parse/type_convert/to_num.py +358 -0
- lionagi/libs/parse/type_convert/to_str.py +195 -0
- lionagi/libs/parse/types.py +9 -0
- lionagi/libs/parse/validate/__init__.py +14 -0
- lionagi/libs/parse/validate/boolean.py +96 -0
- lionagi/libs/parse/validate/keys.py +150 -0
- lionagi/libs/parse/validate/mapping.py +109 -0
- lionagi/libs/parse/validate/params.py +62 -0
- lionagi/libs/parse/xml/__init__.py +10 -0
- lionagi/libs/parse/xml/convert.py +56 -0
- lionagi/libs/parse/xml/parser.py +93 -0
- lionagi/libs/string_similarity/__init__.py +32 -0
- lionagi/libs/string_similarity/algorithms.py +219 -0
- lionagi/libs/string_similarity/matcher.py +102 -0
- lionagi/libs/string_similarity/utils.py +15 -0
- lionagi/libs/utils.py +255 -0
- lionagi/operations/__init__.py +3 -6
- lionagi/operations/brainstorm/__init__.py +3 -0
- lionagi/operations/brainstorm/brainstorm.py +204 -0
- lionagi/operations/brainstorm/prompt.py +1 -0
- lionagi/operations/plan/__init__.py +3 -0
- lionagi/operations/plan/plan.py +172 -0
- lionagi/operations/plan/prompt.py +21 -0
- lionagi/operations/select/__init__.py +3 -0
- lionagi/operations/select/prompt.py +1 -0
- lionagi/operations/select/select.py +100 -0
- lionagi/operations/select/utils.py +107 -0
- lionagi/operations/utils.py +35 -0
- lionagi/protocols/adapters/adapter.py +79 -0
- lionagi/protocols/adapters/json_adapter.py +43 -0
- lionagi/protocols/adapters/pandas_adapter.py +96 -0
- lionagi/protocols/configs/__init__.py +15 -0
- lionagi/protocols/configs/branch_config.py +86 -0
- lionagi/protocols/configs/id_config.py +15 -0
- lionagi/protocols/configs/imodel_config.py +73 -0
- lionagi/protocols/configs/log_config.py +93 -0
- lionagi/protocols/configs/retry_config.py +29 -0
- lionagi/protocols/operatives/__init__.py +15 -0
- lionagi/protocols/operatives/action.py +181 -0
- lionagi/protocols/operatives/instruct.py +196 -0
- lionagi/protocols/operatives/operative.py +182 -0
- lionagi/protocols/operatives/prompts.py +232 -0
- lionagi/protocols/operatives/reason.py +56 -0
- lionagi/protocols/operatives/step.py +217 -0
- lionagi/protocols/registries/_component_registry.py +19 -0
- lionagi/protocols/registries/_pile_registry.py +26 -0
- lionagi/service/__init__.py +13 -0
- lionagi/service/complete_request_info.py +11 -0
- lionagi/service/imodel.py +125 -0
- lionagi/service/rate_limiter.py +108 -0
- lionagi/service/service.py +41 -0
- lionagi/service/service_match_util.py +131 -0
- lionagi/service/service_util.py +72 -0
- lionagi/service/token_calculator.py +51 -0
- lionagi/settings.py +136 -0
- lionagi/strategies/base.py +53 -0
- lionagi/strategies/concurrent.py +71 -0
- lionagi/strategies/concurrent_chunk.py +43 -0
- lionagi/strategies/concurrent_sequential_chunk.py +104 -0
- lionagi/strategies/params.py +128 -0
- lionagi/strategies/sequential.py +23 -0
- lionagi/strategies/sequential_chunk.py +89 -0
- lionagi/strategies/sequential_concurrent_chunk.py +100 -0
- lionagi/strategies/types.py +21 -0
- lionagi/strategies/utils.py +49 -0
- lionagi/version.py +1 -1
- lionagi-0.5.1.dist-info/METADATA +545 -0
- lionagi-0.5.1.dist-info/RECORD +373 -0
- {lionagi-0.4.0.dist-info → lionagi-0.5.1.dist-info}/WHEEL +1 -1
- lionagi/core/_setting/_setting.py +0 -59
- lionagi/core/action/README.md +0 -20
- lionagi/core/action/manual.py +0 -1
- lionagi/core/action/node.py +0 -94
- lionagi/core/action/tool_manager.py +0 -342
- lionagi/core/agent/README.md +0 -1
- lionagi/core/agent/base_agent.py +0 -82
- lionagi/core/agent/eval/README.md +0 -1
- lionagi/core/agent/eval/evaluator.py +0 -1
- lionagi/core/agent/eval/vote.py +0 -40
- lionagi/core/agent/learn/learner.py +0 -59
- lionagi/core/agent/plan/unit_template.py +0 -1
- lionagi/core/collections/README.md +0 -23
- lionagi/core/collections/__init__.py +0 -16
- lionagi/core/collections/_logger.py +0 -312
- lionagi/core/collections/abc/README.md +0 -63
- lionagi/core/collections/abc/__init__.py +0 -53
- lionagi/core/collections/abc/component.py +0 -620
- lionagi/core/collections/abc/concepts.py +0 -277
- lionagi/core/collections/abc/exceptions.py +0 -136
- lionagi/core/collections/abc/util.py +0 -45
- lionagi/core/collections/exchange.py +0 -146
- lionagi/core/collections/flow.py +0 -416
- lionagi/core/collections/model.py +0 -465
- lionagi/core/collections/pile.py +0 -1232
- lionagi/core/collections/progression.py +0 -221
- lionagi/core/collections/util.py +0 -73
- lionagi/core/director/README.md +0 -1
- lionagi/core/director/direct.py +0 -298
- lionagi/core/director/director.py +0 -2
- lionagi/core/director/operations/select.py +0 -3
- lionagi/core/director/operations/utils.py +0 -6
- lionagi/core/engine/branch_engine.py +0 -361
- lionagi/core/engine/instruction_map_engine.py +0 -213
- lionagi/core/engine/sandbox_.py +0 -16
- lionagi/core/engine/script_engine.py +0 -89
- lionagi/core/executor/base_executor.py +0 -97
- lionagi/core/executor/graph_executor.py +0 -335
- lionagi/core/executor/neo4j_executor.py +0 -394
- lionagi/core/generic/README.md +0 -0
- lionagi/core/generic/edge_condition.py +0 -17
- lionagi/core/generic/hyperedge.py +0 -1
- lionagi/core/generic/tree.py +0 -49
- lionagi/core/generic/tree_node.py +0 -85
- lionagi/core/mail/__init__.py +0 -11
- lionagi/core/mail/mail.py +0 -26
- lionagi/core/mail/mail_manager.py +0 -185
- lionagi/core/mail/package.py +0 -49
- lionagi/core/mail/start_mail.py +0 -36
- lionagi/core/message/__init__.py +0 -18
- lionagi/core/message/action_request.py +0 -114
- lionagi/core/message/action_response.py +0 -121
- lionagi/core/message/assistant_response.py +0 -80
- lionagi/core/message/instruction.py +0 -194
- lionagi/core/message/message.py +0 -86
- lionagi/core/message/system.py +0 -71
- lionagi/core/message/util.py +0 -274
- lionagi/core/report/__init__.py +0 -4
- lionagi/core/report/base.py +0 -201
- lionagi/core/report/form.py +0 -212
- lionagi/core/report/report.py +0 -150
- lionagi/core/report/util.py +0 -15
- lionagi/core/rule/_default.py +0 -17
- lionagi/core/rule/action.py +0 -87
- lionagi/core/rule/base.py +0 -234
- lionagi/core/rule/boolean.py +0 -56
- lionagi/core/rule/choice.py +0 -48
- lionagi/core/rule/mapping.py +0 -82
- lionagi/core/rule/number.py +0 -73
- lionagi/core/rule/rulebook.py +0 -45
- lionagi/core/rule/string.py +0 -43
- lionagi/core/rule/util.py +0 -0
- lionagi/core/session/directive_mixin.py +0 -307
- lionagi/core/structure/__init__.py +0 -1
- lionagi/core/structure/chain.py +0 -1
- lionagi/core/structure/forest.py +0 -1
- lionagi/core/structure/graph.py +0 -1
- lionagi/core/structure/tree.py +0 -1
- lionagi/core/unit/__init__.py +0 -4
- lionagi/core/unit/parallel_unit.py +0 -234
- lionagi/core/unit/template/action.py +0 -65
- lionagi/core/unit/template/base.py +0 -35
- lionagi/core/unit/template/plan.py +0 -69
- lionagi/core/unit/template/predict.py +0 -95
- lionagi/core/unit/template/score.py +0 -108
- lionagi/core/unit/template/select.py +0 -91
- lionagi/core/unit/unit.py +0 -452
- lionagi/core/unit/unit_form.py +0 -290
- lionagi/core/unit/unit_mixin.py +0 -1166
- lionagi/core/unit/util.py +0 -103
- lionagi/core/validator/validator.py +0 -376
- lionagi/core/work/work.py +0 -59
- lionagi/core/work/work_edge.py +0 -102
- lionagi/core/work/work_function.py +0 -114
- lionagi/core/work/work_function_node.py +0 -50
- lionagi/core/work/work_queue.py +0 -90
- lionagi/core/work/work_task.py +0 -151
- lionagi/core/work/worker.py +0 -410
- lionagi/core/work/worker_engine.py +0 -208
- lionagi/core/work/worklog.py +0 -108
- lionagi/experimental/compressor/base.py +0 -47
- lionagi/experimental/compressor/llm_compressor.py +0 -265
- lionagi/experimental/compressor/llm_summarizer.py +0 -61
- lionagi/experimental/compressor/util.py +0 -70
- lionagi/experimental/directive/README.md +0 -1
- lionagi/experimental/directive/__init__.py +0 -19
- lionagi/experimental/directive/parser/base_parser.py +0 -294
- lionagi/experimental/directive/parser/base_syntax.txt +0 -200
- lionagi/experimental/directive/template/base_template.py +0 -71
- lionagi/experimental/directive/template/schema.py +0 -36
- lionagi/experimental/directive/tokenizer.py +0 -59
- lionagi/experimental/evaluator/README.md +0 -1
- lionagi/experimental/evaluator/ast_evaluator.py +0 -119
- lionagi/experimental/evaluator/base_evaluator.py +0 -213
- lionagi/experimental/knowledge/__init__.py +0 -0
- lionagi/experimental/knowledge/base.py +0 -10
- lionagi/experimental/knowledge/graph.py +0 -0
- lionagi/experimental/memory/__init__.py +0 -0
- lionagi/experimental/strategies/__init__.py +0 -0
- lionagi/experimental/strategies/base.py +0 -1
- lionagi/integrations/bridge/__init__.py +0 -4
- lionagi/integrations/bridge/autogen_/__init__.py +0 -0
- lionagi/integrations/bridge/autogen_/autogen_.py +0 -127
- lionagi/integrations/bridge/langchain_/__init__.py +0 -0
- lionagi/integrations/bridge/langchain_/documents.py +0 -138
- lionagi/integrations/bridge/langchain_/langchain_bridge.py +0 -68
- lionagi/integrations/bridge/llamaindex_/__init__.py +0 -0
- lionagi/integrations/bridge/llamaindex_/index.py +0 -36
- lionagi/integrations/bridge/llamaindex_/llama_index_bridge.py +0 -108
- lionagi/integrations/bridge/llamaindex_/llama_pack.py +0 -256
- lionagi/integrations/bridge/llamaindex_/node_parser.py +0 -92
- lionagi/integrations/bridge/llamaindex_/reader.py +0 -201
- lionagi/integrations/bridge/llamaindex_/textnode.py +0 -59
- lionagi/integrations/bridge/pydantic_/__init__.py +0 -0
- lionagi/integrations/bridge/pydantic_/pydantic_bridge.py +0 -7
- lionagi/integrations/bridge/transformers_/__init__.py +0 -0
- lionagi/integrations/bridge/transformers_/install_.py +0 -39
- lionagi/integrations/chunker/__init__.py +0 -0
- lionagi/integrations/chunker/chunk.py +0 -314
- lionagi/integrations/config/__init__.py +0 -4
- lionagi/integrations/config/mlx_configs.py +0 -1
- lionagi/integrations/config/oai_configs.py +0 -154
- lionagi/integrations/config/ollama_configs.py +0 -1
- lionagi/integrations/config/openrouter_configs.py +0 -74
- lionagi/integrations/langchain_/__init__.py +0 -0
- lionagi/integrations/llamaindex_/__init__.py +0 -0
- lionagi/integrations/loader/__init__.py +0 -0
- lionagi/integrations/loader/load.py +0 -257
- lionagi/integrations/loader/load_util.py +0 -214
- lionagi/integrations/provider/__init__.py +0 -11
- lionagi/integrations/provider/_mapping.py +0 -47
- lionagi/integrations/provider/litellm.py +0 -53
- lionagi/integrations/provider/mistralai.py +0 -1
- lionagi/integrations/provider/mlx_service.py +0 -55
- lionagi/integrations/provider/oai.py +0 -196
- lionagi/integrations/provider/ollama.py +0 -55
- lionagi/integrations/provider/openrouter.py +0 -170
- lionagi/integrations/provider/services.py +0 -138
- lionagi/integrations/provider/transformers.py +0 -108
- lionagi/integrations/storage/__init__.py +0 -3
- lionagi/integrations/storage/neo4j.py +0 -681
- lionagi/integrations/storage/storage_util.py +0 -302
- lionagi/integrations/storage/structure_excel.py +0 -291
- lionagi/integrations/storage/to_csv.py +0 -70
- lionagi/integrations/storage/to_excel.py +0 -91
- lionagi/libs/ln_api.py +0 -944
- lionagi/libs/ln_async.py +0 -208
- lionagi/libs/ln_context.py +0 -37
- lionagi/libs/ln_convert.py +0 -671
- lionagi/libs/ln_dataframe.py +0 -187
- lionagi/libs/ln_func_call.py +0 -1328
- lionagi/libs/ln_image.py +0 -114
- lionagi/libs/ln_knowledge_graph.py +0 -422
- lionagi/libs/ln_nested.py +0 -822
- lionagi/libs/ln_parse.py +0 -750
- lionagi/libs/ln_queue.py +0 -107
- lionagi/libs/ln_tokenize.py +0 -179
- lionagi/libs/ln_validate.py +0 -299
- lionagi/libs/special_tokens.py +0 -172
- lionagi/libs/sys_util.py +0 -710
- lionagi/lions/__init__.py +0 -0
- lionagi/lions/coder/__init__.py +0 -0
- lionagi/lions/coder/add_feature.py +0 -20
- lionagi/lions/coder/base_prompts.py +0 -22
- lionagi/lions/coder/code_form.py +0 -15
- lionagi/lions/coder/coder.py +0 -184
- lionagi/lions/coder/util.py +0 -101
- lionagi/lions/director/__init__.py +0 -0
- lionagi/lions/judge/__init__.py +0 -0
- lionagi/lions/judge/config.py +0 -8
- lionagi/lions/judge/data/__init__.py +0 -0
- lionagi/lions/judge/data/sample_codes.py +0 -526
- lionagi/lions/judge/data/sample_rurbic.py +0 -48
- lionagi/lions/judge/forms/__init__.py +0 -0
- lionagi/lions/judge/forms/code_analysis_form.py +0 -126
- lionagi/lions/judge/rubric.py +0 -34
- lionagi/lions/judge/services/__init__.py +0 -0
- lionagi/lions/judge/services/judge_code.py +0 -49
- lionagi/lions/researcher/__init__.py +0 -0
- lionagi/lions/researcher/data_source/__init__.py +0 -0
- lionagi/lions/researcher/data_source/finhub_.py +0 -192
- lionagi/lions/researcher/data_source/google_.py +0 -207
- lionagi/lions/researcher/data_source/wiki_.py +0 -98
- lionagi/lions/researcher/data_source/yfinance_.py +0 -21
- lionagi/operations/brainstorm.py +0 -87
- lionagi/operations/config.py +0 -6
- lionagi/operations/rank.py +0 -102
- lionagi/operations/score.py +0 -144
- lionagi/operations/select.py +0 -141
- lionagi-0.4.0.dist-info/METADATA +0 -241
- lionagi-0.4.0.dist-info/RECORD +0 -249
- /lionagi/{core/_setting → integrations/anthropic_/api_endpoints/messages/response}/__init__.py +0 -0
- /lionagi/{core/agent → integrations/groq_/api_endpoints}/__init__.py +0 -0
- /lionagi/{core/agent/eval → integrations/ollama_/api_endpoints/completion}/__init__.py +0 -0
- /lionagi/{core/agent/learn → integrations/ollama_/api_endpoints/embedding}/__init__.py +0 -0
- /lionagi/{core/agent/plan → integrations/openai_}/__init__.py +0 -0
- /lionagi/{core/director → integrations/openai_/api_endpoints/chat_completions/response}/__init__.py +0 -0
- /lionagi/{core/director/operations → integrations/openai_/image_token_calculator}/__init__.py +0 -0
- /lionagi/{core/engine → integrations/perplexity_/api_endpoints}/__init__.py +0 -0
- /lionagi/{core/executor → integrations/perplexity_/api_endpoints/chat_completions}/__init__.py +0 -0
- /lionagi/{core/generic/registry/component_registry → integrations/perplexity_/api_endpoints/chat_completions/request}/__init__.py +0 -0
- /lionagi/{core/rule → integrations/perplexity_/api_endpoints/chat_completions/response}/__init__.py +0 -0
- /lionagi/{core/unit/template → libs/compress}/__init__.py +0 -0
- /lionagi/{core/validator → libs/file}/__init__.py +0 -0
- /lionagi/{core/work → libs/func}/__init__.py +0 -0
- /lionagi/{experimental → libs/package}/__init__.py +0 -0
- /lionagi/{core/agent/plan/plan.py → libs/parse/params.py} +0 -0
- /lionagi/{experimental/compressor → protocols}/__init__.py +0 -0
- /lionagi/{experimental/directive/parser → protocols/adapters}/__init__.py +0 -0
- /lionagi/{experimental/directive/template → protocols/registries}/__init__.py +0 -0
- /lionagi/{experimental/evaluator → strategies}/__init__.py +0 -0
- {lionagi-0.4.0.dist-info → lionagi-0.5.1.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
|