autobyteus 1.2.0__py3-none-any.whl → 1.2.3__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.
- autobyteus/agent/agent.py +15 -5
- autobyteus/agent/bootstrap_steps/__init__.py +1 -3
- autobyteus/agent/bootstrap_steps/agent_bootstrapper.py +3 -59
- autobyteus/agent/bootstrap_steps/base_bootstrap_step.py +1 -4
- autobyteus/agent/bootstrap_steps/mcp_server_prewarming_step.py +1 -3
- autobyteus/agent/bootstrap_steps/system_prompt_processing_step.py +16 -13
- autobyteus/agent/bootstrap_steps/workspace_context_initialization_step.py +2 -4
- autobyteus/agent/context/agent_config.py +43 -20
- autobyteus/agent/context/agent_context.py +23 -18
- autobyteus/agent/context/agent_runtime_state.py +23 -19
- autobyteus/agent/events/__init__.py +16 -1
- autobyteus/agent/events/agent_events.py +43 -3
- autobyteus/agent/events/agent_input_event_queue_manager.py +79 -26
- autobyteus/agent/events/event_store.py +57 -0
- autobyteus/agent/events/notifiers.py +74 -60
- autobyteus/agent/events/worker_event_dispatcher.py +21 -64
- autobyteus/agent/factory/agent_factory.py +52 -0
- autobyteus/agent/handlers/__init__.py +2 -0
- autobyteus/agent/handlers/approved_tool_invocation_event_handler.py +51 -34
- autobyteus/agent/handlers/bootstrap_event_handler.py +155 -0
- autobyteus/agent/handlers/inter_agent_message_event_handler.py +10 -0
- autobyteus/agent/handlers/lifecycle_event_logger.py +19 -11
- autobyteus/agent/handlers/llm_complete_response_received_event_handler.py +10 -15
- autobyteus/agent/handlers/llm_user_message_ready_event_handler.py +188 -48
- autobyteus/agent/handlers/tool_execution_approval_event_handler.py +0 -10
- autobyteus/agent/handlers/tool_invocation_request_event_handler.py +53 -48
- autobyteus/agent/handlers/tool_result_event_handler.py +7 -8
- autobyteus/agent/handlers/user_input_message_event_handler.py +10 -3
- autobyteus/agent/input_processor/memory_ingest_input_processor.py +40 -0
- autobyteus/agent/lifecycle/__init__.py +12 -0
- autobyteus/agent/lifecycle/base_processor.py +109 -0
- autobyteus/agent/lifecycle/events.py +35 -0
- autobyteus/agent/lifecycle/processor_definition.py +36 -0
- autobyteus/agent/lifecycle/processor_registry.py +106 -0
- autobyteus/agent/llm_request_assembler.py +98 -0
- autobyteus/agent/llm_response_processor/__init__.py +1 -8
- autobyteus/agent/message/context_file_type.py +1 -1
- autobyteus/agent/message/send_message_to.py +5 -4
- autobyteus/agent/runtime/agent_runtime.py +29 -21
- autobyteus/agent/runtime/agent_worker.py +98 -19
- autobyteus/agent/shutdown_steps/__init__.py +2 -0
- autobyteus/agent/shutdown_steps/agent_shutdown_orchestrator.py +2 -0
- autobyteus/agent/shutdown_steps/tool_cleanup_step.py +58 -0
- autobyteus/agent/status/__init__.py +14 -0
- autobyteus/agent/status/manager.py +93 -0
- autobyteus/agent/status/status_deriver.py +96 -0
- autobyteus/agent/{phases/phase_enum.py → status/status_enum.py} +16 -16
- autobyteus/agent/status/status_update_utils.py +73 -0
- autobyteus/agent/streaming/__init__.py +52 -5
- autobyteus/agent/streaming/adapters/__init__.py +18 -0
- autobyteus/agent/streaming/adapters/invocation_adapter.py +184 -0
- autobyteus/agent/streaming/adapters/tool_call_parsing.py +163 -0
- autobyteus/agent/streaming/adapters/tool_syntax_registry.py +67 -0
- autobyteus/agent/streaming/agent_event_stream.py +3 -178
- autobyteus/agent/streaming/api_tool_call/__init__.py +16 -0
- autobyteus/agent/streaming/api_tool_call/file_content_streamer.py +56 -0
- autobyteus/agent/streaming/api_tool_call/json_string_field_extractor.py +175 -0
- autobyteus/agent/streaming/api_tool_call_streaming_response_handler.py +4 -0
- autobyteus/agent/streaming/events/__init__.py +6 -0
- autobyteus/agent/streaming/events/stream_event_payloads.py +284 -0
- autobyteus/agent/streaming/events/stream_events.py +141 -0
- autobyteus/agent/streaming/handlers/__init__.py +15 -0
- autobyteus/agent/streaming/handlers/api_tool_call_streaming_response_handler.py +303 -0
- autobyteus/agent/streaming/handlers/parsing_streaming_response_handler.py +107 -0
- autobyteus/agent/streaming/handlers/pass_through_streaming_response_handler.py +107 -0
- autobyteus/agent/streaming/handlers/streaming_handler_factory.py +177 -0
- autobyteus/agent/streaming/handlers/streaming_response_handler.py +58 -0
- autobyteus/agent/streaming/parser/__init__.py +61 -0
- autobyteus/agent/streaming/parser/event_emitter.py +181 -0
- autobyteus/agent/streaming/parser/events.py +4 -0
- autobyteus/agent/streaming/parser/invocation_adapter.py +4 -0
- autobyteus/agent/streaming/parser/json_parsing_strategies/__init__.py +19 -0
- autobyteus/agent/streaming/parser/json_parsing_strategies/base.py +32 -0
- autobyteus/agent/streaming/parser/json_parsing_strategies/default.py +34 -0
- autobyteus/agent/streaming/parser/json_parsing_strategies/gemini.py +31 -0
- autobyteus/agent/streaming/parser/json_parsing_strategies/openai.py +64 -0
- autobyteus/agent/streaming/parser/json_parsing_strategies/registry.py +75 -0
- autobyteus/agent/streaming/parser/parser_context.py +227 -0
- autobyteus/agent/streaming/parser/parser_factory.py +132 -0
- autobyteus/agent/streaming/parser/sentinel_format.py +7 -0
- autobyteus/agent/streaming/parser/state_factory.py +62 -0
- autobyteus/agent/streaming/parser/states/__init__.py +1 -0
- autobyteus/agent/streaming/parser/states/base_state.py +60 -0
- autobyteus/agent/streaming/parser/states/custom_xml_tag_run_bash_parsing_state.py +38 -0
- autobyteus/agent/streaming/parser/states/custom_xml_tag_write_file_parsing_state.py +55 -0
- autobyteus/agent/streaming/parser/states/delimited_content_state.py +146 -0
- autobyteus/agent/streaming/parser/states/json_initialization_state.py +144 -0
- autobyteus/agent/streaming/parser/states/json_tool_parsing_state.py +137 -0
- autobyteus/agent/streaming/parser/states/sentinel_content_state.py +30 -0
- autobyteus/agent/streaming/parser/states/sentinel_initialization_state.py +117 -0
- autobyteus/agent/streaming/parser/states/text_state.py +78 -0
- autobyteus/agent/streaming/parser/states/xml_patch_file_tool_parsing_state.py +328 -0
- autobyteus/agent/streaming/parser/states/xml_run_bash_tool_parsing_state.py +129 -0
- autobyteus/agent/streaming/parser/states/xml_tag_initialization_state.py +151 -0
- autobyteus/agent/streaming/parser/states/xml_tool_parsing_state.py +63 -0
- autobyteus/agent/streaming/parser/states/xml_write_file_tool_parsing_state.py +343 -0
- autobyteus/agent/streaming/parser/strategies/__init__.py +17 -0
- autobyteus/agent/streaming/parser/strategies/base.py +24 -0
- autobyteus/agent/streaming/parser/strategies/json_tool_strategy.py +26 -0
- autobyteus/agent/streaming/parser/strategies/registry.py +28 -0
- autobyteus/agent/streaming/parser/strategies/sentinel_strategy.py +23 -0
- autobyteus/agent/streaming/parser/strategies/xml_tag_strategy.py +21 -0
- autobyteus/agent/streaming/parser/stream_scanner.py +167 -0
- autobyteus/agent/streaming/parser/streaming_parser.py +212 -0
- autobyteus/agent/streaming/parser/tool_call_parsing.py +4 -0
- autobyteus/agent/streaming/parser/tool_constants.py +7 -0
- autobyteus/agent/streaming/parser/tool_syntax_registry.py +4 -0
- autobyteus/agent/streaming/parser/xml_tool_parsing_state_registry.py +55 -0
- autobyteus/agent/streaming/parsing_streaming_response_handler.py +4 -0
- autobyteus/agent/streaming/pass_through_streaming_response_handler.py +4 -0
- autobyteus/agent/streaming/queue_streamer.py +3 -57
- autobyteus/agent/streaming/segments/__init__.py +5 -0
- autobyteus/agent/streaming/segments/segment_events.py +81 -0
- autobyteus/agent/streaming/stream_event_payloads.py +2 -198
- autobyteus/agent/streaming/stream_events.py +3 -128
- autobyteus/agent/streaming/streaming_handler_factory.py +4 -0
- autobyteus/agent/streaming/streaming_response_handler.py +4 -0
- autobyteus/agent/streaming/streams/__init__.py +5 -0
- autobyteus/agent/streaming/streams/agent_event_stream.py +197 -0
- autobyteus/agent/streaming/utils/__init__.py +5 -0
- autobyteus/agent/streaming/utils/queue_streamer.py +59 -0
- autobyteus/agent/system_prompt_processor/__init__.py +2 -0
- autobyteus/agent/system_prompt_processor/available_skills_processor.py +96 -0
- autobyteus/agent/system_prompt_processor/base_processor.py +1 -1
- autobyteus/agent/system_prompt_processor/processor_meta.py +15 -2
- autobyteus/agent/system_prompt_processor/tool_manifest_injector_processor.py +39 -58
- autobyteus/agent/token_budget.py +56 -0
- autobyteus/agent/tool_execution_result_processor/memory_ingest_tool_result_processor.py +29 -0
- autobyteus/agent/tool_invocation.py +16 -40
- autobyteus/agent/tool_invocation_preprocessor/__init__.py +9 -0
- autobyteus/agent/tool_invocation_preprocessor/base_preprocessor.py +45 -0
- autobyteus/agent/tool_invocation_preprocessor/processor_definition.py +15 -0
- autobyteus/agent/tool_invocation_preprocessor/processor_meta.py +33 -0
- autobyteus/agent/tool_invocation_preprocessor/processor_registry.py +60 -0
- autobyteus/agent/utils/wait_for_idle.py +12 -14
- autobyteus/agent/workspace/base_workspace.py +6 -27
- autobyteus/agent_team/agent_team.py +3 -3
- autobyteus/agent_team/agent_team_builder.py +1 -41
- autobyteus/agent_team/bootstrap_steps/__init__.py +0 -4
- autobyteus/agent_team/bootstrap_steps/agent_configuration_preparation_step.py +8 -18
- autobyteus/agent_team/bootstrap_steps/agent_team_bootstrapper.py +4 -16
- autobyteus/agent_team/bootstrap_steps/base_agent_team_bootstrap_step.py +1 -2
- autobyteus/agent_team/bootstrap_steps/coordinator_initialization_step.py +1 -2
- autobyteus/agent_team/bootstrap_steps/task_notifier_initialization_step.py +5 -6
- autobyteus/agent_team/bootstrap_steps/team_context_initialization_step.py +15 -15
- autobyteus/agent_team/context/agent_team_config.py +6 -3
- autobyteus/agent_team/context/agent_team_context.py +25 -3
- autobyteus/agent_team/context/agent_team_runtime_state.py +11 -8
- autobyteus/agent_team/events/__init__.py +11 -0
- autobyteus/agent_team/events/agent_team_event_dispatcher.py +22 -9
- autobyteus/agent_team/events/agent_team_events.py +16 -0
- autobyteus/agent_team/events/event_store.py +57 -0
- autobyteus/agent_team/factory/agent_team_factory.py +8 -0
- autobyteus/agent_team/handlers/inter_agent_message_request_event_handler.py +18 -2
- autobyteus/agent_team/handlers/lifecycle_agent_team_event_handler.py +21 -5
- autobyteus/agent_team/handlers/process_user_message_event_handler.py +17 -8
- autobyteus/agent_team/handlers/tool_approval_team_event_handler.py +19 -4
- autobyteus/agent_team/runtime/agent_team_runtime.py +41 -10
- autobyteus/agent_team/runtime/agent_team_worker.py +69 -5
- autobyteus/agent_team/status/__init__.py +14 -0
- autobyteus/agent_team/status/agent_team_status.py +18 -0
- autobyteus/agent_team/status/agent_team_status_manager.py +33 -0
- autobyteus/agent_team/status/status_deriver.py +62 -0
- autobyteus/agent_team/status/status_update_utils.py +42 -0
- autobyteus/agent_team/streaming/__init__.py +2 -2
- autobyteus/agent_team/streaming/agent_team_event_notifier.py +10 -10
- autobyteus/agent_team/streaming/agent_team_stream_event_payloads.py +7 -7
- autobyteus/agent_team/streaming/agent_team_stream_events.py +11 -11
- autobyteus/agent_team/system_prompt_processor/__init__.py +6 -0
- autobyteus/agent_team/system_prompt_processor/team_manifest_injector_processor.py +76 -0
- autobyteus/agent_team/task_notification/activation_policy.py +1 -1
- autobyteus/agent_team/task_notification/system_event_driven_agent_task_notifier.py +22 -22
- autobyteus/agent_team/task_notification/task_notification_mode.py +20 -1
- autobyteus/agent_team/utils/wait_for_idle.py +4 -4
- autobyteus/cli/agent_cli.py +18 -10
- autobyteus/cli/agent_team_tui/app.py +18 -15
- autobyteus/cli/agent_team_tui/state.py +21 -23
- autobyteus/cli/agent_team_tui/widgets/agent_list_sidebar.py +15 -15
- autobyteus/cli/agent_team_tui/widgets/focus_pane.py +146 -39
- autobyteus/cli/agent_team_tui/widgets/renderables.py +1 -1
- autobyteus/cli/agent_team_tui/widgets/shared.py +26 -26
- autobyteus/cli/agent_team_tui/widgets/{task_board_panel.py → task_plan_panel.py} +5 -5
- autobyteus/cli/cli_display.py +193 -44
- autobyteus/cli/workflow_tui/app.py +9 -10
- autobyteus/cli/workflow_tui/state.py +14 -16
- autobyteus/cli/workflow_tui/widgets/agent_list_sidebar.py +15 -15
- autobyteus/cli/workflow_tui/widgets/focus_pane.py +137 -35
- autobyteus/cli/workflow_tui/widgets/renderables.py +1 -1
- autobyteus/cli/workflow_tui/widgets/shared.py +25 -25
- autobyteus/clients/autobyteus_client.py +94 -1
- autobyteus/events/event_types.py +15 -21
- autobyteus/llm/api/autobyteus_llm.py +33 -29
- autobyteus/llm/api/claude_llm.py +142 -36
- autobyteus/llm/api/gemini_llm.py +163 -59
- autobyteus/llm/api/grok_llm.py +1 -1
- autobyteus/llm/api/minimax_llm.py +26 -0
- autobyteus/llm/api/mistral_llm.py +113 -87
- autobyteus/llm/api/ollama_llm.py +9 -42
- autobyteus/llm/api/openai_compatible_llm.py +127 -91
- autobyteus/llm/api/openai_llm.py +3 -3
- autobyteus/llm/api/openai_responses_llm.py +324 -0
- autobyteus/llm/api/zhipu_llm.py +21 -2
- autobyteus/llm/autobyteus_provider.py +70 -60
- autobyteus/llm/base_llm.py +85 -81
- autobyteus/llm/converters/__init__.py +14 -0
- autobyteus/llm/converters/anthropic_tool_call_converter.py +37 -0
- autobyteus/llm/converters/gemini_tool_call_converter.py +57 -0
- autobyteus/llm/converters/mistral_tool_call_converter.py +37 -0
- autobyteus/llm/converters/openai_tool_call_converter.py +38 -0
- autobyteus/llm/extensions/base_extension.py +6 -12
- autobyteus/llm/extensions/token_usage_tracking_extension.py +45 -18
- autobyteus/llm/llm_factory.py +282 -204
- autobyteus/llm/lmstudio_provider.py +60 -49
- autobyteus/llm/models.py +35 -2
- autobyteus/llm/ollama_provider.py +60 -49
- autobyteus/llm/ollama_provider_resolver.py +0 -1
- autobyteus/llm/prompt_renderers/__init__.py +19 -0
- autobyteus/llm/prompt_renderers/anthropic_prompt_renderer.py +104 -0
- autobyteus/llm/prompt_renderers/autobyteus_prompt_renderer.py +19 -0
- autobyteus/llm/prompt_renderers/base_prompt_renderer.py +10 -0
- autobyteus/llm/prompt_renderers/gemini_prompt_renderer.py +63 -0
- autobyteus/llm/prompt_renderers/mistral_prompt_renderer.py +87 -0
- autobyteus/llm/prompt_renderers/ollama_prompt_renderer.py +51 -0
- autobyteus/llm/prompt_renderers/openai_chat_renderer.py +97 -0
- autobyteus/llm/prompt_renderers/openai_responses_renderer.py +101 -0
- autobyteus/llm/providers.py +1 -3
- autobyteus/llm/token_counter/claude_token_counter.py +56 -25
- autobyteus/llm/token_counter/mistral_token_counter.py +12 -8
- autobyteus/llm/token_counter/openai_token_counter.py +24 -5
- autobyteus/llm/token_counter/token_counter_factory.py +12 -5
- autobyteus/llm/utils/llm_config.py +6 -12
- autobyteus/llm/utils/media_payload_formatter.py +27 -20
- autobyteus/llm/utils/messages.py +55 -3
- autobyteus/llm/utils/response_types.py +3 -0
- autobyteus/llm/utils/tool_call_delta.py +31 -0
- autobyteus/memory/__init__.py +32 -0
- autobyteus/memory/active_transcript.py +69 -0
- autobyteus/memory/compaction/__init__.py +9 -0
- autobyteus/memory/compaction/compaction_result.py +8 -0
- autobyteus/memory/compaction/compactor.py +89 -0
- autobyteus/memory/compaction/summarizer.py +11 -0
- autobyteus/memory/compaction_snapshot_builder.py +84 -0
- autobyteus/memory/memory_manager.py +183 -0
- autobyteus/memory/models/__init__.py +14 -0
- autobyteus/memory/models/episodic_item.py +41 -0
- autobyteus/memory/models/memory_types.py +7 -0
- autobyteus/memory/models/raw_trace_item.py +79 -0
- autobyteus/memory/models/semantic_item.py +41 -0
- autobyteus/memory/models/tool_interaction.py +20 -0
- autobyteus/memory/policies/__init__.py +5 -0
- autobyteus/memory/policies/compaction_policy.py +16 -0
- autobyteus/memory/retrieval/__init__.py +7 -0
- autobyteus/memory/retrieval/memory_bundle.py +11 -0
- autobyteus/memory/retrieval/retriever.py +13 -0
- autobyteus/memory/store/__init__.py +7 -0
- autobyteus/memory/store/base_store.py +14 -0
- autobyteus/memory/store/file_store.py +98 -0
- autobyteus/memory/tool_interaction_builder.py +46 -0
- autobyteus/memory/turn_tracker.py +9 -0
- autobyteus/multimedia/audio/api/__init__.py +3 -2
- autobyteus/multimedia/audio/api/autobyteus_audio_client.py +19 -5
- autobyteus/multimedia/audio/api/gemini_audio_client.py +108 -16
- autobyteus/multimedia/audio/api/openai_audio_client.py +112 -0
- autobyteus/multimedia/audio/audio_client_factory.py +84 -9
- autobyteus/multimedia/audio/audio_model.py +2 -1
- autobyteus/multimedia/image/api/autobyteus_image_client.py +19 -5
- autobyteus/multimedia/image/api/gemini_image_client.py +38 -17
- autobyteus/multimedia/image/api/openai_image_client.py +125 -43
- autobyteus/multimedia/image/autobyteus_image_provider.py +2 -1
- autobyteus/multimedia/image/image_client_factory.py +47 -15
- autobyteus/multimedia/image/image_model.py +5 -2
- autobyteus/multimedia/providers.py +3 -2
- autobyteus/skills/loader.py +71 -0
- autobyteus/skills/model.py +11 -0
- autobyteus/skills/registry.py +70 -0
- autobyteus/task_management/__init__.py +43 -20
- autobyteus/task_management/{base_task_board.py → base_task_plan.py} +16 -13
- autobyteus/task_management/converters/__init__.py +2 -2
- autobyteus/task_management/converters/{task_board_converter.py → task_plan_converter.py} +13 -13
- autobyteus/task_management/events.py +7 -7
- autobyteus/task_management/{in_memory_task_board.py → in_memory_task_plan.py} +34 -22
- autobyteus/task_management/schemas/__init__.py +3 -0
- autobyteus/task_management/schemas/task_status_report.py +2 -2
- autobyteus/task_management/schemas/todo_definition.py +15 -0
- autobyteus/task_management/todo.py +29 -0
- autobyteus/task_management/todo_list.py +75 -0
- autobyteus/task_management/tools/__init__.py +24 -8
- autobyteus/task_management/tools/task_tools/__init__.py +19 -0
- autobyteus/task_management/tools/{assign_task_to.py → task_tools/assign_task_to.py} +18 -18
- autobyteus/task_management/tools/{publish_task.py → task_tools/create_task.py} +16 -18
- autobyteus/task_management/tools/{publish_tasks.py → task_tools/create_tasks.py} +19 -19
- autobyteus/task_management/tools/{get_my_tasks.py → task_tools/get_my_tasks.py} +15 -15
- autobyteus/task_management/tools/{get_task_board_status.py → task_tools/get_task_plan_status.py} +16 -16
- autobyteus/task_management/tools/{update_task_status.py → task_tools/update_task_status.py} +16 -16
- autobyteus/task_management/tools/todo_tools/__init__.py +18 -0
- autobyteus/task_management/tools/todo_tools/add_todo.py +78 -0
- autobyteus/task_management/tools/todo_tools/create_todo_list.py +79 -0
- autobyteus/task_management/tools/todo_tools/get_todo_list.py +55 -0
- autobyteus/task_management/tools/todo_tools/update_todo_status.py +85 -0
- autobyteus/tools/__init__.py +43 -52
- autobyteus/tools/base_tool.py +7 -0
- autobyteus/tools/file/__init__.py +9 -0
- autobyteus/tools/file/patch_file.py +149 -0
- autobyteus/tools/file/{file_reader.py → read_file.py} +38 -7
- autobyteus/tools/file/{file_writer.py → write_file.py} +7 -4
- autobyteus/tools/functional_tool.py +53 -14
- autobyteus/tools/mcp/__init__.py +2 -0
- autobyteus/tools/mcp/config_service.py +5 -1
- autobyteus/tools/mcp/server/__init__.py +2 -0
- autobyteus/tools/mcp/server/http_managed_mcp_server.py +1 -1
- autobyteus/tools/mcp/server/websocket_managed_mcp_server.py +141 -0
- autobyteus/tools/mcp/server_instance_manager.py +8 -1
- autobyteus/tools/mcp/tool.py +3 -3
- autobyteus/tools/mcp/tool_registrar.py +5 -2
- autobyteus/tools/mcp/types.py +61 -0
- autobyteus/tools/multimedia/__init__.py +2 -1
- autobyteus/tools/multimedia/audio_tools.py +72 -19
- autobyteus/tools/{download_media_tool.py → multimedia/download_media_tool.py} +21 -7
- autobyteus/tools/multimedia/image_tools.py +248 -64
- autobyteus/tools/multimedia/media_reader_tool.py +1 -1
- autobyteus/tools/operation_executor/journal_manager.py +107 -0
- autobyteus/tools/operation_executor/operation_event_buffer.py +57 -0
- autobyteus/tools/operation_executor/operation_event_producer.py +29 -0
- autobyteus/tools/operation_executor/operation_executor.py +58 -0
- autobyteus/tools/registry/tool_definition.py +108 -14
- autobyteus/tools/registry/tool_registry.py +29 -0
- autobyteus/tools/search/__init__.py +17 -0
- autobyteus/tools/search/base_strategy.py +35 -0
- autobyteus/tools/search/client.py +24 -0
- autobyteus/tools/search/factory.py +81 -0
- autobyteus/tools/search/google_cse_strategy.py +68 -0
- autobyteus/tools/search/providers.py +10 -0
- autobyteus/tools/search/serpapi_strategy.py +65 -0
- autobyteus/tools/search/serper_strategy.py +87 -0
- autobyteus/tools/search_tool.py +83 -0
- autobyteus/tools/skill/load_skill.py +50 -0
- autobyteus/tools/terminal/__init__.py +45 -0
- autobyteus/tools/terminal/ansi_utils.py +32 -0
- autobyteus/tools/terminal/background_process_manager.py +233 -0
- autobyteus/tools/terminal/output_buffer.py +105 -0
- autobyteus/tools/terminal/prompt_detector.py +63 -0
- autobyteus/tools/terminal/pty_session.py +241 -0
- autobyteus/tools/terminal/session_factory.py +20 -0
- autobyteus/tools/terminal/terminal_session_manager.py +226 -0
- autobyteus/tools/terminal/tools/__init__.py +13 -0
- autobyteus/tools/terminal/tools/get_process_output.py +81 -0
- autobyteus/tools/terminal/tools/run_bash.py +109 -0
- autobyteus/tools/terminal/tools/start_background_process.py +104 -0
- autobyteus/tools/terminal/tools/stop_background_process.py +67 -0
- autobyteus/tools/terminal/types.py +54 -0
- autobyteus/tools/terminal/wsl_tmux_session.py +221 -0
- autobyteus/tools/terminal/wsl_utils.py +156 -0
- autobyteus/tools/tool_meta.py +4 -24
- autobyteus/tools/transaction_management/backup_handler.py +48 -0
- autobyteus/tools/transaction_management/operation_lifecycle_manager.py +62 -0
- autobyteus/tools/usage/__init__.py +1 -2
- autobyteus/tools/usage/formatters/__init__.py +17 -1
- autobyteus/tools/usage/formatters/base_formatter.py +8 -0
- autobyteus/tools/usage/formatters/default_xml_schema_formatter.py +2 -2
- autobyteus/tools/usage/formatters/mistral_json_schema_formatter.py +18 -0
- autobyteus/tools/usage/formatters/patch_file_xml_example_formatter.py +64 -0
- autobyteus/tools/usage/formatters/patch_file_xml_schema_formatter.py +31 -0
- autobyteus/tools/usage/formatters/run_bash_xml_example_formatter.py +32 -0
- autobyteus/tools/usage/formatters/run_bash_xml_schema_formatter.py +36 -0
- autobyteus/tools/usage/formatters/write_file_xml_example_formatter.py +53 -0
- autobyteus/tools/usage/formatters/write_file_xml_schema_formatter.py +31 -0
- autobyteus/tools/usage/providers/tool_manifest_provider.py +10 -10
- autobyteus/tools/usage/registries/__init__.py +1 -3
- autobyteus/tools/usage/registries/tool_formatting_registry.py +115 -8
- autobyteus/tools/usage/tool_schema_provider.py +51 -0
- autobyteus/tools/web/__init__.py +4 -0
- autobyteus/tools/web/read_url_tool.py +80 -0
- autobyteus/utils/diff_utils.py +271 -0
- autobyteus/utils/download_utils.py +109 -0
- autobyteus/utils/file_utils.py +57 -2
- autobyteus/utils/gemini_helper.py +56 -0
- autobyteus/utils/gemini_model_mapping.py +71 -0
- autobyteus/utils/llm_output_formatter.py +75 -0
- autobyteus/utils/tool_call_format.py +36 -0
- autobyteus/workflow/agentic_workflow.py +3 -3
- autobyteus/workflow/bootstrap_steps/agent_tool_injection_step.py +2 -2
- autobyteus/workflow/bootstrap_steps/base_workflow_bootstrap_step.py +2 -2
- autobyteus/workflow/bootstrap_steps/coordinator_initialization_step.py +2 -2
- autobyteus/workflow/bootstrap_steps/coordinator_prompt_preparation_step.py +4 -11
- autobyteus/workflow/bootstrap_steps/workflow_bootstrapper.py +6 -6
- autobyteus/workflow/bootstrap_steps/workflow_runtime_queue_initialization_step.py +2 -2
- autobyteus/workflow/context/workflow_context.py +3 -3
- autobyteus/workflow/context/workflow_runtime_state.py +5 -5
- autobyteus/workflow/events/workflow_event_dispatcher.py +5 -5
- autobyteus/workflow/handlers/lifecycle_workflow_event_handler.py +3 -3
- autobyteus/workflow/handlers/process_user_message_event_handler.py +5 -5
- autobyteus/workflow/handlers/tool_approval_workflow_event_handler.py +2 -2
- autobyteus/workflow/runtime/workflow_runtime.py +8 -8
- autobyteus/workflow/runtime/workflow_worker.py +3 -3
- autobyteus/workflow/status/__init__.py +11 -0
- autobyteus/workflow/status/workflow_status.py +19 -0
- autobyteus/workflow/status/workflow_status_manager.py +48 -0
- autobyteus/workflow/streaming/__init__.py +2 -2
- autobyteus/workflow/streaming/workflow_event_notifier.py +7 -7
- autobyteus/workflow/streaming/workflow_stream_event_payloads.py +4 -4
- autobyteus/workflow/streaming/workflow_stream_events.py +3 -3
- autobyteus/workflow/utils/wait_for_idle.py +4 -4
- autobyteus-1.2.3.dist-info/METADATA +293 -0
- autobyteus-1.2.3.dist-info/RECORD +600 -0
- {autobyteus-1.2.0.dist-info → autobyteus-1.2.3.dist-info}/WHEEL +1 -1
- {autobyteus-1.2.0.dist-info → autobyteus-1.2.3.dist-info}/top_level.txt +0 -1
- autobyteus/agent/bootstrap_steps/agent_runtime_queue_initialization_step.py +0 -57
- autobyteus/agent/hooks/__init__.py +0 -16
- autobyteus/agent/hooks/base_phase_hook.py +0 -78
- autobyteus/agent/hooks/hook_definition.py +0 -36
- autobyteus/agent/hooks/hook_meta.py +0 -37
- autobyteus/agent/hooks/hook_registry.py +0 -106
- autobyteus/agent/llm_response_processor/provider_aware_tool_usage_processor.py +0 -103
- autobyteus/agent/phases/__init__.py +0 -18
- autobyteus/agent/phases/discover.py +0 -53
- autobyteus/agent/phases/manager.py +0 -265
- autobyteus/agent/phases/transition_decorator.py +0 -40
- autobyteus/agent/phases/transition_info.py +0 -33
- autobyteus/agent/remote_agent.py +0 -244
- autobyteus/agent/workspace/workspace_definition.py +0 -36
- autobyteus/agent/workspace/workspace_meta.py +0 -37
- autobyteus/agent/workspace/workspace_registry.py +0 -72
- autobyteus/agent_team/bootstrap_steps/agent_team_runtime_queue_initialization_step.py +0 -25
- autobyteus/agent_team/bootstrap_steps/coordinator_prompt_preparation_step.py +0 -85
- autobyteus/agent_team/phases/__init__.py +0 -11
- autobyteus/agent_team/phases/agent_team_operational_phase.py +0 -19
- autobyteus/agent_team/phases/agent_team_phase_manager.py +0 -48
- autobyteus/llm/api/bedrock_llm.py +0 -92
- autobyteus/llm/api/groq_llm.py +0 -94
- autobyteus/llm/api/nvidia_llm.py +0 -108
- autobyteus/llm/utils/token_pricing_config.py +0 -87
- autobyteus/person/examples/sample_persons.py +0 -14
- autobyteus/person/examples/sample_roles.py +0 -14
- autobyteus/person/person.py +0 -29
- autobyteus/person/role.py +0 -14
- autobyteus/rpc/__init__.py +0 -73
- autobyteus/rpc/client/__init__.py +0 -17
- autobyteus/rpc/client/abstract_client_connection.py +0 -124
- autobyteus/rpc/client/client_connection_manager.py +0 -153
- autobyteus/rpc/client/sse_client_connection.py +0 -306
- autobyteus/rpc/client/stdio_client_connection.py +0 -280
- autobyteus/rpc/config/__init__.py +0 -13
- autobyteus/rpc/config/agent_server_config.py +0 -153
- autobyteus/rpc/config/agent_server_registry.py +0 -152
- autobyteus/rpc/hosting.py +0 -244
- autobyteus/rpc/protocol.py +0 -244
- autobyteus/rpc/server/__init__.py +0 -20
- autobyteus/rpc/server/agent_server_endpoint.py +0 -181
- autobyteus/rpc/server/base_method_handler.py +0 -40
- autobyteus/rpc/server/method_handlers.py +0 -259
- autobyteus/rpc/server/sse_server_handler.py +0 -182
- autobyteus/rpc/server/stdio_server_handler.py +0 -151
- autobyteus/rpc/server_main.py +0 -198
- autobyteus/rpc/transport_type.py +0 -13
- autobyteus/tools/bash/__init__.py +0 -2
- autobyteus/tools/bash/bash_executor.py +0 -100
- autobyteus/tools/browser/__init__.py +0 -2
- autobyteus/tools/browser/session_aware/__init__.py +0 -0
- autobyteus/tools/browser/session_aware/browser_session_aware_navigate_to.py +0 -75
- autobyteus/tools/browser/session_aware/browser_session_aware_tool.py +0 -30
- autobyteus/tools/browser/session_aware/browser_session_aware_web_element_trigger.py +0 -154
- autobyteus/tools/browser/session_aware/browser_session_aware_webpage_reader.py +0 -89
- autobyteus/tools/browser/session_aware/browser_session_aware_webpage_screenshot_taker.py +0 -107
- autobyteus/tools/browser/session_aware/factory/__init__.py +0 -0
- autobyteus/tools/browser/session_aware/factory/browser_session_aware_web_element_trigger_factory.py +0 -14
- autobyteus/tools/browser/session_aware/factory/browser_session_aware_webpage_reader_factory.py +0 -26
- autobyteus/tools/browser/session_aware/factory/browser_session_aware_webpage_screenshot_taker_factory.py +0 -14
- autobyteus/tools/browser/session_aware/shared_browser_session.py +0 -11
- autobyteus/tools/browser/session_aware/shared_browser_session_manager.py +0 -25
- autobyteus/tools/browser/session_aware/web_element_action.py +0 -20
- autobyteus/tools/browser/standalone/__init__.py +0 -6
- autobyteus/tools/browser/standalone/factory/__init__.py +0 -0
- autobyteus/tools/browser/standalone/factory/webpage_reader_factory.py +0 -25
- autobyteus/tools/browser/standalone/factory/webpage_screenshot_taker_factory.py +0 -14
- autobyteus/tools/browser/standalone/navigate_to.py +0 -80
- autobyteus/tools/browser/standalone/web_page_pdf_generator.py +0 -97
- autobyteus/tools/browser/standalone/webpage_image_downloader.py +0 -165
- autobyteus/tools/browser/standalone/webpage_reader.py +0 -101
- autobyteus/tools/browser/standalone/webpage_screenshot_taker.py +0 -101
- autobyteus/tools/file/file_editor.py +0 -200
- autobyteus/tools/google_search.py +0 -149
- autobyteus/tools/timer.py +0 -171
- autobyteus/tools/usage/parsers/__init__.py +0 -22
- autobyteus/tools/usage/parsers/_json_extractor.py +0 -99
- autobyteus/tools/usage/parsers/_string_decoders.py +0 -18
- autobyteus/tools/usage/parsers/anthropic_xml_tool_usage_parser.py +0 -10
- autobyteus/tools/usage/parsers/base_parser.py +0 -41
- autobyteus/tools/usage/parsers/default_json_tool_usage_parser.py +0 -83
- autobyteus/tools/usage/parsers/default_xml_tool_usage_parser.py +0 -316
- autobyteus/tools/usage/parsers/exceptions.py +0 -13
- autobyteus/tools/usage/parsers/gemini_json_tool_usage_parser.py +0 -77
- autobyteus/tools/usage/parsers/openai_json_tool_usage_parser.py +0 -149
- autobyteus/tools/usage/parsers/provider_aware_tool_usage_parser.py +0 -59
- autobyteus/tools/usage/registries/tool_usage_parser_registry.py +0 -62
- autobyteus/workflow/phases/__init__.py +0 -11
- autobyteus/workflow/phases/workflow_operational_phase.py +0 -19
- autobyteus/workflow/phases/workflow_phase_manager.py +0 -48
- autobyteus-1.2.0.dist-info/METADATA +0 -205
- autobyteus-1.2.0.dist-info/RECORD +0 -496
- examples/__init__.py +0 -1
- examples/agent_team/__init__.py +0 -1
- examples/discover_phase_transitions.py +0 -104
- examples/run_browser_agent.py +0 -262
- examples/run_google_slides_agent.py +0 -287
- examples/run_mcp_browser_client.py +0 -174
- examples/run_mcp_google_slides_client.py +0 -270
- examples/run_mcp_list_tools.py +0 -189
- examples/run_poem_writer.py +0 -284
- examples/run_sqlite_agent.py +0 -295
- /autobyteus/{person → skills}/__init__.py +0 -0
- /autobyteus/{person/examples → tools/skill}/__init__.py +0 -0
- {autobyteus-1.2.0.dist-info → autobyteus-1.2.3.dist-info}/licenses/LICENSE +0 -0
|
@@ -1,104 +0,0 @@
|
|
|
1
|
-
# file: autobyteus/examples/discover_phase_transitions.py
|
|
2
|
-
"""
|
|
3
|
-
This example script demonstrates how to use the PhaseTransitionDiscoverer
|
|
4
|
-
to programmatically find all valid phase transitions within the agent lifecycle.
|
|
5
|
-
|
|
6
|
-
This is useful for developers who want to create their own BasePhaseHook
|
|
7
|
-
subclasses, as it provides a definitive list of the source and target phases
|
|
8
|
-
they can hook into.
|
|
9
|
-
"""
|
|
10
|
-
import sys
|
|
11
|
-
from pathlib import Path
|
|
12
|
-
from typing import List, Dict, Any
|
|
13
|
-
|
|
14
|
-
# --- Boilerplate to make the script runnable from the project root ---
|
|
15
|
-
SCRIPT_DIR = Path(__file__).resolve().parent
|
|
16
|
-
PACKAGE_ROOT = SCRIPT_DIR.parent
|
|
17
|
-
if str(PACKAGE_ROOT) not in sys.path:
|
|
18
|
-
sys.path.insert(0, str(PACKAGE_ROOT))
|
|
19
|
-
|
|
20
|
-
try:
|
|
21
|
-
from autobyteus.agent.phases import PhaseTransitionDiscoverer, PhaseTransitionInfo
|
|
22
|
-
except ImportError as e:
|
|
23
|
-
print(f"Error importing autobyteus components: {e}", file=sys.stderr)
|
|
24
|
-
print("Please ensure that the autobyteus library is installed and accessible.", file=sys.stderr)
|
|
25
|
-
sys.exit(1)
|
|
26
|
-
|
|
27
|
-
|
|
28
|
-
def _prepare_table_data(transitions: List[PhaseTransitionInfo]) -> List[Dict[str, str]]:
|
|
29
|
-
"""Transforms transition info objects into a list of dictionaries for printing."""
|
|
30
|
-
table_data = []
|
|
31
|
-
for t in transitions:
|
|
32
|
-
from_str = "\n".join([p.value for p in t.source_phases])
|
|
33
|
-
table_data.append({
|
|
34
|
-
"From": from_str,
|
|
35
|
-
"To": t.target_phase.value,
|
|
36
|
-
"Trigger": f"AgentPhaseManager.{t.triggering_method}()",
|
|
37
|
-
"Description": t.description,
|
|
38
|
-
})
|
|
39
|
-
return table_data
|
|
40
|
-
|
|
41
|
-
def _print_as_table(data: List[Dict[str, str]]):
|
|
42
|
-
"""Calculates column widths and prints the data in a formatted table."""
|
|
43
|
-
if not data:
|
|
44
|
-
return
|
|
45
|
-
|
|
46
|
-
headers = list(data[0].keys())
|
|
47
|
-
|
|
48
|
-
# Calculate max widths for each column
|
|
49
|
-
widths = {h: len(h) for h in headers}
|
|
50
|
-
for row in data:
|
|
51
|
-
for h in headers:
|
|
52
|
-
# Check max width line-by-line for multiline content
|
|
53
|
-
max_line_width = max(len(line) for line in row[h].split('\n')) if row[h] else 0
|
|
54
|
-
widths[h] = max(widths[h], max_line_width)
|
|
55
|
-
|
|
56
|
-
# --- Print Header ---
|
|
57
|
-
header_line = " | ".join(h.ljust(widths[h]) for h in headers)
|
|
58
|
-
separator_line = "-+-".join("-" * widths[h] for h in headers)
|
|
59
|
-
print(header_line)
|
|
60
|
-
print(separator_line)
|
|
61
|
-
|
|
62
|
-
# --- Print Rows ---
|
|
63
|
-
for row in data:
|
|
64
|
-
# Split multiline content to handle rowspan alignment
|
|
65
|
-
split_rows = [
|
|
66
|
-
{h: row[h].split('\n') for h in headers}
|
|
67
|
-
]
|
|
68
|
-
|
|
69
|
-
num_lines = max(len(split_rows[0][h]) for h in headers)
|
|
70
|
-
|
|
71
|
-
for i in range(num_lines):
|
|
72
|
-
line_parts = []
|
|
73
|
-
for h in headers:
|
|
74
|
-
cell_lines = split_rows[0][h]
|
|
75
|
-
part = cell_lines[i] if i < len(cell_lines) else ""
|
|
76
|
-
line_parts.append(part.ljust(widths[h]))
|
|
77
|
-
print(" | ".join(line_parts))
|
|
78
|
-
|
|
79
|
-
# Print a thin separator between table rows for readability
|
|
80
|
-
print("-+-".join("-" * (w) for w in widths.values()))
|
|
81
|
-
|
|
82
|
-
|
|
83
|
-
def main():
|
|
84
|
-
"""
|
|
85
|
-
Discovers and prints all available agent phase transitions.
|
|
86
|
-
"""
|
|
87
|
-
print("--- Discovering all available agent phase transitions ---")
|
|
88
|
-
print("This table shows every possible transition you can create a custom hook for.\n")
|
|
89
|
-
|
|
90
|
-
all_transitions = PhaseTransitionDiscoverer.discover()
|
|
91
|
-
|
|
92
|
-
if not all_transitions:
|
|
93
|
-
print("No transitions were discovered. This is unexpected.")
|
|
94
|
-
return
|
|
95
|
-
|
|
96
|
-
# Prepare and print the data
|
|
97
|
-
table_data = _prepare_table_data(all_transitions)
|
|
98
|
-
_print_as_table(table_data)
|
|
99
|
-
|
|
100
|
-
print(f"\nTotal of {len(all_transitions)} unique transitions discovered.")
|
|
101
|
-
|
|
102
|
-
|
|
103
|
-
if __name__ == "__main__":
|
|
104
|
-
main()
|
examples/run_browser_agent.py
DELETED
|
@@ -1,262 +0,0 @@
|
|
|
1
|
-
# file: autobyteus/examples/run_browser_agent.py
|
|
2
|
-
import asyncio
|
|
3
|
-
import logging
|
|
4
|
-
import argparse
|
|
5
|
-
from pathlib import Path
|
|
6
|
-
import sys
|
|
7
|
-
import os
|
|
8
|
-
|
|
9
|
-
# --- Boilerplate to make the script runnable from the project root ---
|
|
10
|
-
SCRIPT_DIR = Path(__file__).resolve().parent
|
|
11
|
-
PACKAGE_ROOT = SCRIPT_DIR.parent
|
|
12
|
-
if str(PACKAGE_ROOT) not in sys.path:
|
|
13
|
-
sys.path.insert(0, str(PACKAGE_ROOT))
|
|
14
|
-
|
|
15
|
-
# Load environment variables from .env file in the project root
|
|
16
|
-
try:
|
|
17
|
-
from dotenv import load_dotenv
|
|
18
|
-
env_file_path = PACKAGE_ROOT / ".env"
|
|
19
|
-
if env_file_path.exists():
|
|
20
|
-
load_dotenv(env_file_path)
|
|
21
|
-
print(f"Loaded environment variables from: {env_file_path}")
|
|
22
|
-
else:
|
|
23
|
-
print(f"Info: No .env file found at: {env_file_path}. Relying on exported environment variables.")
|
|
24
|
-
except ImportError:
|
|
25
|
-
print("Warning: python-dotenv not installed. Cannot load .env file.")
|
|
26
|
-
|
|
27
|
-
# --- Imports for the Browser Agent Example ---
|
|
28
|
-
try:
|
|
29
|
-
# For MCP Tool Integration
|
|
30
|
-
from autobyteus.tools.mcp import McpToolRegistrar
|
|
31
|
-
from autobyteus.tools.registry import default_tool_registry
|
|
32
|
-
|
|
33
|
-
# For Agent creation
|
|
34
|
-
from autobyteus.agent.context.agent_config import AgentConfig
|
|
35
|
-
from autobyteus.llm.models import LLMModel
|
|
36
|
-
from autobyteus.llm.llm_factory import default_llm_factory, LLMFactory
|
|
37
|
-
from autobyteus.agent.factory.agent_factory import AgentFactory
|
|
38
|
-
from autobyteus.cli import agent_cli
|
|
39
|
-
except ImportError as e:
|
|
40
|
-
print(f"Error importing autobyteus components: {e}", file=sys.stderr)
|
|
41
|
-
print("Please ensure that the autobyteus library is installed and accessible.", file=sys.stderr)
|
|
42
|
-
sys.exit(1)
|
|
43
|
-
|
|
44
|
-
# --- Logging Setup ---
|
|
45
|
-
logger = logging.getLogger("browser_agent_example")
|
|
46
|
-
interactive_logger = logging.getLogger("autobyteus.cli.interactive")
|
|
47
|
-
|
|
48
|
-
def setup_logging(args: argparse.Namespace):
|
|
49
|
-
"""
|
|
50
|
-
Configures logging for the interactive session.
|
|
51
|
-
"""
|
|
52
|
-
loggers_to_clear = [
|
|
53
|
-
logging.getLogger(),
|
|
54
|
-
logging.getLogger("autobyteus"),
|
|
55
|
-
logging.getLogger("autobyteus.cli"),
|
|
56
|
-
interactive_logger,
|
|
57
|
-
]
|
|
58
|
-
for l in loggers_to_clear:
|
|
59
|
-
if l.hasHandlers():
|
|
60
|
-
for handler in l.handlers[:]:
|
|
61
|
-
l.removeHandler(handler)
|
|
62
|
-
if hasattr(handler, 'close'): handler.close()
|
|
63
|
-
|
|
64
|
-
script_log_level = logging.DEBUG if args.debug else logging.INFO
|
|
65
|
-
|
|
66
|
-
# 1. Handler for unformatted interactive output
|
|
67
|
-
interactive_handler = logging.StreamHandler(sys.stdout)
|
|
68
|
-
interactive_logger.addHandler(interactive_handler)
|
|
69
|
-
interactive_logger.setLevel(logging.INFO)
|
|
70
|
-
interactive_logger.propagate = False
|
|
71
|
-
|
|
72
|
-
# 2. Handler for formatted console logs
|
|
73
|
-
console_formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(name)s - %(message)s')
|
|
74
|
-
|
|
75
|
-
class FormattedConsoleFilter(logging.Filter):
|
|
76
|
-
def filter(self, record):
|
|
77
|
-
if record.name.startswith("browser_agent_example") or record.name.startswith("autobyteus.cli"):
|
|
78
|
-
return True
|
|
79
|
-
if record.levelno >= logging.CRITICAL:
|
|
80
|
-
return True
|
|
81
|
-
return False
|
|
82
|
-
|
|
83
|
-
formatted_console_handler = logging.StreamHandler(sys.stdout)
|
|
84
|
-
formatted_console_handler.setFormatter(console_formatter)
|
|
85
|
-
formatted_console_handler.addFilter(FormattedConsoleFilter())
|
|
86
|
-
|
|
87
|
-
root_logger = logging.getLogger()
|
|
88
|
-
root_logger.addHandler(formatted_console_handler)
|
|
89
|
-
root_logger.setLevel(script_log_level)
|
|
90
|
-
|
|
91
|
-
# 3. Handler for the main agent log file
|
|
92
|
-
log_file_path = Path(args.agent_log_file).resolve()
|
|
93
|
-
log_file_path.parent.mkdir(parents=True, exist_ok=True)
|
|
94
|
-
agent_file_handler = logging.FileHandler(log_file_path, mode='w')
|
|
95
|
-
agent_file_formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(name)s:%(lineno)d - %(message)s')
|
|
96
|
-
agent_file_handler.setFormatter(agent_file_formatter)
|
|
97
|
-
file_log_level = logging.DEBUG if args.debug else logging.INFO
|
|
98
|
-
|
|
99
|
-
autobyteus_logger = logging.getLogger("autobyteus")
|
|
100
|
-
autobyteus_logger.addHandler(agent_file_handler)
|
|
101
|
-
autobyteus_logger.setLevel(file_log_level)
|
|
102
|
-
autobyteus_logger.propagate = True
|
|
103
|
-
|
|
104
|
-
# 4. Isolate noisy queue manager logs to a separate file in debug mode
|
|
105
|
-
if args.debug:
|
|
106
|
-
queue_log_file_path = Path(log_file_path.parent / f"{log_file_path.stem}_queue.log").resolve()
|
|
107
|
-
|
|
108
|
-
queue_file_handler = logging.FileHandler(queue_log_file_path, mode='w')
|
|
109
|
-
queue_file_formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(name)s - %(message)s')
|
|
110
|
-
queue_file_handler.setFormatter(queue_file_formatter)
|
|
111
|
-
|
|
112
|
-
queue_logger = logging.getLogger("autobyteus.agent.events.agent_input_event_queue_manager")
|
|
113
|
-
|
|
114
|
-
queue_logger.setLevel(logging.DEBUG)
|
|
115
|
-
queue_logger.addHandler(queue_file_handler)
|
|
116
|
-
queue_logger.propagate = False # IMPORTANT: Stop logs from bubbling up to the main agent_logs.txt
|
|
117
|
-
|
|
118
|
-
logger.info(f"Debug mode: Redirecting noisy queue manager DEBUG logs to: {queue_log_file_path}")
|
|
119
|
-
|
|
120
|
-
# 5. Configure `autobyteus.cli` package logging
|
|
121
|
-
cli_logger = logging.getLogger("autobyteus.cli")
|
|
122
|
-
cli_logger.setLevel(script_log_level)
|
|
123
|
-
cli_logger.propagate = True
|
|
124
|
-
|
|
125
|
-
logger.info(f"Core library logs (excluding CLI) redirected to: {log_file_path} (level: {logging.getLevelName(file_log_level)})")
|
|
126
|
-
|
|
127
|
-
# --- Environment Variable Checks ---
|
|
128
|
-
def check_required_env_vars():
|
|
129
|
-
"""Checks for environment variables required by this example. None are required for the browser agent."""
|
|
130
|
-
# The browser MCP server is self-contained via npx and doesn't require env vars.
|
|
131
|
-
# This function is kept for structural consistency.
|
|
132
|
-
logger.info("No specific environment variables are required for the browser agent example.")
|
|
133
|
-
return {}
|
|
134
|
-
|
|
135
|
-
async def main(args: argparse.Namespace):
|
|
136
|
-
"""Main function to configure and run the BrowserAgent."""
|
|
137
|
-
logger.info("--- Starting Browser Agent Example ---")
|
|
138
|
-
check_required_env_vars()
|
|
139
|
-
|
|
140
|
-
# 1. Instantiate the core MCP and registry components.
|
|
141
|
-
tool_registry = default_tool_registry
|
|
142
|
-
registrar = McpToolRegistrar()
|
|
143
|
-
|
|
144
|
-
# 2. Define the configuration for the MCP server as a dictionary.
|
|
145
|
-
server_id = "browsermcp"
|
|
146
|
-
browser_mcp_config_dict = {
|
|
147
|
-
server_id: {
|
|
148
|
-
"transport_type": "stdio",
|
|
149
|
-
"stdio_params": {
|
|
150
|
-
"command": "npx",
|
|
151
|
-
"args": ["@browsermcp/mcp@latest"],
|
|
152
|
-
"env": {},
|
|
153
|
-
},
|
|
154
|
-
"enabled": True,
|
|
155
|
-
}
|
|
156
|
-
}
|
|
157
|
-
|
|
158
|
-
try:
|
|
159
|
-
# 3. Discover and register tools by passing the config dictionary directly.
|
|
160
|
-
logger.info(f"Performing targeted discovery for remote Browser tools from server: '{server_id}'...")
|
|
161
|
-
await registrar.discover_and_register_tools(mcp_config=browser_mcp_config_dict)
|
|
162
|
-
logger.info("Remote tool registration complete.")
|
|
163
|
-
|
|
164
|
-
# 4. Create tool instances from the registry for our agent.
|
|
165
|
-
# Use the ToolRegistry to get tools by their source server ID.
|
|
166
|
-
browser_tool_defs = tool_registry.get_tools_by_mcp_server(server_id)
|
|
167
|
-
browser_tool_names = [tool_def.name for tool_def in browser_tool_defs]
|
|
168
|
-
|
|
169
|
-
if not browser_tool_names:
|
|
170
|
-
logger.error(f"No Browser tools were found in the registry for server '{server_id}' after discovery. Cannot create agent.")
|
|
171
|
-
return
|
|
172
|
-
|
|
173
|
-
logger.info(f"Creating instances for registered Browser tools: {browser_tool_names}")
|
|
174
|
-
tools_for_agent = [tool_registry.create_tool(name) for name in browser_tool_names]
|
|
175
|
-
|
|
176
|
-
# 5. Configure and create the agent.
|
|
177
|
-
try:
|
|
178
|
-
_ = LLMModel[args.llm_model]
|
|
179
|
-
except (KeyError, ValueError):
|
|
180
|
-
logger.error(f"LLM Model '{args.llm_model}' is not valid or ambiguous.", file=sys.stderr)
|
|
181
|
-
try:
|
|
182
|
-
LLMFactory.ensure_initialized()
|
|
183
|
-
print("\nAvailable LLM Models (use the 'Identifier' with --llm-model):")
|
|
184
|
-
all_models = sorted(list(LLMModel), key=lambda m: m.model_identifier)
|
|
185
|
-
if not all_models:
|
|
186
|
-
print(" No models found.")
|
|
187
|
-
for model in all_models:
|
|
188
|
-
print(f" - Display Name: {model.name:<30} Identifier: {model.model_identifier}")
|
|
189
|
-
except Exception as e:
|
|
190
|
-
print(f"Additionally, an error occurred while listing models: {e}", file=sys.stderr)
|
|
191
|
-
sys.exit(1)
|
|
192
|
-
|
|
193
|
-
logger.info(f"Creating LLM instance for model: {args.llm_model}")
|
|
194
|
-
llm_instance = default_llm_factory.create_llm(model_identifier=args.llm_model)
|
|
195
|
-
|
|
196
|
-
system_prompt = (
|
|
197
|
-
"You are a helpful assistant that can browse the web to find information. "
|
|
198
|
-
"You have access to a set of specialized tools for this purpose.\n"
|
|
199
|
-
"When asked to perform a task on a browser, you should understand the user's intent and use the available tools to fulfill it.\n\n"
|
|
200
|
-
"Here is the manifest of tools available to you, including their definitions and examples:\n"
|
|
201
|
-
"{{tools}}"
|
|
202
|
-
)
|
|
203
|
-
|
|
204
|
-
browser_agent_config = AgentConfig(
|
|
205
|
-
name="BrowserAgent",
|
|
206
|
-
role="WebNavigator",
|
|
207
|
-
description="An agent that can browse the web using a set of remote tools.",
|
|
208
|
-
llm_instance=llm_instance,
|
|
209
|
-
system_prompt=system_prompt,
|
|
210
|
-
tools=tools_for_agent,
|
|
211
|
-
auto_execute_tools=False
|
|
212
|
-
)
|
|
213
|
-
|
|
214
|
-
agent = AgentFactory().create_agent(config=browser_agent_config)
|
|
215
|
-
logger.info(f"Browser Agent instance created: {agent.agent_id}")
|
|
216
|
-
|
|
217
|
-
# 6. Run the agent in an interactive CLI session.
|
|
218
|
-
logger.info(f"Starting interactive session for agent {agent.agent_id}...")
|
|
219
|
-
await agent_cli.run(agent=agent)
|
|
220
|
-
logger.info(f"Interactive session for agent {agent.agent_id} finished.")
|
|
221
|
-
|
|
222
|
-
except Exception as e:
|
|
223
|
-
logger.error(f"An error occurred during the agent workflow: {e}", exc_info=True)
|
|
224
|
-
|
|
225
|
-
logger.info("--- Browser Agent Example Finished ---")
|
|
226
|
-
|
|
227
|
-
if __name__ == "__main__":
|
|
228
|
-
parser = argparse.ArgumentParser(description="Run the BrowserAgent interactively.")
|
|
229
|
-
parser.add_argument("--llm-model", type=str, default="gemini-2.0-flash-", help=f"The LLM model identifier to use. Call --help-models for list.")
|
|
230
|
-
parser.add_argument("--help-models", action="store_true", help="Display available LLM models and exit.")
|
|
231
|
-
parser.add_argument("--debug", action="store_true", help="Enable debug logging.")
|
|
232
|
-
parser.add_argument("--agent-log-file", type=str, default="./agent_logs_browser.txt",
|
|
233
|
-
help="Path to the log file for autobyteus.* library logs. (Default: ./agent_logs_browser.txt)")
|
|
234
|
-
parser.add_argument("--no-tool-logs", action="store_true",
|
|
235
|
-
help="Disable display of [Tool Log (...)] messages on the console by the agent_cli.")
|
|
236
|
-
|
|
237
|
-
if "--help-models" in sys.argv:
|
|
238
|
-
try:
|
|
239
|
-
LLMFactory.ensure_initialized()
|
|
240
|
-
print("Available LLM Models (use the 'Identifier' with --llm-model):")
|
|
241
|
-
all_models = sorted(list(LLMModel), key=lambda m: m.model_identifier)
|
|
242
|
-
if not all_models:
|
|
243
|
-
print(" No models found.")
|
|
244
|
-
for model in all_models:
|
|
245
|
-
print(f" - Display Name: {model.name:<30} Identifier: {model.model_identifier}")
|
|
246
|
-
except Exception as e:
|
|
247
|
-
print(f"Error listing models: {e}")
|
|
248
|
-
sys.exit(0)
|
|
249
|
-
|
|
250
|
-
parsed_args = parser.parse_args()
|
|
251
|
-
|
|
252
|
-
setup_logging(parsed_args)
|
|
253
|
-
check_required_env_vars()
|
|
254
|
-
|
|
255
|
-
try:
|
|
256
|
-
asyncio.run(main(parsed_args))
|
|
257
|
-
except (KeyboardInterrupt, SystemExit):
|
|
258
|
-
logger.info("Script interrupted by user. Exiting.")
|
|
259
|
-
except Exception as e:
|
|
260
|
-
logger.error(f"An unhandled error occurred at the top level: {e}", exc_info=True)
|
|
261
|
-
finally:
|
|
262
|
-
logger.info("Exiting script.")
|
|
@@ -1,287 +0,0 @@
|
|
|
1
|
-
# file: autobyteus/examples/run_google_slides_agent.py
|
|
2
|
-
import asyncio
|
|
3
|
-
import logging
|
|
4
|
-
import argparse
|
|
5
|
-
from pathlib import Path
|
|
6
|
-
import sys
|
|
7
|
-
import os
|
|
8
|
-
|
|
9
|
-
# --- Boilerplate to make the script runnable from the project root ---
|
|
10
|
-
SCRIPT_DIR = Path(__file__).resolve().parent
|
|
11
|
-
PACKAGE_ROOT = SCRIPT_DIR.parent
|
|
12
|
-
if str(PACKAGE_ROOT) not in sys.path:
|
|
13
|
-
sys.path.insert(0, str(PACKAGE_ROOT))
|
|
14
|
-
|
|
15
|
-
# Load environment variables from .env file in the project root
|
|
16
|
-
try:
|
|
17
|
-
from dotenv import load_dotenv
|
|
18
|
-
env_file_path = PACKAGE_ROOT / ".env"
|
|
19
|
-
if env_file_path.exists():
|
|
20
|
-
load_dotenv(env_file_path)
|
|
21
|
-
print(f"Loaded environment variables from: {env_file_path}")
|
|
22
|
-
else:
|
|
23
|
-
print(f"Info: No .env file found at: {env_file_path}. Relying on exported environment variables.")
|
|
24
|
-
except ImportError:
|
|
25
|
-
print("Warning: python-dotenv not installed. Cannot load .env file.")
|
|
26
|
-
|
|
27
|
-
# --- Imports for the Google Slides Agent Example ---
|
|
28
|
-
try:
|
|
29
|
-
# For MCP Tool Integration
|
|
30
|
-
from autobyteus.tools.mcp import McpToolRegistrar
|
|
31
|
-
from autobyteus.tools.registry import default_tool_registry
|
|
32
|
-
|
|
33
|
-
# For Agent creation
|
|
34
|
-
from autobyteus.agent.context.agent_config import AgentConfig
|
|
35
|
-
from autobyteus.llm.models import LLMModel
|
|
36
|
-
from autobyteus.llm.llm_factory import default_llm_factory, LLMFactory
|
|
37
|
-
from autobyteus.agent.factory.agent_factory import AgentFactory
|
|
38
|
-
from autobyteus.cli import agent_cli
|
|
39
|
-
except ImportError as e:
|
|
40
|
-
print(f"Error importing autobyteus components: {e}", file=sys.stderr)
|
|
41
|
-
print("Please ensure that the autobyteus library is installed and accessible.", file=sys.stderr)
|
|
42
|
-
sys.exit(1)
|
|
43
|
-
|
|
44
|
-
# --- Logging Setup ---
|
|
45
|
-
logger = logging.getLogger("google_slides_agent_example")
|
|
46
|
-
interactive_logger = logging.getLogger("autobyteus.cli.interactive")
|
|
47
|
-
|
|
48
|
-
def setup_logging(args: argparse.Namespace):
|
|
49
|
-
"""
|
|
50
|
-
Configures logging for the interactive session.
|
|
51
|
-
"""
|
|
52
|
-
loggers_to_clear = [
|
|
53
|
-
logging.getLogger(),
|
|
54
|
-
logging.getLogger("autobyteus"),
|
|
55
|
-
logging.getLogger("autobyteus.cli"),
|
|
56
|
-
interactive_logger,
|
|
57
|
-
]
|
|
58
|
-
for l in loggers_to_clear:
|
|
59
|
-
if l.hasHandlers():
|
|
60
|
-
for handler in l.handlers[:]:
|
|
61
|
-
l.removeHandler(handler)
|
|
62
|
-
if hasattr(handler, 'close'): handler.close()
|
|
63
|
-
|
|
64
|
-
script_log_level = logging.DEBUG if args.debug else logging.INFO
|
|
65
|
-
|
|
66
|
-
# 1. Handler for unformatted interactive output
|
|
67
|
-
interactive_handler = logging.StreamHandler(sys.stdout)
|
|
68
|
-
interactive_logger.addHandler(interactive_handler)
|
|
69
|
-
interactive_logger.setLevel(logging.INFO)
|
|
70
|
-
interactive_logger.propagate = False
|
|
71
|
-
|
|
72
|
-
# 2. Handler for formatted console logs
|
|
73
|
-
console_formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(name)s - %(message)s')
|
|
74
|
-
|
|
75
|
-
class FormattedConsoleFilter(logging.Filter):
|
|
76
|
-
def filter(self, record):
|
|
77
|
-
if record.name.startswith("google_slides_agent_example") or record.name.startswith("autobyteus.cli"):
|
|
78
|
-
return True
|
|
79
|
-
if record.levelno >= logging.CRITICAL:
|
|
80
|
-
return True
|
|
81
|
-
return False
|
|
82
|
-
|
|
83
|
-
formatted_console_handler = logging.StreamHandler(sys.stdout)
|
|
84
|
-
formatted_console_handler.setFormatter(console_formatter)
|
|
85
|
-
formatted_console_handler.addFilter(FormattedConsoleFilter())
|
|
86
|
-
|
|
87
|
-
root_logger = logging.getLogger()
|
|
88
|
-
root_logger.addHandler(formatted_console_handler)
|
|
89
|
-
root_logger.setLevel(script_log_level)
|
|
90
|
-
|
|
91
|
-
# 3. Handler for the main agent log file
|
|
92
|
-
log_file_path = Path(args.agent_log_file).resolve()
|
|
93
|
-
log_file_path.parent.mkdir(parents=True, exist_ok=True)
|
|
94
|
-
agent_file_handler = logging.FileHandler(log_file_path, mode='w')
|
|
95
|
-
agent_file_formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(name)s:%(lineno)d - %(message)s')
|
|
96
|
-
agent_file_handler.setFormatter(agent_file_formatter)
|
|
97
|
-
file_log_level = logging.DEBUG if args.debug else logging.INFO
|
|
98
|
-
|
|
99
|
-
autobyteus_logger = logging.getLogger("autobyteus")
|
|
100
|
-
autobyteus_logger.addHandler(agent_file_handler)
|
|
101
|
-
autobyteus_logger.setLevel(file_log_level)
|
|
102
|
-
autobyteus_logger.propagate = True
|
|
103
|
-
|
|
104
|
-
# 4. Isolate noisy queue manager logs to a separate file in debug mode
|
|
105
|
-
if args.debug:
|
|
106
|
-
queue_log_file_path = Path(log_file_path.parent / f"{log_file_path.stem}_queue.log").resolve()
|
|
107
|
-
|
|
108
|
-
queue_file_handler = logging.FileHandler(queue_log_file_path, mode='w')
|
|
109
|
-
queue_file_formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(name)s - %(message)s')
|
|
110
|
-
queue_file_handler.setFormatter(queue_file_formatter)
|
|
111
|
-
|
|
112
|
-
queue_logger = logging.getLogger("autobyteus.agent.events.agent_input_event_queue_manager")
|
|
113
|
-
|
|
114
|
-
queue_logger.setLevel(logging.DEBUG)
|
|
115
|
-
queue_logger.addHandler(queue_file_handler)
|
|
116
|
-
queue_logger.propagate = False # IMPORTANT: Stop logs from bubbling up to the main agent_logs.txt
|
|
117
|
-
|
|
118
|
-
logger.info(f"Debug mode: Redirecting noisy queue manager DEBUG logs to: {queue_log_file_path}")
|
|
119
|
-
|
|
120
|
-
# 5. Configure `autobyteus.cli` package logging
|
|
121
|
-
cli_logger = logging.getLogger("autobyteus.cli")
|
|
122
|
-
cli_logger.setLevel(script_log_level)
|
|
123
|
-
cli_logger.propagate = True
|
|
124
|
-
|
|
125
|
-
logger.info(f"Core library logs (excluding CLI) redirected to: {log_file_path} (level: {logging.getLevelName(file_log_level)})")
|
|
126
|
-
|
|
127
|
-
# --- Environment Variable Checks ---
|
|
128
|
-
def check_required_env_vars():
|
|
129
|
-
"""Checks for environment variables required by this example and returns them."""
|
|
130
|
-
required_vars = {
|
|
131
|
-
"script_path": "TEST_GOOGLE_SLIDES_MCP_SCRIPT_PATH",
|
|
132
|
-
"google_client_id": "GOOGLE_CLIENT_ID",
|
|
133
|
-
"google_client_secret": "GOOGLE_CLIENT_SECRET",
|
|
134
|
-
"google_refresh_token": "GOOGLE_REFRESH_TOKEN",
|
|
135
|
-
}
|
|
136
|
-
env_values = {}
|
|
137
|
-
missing_vars = []
|
|
138
|
-
for key, var_name in required_vars.items():
|
|
139
|
-
value = os.environ.get(var_name)
|
|
140
|
-
if not value:
|
|
141
|
-
missing_vars.append(var_name)
|
|
142
|
-
else:
|
|
143
|
-
env_values[key] = value
|
|
144
|
-
if missing_vars:
|
|
145
|
-
logger.error("This example requires the following environment variables to be set: %s", missing_vars)
|
|
146
|
-
sys.exit(1)
|
|
147
|
-
if not Path(env_values["script_path"]).exists():
|
|
148
|
-
logger.error(f"The script path specified by TEST_GOOGLE_SLIDES_MCP_SCRIPT_PATH does not exist: {env_values['script_path']}")
|
|
149
|
-
sys.exit(1)
|
|
150
|
-
return env_values
|
|
151
|
-
|
|
152
|
-
async def main(args: argparse.Namespace):
|
|
153
|
-
"""Main function to configure and run the GoogleSlidesAgent."""
|
|
154
|
-
logger.info("--- Starting Google Slides Agent Example ---")
|
|
155
|
-
env_vars = check_required_env_vars()
|
|
156
|
-
|
|
157
|
-
# 1. Instantiate the core MCP and registry components.
|
|
158
|
-
tool_registry = default_tool_registry
|
|
159
|
-
registrar = McpToolRegistrar()
|
|
160
|
-
|
|
161
|
-
# 2. Define the configuration for the MCP server using a clear dictionary format.
|
|
162
|
-
server_id = "google-slides-mcp"
|
|
163
|
-
google_slides_mcp_config_dict = {
|
|
164
|
-
server_id: {
|
|
165
|
-
"transport_type": "stdio",
|
|
166
|
-
"stdio_params": {
|
|
167
|
-
"command": "node",
|
|
168
|
-
"args": [env_vars["script_path"]],
|
|
169
|
-
"env": {
|
|
170
|
-
"GOOGLE_CLIENT_ID": env_vars["google_client_id"],
|
|
171
|
-
"GOOGLE_CLIENT_SECRET": env_vars["google_client_secret"],
|
|
172
|
-
"GOOGLE_REFRESH_TOKEN": env_vars["google_refresh_token"],
|
|
173
|
-
}
|
|
174
|
-
},
|
|
175
|
-
"enabled": True,
|
|
176
|
-
"tool_name_prefix": "gslide",
|
|
177
|
-
}
|
|
178
|
-
}
|
|
179
|
-
|
|
180
|
-
try:
|
|
181
|
-
# 3. Discover and register tools by passing the config dictionary directly.
|
|
182
|
-
# The registrar will handle parsing, validation, and storage.
|
|
183
|
-
logger.info(f"Performing targeted discovery for remote Google Slides tools from server: '{server_id}'...")
|
|
184
|
-
await registrar.load_and_register_server(config_dict=google_slides_mcp_config_dict)
|
|
185
|
-
logger.info("Remote tool registration complete.")
|
|
186
|
-
|
|
187
|
-
# 4. Create tool instances from the registry for our agent.
|
|
188
|
-
# Use the ToolRegistry to get tools by their source server ID.
|
|
189
|
-
gslides_tool_defs = tool_registry.get_tools_by_mcp_server(server_id)
|
|
190
|
-
gslides_tool_names = [tool_def.name for tool_def in gslides_tool_defs]
|
|
191
|
-
|
|
192
|
-
if not gslides_tool_names:
|
|
193
|
-
logger.error(f"No Google Slides tools were found in the registry for server '{server_id}' after discovery. Cannot create agent.")
|
|
194
|
-
return
|
|
195
|
-
|
|
196
|
-
logger.info(f"Creating instances for registered Google Slides tools: {gslides_tool_names}")
|
|
197
|
-
tools_for_agent = [tool_registry.create_tool(name) for name in gslides_tool_names]
|
|
198
|
-
|
|
199
|
-
# 5. Configure and create the agent.
|
|
200
|
-
try:
|
|
201
|
-
_ = LLMModel[args.llm_model]
|
|
202
|
-
except (KeyError, ValueError):
|
|
203
|
-
logger.error(f"LLM Model '{args.llm_model}' is not valid or ambiguous.", file=sys.stderr)
|
|
204
|
-
try:
|
|
205
|
-
LLMFactory.ensure_initialized()
|
|
206
|
-
print("\nAvailable LLM Models (use the 'Identifier' with --llm-model):")
|
|
207
|
-
all_models = sorted(list(LLMModel), key=lambda m: m.model_identifier)
|
|
208
|
-
if not all_models:
|
|
209
|
-
print(" No models found.")
|
|
210
|
-
for model in all_models:
|
|
211
|
-
print(f" - Display Name: {model.name:<30} Identifier: {model.model_identifier}")
|
|
212
|
-
except Exception as e:
|
|
213
|
-
print(f"Additionally, an error occurred while listing models: {e}", file=sys.stderr)
|
|
214
|
-
sys.exit(1)
|
|
215
|
-
|
|
216
|
-
logger.info(f"Creating LLM instance for model: {args.llm_model}")
|
|
217
|
-
llm_instance = default_llm_factory.create_llm(model_identifier=args.llm_model)
|
|
218
|
-
|
|
219
|
-
system_prompt = (
|
|
220
|
-
"You are a helpful assistant with expertise in creating and managing Google Slides presentations.\n"
|
|
221
|
-
"You have access to a set of specialized tools for this purpose.\n\n"
|
|
222
|
-
"When asked to create a presentation, you should use the 'gslides_create_presentation' tool.\n"
|
|
223
|
-
"When asked to add content, you should find out what kind of content and use the 'gslides_batch_update_presentation' tool with the correct request objects.\n"
|
|
224
|
-
"When asked to summarize a presentation, use the 'gslides_summarize_presentation' tool.\n\n"
|
|
225
|
-
"Here is the manifest of tools available to you, including their definitions and examples:\n"
|
|
226
|
-
"{{tools}}"
|
|
227
|
-
)
|
|
228
|
-
|
|
229
|
-
gslides_agent_config = AgentConfig(
|
|
230
|
-
name="GoogleSlidesAgent",
|
|
231
|
-
role="GoogleSlidesExpert",
|
|
232
|
-
description="An agent that can create and manage Google Slides presentations using a set of remote tools.",
|
|
233
|
-
llm_instance=llm_instance,
|
|
234
|
-
system_prompt=system_prompt,
|
|
235
|
-
tools=tools_for_agent,
|
|
236
|
-
auto_execute_tools=False
|
|
237
|
-
)
|
|
238
|
-
|
|
239
|
-
agent = AgentFactory().create_agent(config=gslides_agent_config)
|
|
240
|
-
logger.info(f"Google Slides Agent instance created: {agent.agent_id}")
|
|
241
|
-
|
|
242
|
-
# 6. Run the agent in an interactive CLI session.
|
|
243
|
-
logger.info(f"Starting interactive session for agent {agent.agent_id}...")
|
|
244
|
-
await agent_cli.run(agent=agent)
|
|
245
|
-
logger.info(f"Interactive session for agent {agent.agent_id} finished.")
|
|
246
|
-
|
|
247
|
-
except Exception as e:
|
|
248
|
-
logger.error(f"An error occurred during the agent workflow: {e}", exc_info=True)
|
|
249
|
-
|
|
250
|
-
logger.info("--- Google Slides Agent Example Finished ---")
|
|
251
|
-
|
|
252
|
-
if __name__ == "__main__":
|
|
253
|
-
parser = argparse.ArgumentParser(description="Run the GoogleSlidesAgent interactively.")
|
|
254
|
-
parser.add_argument("--llm-model", type=str, default="kimi-latest", help=f"The LLM model identifier to use. Call --help-models for list.")
|
|
255
|
-
parser.add_argument("--help-models", action="store_true", help="Display available LLM models and exit.")
|
|
256
|
-
parser.add_argument("--debug", action="store_true", help="Enable debug logging.")
|
|
257
|
-
parser.add_argument("--agent-log-file", type=str, default="./agent_logs_gslides.txt",
|
|
258
|
-
help="Path to the log file for autobyteus.* library logs. (Default: ./agent_logs_gslides.txt)")
|
|
259
|
-
parser.add_argument("--no-tool-logs", action="store_true",
|
|
260
|
-
help="Disable display of [Tool Log (...)] messages on the console by the agent_cli.")
|
|
261
|
-
|
|
262
|
-
if "--help-models" in sys.argv:
|
|
263
|
-
try:
|
|
264
|
-
LLMFactory.ensure_initialized()
|
|
265
|
-
print("Available LLM Models (use the 'Identifier' with --llm-model):")
|
|
266
|
-
all_models = sorted(list(LLMModel), key=lambda m: m.model_identifier)
|
|
267
|
-
if not all_models:
|
|
268
|
-
print(" No models found.")
|
|
269
|
-
for model in all_models:
|
|
270
|
-
print(f" - Display Name: {model.name:<30} Identifier: {model.model_identifier}")
|
|
271
|
-
except Exception as e:
|
|
272
|
-
print(f"Error listing models: {e}")
|
|
273
|
-
sys.exit(0)
|
|
274
|
-
|
|
275
|
-
parsed_args = parser.parse_args()
|
|
276
|
-
|
|
277
|
-
setup_logging(parsed_args)
|
|
278
|
-
check_required_env_vars()
|
|
279
|
-
|
|
280
|
-
try:
|
|
281
|
-
asyncio.run(main(parsed_args))
|
|
282
|
-
except (KeyboardInterrupt, SystemExit):
|
|
283
|
-
logger.info("Script interrupted by user. Exiting.")
|
|
284
|
-
except Exception as e:
|
|
285
|
-
logger.error(f"An unhandled error occurred at the top level: {e}", exc_info=True)
|
|
286
|
-
finally:
|
|
287
|
-
logger.info("Exiting script.")
|