pyfect-dev 0.1.0__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- pyfect/__init__.py +729 -0
- pyfect/_all.py +242 -0
- pyfect/_background.py +28 -0
- pyfect/_binary_codecs.py +124 -0
- pyfect/_cached.py +165 -0
- pyfect/_catch.py +128 -0
- pyfect/_catch_reason.py +574 -0
- pyfect/_concurrency.py +291 -0
- pyfect/_context_effects.py +163 -0
- pyfect/_context_managers.py +76 -0
- pyfect/_core.py +368 -0
- pyfect/_digits.py +71 -0
- pyfect/_effect_args.py +46 -0
- pyfect/_effect_base.py +71 -0
- pyfect/_effect_concurrency.py +133 -0
- pyfect/_effect_eager.py +103 -0
- pyfect/_effect_errors.py +344 -0
- pyfect/_effect_folding.py +110 -0
- pyfect/_effect_hash.py +140 -0
- pyfect/_effect_logging.py +57 -0
- pyfect/_effect_metrics.py +102 -0
- pyfect/_effect_resources.py +179 -0
- pyfect/_effect_services.py +25 -0
- pyfect/_effect_timing.py +199 -0
- pyfect/_effect_tracing.py +99 -0
- pyfect/_equal.py +195 -0
- pyfect/_extras.py +43 -0
- pyfect/_fiber.py +969 -0
- pyfect/_fiber_all.py +134 -0
- pyfect/_fiber_collections.py +106 -0
- pyfect/_fn.py +706 -0
- pyfect/_hamt.py +174 -0
- pyfect/_interrupt_reason.py +33 -0
- pyfect/_isaac.py +149 -0
- pyfect/_iterables.py +231 -0
- pyfect/_js.py +263 -0
- pyfect/_js_date.py +87 -0
- pyfect/_json_text.py +286 -0
- pyfect/_json_value.py +9 -0
- pyfect/_layer_core.py +403 -0
- pyfect/_layer_for_life.py +33 -0
- pyfect/_layer_tracing.py +70 -0
- pyfect/_locked_ref.py +147 -0
- pyfect/_loop_thread.py +152 -0
- pyfect/_metric_ref.py +29 -0
- pyfect/_one_or_many.py +14 -0
- pyfect/_os_wait.py +118 -0
- pyfect/_path_posix.py +129 -0
- pyfect/_primitives.py +434 -0
- pyfect/_public.py +36 -0
- pyfect/_reasons.py +64 -0
- pyfect/_released.py +67 -0
- pyfect/_root.py +87 -0
- pyfect/_round_robin.py +73 -0
- pyfect/_some.py +17 -0
- pyfect/_stack.py +52 -0
- pyfect/_stack_frame.py +114 -0
- pyfect/_string_hash.py +35 -0
- pyfect/_subscriber_queue.py +47 -0
- pyfect/_tags.py +14 -0
- pyfect/_tracer_ref.py +21 -0
- pyfect/_tx_ops.py +77 -0
- pyfect/_unsubscripted.py +23 -0
- pyfect/_url.py +54 -0
- pyfect/_waiters.py +127 -0
- pyfect/_wake_tracking.py +98 -0
- pyfect/_wire_codec.py +123 -0
- pyfect/ai/__init__.py +73 -0
- pyfect/ai/_approvals.py +249 -0
- pyfect/ai/_decision_answers.py +185 -0
- pyfect/ai/_generated_client.py +298 -0
- pyfect/ai/_generated_support.py +84 -0
- pyfect/ai/_mcp/__init__.py +1 -0
- pyfect/ai/_mcp/batch.py +68 -0
- pyfect/ai/_mcp/client_transport.py +412 -0
- pyfect/ai/_mcp/client_wire.py +284 -0
- pyfect/ai/_mcp/core.py +490 -0
- pyfect/ai/_mcp/dispatch.py +1048 -0
- pyfect/ai/_mcp/http.py +462 -0
- pyfect/ai/_mcp/http_admission.py +269 -0
- pyfect/ai/_mcp/http_status.py +27 -0
- pyfect/ai/_mcp/input_wire.py +226 -0
- pyfect/ai/_mcp/log_levels.py +55 -0
- pyfect/ai/_mcp/param_headers.py +63 -0
- pyfect/ai/_mcp/projection.py +248 -0
- pyfect/ai/_mcp/register.py +683 -0
- pyfect/ai/_mcp/reverse_policy.py +155 -0
- pyfect/ai/_mcp/routing_header.py +24 -0
- pyfect/ai/_mcp/sessions.py +114 -0
- pyfect/ai/_mcp/stdio.py +110 -0
- pyfect/ai/_mcp/subscriptions.py +134 -0
- pyfect/ai/_mcp/tool_schema.py +35 -0
- pyfect/ai/_noul_answers.py +64 -0
- pyfect/ai/_openai_all_of.py +137 -0
- pyfect/ai/_openai_errors.py +233 -0
- pyfect/ai/_provider.py +439 -0
- pyfect/ai/_provider_tools.py +129 -0
- pyfect/ai/_stream_resolution.py +231 -0
- pyfect/ai/_structured_output_checks.py +138 -0
- pyfect/ai/_structured_output_codec.py +229 -0
- pyfect/ai/_structured_output_json.py +109 -0
- pyfect/ai/_tool_call_ids.py +80 -0
- pyfect/ai/ai_error.py +785 -0
- pyfect/ai/anthropic/__init__.py +28 -0
- pyfect/ai/anthropic/_errors.py +164 -0
- pyfect/ai/anthropic/_prompt.py +375 -0
- pyfect/ai/anthropic/_response.py +443 -0
- pyfect/ai/anthropic/_schema.py +206 -0
- pyfect/ai/anthropic/_stream.py +377 -0
- pyfect/ai/anthropic/_tools.py +248 -0
- pyfect/ai/anthropic/client.py +289 -0
- pyfect/ai/anthropic/config.py +32 -0
- pyfect/ai/anthropic/error.py +27 -0
- pyfect/ai/anthropic/language_model.py +366 -0
- pyfect/ai/anthropic/telemetry.py +51 -0
- pyfect/ai/anthropic/tool.py +570 -0
- pyfect/ai/anthropic_structured_output.py +138 -0
- pyfect/ai/chat.py +548 -0
- pyfect/ai/cloudflare/__init__.py +15 -0
- pyfect/ai/cloudflare/_errors.py +108 -0
- pyfect/ai/cloudflare/client.py +174 -0
- pyfect/ai/cloudflare/decision_model.py +117 -0
- pyfect/ai/cloudflare/schema.py +144 -0
- pyfect/ai/decision.py +290 -0
- pyfect/ai/decision_model.py +283 -0
- pyfect/ai/embedding_model.py +177 -0
- pyfect/ai/id_generator.py +79 -0
- pyfect/ai/language_model.py +1077 -0
- pyfect/ai/mcp_client.py +682 -0
- pyfect/ai/mcp_client_error.py +223 -0
- pyfect/ai/mcp_protocol.py +205 -0
- pyfect/ai/mcp_schema.py +1300 -0
- pyfect/ai/mcp_server.py +404 -0
- pyfect/ai/model.py +56 -0
- pyfect/ai/openai/__init__.py +49 -0
- pyfect/ai/openai/_errors.py +89 -0
- pyfect/ai/openai/_prompt.py +527 -0
- pyfect/ai/openai/_requests.py +34 -0
- pyfect/ai/openai/_response.py +569 -0
- pyfect/ai/openai/_schema.py +769 -0
- pyfect/ai/openai/_stream.py +771 -0
- pyfect/ai/openai/_tools.py +128 -0
- pyfect/ai/openai/_web_socket.py +295 -0
- pyfect/ai/openai/client.py +296 -0
- pyfect/ai/openai/client_generated.py +123 -0
- pyfect/ai/openai/config.py +34 -0
- pyfect/ai/openai/decision_model.py +177 -0
- pyfect/ai/openai/embedding_model.py +116 -0
- pyfect/ai/openai/error.py +8 -0
- pyfect/ai/openai/generated/__init__.py +7213 -0
- pyfect/ai/openai/generated/_recursive.py +18 -0
- pyfect/ai/openai/generated/_schemas_0.py +1015 -0
- pyfect/ai/openai/generated/_schemas_1.py +1016 -0
- pyfect/ai/openai/generated/_schemas_2.py +1017 -0
- pyfect/ai/openai/generated/_schemas_3.py +615 -0
- pyfect/ai/openai/generated/client.py +5298 -0
- pyfect/ai/openai/generated/types.py +8123 -0
- pyfect/ai/openai/language_model.py +304 -0
- pyfect/ai/openai/telemetry.py +19 -0
- pyfect/ai/openai/tool.py +391 -0
- pyfect/ai/openai_compat/__init__.py +25 -0
- pyfect/ai/openai_compat/_prompt.py +191 -0
- pyfect/ai/openai_compat/_request.py +153 -0
- pyfect/ai/openai_compat/_response.py +407 -0
- pyfect/ai/openai_compat/_schema.py +97 -0
- pyfect/ai/openai_compat/client.py +234 -0
- pyfect/ai/openai_compat/config.py +32 -0
- pyfect/ai/openai_compat/embedding_model.py +139 -0
- pyfect/ai/openai_compat/error.py +7 -0
- pyfect/ai/openai_compat/language_model.py +245 -0
- pyfect/ai/openai_compat/telemetry.py +51 -0
- pyfect/ai/openai_structured_output.py +118 -0
- pyfect/ai/openrouter/__init__.py +36 -0
- pyfect/ai/openrouter/_errors.py +186 -0
- pyfect/ai/openrouter/_prompt.py +272 -0
- pyfect/ai/openrouter/_response.py +226 -0
- pyfect/ai/openrouter/_stream.py +320 -0
- pyfect/ai/openrouter/client.py +283 -0
- pyfect/ai/openrouter/config.py +36 -0
- pyfect/ai/openrouter/decision_model.py +161 -0
- pyfect/ai/openrouter/error.py +23 -0
- pyfect/ai/openrouter/generated/__init__.py +3029 -0
- pyfect/ai/openrouter/generated/_recursive.py +15 -0
- pyfect/ai/openrouter/generated/_schemas_0.py +1015 -0
- pyfect/ai/openrouter/generated/_schemas_1.py +522 -0
- pyfect/ai/openrouter/generated/client.py +1445 -0
- pyfect/ai/openrouter/generated/types.py +4769 -0
- pyfect/ai/openrouter/language_model.py +345 -0
- pyfect/ai/openrouter/schema.py +165 -0
- pyfect/ai/prompt.py +664 -0
- pyfect/ai/response.py +768 -0
- pyfect/ai/response_id_tracker.py +82 -0
- pyfect/ai/telemetry.py +126 -0
- pyfect/ai/tokenizer.py +61 -0
- pyfect/ai/tool.py +536 -0
- pyfect/ai/toolkit.py +490 -0
- pyfect/ai/typesafe/__init__.py +17 -0
- pyfect/ai/typesafe/_errors.py +120 -0
- pyfect/ai/typesafe/client.py +146 -0
- pyfect/ai/typesafe/config.py +34 -0
- pyfect/ai/typesafe/decision_model.py +90 -0
- pyfect/ai/typesafe/schema.py +140 -0
- pyfect/batching/__init__.py +58 -0
- pyfect/batching/_api.py +71 -0
- pyfect/batching/_engine.py +229 -0
- pyfect/batching/_from_effect.py +76 -0
- pyfect/batching/_persisted.py +98 -0
- pyfect/batching/_request.py +133 -0
- pyfect/batching/_resolver.py +477 -0
- pyfect/batching/_with_cache.py +90 -0
- pyfect/big_decimal.py +500 -0
- pyfect/brand.py +114 -0
- pyfect/byte_size.py +354 -0
- pyfect/cache/__init__.py +29 -0
- pyfect/cache/_cache.py +426 -0
- pyfect/cache/_expiry.py +69 -0
- pyfect/cache/_scoped_cache.py +546 -0
- pyfect/case.py +174 -0
- pyfect/cause.py +462 -0
- pyfect/channel/__init__.py +155 -0
- pyfect/channel/_base.py +92 -0
- pyfect/channel/_catch.py +158 -0
- pyfect/channel/_catch_reason.py +572 -0
- pyfect/channel/_channel.py +158 -0
- pyfect/channel/_combinators.py +220 -0
- pyfect/channel/_concurrency_methods.py +217 -0
- pyfect/channel/_concurrent_map.py +118 -0
- pyfect/channel/_constructors.py +349 -0
- pyfect/channel/_context.py +232 -0
- pyfect/channel/_error_methods.py +336 -0
- pyfect/channel/_errors.py +180 -0
- pyfect/channel/_filters.py +225 -0
- pyfect/channel/_lifecycle.py +71 -0
- pyfect/channel/_lifecycle_methods.py +206 -0
- pyfect/channel/_loop.py +100 -0
- pyfect/channel/_merge.py +246 -0
- pyfect/channel/_repeat.py +154 -0
- pyfect/channel/_run.py +149 -0
- pyfect/channel/_run_methods.py +341 -0
- pyfect/channel/_sinks.py +88 -0
- pyfect/channel/_sources.py +238 -0
- pyfect/channel/_text.py +176 -0
- pyfect/channel/_transform_methods.py +415 -0
- pyfect/channel/_upstream.py +67 -0
- pyfect/channel_schema.py +131 -0
- pyfect/cli/__init__.py +79 -0
- pyfect/cli/_ansi.py +82 -0
- pyfect/cli/_command.py +469 -0
- pyfect/cli/_config.py +164 -0
- pyfect/cli/_environment.py +10 -0
- pyfect/cli/_help.py +195 -0
- pyfect/cli/_lexer.py +81 -0
- pyfect/cli/_parser.py +465 -0
- pyfect/cli/_settings.py +85 -0
- pyfect/cli/_suggest.py +33 -0
- pyfect/cli/_width.py +134 -0
- pyfect/cli/_wizard.py +317 -0
- pyfect/cli/argument.py +212 -0
- pyfect/cli/cli_config.py +40 -0
- pyfect/cli/cli_error.py +242 -0
- pyfect/cli/cli_output.py +305 -0
- pyfect/cli/command.py +440 -0
- pyfect/cli/completions/__init__.py +50 -0
- pyfect/cli/completions/_completer.ps1 +190 -0
- pyfect/cli/completions/_descriptor.py +182 -0
- pyfect/cli/completions/bash.py +300 -0
- pyfect/cli/completions/fish.py +164 -0
- pyfect/cli/completions/powershell.py +156 -0
- pyfect/cli/completions/zsh.py +189 -0
- pyfect/cli/flag.py +246 -0
- pyfect/cli/global_flag.py +99 -0
- pyfect/cli/help_doc.py +63 -0
- pyfect/cli/param.py +789 -0
- pyfect/cli/primitive.py +308 -0
- pyfect/cli/prompt/__init__.py +75 -0
- pyfect/cli/prompt/_auto_complete.py +205 -0
- pyfect/cli/prompt/_choices.py +104 -0
- pyfect/cli/prompt/_confirm.py +195 -0
- pyfect/cli/prompt/_core.py +512 -0
- pyfect/cli/prompt/_date.py +173 -0
- pyfect/cli/prompt/_date_parts.py +387 -0
- pyfect/cli/prompt/_file.py +372 -0
- pyfect/cli/prompt/_multi_select.py +204 -0
- pyfect/cli/prompt/_number.py +257 -0
- pyfect/cli/prompt/_select.py +117 -0
- pyfect/cli/prompt/_text.py +239 -0
- pyfect/cli/terminal_capabilities.py +118 -0
- pyfect/clock.py +150 -0
- pyfect/cluster/__init__.py +155 -0
- pyfect/cluster/_cluster_abandon.py +36 -0
- pyfect/cluster/_cluster_layer.py +122 -0
- pyfect/cluster/_entity_activation.py +16 -0
- pyfect/cluster/_entity_manager/__init__.py +22 -0
- pyfect/cluster/_entity_manager/_activation.py +540 -0
- pyfect/cluster/_entity_manager/_core.py +186 -0
- pyfect/cluster/_entity_manager/_decode.py +60 -0
- pyfect/cluster/_entity_manager/_manager.py +423 -0
- pyfect/cluster/_entity_manager/_state.py +113 -0
- pyfect/cluster/_entity_reaper.py +94 -0
- pyfect/cluster/_hash.py +13 -0
- pyfect/cluster/_interruptors.py +88 -0
- pyfect/cluster/_resource_map.py +179 -0
- pyfect/cluster/_resource_ref.py +266 -0
- pyfect/cluster/_shard_lock.py +14 -0
- pyfect/cluster/cluster_cron.py +230 -0
- pyfect/cluster/cluster_error.py +81 -0
- pyfect/cluster/cluster_http.py +210 -0
- pyfect/cluster/cluster_metrics.py +54 -0
- pyfect/cluster/cluster_schema.py +137 -0
- pyfect/cluster/cluster_socket.py +197 -0
- pyfect/cluster/cluster_workflow_engine/__init__.py +24 -0
- pyfect/cluster/cluster_workflow_engine/_engine.py +1099 -0
- pyfect/cluster/cluster_workflow_engine/_entities.py +298 -0
- pyfect/cluster/deliver_at.py +31 -0
- pyfect/cluster/entity/__init__.py +39 -0
- pyfect/cluster/entity/_entity.py +332 -0
- pyfect/cluster/entity/_handlers.py +168 -0
- pyfect/cluster/entity/_keep_alive.py +74 -0
- pyfect/cluster/entity/_queue.py +103 -0
- pyfect/cluster/entity/_request.py +34 -0
- pyfect/cluster/entity/_services.py +23 -0
- pyfect/cluster/entity/_test_client.py +237 -0
- pyfect/cluster/entity_address.py +25 -0
- pyfect/cluster/entity_id.py +20 -0
- pyfect/cluster/entity_proxy.py +161 -0
- pyfect/cluster/entity_proxy_server.py +103 -0
- pyfect/cluster/entity_resource.py +122 -0
- pyfect/cluster/entity_type.py +20 -0
- pyfect/cluster/envelope.py +307 -0
- pyfect/cluster/http_runner.py +254 -0
- pyfect/cluster/k8s_http_client.py +281 -0
- pyfect/cluster/k8s_types.py +1394 -0
- pyfect/cluster/machine_id.py +21 -0
- pyfect/cluster/message.py +219 -0
- pyfect/cluster/message_storage/__init__.py +67 -0
- pyfect/cluster/message_storage/_encoded.py +393 -0
- pyfect/cluster/message_storage/_memory.py +386 -0
- pyfect/cluster/message_storage/_noop.py +109 -0
- pyfect/cluster/message_storage/_service.py +388 -0
- pyfect/cluster/reply.py +236 -0
- pyfect/cluster/runner.py +32 -0
- pyfect/cluster/runner_address.py +23 -0
- pyfect/cluster/runner_health.py +93 -0
- pyfect/cluster/runner_server.py +286 -0
- pyfect/cluster/runner_storage.py +212 -0
- pyfect/cluster/runners/__init__.py +55 -0
- pyfect/cluster/runners/_local.py +460 -0
- pyfect/cluster/runners/_noop.py +57 -0
- pyfect/cluster/runners/_rpc.py +290 -0
- pyfect/cluster/runners/_service.py +113 -0
- pyfect/cluster/shard_id.py +39 -0
- pyfect/cluster/sharding/__init__.py +12 -0
- pyfect/cluster/sharding/_acquisition.py +490 -0
- pyfect/cluster/sharding/_clients.py +271 -0
- pyfect/cluster/sharding/_make.py +388 -0
- pyfect/cluster/sharding/_routing.py +296 -0
- pyfect/cluster/sharding/_runners_sync.py +198 -0
- pyfect/cluster/sharding/_service.py +91 -0
- pyfect/cluster/sharding/_singletons.py +183 -0
- pyfect/cluster/sharding/_state.py +296 -0
- pyfect/cluster/sharding/_storage_loop.py +421 -0
- pyfect/cluster/sharding_config.py +230 -0
- pyfect/cluster/sharding_registration_event.py +28 -0
- pyfect/cluster/single_runner.py +49 -0
- pyfect/cluster/singleton.py +36 -0
- pyfect/cluster/singleton_address.py +20 -0
- pyfect/cluster/snowflake.py +133 -0
- pyfect/cluster/socket_runner.py +67 -0
- pyfect/cluster/sql_message_storage.py +1001 -0
- pyfect/cluster/sql_runner_storage.py +841 -0
- pyfect/cluster/test_runner.py +23 -0
- pyfect/config/__init__.py +93 -0
- pyfect/config/_all.py +96 -0
- pyfect/config/_collections.py +291 -0
- pyfect/config/_config.py +261 -0
- pyfect/config/_constructors.py +106 -0
- pyfect/config/_error.py +27 -0
- pyfect/config/_leaf.py +92 -0
- pyfect/config/_scalars.py +193 -0
- pyfect/config/_schema.py +340 -0
- pyfect/config_provider/__init__.py +60 -0
- pyfect/config_provider/_dir.py +84 -0
- pyfect/config_provider/_dotenv.py +127 -0
- pyfect/config_provider/_env.py +79 -0
- pyfect/config_provider/_layers.py +80 -0
- pyfect/config_provider/_node.py +90 -0
- pyfect/config_provider/_provider.py +118 -0
- pyfect/config_provider/_unknown.py +62 -0
- pyfect/console/__init__.py +63 -0
- pyfect/console/_effects.py +134 -0
- pyfect/console/_live.py +156 -0
- pyfect/console/_service.py +49 -0
- pyfect/console/_table.py +90 -0
- pyfect/context.py +414 -0
- pyfect/cron/__init__.py +30 -0
- pyfect/cron/_model.py +183 -0
- pyfect/cron/_search.py +268 -0
- pyfect/cron/_text.py +202 -0
- pyfect/cron/_wall.py +42 -0
- pyfect/crypto.py +257 -0
- pyfect/date_time/__init__.py +77 -0
- pyfect/date_time/_calendar.py +209 -0
- pyfect/date_time/_current.py +118 -0
- pyfect/date_time/_epoch.py +22 -0
- pyfect/date_time/_iso.py +80 -0
- pyfect/date_time/_zones.py +135 -0
- pyfect/date_time/intl.py +144 -0
- pyfect/deferred.py +145 -0
- pyfect/devtools/__init__.py +24 -0
- pyfect/devtools/client.py +323 -0
- pyfect/devtools/layers.py +32 -0
- pyfect/devtools/schema.py +301 -0
- pyfect/devtools/server.py +72 -0
- pyfect/duration.py +295 -0
- pyfect/effect.py +565 -0
- pyfect/effectable.py +37 -0
- pyfect/encoding/__init__.py +62 -0
- pyfect/encoding/_archive.py +197 -0
- pyfect/encoding/_json_array_scanner.py +477 -0
- pyfect/encoding/base64.py +23 -0
- pyfect/encoding/base64_url.py +19 -0
- pyfect/encoding/encoding_error.py +17 -0
- pyfect/encoding/hex.py +28 -0
- pyfect/encoding/ini.py +188 -0
- pyfect/encoding/json_array.py +246 -0
- pyfect/encoding/ndjson.py +461 -0
- pyfect/encoding/schema_binary/__init__.py +257 -0
- pyfect/encoding/schema_binary/_channels.py +173 -0
- pyfect/encoding/schema_binary/_compile.py +584 -0
- pyfect/encoding/schema_binary/_decode.py +848 -0
- pyfect/encoding/schema_binary/_dictionary.py +90 -0
- pyfect/encoding/schema_binary/_encode.py +838 -0
- pyfect/encoding/schema_binary/_fingerprint.py +168 -0
- pyfect/encoding/schema_binary/_layout.py +350 -0
- pyfect/encoding/schema_binary/_literals.py +47 -0
- pyfect/encoding/schema_binary/_parser.py +163 -0
- pyfect/encoding/schema_binary/_reader.py +148 -0
- pyfect/encoding/schema_binary/_target.py +310 -0
- pyfect/encoding/schema_binary/_values.py +133 -0
- pyfect/encoding/schema_binary/_wire.py +253 -0
- pyfect/encoding/sse.py +476 -0
- pyfect/encoding/tar/__init__.py +396 -0
- pyfect/encoding/tar/_decoder.py +220 -0
- pyfect/encoding/tar/_encoder.py +83 -0
- pyfect/encoding/tar/_header.py +494 -0
- pyfect/encoding/tar/_model.py +82 -0
- pyfect/encoding/toml.py +375 -0
- pyfect/encoding/xml/__init__.py +222 -0
- pyfect/encoding/xml/_chars.py +119 -0
- pyfect/encoding/xml/_collector.py +106 -0
- pyfect/encoding/xml/_inputs.py +126 -0
- pyfect/encoding/xml/_model.py +114 -0
- pyfect/encoding/xml/_parser.py +694 -0
- pyfect/encoding/xml/_scanner.py +180 -0
- pyfect/encoding/xml/_writer.py +238 -0
- pyfect/encoding/yaml.py +739 -0
- pyfect/encoding/zip/__init__.py +392 -0
- pyfect/encoding/zip/_decoder.py +378 -0
- pyfect/encoding/zip/_format.py +122 -0
- pyfect/encoding/zip/_inflate.py +65 -0
- pyfect/encoding/zip/_model.py +86 -0
- pyfect/encoding/zip/_writer.py +311 -0
- pyfect/error_reporter.py +240 -0
- pyfect/errors.py +321 -0
- pyfect/eventlog/__init__.py +77 -0
- pyfect/eventlog/_conflicts.py +94 -0
- pyfect/eventlog/_crypto.py +64 -0
- pyfect/eventlog/_identity.py +26 -0
- pyfect/eventlog/_identity_derivation.py +79 -0
- pyfect/eventlog/_transferable_bytes.py +55 -0
- pyfect/eventlog/_uuid.py +33 -0
- pyfect/eventlog/event.py +115 -0
- pyfect/eventlog/event_group.py +55 -0
- pyfect/eventlog/event_journal.py +630 -0
- pyfect/eventlog/event_log.py +1039 -0
- pyfect/eventlog/event_log_encryption.py +219 -0
- pyfect/eventlog/event_log_message.py +357 -0
- pyfect/eventlog/event_log_remote.py +465 -0
- pyfect/eventlog/event_log_server.py +442 -0
- pyfect/eventlog/event_log_server_encrypted.py +322 -0
- pyfect/eventlog/event_log_server_unencrypted.py +814 -0
- pyfect/eventlog/event_log_session_auth.py +250 -0
- pyfect/eventlog/event_row.py +423 -0
- pyfect/eventlog/sql_event_journal.py +692 -0
- pyfect/eventlog/sql_event_log_server_encrypted.py +440 -0
- pyfect/eventlog/sql_event_log_server_unencrypted.py +568 -0
- pyfect/execution_plan/__init__.py +42 -0
- pyfect/execution_plan/_events.py +128 -0
- pyfect/execution_plan/_model.py +108 -0
- pyfect/execution_plan/_run.py +176 -0
- pyfect/exit.py +205 -0
- pyfect/fiber_handle.py +305 -0
- pyfect/fiber_map.py +350 -0
- pyfect/fiber_set.py +232 -0
- pyfect/file_system/__init__.py +69 -0
- pyfect/file_system/_copy.py +116 -0
- pyfect/file_system/_errors.py +32 -0
- pyfect/file_system/_file.py +177 -0
- pyfect/file_system/_live.py +351 -0
- pyfect/file_system/_model.py +72 -0
- pyfect/file_system/_noop.py +189 -0
- pyfect/file_system/_service.py +336 -0
- pyfect/file_system/_stat.py +69 -0
- pyfect/file_system/_watch.py +107 -0
- pyfect/filters.py +245 -0
- pyfect/graph/__init__.py +101 -0
- pyfect/graph/_all_pairs.py +150 -0
- pyfect/graph/_combine.py +259 -0
- pyfect/graph/_compact.py +143 -0
- pyfect/graph/_components.py +200 -0
- pyfect/graph/_cycles.py +144 -0
- pyfect/graph/_enumerate.py +168 -0
- pyfect/graph/_errors.py +51 -0
- pyfect/graph/_flow.py +164 -0
- pyfect/graph/_forest.py +109 -0
- pyfect/graph/_graph.py +269 -0
- pyfect/graph/_lowlink.py +98 -0
- pyfect/graph/_matching.py +101 -0
- pyfect/graph/_mutate.py +211 -0
- pyfect/graph/_query.py +138 -0
- pyfect/graph/_records.py +119 -0
- pyfect/graph/_render.py +109 -0
- pyfect/graph/_shortest.py +274 -0
- pyfect/graph/_snapshot.py +50 -0
- pyfect/graph/_store.py +136 -0
- pyfect/graph/_traverse.py +289 -0
- pyfect/graph/_view.py +384 -0
- pyfect/graph/_walker.py +70 -0
- pyfect/graphql/__init__.py +87 -0
- pyfect/graphql/_codec.py +117 -0
- pyfect/graphql/_http.py +309 -0
- pyfect/graphql/_schemas.py +39 -0
- pyfect/graphql/_slices.py +110 -0
- pyfect/graphql/_template.py +111 -0
- pyfect/graphql/_types.py +115 -0
- pyfect/graphql/_ws.py +395 -0
- pyfect/graphql/_ws_messages.py +120 -0
- pyfect/graphql/client.py +428 -0
- pyfect/graphql/client_error.py +285 -0
- pyfect/graphql/group.py +142 -0
- pyfect/graphql/middleware.py +98 -0
- pyfect/graphql/operation.py +409 -0
- pyfect/graphql/protocol.py +119 -0
- pyfect/graphql/resolver.py +574 -0
- pyfect/graphql/template.py +61 -0
- pyfect/hash_map.py +317 -0
- pyfect/hash_ring.py +206 -0
- pyfect/hash_set.py +163 -0
- pyfect/http/__init__.py +125 -0
- pyfect/http/_asgi_listener.py +272 -0
- pyfect/http/_authority.py +63 -0
- pyfect/http/_brotli.py +27 -0
- pyfect/http/_compression.py +52 -0
- pyfect/http/_field.py +18 -0
- pyfect/http/_foreign_path.py +27 -0
- pyfect/http/_form_data.py +34 -0
- pyfect/http/_httpx_options.py +120 -0
- pyfect/http/_in_memory.py +76 -0
- pyfect/http/_request_state.py +59 -0
- pyfect/http/_vary.py +14 -0
- pyfect/http/abort.py +65 -0
- pyfect/http/asgi/__init__.py +59 -0
- pyfect/http/asgi/_app.py +155 -0
- pyfect/http/asgi/_http.py +201 -0
- pyfect/http/asgi/_mount.py +263 -0
- pyfect/http/asgi/_scope.py +86 -0
- pyfect/http/asgi/_serve.py +72 -0
- pyfect/http/asgi/_types.py +13 -0
- pyfect/http/asgi/_web_socket.py +325 -0
- pyfect/http/body.py +437 -0
- pyfect/http/client/__init__.py +111 -0
- pyfect/http/client/_circuit_breaker.py +113 -0
- pyfect/http/client/_cookie_jar.py +120 -0
- pyfect/http/client/_make.py +252 -0
- pyfect/http/client/_rate_limit.py +423 -0
- pyfect/http/client/_redirects.py +93 -0
- pyfect/http/client/_retry.py +115 -0
- pyfect/http/client/_service.py +497 -0
- pyfect/http/cookies.py +444 -0
- pyfect/http/error.py +188 -0
- pyfect/http/etag.py +96 -0
- pyfect/http/file_like.py +49 -0
- pyfect/http/find_my_way/__init__.py +10 -0
- pyfect/http/find_my_way/_nodes.py +186 -0
- pyfect/http/find_my_way/_paths.py +43 -0
- pyfect/http/find_my_way/_query_string.py +67 -0
- pyfect/http/find_my_way/_router.py +401 -0
- pyfect/http/headers.py +343 -0
- pyfect/http/http_effect.py +198 -0
- pyfect/http/httpx_client.py +291 -0
- pyfect/http/hypercorn_server.py +122 -0
- pyfect/http/incoming_message.py +140 -0
- pyfect/http/live_server/__init__.py +191 -0
- pyfect/http/live_server/_body.py +164 -0
- pyfect/http/live_server/_exchange.py +321 -0
- pyfect/http/live_server/_head.py +194 -0
- pyfect/http/live_server/_inbound.py +156 -0
- pyfect/http/live_server/_listener.py +211 -0
- pyfect/http/live_server/_serve.py +225 -0
- pyfect/http/live_server/_timers.py +20 -0
- pyfect/http/live_server/_upgrade.py +108 -0
- pyfect/http/live_server/_websocket.py +286 -0
- pyfect/http/live_server/_ws_frames.py +242 -0
- pyfect/http/method.py +30 -0
- pyfect/http/middleware/__init__.py +61 -0
- pyfect/http/middleware/_compression.py +156 -0
- pyfect/http/middleware/_cors.py +129 -0
- pyfect/http/middleware/_decompression.py +274 -0
- pyfect/http/middleware/_forwarded.py +113 -0
- pyfect/http/middleware/_logger.py +73 -0
- pyfect/http/middleware/_search_params.py +34 -0
- pyfect/http/middleware/_tracer.py +153 -0
- pyfect/http/middleware/_types.py +16 -0
- pyfect/http/mime/__init__.py +69 -0
- pyfect/http/mime/_types.py +361 -0
- pyfect/http/multipart.py +613 -0
- pyfect/http/multipart_parser/__init__.py +348 -0
- pyfect/http/multipart_parser/_content_type.py +109 -0
- pyfect/http/multipart_parser/_headers.py +206 -0
- pyfect/http/multipart_parser/_search.py +85 -0
- pyfect/http/origin.py +146 -0
- pyfect/http/platform.py +593 -0
- pyfect/http/prometheus.py +34 -0
- pyfect/http/request.py +384 -0
- pyfect/http/response.py +474 -0
- pyfect/http/router/__init__.py +82 -0
- pyfect/http/router/_api.py +462 -0
- pyfect/http/router/_in_memory.py +76 -0
- pyfect/http/router/_markers.py +47 -0
- pyfect/http/router/_middleware.py +181 -0
- pyfect/http/router/_middleware_stack.py +51 -0
- pyfect/http/router/_provided.py +31 -0
- pyfect/http/router/_route.py +114 -0
- pyfect/http/router/_router.py +308 -0
- pyfect/http/schema.py +276 -0
- pyfect/http/server.py +199 -0
- pyfect/http/server_error.py +261 -0
- pyfect/http/server_request.py +729 -0
- pyfect/http/server_respondable.py +77 -0
- pyfect/http/server_response.py +569 -0
- pyfect/http/static_server/__init__.py +257 -0
- pyfect/http/static_server/_conditional.py +72 -0
- pyfect/http/static_server/_mime.py +55 -0
- pyfect/http/static_server/_range.py +53 -0
- pyfect/http/status.py +51 -0
- pyfect/http/template.py +161 -0
- pyfect/http/trace_context.py +91 -0
- pyfect/http/url.py +442 -0
- pyfect/http/url_params.py +304 -0
- pyfect/http/uvicorn_server.py +119 -0
- pyfect/http/web.py +171 -0
- pyfect/http/web_handler.py +463 -0
- pyfect/http/wire.py +209 -0
- pyfect/http_api/__init__.py +83 -0
- pyfect/http_api/_assets/scalar.js +8 -0
- pyfect/http_api/_assets/swagger-ui.css +3 -0
- pyfect/http_api/_assets/swagger-ui.js +4 -0
- pyfect/http_api/_client_method.py +220 -0
- pyfect/http_api/_client_payload.py +144 -0
- pyfect/http_api/_client_response.py +377 -0
- pyfect/http_api/_handlers.py +282 -0
- pyfect/http_api/_html.py +38 -0
- pyfect/http_api/_media_type.py +6 -0
- pyfect/http_api/_openapi_responses.py +164 -0
- pyfect/http_api/_openapi_schemas.py +163 -0
- pyfect/http_api/_path.py +102 -0
- pyfect/http_api/_payload.py +260 -0
- pyfect/http_api/_response.py +363 -0
- pyfect/http_api/_security.py +174 -0
- pyfect/http_api/api.py +265 -0
- pyfect/http_api/builder.py +447 -0
- pyfect/http_api/client.py +474 -0
- pyfect/http_api/endpoint.py +969 -0
- pyfect/http_api/error.py +280 -0
- pyfect/http_api/group.py +215 -0
- pyfect/http_api/middleware.py +232 -0
- pyfect/http_api/openapi.py +396 -0
- pyfect/http_api/scalar.py +143 -0
- pyfect/http_api/schema.py +713 -0
- pyfect/http_api/security.py +91 -0
- pyfect/http_api/swagger.py +69 -0
- pyfect/http_api/test.py +105 -0
- pyfect/json_patch.py +255 -0
- pyfect/json_pointer.py +100 -0
- pyfect/json_schema/__init__.py +66 -0
- pyfect/json_schema/_api.py +96 -0
- pyfect/json_schema/_converter.py +258 -0
- pyfect/json_schema/_draft04.py +198 -0
- pyfect/json_schema/_draft07.py +203 -0
- pyfect/json_schema/_keywords.py +202 -0
- pyfect/json_schema/_model.py +201 -0
- pyfect/json_schema/_openapi.py +208 -0
- pyfect/latch.py +127 -0
- pyfect/layer.py +479 -0
- pyfect/layer_map.py +279 -0
- pyfect/layer_ref.py +208 -0
- pyfect/logger/__init__.py +91 -0
- pyfect/logger/_batched.py +49 -0
- pyfect/logger/_console.py +116 -0
- pyfect/logger/_file.py +31 -0
- pyfect/logger/_formats.py +104 -0
- pyfect/logger/_layer.py +50 -0
- pyfect/logger/_level.py +53 -0
- pyfect/logger/_log.py +142 -0
- pyfect/logger/_model.py +163 -0
- pyfect/logger/_refs.py +78 -0
- pyfect/logger/_stdlib.py +54 -0
- pyfect/logger/_text.py +263 -0
- pyfect/logger/_tracer.py +35 -0
- pyfect/managed_runtime.py +393 -0
- pyfect/metric/__init__.py +119 -0
- pyfect/metric/_boundaries.py +27 -0
- pyfect/metric/_kinds.py +417 -0
- pyfect/metric/_metric.py +302 -0
- pyfect/metric/_prometheus.py +150 -0
- pyfect/metric/_refs.py +78 -0
- pyfect/metric/_runtime.py +75 -0
- pyfect/metric/_snapshot.py +97 -0
- pyfect/metric/_state.py +74 -0
- pyfect/msgspec.py +98 -0
- pyfect/mutable_hash_map.py +140 -0
- pyfect/mutable_hash_set.py +95 -0
- pyfect/net/__init__.py +18 -0
- pyfect/net/_errors.py +29 -0
- pyfect/net/address_resolver.py +179 -0
- pyfect/net/dns/__init__.py +92 -0
- pyfect/net/dns/_error.py +43 -0
- pyfect/net/dns/_records.py +335 -0
- pyfect/net/dns/_service.py +287 -0
- pyfect/net/dns/_system.py +89 -0
- pyfect/net/host.py +311 -0
- pyfect/net/ip_interface.py +202 -0
- pyfect/net/ip_network.py +250 -0
- pyfect/net/net_address/__init__.py +245 -0
- pyfect/net/net_address/_inet.py +412 -0
- pyfect/net/net_address/_ip.py +348 -0
- pyfect/net/net_address/_mac.py +119 -0
- pyfect/optic/__init__.py +50 -0
- pyfect/optic/_builtins.py +90 -0
- pyfect/optic/_data.py +97 -0
- pyfect/optic/_optic.py +422 -0
- pyfect/optic/_steps.py +151 -0
- pyfect/option.py +362 -0
- pyfect/otel/__init__.py +22 -0
- pyfect/otel/_attributes.py +67 -0
- pyfect/otel/_producer.py +322 -0
- pyfect/otel/_shutdown.py +26 -0
- pyfect/otel/_span.py +292 -0
- pyfect/otel/_wrapper.py +92 -0
- pyfect/otel/logger.py +154 -0
- pyfect/otel/metrics.py +126 -0
- pyfect/otel/resource.py +111 -0
- pyfect/otel/sdk.py +195 -0
- pyfect/otel/tracer.py +157 -0
- pyfect/partitioned_semaphore.py +211 -0
- pyfect/path.py +304 -0
- pyfect/persistence/__init__.py +77 -0
- pyfect/persistence/_columns.py +32 -0
- pyfect/persistence/_error.py +14 -0
- pyfect/persistence/_seconds.py +68 -0
- pyfect/persistence/_sql_keys.py +47 -0
- pyfect/persistence/circuit_breaker/__init__.py +120 -0
- pyfect/persistence/circuit_breaker/_breaker.py +317 -0
- pyfect/persistence/circuit_breaker/_errors.py +67 -0
- pyfect/persistence/circuit_breaker/_memory.py +243 -0
- pyfect/persistence/circuit_breaker/_redis.py +509 -0
- pyfect/persistence/circuit_breaker/_store.py +60 -0
- pyfect/persistence/circuit_breaker/_types.py +191 -0
- pyfect/persistence/key_value_store/__init__.py +51 -0
- pyfect/persistence/key_value_store/_layers.py +216 -0
- pyfect/persistence/key_value_store/_schema_store.py +91 -0
- pyfect/persistence/key_value_store/_sql.py +197 -0
- pyfect/persistence/key_value_store/_store.py +212 -0
- pyfect/persistence/persistable.py +164 -0
- pyfect/persistence/persisted_cache.py +159 -0
- pyfect/persistence/persisted_queue/__init__.py +75 -0
- pyfect/persistence/persisted_queue/_error.py +42 -0
- pyfect/persistence/persisted_queue/_mailbox.py +101 -0
- pyfect/persistence/persisted_queue/_memory.py +182 -0
- pyfect/persistence/persisted_queue/_queue.py +272 -0
- pyfect/persistence/persisted_queue/_redis.py +455 -0
- pyfect/persistence/persisted_queue/_redis_scripts.py +395 -0
- pyfect/persistence/persisted_queue/_sql.py +334 -0
- pyfect/persistence/persisted_queue/_sql_migrations.py +319 -0
- pyfect/persistence/persisted_queue/_sql_statements.py +441 -0
- pyfect/persistence/persisted_queue/_store.py +147 -0
- pyfect/persistence/persistence/__init__.py +84 -0
- pyfect/persistence/persistence/_kvs.py +135 -0
- pyfect/persistence/persistence/_memory.py +62 -0
- pyfect/persistence/persistence/_redis.py +162 -0
- pyfect/persistence/persistence/_service.py +276 -0
- pyfect/persistence/persistence/_sql.py +499 -0
- pyfect/persistence/rate_limiter/__init__.py +82 -0
- pyfect/persistence/rate_limiter/_errors.py +59 -0
- pyfect/persistence/rate_limiter/_limiter.py +325 -0
- pyfect/persistence/rate_limiter/_memory.py +217 -0
- pyfect/persistence/rate_limiter/_redis.py +485 -0
- pyfect/persistence/rate_limiter/_store.py +89 -0
- pyfect/persistence/redis/__init__.py +52 -0
- pyfect/persistence/redis/_error.py +12 -0
- pyfect/persistence/redis/_make.py +234 -0
- pyfect/persistence/redis/_script.py +78 -0
- pyfect/persistence/redis/_service.py +104 -0
- pyfect/persistence/redis_py.py +303 -0
- pyfect/persistence/user_directories.py +195 -0
- pyfect/platform.py +26 -0
- pyfect/platform_error.py +256 -0
- pyfect/pool/__init__.py +36 -0
- pyfect/pool/_pool.py +707 -0
- pyfect/pool/_state.py +161 -0
- pyfect/pool/_ttl.py +105 -0
- pyfect/pool/_waiters.py +116 -0
- pyfect/primary_key.py +35 -0
- pyfect/process/__init__.py +82 -0
- pyfect/process/_child.py +213 -0
- pyfect/process/_command.py +358 -0
- pyfect/process/_handle.py +77 -0
- pyfect/process/_live.py +455 -0
- pyfect/process/_os.py +78 -0
- pyfect/process/_pipes.py +291 -0
- pyfect/process/_spawner.py +75 -0
- pyfect/pubsub/__init__.py +33 -0
- pyfect/pubsub/_atomic.py +181 -0
- pyfect/pubsub/_hub.py +214 -0
- pyfect/pubsub/_pubsub.py +214 -0
- pyfect/pubsub/_replay.py +107 -0
- pyfect/pubsub/_subscription.py +226 -0
- pyfect/pull.py +132 -0
- pyfect/py.typed +0 -0
- pyfect/pydantic.py +188 -0
- pyfect/queue/__init__.py +33 -0
- pyfect/queue/_queue.py +525 -0
- pyfect/queue/_state.py +111 -0
- pyfect/queue/_views.py +55 -0
- pyfect/random.py +146 -0
- pyfect/rc_map.py +363 -0
- pyfect/rc_ref.py +271 -0
- pyfect/reactivity/__init__.py +53 -0
- pyfect/reactivity/_async_result_schema.py +227 -0
- pyfect/reactivity/_hydration.py +18 -0
- pyfect/reactivity/_live.py +21 -0
- pyfect/reactivity/_reactivity.py +256 -0
- pyfect/reactivity/_result_effect.py +17 -0
- pyfect/reactivity/_time.py +11 -0
- pyfect/reactivity/async_result.py +679 -0
- pyfect/reactivity/atom/__init__.py +143 -0
- pyfect/reactivity/atom/_combinators.py +223 -0
- pyfect/reactivity/atom/_constructors.py +80 -0
- pyfect/reactivity/atom/_conversions.py +79 -0
- pyfect/reactivity/atom/_effect.py +252 -0
- pyfect/reactivity/atom/_fn.py +341 -0
- pyfect/reactivity/atom/_kvs.py +101 -0
- pyfect/reactivity/atom/_optimistic.py +162 -0
- pyfect/reactivity/atom/_runtime.py +323 -0
- pyfect/reactivity/atom/_serializable.py +65 -0
- pyfect/reactivity/atom/_subscription_ref.py +102 -0
- pyfect/reactivity/atom/_swr.py +69 -0
- pyfect/reactivity/atom/_types.py +435 -0
- pyfect/reactivity/atom_http_api.py +667 -0
- pyfect/reactivity/atom_ref.py +260 -0
- pyfect/reactivity/atom_registry/__init__.py +219 -0
- pyfect/reactivity/atom_registry/_batch.py +96 -0
- pyfect/reactivity/atom_registry/_dispatcher.py +60 -0
- pyfect/reactivity/atom_registry/_lifetime.py +202 -0
- pyfect/reactivity/atom_registry/_node.py +423 -0
- pyfect/reactivity/atom_registry/_registry.py +380 -0
- pyfect/reactivity/atom_rpc.py +283 -0
- pyfect/reactivity/hydration.py +200 -0
- pyfect/redacted.py +83 -0
- pyfect/ref.py +129 -0
- pyfect/resource.py +98 -0
- pyfect/result.py +306 -0
- pyfect/rpc/__init__.py +64 -0
- pyfect/rpc/_encoded_schema.py +86 -0
- pyfect/rpc/_json_rpc.py +246 -0
- pyfect/rpc/_methods.py +177 -0
- pyfect/rpc/_rpc.py +498 -0
- pyfect/rpc/client/__init__.py +100 -0
- pyfect/rpc/client/_core.py +449 -0
- pyfect/rpc/client/_http.py +257 -0
- pyfect/rpc/client/_protocol.py +171 -0
- pyfect/rpc/client/_serialized.py +321 -0
- pyfect/rpc/client/_transports.py +328 -0
- pyfect/rpc/client/_worker.py +283 -0
- pyfect/rpc/client_error.py +35 -0
- pyfect/rpc/group.py +296 -0
- pyfect/rpc/message.py +299 -0
- pyfect/rpc/middleware.py +182 -0
- pyfect/rpc/schema.py +89 -0
- pyfect/rpc/serialization.py +366 -0
- pyfect/rpc/server/__init__.py +181 -0
- pyfect/rpc/server/_core.py +549 -0
- pyfect/rpc/server/_protocol.py +123 -0
- pyfect/rpc/server/_serialized.py +363 -0
- pyfect/rpc/server/_stdio.py +101 -0
- pyfect/rpc/server/_transports.py +427 -0
- pyfect/rpc/server/_worker.py +108 -0
- pyfect/rpc/test.py +74 -0
- pyfect/rpc/worker.py +92 -0
- pyfect/run.py +393 -0
- pyfect/schedule/__init__.py +91 -0
- pyfect/schedule/_combinators.py +168 -0
- pyfect/schedule/_constructors.py +223 -0
- pyfect/schedule/_cron.py +49 -0
- pyfect/schedule/_drive.py +191 -0
- pyfect/schedule/_model.py +405 -0
- pyfect/scheduler.py +575 -0
- pyfect/schema/__init__.py +866 -0
- pyfect/schema/_adapters.py +39 -0
- pyfect/schema/_arbitrary_constraint.py +242 -0
- pyfect/schema/_arbitrary_graph.py +325 -0
- pyfect/schema/_arbitrary_links.py +179 -0
- pyfect/schema/_arbitrary_number.py +112 -0
- pyfect/schema/_arrays.py +287 -0
- pyfect/schema/_assignable.py +118 -0
- pyfect/schema/_boolean_literals.py +31 -0
- pyfect/schema/_checks_number.py +216 -0
- pyfect/schema/_checks_order.py +140 -0
- pyfect/schema/_checks_size.py +277 -0
- pyfect/schema/_checks_string.py +241 -0
- pyfect/schema/_class.py +569 -0
- pyfect/schema/_codec_derivation.py +227 -0
- pyfect/schema/_codec_links.py +221 -0
- pyfect/schema/_codecs.py +316 -0
- pyfect/schema/_collections.py +172 -0
- pyfect/schema/_constructor_default.py +89 -0
- pyfect/schema/_data.py +440 -0
- pyfect/schema/_data_codecs.py +145 -0
- pyfect/schema/_decimal_regexp.py +243 -0
- pyfect/schema/_declare.py +68 -0
- pyfect/schema/_derive.py +292 -0
- pyfect/schema/_duration_codecs.py +27 -0
- pyfect/schema/_ecma_pattern.py +416 -0
- pyfect/schema/_error.py +26 -0
- pyfect/schema/_error_instance.py +222 -0
- pyfect/schema/_graph.py +271 -0
- pyfect/schema/_instances.py +240 -0
- pyfect/schema/_jsonschema/__init__.py +90 -0
- pyfect/schema/_jsonschema/_annotations.py +114 -0
- pyfect/schema/_jsonschema/_checks.py +253 -0
- pyfect/schema/_jsonschema/_compile.py +346 -0
- pyfect/schema/_jsonschema/_exact.py +19 -0
- pyfect/schema/_jsonschema/_merge.py +211 -0
- pyfect/schema/_jsonschema/_objects.py +231 -0
- pyfect/schema/_jsonschema/_references.py +221 -0
- pyfect/schema/_jsonschema/_template.py +69 -0
- pyfect/schema/_literals.py +147 -0
- pyfect/schema/_middleware.py +209 -0
- pyfect/schema/_net.py +522 -0
- pyfect/schema/_options.py +49 -0
- pyfect/schema/_parse.py +351 -0
- pyfect/schema/_parsed_text.py +37 -0
- pyfect/schema/_primitives.py +48 -0
- pyfect/schema/_record.py +41 -0
- pyfect/schema/_refine.py +81 -0
- pyfect/schema/_schema.py +158 -0
- pyfect/schema/_semver.py +144 -0
- pyfect/schema/_strategy/__init__.py +258 -0
- pyfect/schema/_strategy/_big_decimal.py +203 -0
- pyfect/schema/_strategy/_composites.py +119 -0
- pyfect/schema/_strategy/_leaves.py +94 -0
- pyfect/schema/_strategy/_net.py +133 -0
- pyfect/schema/_struct.py +87 -0
- pyfect/schema/_struct_keys.py +128 -0
- pyfect/schema/_tagged.py +254 -0
- pyfect/schema/_template_literal.py +172 -0
- pyfect/schema/_to_codec_array_from_single.py +115 -0
- pyfect/schema/_to_codec_go_json.py +174 -0
- pyfect/schema/_to_codec_json.py +73 -0
- pyfect/schema/_to_codec_string_tree.py +115 -0
- pyfect/schema/_to_differ_json_patch.py +44 -0
- pyfect/schema/_to_encoder_xml.py +119 -0
- pyfect/schema/_to_equivalence.py +186 -0
- pyfect/schema/_to_formatter.py +180 -0
- pyfect/schema/_to_iso.py +73 -0
- pyfect/schema/_to_strategy.py +31 -0
- pyfect/schema/_transform.py +210 -0
- pyfect/schema/_tree.py +27 -0
- pyfect/schema/_type_walk.py +53 -0
- pyfect/schema/_undefined.py +58 -0
- pyfect/schema/_value_codecs.py +84 -0
- pyfect/schema/_values.py +123 -0
- pyfect/schema/_void.py +86 -0
- pyfect/schema/ast/__init__.py +227 -0
- pyfect/schema/ast/_annotations.py +43 -0
- pyfect/schema/ast/_arrays.py +210 -0
- pyfect/schema/ast/_base.py +295 -0
- pyfect/schema/ast/_candidates.py +309 -0
- pyfect/schema/ast/_check.py +181 -0
- pyfect/schema/ast/_class.py +159 -0
- pyfect/schema/ast/_constructors.py +65 -0
- pyfect/schema/ast/_declaration.py +97 -0
- pyfect/schema/ast/_encoding.py +111 -0
- pyfect/schema/ast/_json.py +129 -0
- pyfect/schema/ast/_keywords.py +125 -0
- pyfect/schema/ast/_literal.py +157 -0
- pyfect/schema/ast/_objects.py +356 -0
- pyfect/schema/ast/_record.py +36 -0
- pyfect/schema/ast/_result.py +116 -0
- pyfect/schema/ast/_steps.py +209 -0
- pyfect/schema/ast/_suspend.py +65 -0
- pyfect/schema/ast/_template_literal.py +256 -0
- pyfect/schema/ast/_union.py +172 -0
- pyfect/schema/getter/__init__.py +136 -0
- pyfect/schema/getter/_binary.py +104 -0
- pyfect/schema/getter/_coerce.py +196 -0
- pyfect/schema/getter/_getter.py +243 -0
- pyfect/schema/getter/_make.py +151 -0
- pyfect/schema/getter/_text.py +141 -0
- pyfect/schema/getter/_tree_record.py +170 -0
- pyfect/schema/issue.py +420 -0
- pyfect/schema/model/__init__.py +120 -0
- pyfect/schema/model/_class.py +245 -0
- pyfect/schema/model/_dates.py +103 -0
- pyfect/schema/model/_fields.py +77 -0
- pyfect/schema/model/_toolkit.py +114 -0
- pyfect/schema/model/_uuid.py +69 -0
- pyfect/schema/representation/__init__.py +424 -0
- pyfect/schema/representation/_builtin_revivers.py +331 -0
- pyfect/schema/representation/_code.py +106 -0
- pyfect/schema/representation/_code_builtins.py +281 -0
- pyfect/schema/representation/_code_document.py +563 -0
- pyfect/schema/representation/_code_sort.py +111 -0
- pyfect/schema/representation/_from.py +278 -0
- pyfect/schema/representation/_import/__init__.py +78 -0
- pyfect/schema/representation/_import/_finalize.py +121 -0
- pyfect/schema/representation/_import/_intersect.py +467 -0
- pyfect/schema/representation/_import/_js_order.py +66 -0
- pyfect/schema/representation/_import/_keywords.py +224 -0
- pyfect/schema/representation/_import/_nodes.py +199 -0
- pyfect/schema/representation/_import/_translate.py +455 -0
- pyfect/schema/representation/_json.py +520 -0
- pyfect/schema/representation/_json_schema.py +178 -0
- pyfect/schema/representation/_model.py +284 -0
- pyfect/schema/representation/_revivers.py +89 -0
- pyfect/schema/representation/_to.py +249 -0
- pyfect/schema/transformation/__init__.py +99 -0
- pyfect/schema/transformation/_builtins.py +314 -0
- pyfect/schema/transformation/_durations.py +150 -0
- pyfect/schema/transformation/_transformation.py +177 -0
- pyfect/schema/variant_schema/__init__.py +69 -0
- pyfect/schema/variant_schema/_class.py +188 -0
- pyfect/schema/variant_schema/_defaults.py +30 -0
- pyfect/schema/variant_schema/_field.py +32 -0
- pyfect/schema/variant_schema/_struct.py +130 -0
- pyfect/schema/variant_schema/_toolkit.py +160 -0
- pyfect/scope.py +374 -0
- pyfect/scoped_ref.py +124 -0
- pyfect/semaphore.py +276 -0
- pyfect/semver.py +403 -0
- pyfect/semver_range.py +420 -0
- pyfect/sink/__init__.py +131 -0
- pyfect/sink/_combinators.py +151 -0
- pyfect/sink/_constructors.py +187 -0
- pyfect/sink/_context.py +87 -0
- pyfect/sink/_folds.py +336 -0
- pyfect/sink/_for_each.py +87 -0
- pyfect/sink/_sink.py +265 -0
- pyfect/sink/_steps.py +34 -0
- pyfect/sink/_taking.py +183 -0
- pyfect/socket/__init__.py +146 -0
- pyfect/socket/_accepted.py +65 -0
- pyfect/socket/_connection.py +384 -0
- pyfect/socket/_errors.py +135 -0
- pyfect/socket/_loop.py +66 -0
- pyfect/socket/_net.py +281 -0
- pyfect/socket/_socket.py +245 -0
- pyfect/socket/_tls.py +88 -0
- pyfect/socket/_tls_session.py +117 -0
- pyfect/socket/_web_socket.py +555 -0
- pyfect/socket/datagram/__init__.py +110 -0
- pyfect/socket/datagram/_core.py +460 -0
- pyfect/socket/datagram/_errors.py +102 -0
- pyfect/socket/datagram/_live.py +675 -0
- pyfect/socket/datagram/_model.py +217 -0
- pyfect/socket/server.py +258 -0
- pyfect/socket/ws.py +484 -0
- pyfect/sql/__init__.py +60 -0
- pyfect/sql/_loop.py +134 -0
- pyfect/sql/clickhouse/__init__.py +74 -0
- pyfect/sql/clickhouse/_connection.py +237 -0
- pyfect/sql/clickhouse/_rows.py +87 -0
- pyfect/sql/clickhouse/client.py +322 -0
- pyfect/sql/clickhouse/config.py +241 -0
- pyfect/sql/clickhouse/driver.py +434 -0
- pyfect/sql/clickhouse/error.py +106 -0
- pyfect/sql/clickhouse/migrator.py +47 -0
- pyfect/sql/clickhouse/params.py +168 -0
- pyfect/sql/client/__init__.py +35 -0
- pyfect/sql/client/_client.py +309 -0
- pyfect/sql/client/_sqlite_acquirers.py +130 -0
- pyfect/sql/client/_transaction.py +410 -0
- pyfect/sql/connection.py +87 -0
- pyfect/sql/d1/__init__.py +48 -0
- pyfect/sql/d1/api.py +270 -0
- pyfect/sql/d1/client.py +344 -0
- pyfect/sql/d1/config.py +55 -0
- pyfect/sql/error.py +311 -0
- pyfect/sql/libsql/__init__.py +58 -0
- pyfect/sql/libsql/_connection.py +110 -0
- pyfect/sql/libsql/_hrana.py +258 -0
- pyfect/sql/libsql/_local.py +251 -0
- pyfect/sql/libsql/_targets.py +28 -0
- pyfect/sql/libsql/_values.py +46 -0
- pyfect/sql/libsql/client.py +225 -0
- pyfect/sql/libsql/config.py +170 -0
- pyfect/sql/libsql/migrator.py +43 -0
- pyfect/sql/libsql/sdk.py +123 -0
- pyfect/sql/migrator.py +434 -0
- pyfect/sql/model.py +324 -0
- pyfect/sql/mssql/__init__.py +79 -0
- pyfect/sql/mssql/_connection.py +169 -0
- pyfect/sql/mssql/_driver.py +103 -0
- pyfect/sql/mssql/_names.py +9 -0
- pyfect/sql/mssql/_pool.py +125 -0
- pyfect/sql/mssql/_tds.py +447 -0
- pyfect/sql/mssql/_tds_params.py +155 -0
- pyfect/sql/mssql/_worker.py +155 -0
- pyfect/sql/mssql/client.py +344 -0
- pyfect/sql/mssql/config.py +118 -0
- pyfect/sql/mssql/error.py +111 -0
- pyfect/sql/mssql/migrator.py +43 -0
- pyfect/sql/mssql/parameter.py +44 -0
- pyfect/sql/mssql/procedure.py +119 -0
- pyfect/sql/mssql/types.py +137 -0
- pyfect/sql/mysql/__init__.py +60 -0
- pyfect/sql/mysql/_asyncmy.py +168 -0
- pyfect/sql/mysql/_bind.py +76 -0
- pyfect/sql/mysql/_connection.py +156 -0
- pyfect/sql/mysql/_pool.py +63 -0
- pyfect/sql/mysql/_session.py +262 -0
- pyfect/sql/mysql/client.py +244 -0
- pyfect/sql/mysql/config.py +108 -0
- pyfect/sql/mysql/error.py +104 -0
- pyfect/sql/mysql/migrator.py +44 -0
- pyfect/sql/pg/__init__.py +82 -0
- pyfect/sql/pg/_resolve.py +284 -0
- pyfect/sql/pg/_session.py +380 -0
- pyfect/sql/pg/client.py +398 -0
- pyfect/sql/pg/config.py +142 -0
- pyfect/sql/pg/connection.py +585 -0
- pyfect/sql/pg/error.py +131 -0
- pyfect/sql/pg/migrator.py +139 -0
- pyfect/sql/pg/pool.py +265 -0
- pyfect/sql/resolver.py +294 -0
- pyfect/sql/result.py +24 -0
- pyfect/sql/schema.py +116 -0
- pyfect/sql/sqlite/__init__.py +26 -0
- pyfect/sql/sqlite/_connection.py +230 -0
- pyfect/sql/sqlite/_worker.py +71 -0
- pyfect/sql/sqlite/client.py +248 -0
- pyfect/sql/sqlite/migrator.py +43 -0
- pyfect/sql/statement/__init__.py +144 -0
- pyfect/sql/statement/_compiler.py +384 -0
- pyfect/sql/statement/_segments.py +487 -0
- pyfect/sql/statement/_statement.py +526 -0
- pyfect/sql/statement/_transforms.py +93 -0
- pyfect/sql/stream.py +88 -0
- pyfect/stdio.py +206 -0
- pyfect/stream/__init__.py +151 -0
- pyfect/stream/_async_iterable.py +129 -0
- pyfect/stream/_base.py +69 -0
- pyfect/stream/_bridge_methods.py +139 -0
- pyfect/stream/_catch.py +135 -0
- pyfect/stream/_catch_reason.py +558 -0
- pyfect/stream/_channel_methods.py +61 -0
- pyfect/stream/_chunk_methods.py +169 -0
- pyfect/stream/_chunking.py +310 -0
- pyfect/stream/_concat.py +65 -0
- pyfect/stream/_constructors.py +341 -0
- pyfect/stream/_context.py +147 -0
- pyfect/stream/_error_methods.py +254 -0
- pyfect/stream/_execution_plan.py +199 -0
- pyfect/stream/_fanout.py +154 -0
- pyfect/stream/_fanout_methods.py +183 -0
- pyfect/stream/_filter_methods.py +203 -0
- pyfect/stream/_grouping.py +168 -0
- pyfect/stream/_many.py +65 -0
- pyfect/stream/_map_methods.py +233 -0
- pyfect/stream/_mapping.py +126 -0
- pyfect/stream/_run.py +114 -0
- pyfect/stream/_run_methods.py +114 -0
- pyfect/stream/_service_methods.py +36 -0
- pyfect/stream/_sink_methods.py +151 -0
- pyfect/stream/_sinking.py +277 -0
- pyfect/stream/_sources.py +124 -0
- pyfect/stream/_stream.py +162 -0
- pyfect/stream/_taking.py +330 -0
- pyfect/stream/_text.py +43 -0
- pyfect/stream/_timing.py +294 -0
- pyfect/stream/_timing_methods.py +252 -0
- pyfect/stream/_zip_methods.py +279 -0
- pyfect/stream/_zipping.py +213 -0
- pyfect/subscription_ref.py +66 -0
- pyfect/synchronized_ref.py +38 -0
- pyfect/take.py +36 -0
- pyfect/terminal/__init__.py +41 -0
- pyfect/terminal/_keys.py +113 -0
- pyfect/terminal/_live.py +223 -0
- pyfect/terminal/_service.py +73 -0
- pyfect/testing/__init__.py +122 -0
- pyfect/testing/_assertions.py +92 -0
- pyfect/testing/_clock.py +309 -0
- pyfect/testing/_console.py +164 -0
- pyfect/testing/_crypto.py +44 -0
- pyfect/testing/_decorators.py +211 -0
- pyfect/testing/_flaky.py +34 -0
- pyfect/testing/_layers.py +153 -0
- pyfect/testing/_prop.py +255 -0
- pyfect/testing/_runners.py +84 -0
- pyfect/testing/_runtime.py +174 -0
- pyfect/testing/_schema.py +238 -0
- pyfect/testing/plugin.py +59 -0
- pyfect/tracer/__init__.py +107 -0
- pyfect/tracer/_model.py +176 -0
- pyfect/tracer/_native.py +172 -0
- pyfect/tracer/_refs.py +87 -0
- pyfect/tracer/_spans.py +269 -0
- pyfect/transaction.py +340 -0
- pyfect/trie.py +317 -0
- pyfect/tx_chunk.py +127 -0
- pyfect/tx_deferred.py +75 -0
- pyfect/tx_hash_map.py +265 -0
- pyfect/tx_hash_set.py +151 -0
- pyfect/tx_priority_queue.py +224 -0
- pyfect/tx_pubsub.py +194 -0
- pyfect/tx_queue/__init__.py +27 -0
- pyfect/tx_queue/_queue.py +359 -0
- pyfect/tx_queue/_views.py +60 -0
- pyfect/tx_reentrant_lock.py +201 -0
- pyfect/tx_ref.py +79 -0
- pyfect/tx_semaphore.py +136 -0
- pyfect/tx_subscription_ref.py +118 -0
- pyfect/workers/__init__.py +62 -0
- pyfect/workers/_frames.py +82 -0
- pyfect/workers/_mailbox.py +51 -0
- pyfect/workers/_port.py +106 -0
- pyfect/workers/_process.py +155 -0
- pyfect/workers/_thread.py +100 -0
- pyfect/workers/python_worker.py +133 -0
- pyfect/workers/python_worker_runner.py +112 -0
- pyfect/workers/transferable.py +103 -0
- pyfect/workers/worker.py +273 -0
- pyfect/workers/worker_error.py +86 -0
- pyfect/workers/worker_runner.py +85 -0
- pyfect/workflow/__init__.py +78 -0
- pyfect/workflow/_crypto.py +23 -0
- pyfect/workflow/_encoded.py +381 -0
- pyfect/workflow/_memory.py +424 -0
- pyfect/workflow/_result.py +43 -0
- pyfect/workflow/_services.py +241 -0
- pyfect/workflow/_suspension.py +153 -0
- pyfect/workflow/activity.py +427 -0
- pyfect/workflow/durable_clock.py +72 -0
- pyfect/workflow/durable_deferred.py +403 -0
- pyfect/workflow/durable_queue.py +259 -0
- pyfect/workflow/workflow.py +509 -0
- pyfect/workflow/workflow_engine.py +42 -0
- pyfect/workflow/workflow_proxy.py +161 -0
- pyfect/workflow/workflow_proxy_server.py +146 -0
- pyfect_dev-0.1.0.dist-info/METADATA +296 -0
- pyfect_dev-0.1.0.dist-info/RECORD +1265 -0
- pyfect_dev-0.1.0.dist-info/WHEEL +4 -0
- pyfect_dev-0.1.0.dist-info/entry_points.txt +3 -0
- pyfect_dev-0.1.0.dist-info/licenses/LICENSE +21 -0
|
@@ -0,0 +1,412 @@
|
|
|
1
|
+
"""The MCP client's transports: a child process on stdio, or Streamable
|
|
2
|
+
HTTP. A transport sends a request and waits for its response, sends a
|
|
3
|
+
notification, and says which tools it can call."""
|
|
4
|
+
|
|
5
|
+
from __future__ import annotations
|
|
6
|
+
|
|
7
|
+
from collections.abc import Callable, Mapping
|
|
8
|
+
from dataclasses import dataclass
|
|
9
|
+
from typing import Any, Literal, Never, cast
|
|
10
|
+
|
|
11
|
+
from pyfect import stream
|
|
12
|
+
from pyfect._core import Effect
|
|
13
|
+
from pyfect.ai._mcp.client_wire import (
|
|
14
|
+
cancelled,
|
|
15
|
+
encode_message,
|
|
16
|
+
failure,
|
|
17
|
+
header_parameters,
|
|
18
|
+
is_request,
|
|
19
|
+
is_response,
|
|
20
|
+
limit_error,
|
|
21
|
+
media_type,
|
|
22
|
+
parse_message,
|
|
23
|
+
reply_to_server,
|
|
24
|
+
routing_headers,
|
|
25
|
+
)
|
|
26
|
+
from pyfect.ai.mcp_client_error import (
|
|
27
|
+
ClosedError,
|
|
28
|
+
ConfigurationError,
|
|
29
|
+
HttpError,
|
|
30
|
+
McpClientError,
|
|
31
|
+
ProtocolError,
|
|
32
|
+
TransportError,
|
|
33
|
+
)
|
|
34
|
+
from pyfect.ai.mcp_protocol import ProtocolAdapter
|
|
35
|
+
from pyfect.ai.mcp_schema import JsonObject, RequestId, Tool, parse_request_id
|
|
36
|
+
from pyfect.byte_size import ByteSize
|
|
37
|
+
from pyfect.channel import LineTooLongError, split_lines_within
|
|
38
|
+
from pyfect.deferred import Deferred
|
|
39
|
+
from pyfect.duration import Seconds
|
|
40
|
+
from pyfect.effect import fail, gen, succeed, suspend, sync
|
|
41
|
+
from pyfect.encoding import sse
|
|
42
|
+
from pyfect.errors import Done
|
|
43
|
+
from pyfect.http import request as http_request
|
|
44
|
+
from pyfect.http.client import HttpClient
|
|
45
|
+
from pyfect.http.error import HttpClientError
|
|
46
|
+
from pyfect.http.response import HttpClientResponse, is_too_large, with_body_limit
|
|
47
|
+
from pyfect.process import ChildProcessSpawner, Command
|
|
48
|
+
from pyfect.queue import Queue
|
|
49
|
+
from pyfect.scope import Scope, add_finalizer
|
|
50
|
+
|
|
51
|
+
type ClientProtocolVersion = Literal["2025-11-25", "2026-07-28"]
|
|
52
|
+
type ClientProtocol = ProtocolAdapter[ClientProtocolVersion]
|
|
53
|
+
|
|
54
|
+
CLOSE_TIMEOUT = Seconds(5)
|
|
55
|
+
DEFAULT_MAX_MESSAGE_BYTES = 16 * 1024 * 1024
|
|
56
|
+
|
|
57
|
+
_VOID: Effect[None] = succeed(None)
|
|
58
|
+
|
|
59
|
+
|
|
60
|
+
@dataclass(frozen=True, slots=True)
|
|
61
|
+
class TransportService:
|
|
62
|
+
"""How a client reaches its server (Effect's ``McpClient.Transport``)."""
|
|
63
|
+
|
|
64
|
+
protocol: ClientProtocol
|
|
65
|
+
request: Callable[[JsonObject, Tool | None], Effect[JsonObject, McpClientError]]
|
|
66
|
+
notify: Callable[[JsonObject], Effect[None, McpClientError]]
|
|
67
|
+
is_callable: Callable[[Tool], bool]
|
|
68
|
+
|
|
69
|
+
|
|
70
|
+
def message_limit(max_message_bytes: int | None) -> Effect[ByteSize, McpClientError]:
|
|
71
|
+
limit = DEFAULT_MAX_MESSAGE_BYTES if max_message_bytes is None else max_message_bytes
|
|
72
|
+
if limit <= 0:
|
|
73
|
+
return fail(failure(ConfigurationError("maxMessageBytes must be a positive safe integer")))
|
|
74
|
+
return succeed(ByteSize(limit))
|
|
75
|
+
|
|
76
|
+
|
|
77
|
+
class _Closing:
|
|
78
|
+
"""The first error a transport closed with, failing every later call."""
|
|
79
|
+
|
|
80
|
+
def __init__(self) -> None:
|
|
81
|
+
self.failed: McpClientError | None = None
|
|
82
|
+
self.signal = Deferred[Never, McpClientError]()
|
|
83
|
+
|
|
84
|
+
def close(self, error: McpClientError) -> Effect[None]:
|
|
85
|
+
def run() -> Effect[None]:
|
|
86
|
+
if self.failed is None:
|
|
87
|
+
self.failed = error
|
|
88
|
+
return self.signal.fail(self.failed).as_void()
|
|
89
|
+
|
|
90
|
+
return suspend(run)
|
|
91
|
+
|
|
92
|
+
def check(self) -> Effect[None, McpClientError]:
|
|
93
|
+
return _VOID if self.failed is None else fail(self.failed)
|
|
94
|
+
|
|
95
|
+
|
|
96
|
+
def stdio(
|
|
97
|
+
*, command: Command, protocol: ClientProtocol, max_message_bytes: int | None = None
|
|
98
|
+
) -> Effect[TransportService, McpClientError, ChildProcessSpawner | Scope]:
|
|
99
|
+
"""A transport to ``command``'s process, on its standard input and
|
|
100
|
+
output. Closing the scope closes its input, then terminates it if it
|
|
101
|
+
has not exited within five seconds."""
|
|
102
|
+
|
|
103
|
+
@gen
|
|
104
|
+
def run():
|
|
105
|
+
max_bytes = yield from message_limit(max_message_bytes)
|
|
106
|
+
spawner = yield from ChildProcessSpawner()
|
|
107
|
+
process = yield from spawner.spawn(command).map_error(
|
|
108
|
+
lambda cause: failure(TransportError("Cannot spawn MCP server", cause))
|
|
109
|
+
)
|
|
110
|
+
pending: dict[RequestId, Deferred[JsonObject]] = {}
|
|
111
|
+
closing = _Closing()
|
|
112
|
+
lines = Queue[str, Done]()
|
|
113
|
+
|
|
114
|
+
def write(message: JsonObject) -> Effect[None, McpClientError]:
|
|
115
|
+
return closing.check().and_then(
|
|
116
|
+
encode_message(message, max_bytes).flat_map(
|
|
117
|
+
lambda text: lines.offer(text + "\n").as_void()
|
|
118
|
+
)
|
|
119
|
+
)
|
|
120
|
+
|
|
121
|
+
def receive(message: JsonObject) -> Effect[None, McpClientError]:
|
|
122
|
+
if is_request(message):
|
|
123
|
+
return write(reply_to_server(message))
|
|
124
|
+
if not is_response(message):
|
|
125
|
+
return _VOID
|
|
126
|
+
id = parse_request_id(message.get("id"))
|
|
127
|
+
if id is None:
|
|
128
|
+
return fail(
|
|
129
|
+
failure(
|
|
130
|
+
ProtocolError(description="Server returned an uncorrelated JSON-RPC error")
|
|
131
|
+
)
|
|
132
|
+
)
|
|
133
|
+
response = pending.get(id)
|
|
134
|
+
return _VOID if response is None else response.succeed(message).as_void()
|
|
135
|
+
|
|
136
|
+
def line(text: str) -> Effect[None, McpClientError]:
|
|
137
|
+
if not text.strip():
|
|
138
|
+
return _VOID
|
|
139
|
+
return parse_message(text, max_bytes).flat_map(receive)
|
|
140
|
+
|
|
141
|
+
def as_client_error(cause: object) -> McpClientError:
|
|
142
|
+
if isinstance(cause, McpClientError):
|
|
143
|
+
return cause
|
|
144
|
+
if isinstance(cause, LineTooLongError):
|
|
145
|
+
return limit_error("MCP message exceeds maxMessageBytes")
|
|
146
|
+
return failure(TransportError("MCP stdout failed", cause))
|
|
147
|
+
|
|
148
|
+
yield from (
|
|
149
|
+
stream.from_queue(lines)
|
|
150
|
+
.encode_text()
|
|
151
|
+
.run(process.stdin)
|
|
152
|
+
.catch_all(
|
|
153
|
+
lambda cause: closing.close(failure(TransportError("MCP stdin failed", cause)))
|
|
154
|
+
)
|
|
155
|
+
.fork_scoped()
|
|
156
|
+
)
|
|
157
|
+
yield from process.stderr.run_drain().ignore().fork_scoped()
|
|
158
|
+
drained = Deferred[None]()
|
|
159
|
+
yield from (
|
|
160
|
+
process.stdout.decode_text()
|
|
161
|
+
.pipe_through_channel(split_lines_within(max_bytes))
|
|
162
|
+
.run_for_each(line)
|
|
163
|
+
.map_error(as_client_error)
|
|
164
|
+
.and_then(fail(failure(TransportError("MCP server closed stdout"))))
|
|
165
|
+
.catch_all(closing.close)
|
|
166
|
+
.ensuring(drained.succeed(None))
|
|
167
|
+
.fork_scoped()
|
|
168
|
+
)
|
|
169
|
+
yield from (
|
|
170
|
+
process.exit_code.map(lambda code: f"MCP process exited ({code})")
|
|
171
|
+
.or_else_succeed(lambda _: "MCP process exited")
|
|
172
|
+
.tap(lambda _: drained.wait().timeout_option(CLOSE_TIMEOUT))
|
|
173
|
+
.flat_map(lambda description: closing.close(failure(TransportError(description))))
|
|
174
|
+
.fork_scoped()
|
|
175
|
+
)
|
|
176
|
+
|
|
177
|
+
@gen
|
|
178
|
+
def finalize():
|
|
179
|
+
yield from closing.close(failure(ClosedError(description="MCP transport closed")))
|
|
180
|
+
yield from lines.end()
|
|
181
|
+
exited = yield from process.exit_code.timeout_option(CLOSE_TIMEOUT).option()
|
|
182
|
+
if exited is None or exited.value is None:
|
|
183
|
+
yield from process.kill(force_kill_after=CLOSE_TIMEOUT).ignore()
|
|
184
|
+
|
|
185
|
+
yield from add_finalizer(lambda _: finalize)
|
|
186
|
+
|
|
187
|
+
def request(message: JsonObject, _tool: Tool | None) -> Effect[JsonObject, McpClientError]:
|
|
188
|
+
id = cast("RequestId", message["id"])
|
|
189
|
+
response = Deferred[JsonObject]()
|
|
190
|
+
|
|
191
|
+
def cancel() -> Effect[None]:
|
|
192
|
+
if message.get("method") == "initialize":
|
|
193
|
+
return _VOID
|
|
194
|
+
return response.is_done().flat_map(
|
|
195
|
+
lambda done: _VOID if done else write(cancelled(id)).ignore()
|
|
196
|
+
)
|
|
197
|
+
|
|
198
|
+
return (
|
|
199
|
+
sync(lambda: pending.__setitem__(id, response))
|
|
200
|
+
.and_then(write(message))
|
|
201
|
+
.and_then(response.wait())
|
|
202
|
+
.race_first(closing.signal.wait())
|
|
203
|
+
.on_interrupt(cancel())
|
|
204
|
+
.ensuring(sync(lambda: pending.pop(id, None)))
|
|
205
|
+
)
|
|
206
|
+
|
|
207
|
+
return TransportService(protocol, request, write, lambda _: True)
|
|
208
|
+
|
|
209
|
+
return run
|
|
210
|
+
|
|
211
|
+
|
|
212
|
+
def _read_failed(cause: HttpClientError) -> McpClientError:
|
|
213
|
+
if is_too_large(cause):
|
|
214
|
+
return limit_error("MCP message exceeds maxMessageBytes")
|
|
215
|
+
return failure(TransportError("Cannot read MCP response", cause))
|
|
216
|
+
|
|
217
|
+
|
|
218
|
+
def http( # noqa: PLR0915
|
|
219
|
+
*,
|
|
220
|
+
url: str,
|
|
221
|
+
protocol: ClientProtocol,
|
|
222
|
+
headers: Mapping[str, str] | None = None,
|
|
223
|
+
max_message_bytes: int | None = None,
|
|
224
|
+
) -> Effect[TransportService, McpClientError, HttpClient | Scope]:
|
|
225
|
+
"""A Streamable HTTP transport to ``url``. A 2025-11-25 session is
|
|
226
|
+
deleted when the scope closes; a server forgetting it fails every
|
|
227
|
+
later call with a ``ClosedError`` whose ``session_expired`` is set."""
|
|
228
|
+
|
|
229
|
+
@gen
|
|
230
|
+
def run(): # noqa: PLR0915
|
|
231
|
+
max_bytes = yield from message_limit(max_message_bytes)
|
|
232
|
+
client = yield from HttpClient()
|
|
233
|
+
modern = protocol.protocol_version == "2026-07-28"
|
|
234
|
+
closing = _Closing()
|
|
235
|
+
session: list[str] = []
|
|
236
|
+
base_headers = {
|
|
237
|
+
**(headers or {}),
|
|
238
|
+
"accept": "application/json, text/event-stream",
|
|
239
|
+
"mcp-protocol-version": protocol.protocol_version,
|
|
240
|
+
}
|
|
241
|
+
|
|
242
|
+
def current_headers() -> dict[str, str]:
|
|
243
|
+
return {**base_headers, "mcp-session-id": session[0]} if session else dict(base_headers)
|
|
244
|
+
|
|
245
|
+
def http_error(response: HttpClientResponse) -> Effect[Never, McpClientError]:
|
|
246
|
+
def with_remote(remote: Mapping[str, Any] | None) -> Effect[Never, McpClientError]:
|
|
247
|
+
return fail(
|
|
248
|
+
failure(
|
|
249
|
+
HttpError(
|
|
250
|
+
response.status,
|
|
251
|
+
code=None if remote is None else remote.get("code"),
|
|
252
|
+
data=None if remote is None else remote.get("data"),
|
|
253
|
+
description=None if remote is None else remote.get("message"),
|
|
254
|
+
www_authenticate=response.headers.get("www-authenticate"),
|
|
255
|
+
retry_after=response.headers.get("retry-after"),
|
|
256
|
+
)
|
|
257
|
+
)
|
|
258
|
+
)
|
|
259
|
+
|
|
260
|
+
if media_type(response.headers) != "application/json":
|
|
261
|
+
return with_remote(None)
|
|
262
|
+
remote = (
|
|
263
|
+
with_body_limit(response, max_bytes)
|
|
264
|
+
.text.map_error(lambda _: None)
|
|
265
|
+
.flat_map(lambda text: parse_message(text, max_bytes))
|
|
266
|
+
.map(lambda message: cast("Mapping[str, Any] | None", message.get("error")))
|
|
267
|
+
.or_else_succeed(lambda _: None)
|
|
268
|
+
)
|
|
269
|
+
return remote.flat_map(with_remote)
|
|
270
|
+
|
|
271
|
+
def send(
|
|
272
|
+
message: JsonObject, tool: Tool | None
|
|
273
|
+
) -> Effect[HttpClientResponse, McpClientError]:
|
|
274
|
+
@gen
|
|
275
|
+
def go():
|
|
276
|
+
yield from closing.check()
|
|
277
|
+
body = yield from encode_message(message, max_bytes)
|
|
278
|
+
routing = (yield from routing_headers(message, tool)) if modern else {}
|
|
279
|
+
request = (
|
|
280
|
+
http_request.post(url)
|
|
281
|
+
.set_headers({**current_headers(), **routing})
|
|
282
|
+
.body_text(body, "application/json")
|
|
283
|
+
)
|
|
284
|
+
response = yield from client.execute(request).map_error(
|
|
285
|
+
lambda cause: failure(TransportError("MCP HTTP request failed", cause))
|
|
286
|
+
)
|
|
287
|
+
if session and response.status == 404:
|
|
288
|
+
if closing.failed is None:
|
|
289
|
+
closing.failed = failure(ClosedError(session_expired=True))
|
|
290
|
+
return (yield from fail(closing.failed))
|
|
291
|
+
if not 200 <= response.status < 300:
|
|
292
|
+
return (yield from http_error(response))
|
|
293
|
+
assigned = response.headers.get("mcp-session-id")
|
|
294
|
+
if not modern and assigned is not None:
|
|
295
|
+
session[:] = [assigned]
|
|
296
|
+
return response
|
|
297
|
+
|
|
298
|
+
return go
|
|
299
|
+
|
|
300
|
+
def exchange[A](effect: Effect[A, McpClientError]) -> Effect[A, McpClientError]:
|
|
301
|
+
return effect.race_first(closing.signal.wait())
|
|
302
|
+
|
|
303
|
+
def notify(message: JsonObject) -> Effect[None, McpClientError]:
|
|
304
|
+
return exchange(send(message, None).as_void())
|
|
305
|
+
|
|
306
|
+
def match(
|
|
307
|
+
message: JsonObject, request: JsonObject
|
|
308
|
+
) -> Effect[JsonObject | None, McpClientError]:
|
|
309
|
+
if is_request(message):
|
|
310
|
+
return notify(reply_to_server(message)).as_(None)
|
|
311
|
+
if not is_response(message):
|
|
312
|
+
return succeed(None)
|
|
313
|
+
if message.get("id") != request["id"] and not (
|
|
314
|
+
"error" in message and message.get("id") is None
|
|
315
|
+
):
|
|
316
|
+
return fail(
|
|
317
|
+
failure(
|
|
318
|
+
ProtocolError(description="HTTP MCP response id does not match its request")
|
|
319
|
+
)
|
|
320
|
+
)
|
|
321
|
+
return succeed(message)
|
|
322
|
+
|
|
323
|
+
def read_response(
|
|
324
|
+
response: HttpClientResponse, request: JsonObject
|
|
325
|
+
) -> Effect[JsonObject, McpClientError]:
|
|
326
|
+
ended = failure(TransportError("MCP response ended before the terminal response"))
|
|
327
|
+
kind = media_type(response.headers)
|
|
328
|
+
if kind == "application/json":
|
|
329
|
+
return (
|
|
330
|
+
with_body_limit(response, max_bytes)
|
|
331
|
+
.text.map_error(_read_failed)
|
|
332
|
+
.flat_map(lambda text: parse_message(text, max_bytes))
|
|
333
|
+
.flat_map(lambda message: match(message, request))
|
|
334
|
+
.flat_map(lambda found: fail(ended) if found is None else succeed(found))
|
|
335
|
+
)
|
|
336
|
+
if kind == "text/event-stream":
|
|
337
|
+
events: list[str] = []
|
|
338
|
+
|
|
339
|
+
def on_event(event: object) -> None:
|
|
340
|
+
if isinstance(event, sse.Event) and event.data:
|
|
341
|
+
events.append(event.data)
|
|
342
|
+
|
|
343
|
+
parser = sse.make_parser(on_event, max_event_size=max_bytes)
|
|
344
|
+
|
|
345
|
+
def feed(chunk: str) -> stream.Stream[str, McpClientError]:
|
|
346
|
+
error = parser.feed(chunk)
|
|
347
|
+
parsed = stream.from_iterable(events[:])
|
|
348
|
+
events.clear()
|
|
349
|
+
if error is None:
|
|
350
|
+
return parsed
|
|
351
|
+
return parsed.concat(
|
|
352
|
+
stream.fail(limit_error("MCP SSE event exceeds maxMessageBytes"))
|
|
353
|
+
)
|
|
354
|
+
|
|
355
|
+
found = (
|
|
356
|
+
response.stream.map_error(_read_failed)
|
|
357
|
+
.decode_text()
|
|
358
|
+
.flat_map(feed)
|
|
359
|
+
.map_effect(
|
|
360
|
+
lambda data: parse_message(data, max_bytes).flat_map(
|
|
361
|
+
lambda message: match(message, request)
|
|
362
|
+
)
|
|
363
|
+
)
|
|
364
|
+
.filter(lambda message: message is not None)
|
|
365
|
+
.run_head()
|
|
366
|
+
)
|
|
367
|
+
return found.flat_map(
|
|
368
|
+
lambda head: (
|
|
369
|
+
fail(ended) if head is None else succeed(cast("JsonObject", head.value))
|
|
370
|
+
)
|
|
371
|
+
)
|
|
372
|
+
return fail(failure(ProtocolError(description="Unsupported MCP response Content-Type")))
|
|
373
|
+
|
|
374
|
+
@gen
|
|
375
|
+
def finalize():
|
|
376
|
+
if closing.failed is None:
|
|
377
|
+
closing.failed = failure(ClosedError(description="MCP HTTP transport closed"))
|
|
378
|
+
yield from closing.signal.fail(closing.failed)
|
|
379
|
+
reason = closing.failed.reason
|
|
380
|
+
if not session or not isinstance(reason, ClosedError) or reason.session_expired:
|
|
381
|
+
return None
|
|
382
|
+
delete = http_request.delete(url).set_headers(current_headers())
|
|
383
|
+
yield from client.execute(delete).timeout_option(CLOSE_TIMEOUT).ignore()
|
|
384
|
+
return None
|
|
385
|
+
|
|
386
|
+
yield from add_finalizer(lambda _: finalize)
|
|
387
|
+
|
|
388
|
+
def request(message: JsonObject, tool: Tool | None) -> Effect[JsonObject, McpClientError]:
|
|
389
|
+
def received(response: HttpClientResponse) -> Effect[JsonObject, McpClientError]:
|
|
390
|
+
if response.status in (202, 204):
|
|
391
|
+
return fail(
|
|
392
|
+
failure(ProtocolError(description="MCP request received no response"))
|
|
393
|
+
)
|
|
394
|
+
return read_response(response, message)
|
|
395
|
+
|
|
396
|
+
def cancel() -> Effect[None]:
|
|
397
|
+
if modern or message.get("method") == "initialize":
|
|
398
|
+
return _VOID
|
|
399
|
+
return (
|
|
400
|
+
notify(cancelled(cast("RequestId", message["id"])))
|
|
401
|
+
.timeout_option(CLOSE_TIMEOUT)
|
|
402
|
+
.ignore()
|
|
403
|
+
)
|
|
404
|
+
|
|
405
|
+
return exchange(send(message, tool).flat_map(received)).on_interrupt(cancel())
|
|
406
|
+
|
|
407
|
+
def is_callable(tool: Tool) -> bool:
|
|
408
|
+
return not modern or header_parameters(tool) is not None
|
|
409
|
+
|
|
410
|
+
return TransportService(protocol, request, notify, is_callable)
|
|
411
|
+
|
|
412
|
+
return run
|
|
@@ -0,0 +1,284 @@
|
|
|
1
|
+
"""What the MCP client transports share: message encoding and parsing
|
|
2
|
+
within the size limit, answers to server requests, routing headers and
|
|
3
|
+
the text a tool result shows a model."""
|
|
4
|
+
|
|
5
|
+
from __future__ import annotations
|
|
6
|
+
|
|
7
|
+
import base64
|
|
8
|
+
import re
|
|
9
|
+
from collections.abc import Mapping, Sequence
|
|
10
|
+
from dataclasses import dataclass
|
|
11
|
+
from typing import Any, cast
|
|
12
|
+
|
|
13
|
+
from pyfect import _json_text
|
|
14
|
+
from pyfect._core import Effect
|
|
15
|
+
from pyfect._js import text_encode
|
|
16
|
+
from pyfect._json_text import parse as parse_json_text
|
|
17
|
+
from pyfect.ai.mcp_client_error import (
|
|
18
|
+
AnyReason,
|
|
19
|
+
LimitError,
|
|
20
|
+
McpClientError,
|
|
21
|
+
ProtocolError,
|
|
22
|
+
TransportError,
|
|
23
|
+
UnsupportedError,
|
|
24
|
+
ValidationError,
|
|
25
|
+
)
|
|
26
|
+
from pyfect.ai.mcp_schema import (
|
|
27
|
+
METHOD_NOT_FOUND_ERROR_CODE,
|
|
28
|
+
CallToolResult,
|
|
29
|
+
JsonObject,
|
|
30
|
+
RequestId,
|
|
31
|
+
Tool,
|
|
32
|
+
)
|
|
33
|
+
from pyfect.effect import fail, succeed
|
|
34
|
+
|
|
35
|
+
BASE64_PREFIX = "=?base64?"
|
|
36
|
+
BASE64_SUFFIX = "?="
|
|
37
|
+
_HEADER_TOKEN = re.compile(r"^[!#$%&'*+\-.^_`|~0-9A-Za-z]+$")
|
|
38
|
+
_VISIBLE_ASCII = re.compile(r"^[\x20-\x7e]*$")
|
|
39
|
+
|
|
40
|
+
_SCHEMA_KEYWORDS = frozenset(
|
|
41
|
+
{
|
|
42
|
+
"additionalProperties",
|
|
43
|
+
"unevaluatedProperties",
|
|
44
|
+
"propertyNames",
|
|
45
|
+
"items",
|
|
46
|
+
"additionalItems",
|
|
47
|
+
"unevaluatedItems",
|
|
48
|
+
"contains",
|
|
49
|
+
"not",
|
|
50
|
+
"if",
|
|
51
|
+
"then",
|
|
52
|
+
"else",
|
|
53
|
+
"contentSchema",
|
|
54
|
+
}
|
|
55
|
+
)
|
|
56
|
+
_SCHEMA_MAP_KEYWORDS = frozenset(
|
|
57
|
+
{"properties", "patternProperties", "$defs", "definitions", "dependentSchemas", "dependencies"}
|
|
58
|
+
)
|
|
59
|
+
_SCHEMA_ARRAY_KEYWORDS = frozenset({"allOf", "anyOf", "oneOf", "prefixItems", "items"})
|
|
60
|
+
|
|
61
|
+
|
|
62
|
+
def failure(reason: AnyReason) -> McpClientError:
|
|
63
|
+
return McpClientError(reason)
|
|
64
|
+
|
|
65
|
+
|
|
66
|
+
def limit_error(description: str) -> McpClientError:
|
|
67
|
+
return failure(LimitError(description))
|
|
68
|
+
|
|
69
|
+
|
|
70
|
+
def encode_message(message: Mapping[str, Any], max_bytes: int) -> Effect[str, McpClientError]:
|
|
71
|
+
try:
|
|
72
|
+
text = _json_text.stringify(message)
|
|
73
|
+
except (TypeError, ValueError) as cause:
|
|
74
|
+
return fail(failure(TransportError("Cannot encode MCP message", cause)))
|
|
75
|
+
if len(text_encode(text)) > max_bytes:
|
|
76
|
+
return fail(limit_error("MCP message exceeds maxMessageBytes"))
|
|
77
|
+
return succeed(text)
|
|
78
|
+
|
|
79
|
+
|
|
80
|
+
def _is_message(value: object) -> bool:
|
|
81
|
+
if not isinstance(value, dict):
|
|
82
|
+
return False
|
|
83
|
+
body = cast("dict[str, object]", value)
|
|
84
|
+
if body.get("jsonrpc") != "2.0":
|
|
85
|
+
return False
|
|
86
|
+
if "method" in body:
|
|
87
|
+
return isinstance(body["method"], str)
|
|
88
|
+
return "result" in body or "error" in body
|
|
89
|
+
|
|
90
|
+
|
|
91
|
+
def parse_message(text: str, max_bytes: int) -> Effect[JsonObject, McpClientError]:
|
|
92
|
+
if len(text_encode(text)) > max_bytes:
|
|
93
|
+
return fail(limit_error("MCP message exceeds maxMessageBytes"))
|
|
94
|
+
try:
|
|
95
|
+
parsed: object = parse_json_text(text)
|
|
96
|
+
except ValueError as cause:
|
|
97
|
+
return fail(failure(ProtocolError(description="Invalid MCP JSON", cause=cause)))
|
|
98
|
+
if not _is_message(parsed):
|
|
99
|
+
return fail(failure(ProtocolError(description="Invalid JSON-RPC message")))
|
|
100
|
+
return succeed(cast("JsonObject", parsed))
|
|
101
|
+
|
|
102
|
+
|
|
103
|
+
def is_request(message: Mapping[str, Any]) -> bool:
|
|
104
|
+
return "method" in message and "id" in message
|
|
105
|
+
|
|
106
|
+
|
|
107
|
+
def is_response(message: Mapping[str, Any]) -> bool:
|
|
108
|
+
return "method" not in message
|
|
109
|
+
|
|
110
|
+
|
|
111
|
+
def reply_to_server(request: Mapping[str, Any]) -> JsonObject:
|
|
112
|
+
"""The client advertises no capabilities: it answers pings and
|
|
113
|
+
declines every other server request."""
|
|
114
|
+
id = request.get("id")
|
|
115
|
+
if request.get("method") == "ping":
|
|
116
|
+
return {"jsonrpc": "2.0", "id": id, "result": {}}
|
|
117
|
+
return {
|
|
118
|
+
"jsonrpc": "2.0",
|
|
119
|
+
"id": id,
|
|
120
|
+
"error": {
|
|
121
|
+
"code": METHOD_NOT_FOUND_ERROR_CODE,
|
|
122
|
+
"message": f"Unsupported MCP method: {request.get('method')}",
|
|
123
|
+
},
|
|
124
|
+
}
|
|
125
|
+
|
|
126
|
+
|
|
127
|
+
def cancelled(id: RequestId) -> JsonObject:
|
|
128
|
+
return {"jsonrpc": "2.0", "method": "notifications/cancelled", "params": {"requestId": id}}
|
|
129
|
+
|
|
130
|
+
|
|
131
|
+
def header_value(value: str) -> str:
|
|
132
|
+
"""``value`` as a header: as is when plain visible ASCII, else Base64
|
|
133
|
+
between ``=?base64?`` and ``?=``."""
|
|
134
|
+
plain = (
|
|
135
|
+
_VISIBLE_ASCII.match(value) is not None
|
|
136
|
+
and value.strip() == value
|
|
137
|
+
and not (value.startswith(BASE64_PREFIX) and value.endswith(BASE64_SUFFIX))
|
|
138
|
+
)
|
|
139
|
+
if plain:
|
|
140
|
+
return value
|
|
141
|
+
return f"{BASE64_PREFIX}{base64.b64encode(text_encode(value)).decode('ascii')}{BASE64_SUFFIX}"
|
|
142
|
+
|
|
143
|
+
|
|
144
|
+
@dataclass(frozen=True, slots=True)
|
|
145
|
+
class HeaderParameter:
|
|
146
|
+
path: tuple[str, ...]
|
|
147
|
+
header: str
|
|
148
|
+
type: str
|
|
149
|
+
|
|
150
|
+
|
|
151
|
+
def header_parameters(tool: Tool) -> list[HeaderParameter] | None:
|
|
152
|
+
"""The ``x-mcp-header`` annotations along chains of ``properties``, or
|
|
153
|
+
``None`` when one is invalid or found anywhere else."""
|
|
154
|
+
parameters: list[HeaderParameter] = []
|
|
155
|
+
names: set[str] = set()
|
|
156
|
+
|
|
157
|
+
def visit(schema: object, path: tuple[str, ...] | None) -> bool:
|
|
158
|
+
if not isinstance(schema, dict):
|
|
159
|
+
return True
|
|
160
|
+
node = cast("dict[str, object]", schema)
|
|
161
|
+
name = node.get("x-mcp-header")
|
|
162
|
+
if name is not None:
|
|
163
|
+
kind = node.get("type")
|
|
164
|
+
if (
|
|
165
|
+
not path
|
|
166
|
+
or not isinstance(name, str)
|
|
167
|
+
or _HEADER_TOKEN.match(name) is None
|
|
168
|
+
or name.lower() in names
|
|
169
|
+
or kind not in ("string", "integer", "boolean")
|
|
170
|
+
):
|
|
171
|
+
return False
|
|
172
|
+
names.add(name.lower())
|
|
173
|
+
parameters.append(HeaderParameter(path, f"mcp-param-{name.lower()}", cast("str", kind)))
|
|
174
|
+
for keyword, value in node.items():
|
|
175
|
+
if keyword in _SCHEMA_MAP_KEYWORDS and isinstance(value, dict):
|
|
176
|
+
children = cast("dict[str, object]", value)
|
|
177
|
+
for key, child in children.items():
|
|
178
|
+
child_path = (
|
|
179
|
+
(*path, key) if keyword == "properties" and path is not None else None
|
|
180
|
+
)
|
|
181
|
+
if not visit(child, child_path):
|
|
182
|
+
return False
|
|
183
|
+
elif isinstance(value, list):
|
|
184
|
+
if keyword in _SCHEMA_ARRAY_KEYWORDS and not all(
|
|
185
|
+
visit(child, None) for child in cast("list[object]", value)
|
|
186
|
+
):
|
|
187
|
+
return False
|
|
188
|
+
elif keyword in _SCHEMA_KEYWORDS and not visit(value, None):
|
|
189
|
+
return False
|
|
190
|
+
return True
|
|
191
|
+
|
|
192
|
+
return parameters if visit(tool["inputSchema"], ()) else None
|
|
193
|
+
|
|
194
|
+
|
|
195
|
+
def routing_headers(
|
|
196
|
+
message: Mapping[str, Any], tool: Tool | None
|
|
197
|
+
) -> Effect[dict[str, str], McpClientError]:
|
|
198
|
+
"""The 2026-07-28 HTTP routing headers of a request."""
|
|
199
|
+
method = message.get("method")
|
|
200
|
+
if not isinstance(method, str):
|
|
201
|
+
return succeed({})
|
|
202
|
+
headers = {"mcp-method": method}
|
|
203
|
+
params = cast("Mapping[str, Any]", message.get("params") or {})
|
|
204
|
+
name = params.get("uri") if method == "resources/read" else params.get("name")
|
|
205
|
+
if isinstance(name, str) and method in ("tools/call", "prompts/get", "resources/read"):
|
|
206
|
+
headers["mcp-name"] = header_value(name)
|
|
207
|
+
if method != "tools/call":
|
|
208
|
+
return succeed(headers)
|
|
209
|
+
parameters = None if tool is None else header_parameters(tool)
|
|
210
|
+
if parameters is None:
|
|
211
|
+
return fail(failure(UnsupportedError("Tool has invalid x-mcp-header annotations")))
|
|
212
|
+
for parameter in parameters:
|
|
213
|
+
value: object = params.get("arguments")
|
|
214
|
+
for key in parameter.path:
|
|
215
|
+
value = cast("dict[str, object]", value).get(key) if isinstance(value, dict) else None
|
|
216
|
+
if value is None:
|
|
217
|
+
continue
|
|
218
|
+
valid = (
|
|
219
|
+
isinstance(value, int) and not isinstance(value, bool)
|
|
220
|
+
if parameter.type == "integer"
|
|
221
|
+
else isinstance(value, bool)
|
|
222
|
+
if parameter.type == "boolean"
|
|
223
|
+
else isinstance(value, str)
|
|
224
|
+
)
|
|
225
|
+
if not valid:
|
|
226
|
+
return fail(
|
|
227
|
+
failure(
|
|
228
|
+
ValidationError(f"Invalid value for MCP header parameter {parameter.header}")
|
|
229
|
+
)
|
|
230
|
+
)
|
|
231
|
+
text = ("true" if value else "false") if isinstance(value, bool) else str(value)
|
|
232
|
+
headers[parameter.header] = header_value(text)
|
|
233
|
+
return succeed(headers)
|
|
234
|
+
|
|
235
|
+
|
|
236
|
+
def to_text(result: CallToolResult) -> Effect[str, McpClientError]:
|
|
237
|
+
"""What a model sees of a tool result: its text, resource links and
|
|
238
|
+
text resources, or the JSON of ``structuredContent`` when it has none."""
|
|
239
|
+
|
|
240
|
+
def unsupported(content: str) -> Effect[str, McpClientError]:
|
|
241
|
+
return fail(
|
|
242
|
+
failure(
|
|
243
|
+
UnsupportedError(f"MCP {content} requires an application text converter", result)
|
|
244
|
+
)
|
|
245
|
+
)
|
|
246
|
+
|
|
247
|
+
parts: list[str] = []
|
|
248
|
+
for block in result["content"]:
|
|
249
|
+
if block["type"] == "text":
|
|
250
|
+
parts.append(block["text"])
|
|
251
|
+
elif block["type"] == "resource_link":
|
|
252
|
+
label = block.get("title") or block["name"]
|
|
253
|
+
description = block.get("description")
|
|
254
|
+
parts.append(
|
|
255
|
+
f"{label}: {block['uri']}" + ("" if description is None else f"\n{description}")
|
|
256
|
+
)
|
|
257
|
+
elif block["type"] == "resource":
|
|
258
|
+
resource = block["resource"]
|
|
259
|
+
if "text" not in resource:
|
|
260
|
+
return unsupported("binary resources")
|
|
261
|
+
parts.append(cast("Any", resource)["text"])
|
|
262
|
+
else:
|
|
263
|
+
return unsupported(f"{block['type']} content")
|
|
264
|
+
if not parts and "structuredContent" in result:
|
|
265
|
+
return succeed(_json_text.stringify(result["structuredContent"]))
|
|
266
|
+
return succeed("\n".join(parts))
|
|
267
|
+
|
|
268
|
+
|
|
269
|
+
def requires_tasks(tool: object) -> bool:
|
|
270
|
+
if not isinstance(tool, dict):
|
|
271
|
+
return False
|
|
272
|
+
execution = cast("dict[str, object]", tool).get("execution")
|
|
273
|
+
return (
|
|
274
|
+
isinstance(execution, dict)
|
|
275
|
+
and cast("dict[str, object]", execution).get("taskSupport") == "required"
|
|
276
|
+
)
|
|
277
|
+
|
|
278
|
+
|
|
279
|
+
def media_type(headers: Mapping[str, str]) -> str | None:
|
|
280
|
+
content_type = headers.get("content-type")
|
|
281
|
+
return None if content_type is None else content_type.split(";", 1)[0].strip().lower()
|
|
282
|
+
|
|
283
|
+
|
|
284
|
+
__all__: Sequence[str] = ()
|