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