aip-agents-binary 0.0.0b2__py3-none-any.whl → 0.5.12__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.
- aip_agents/__init__.py +65 -0
- aip_agents/a2a/__init__.py +19 -0
- aip_agents/a2a/server/__init__.py +10 -0
- aip_agents/a2a/server/base_executor.py +1086 -0
- aip_agents/a2a/server/google_adk_executor.py +198 -0
- aip_agents/a2a/server/langflow_executor.py +180 -0
- aip_agents/a2a/server/langgraph_executor.py +270 -0
- aip_agents/a2a/types.py +232 -0
- aip_agents/agent/__init__.py +27 -0
- aip_agents/agent/base_agent.py +970 -0
- aip_agents/agent/base_langgraph_agent.py +2942 -0
- aip_agents/agent/google_adk_agent.py +926 -0
- aip_agents/agent/google_adk_constants.py +6 -0
- aip_agents/agent/hitl/__init__.py +24 -0
- aip_agents/agent/hitl/config.py +28 -0
- aip_agents/agent/hitl/langgraph_hitl_mixin.py +515 -0
- aip_agents/agent/hitl/manager.py +532 -0
- aip_agents/agent/hitl/models.py +18 -0
- aip_agents/agent/hitl/prompt/__init__.py +9 -0
- aip_agents/agent/hitl/prompt/base.py +42 -0
- aip_agents/agent/hitl/prompt/deferred.py +73 -0
- aip_agents/agent/hitl/registry.py +149 -0
- aip_agents/agent/{interface.pyi → interface.py} +70 -13
- aip_agents/agent/interfaces.py +65 -0
- aip_agents/agent/langflow_agent.py +464 -0
- aip_agents/agent/langgraph_memory_enhancer_agent.py +433 -0
- aip_agents/agent/langgraph_react_agent.py +2514 -0
- aip_agents/agent/system_instruction_context.py +34 -0
- aip_agents/clients/__init__.py +10 -0
- aip_agents/clients/langflow/__init__.py +10 -0
- aip_agents/clients/langflow/client.py +477 -0
- aip_agents/clients/langflow/types.py +18 -0
- aip_agents/constants.py +23 -0
- aip_agents/credentials/manager.py +132 -0
- aip_agents/examples/__init__.py +5 -0
- aip_agents/examples/compare_streaming_client.py +783 -0
- aip_agents/examples/compare_streaming_server.py +142 -0
- aip_agents/examples/demo_memory_recall.py +401 -0
- aip_agents/examples/hello_world_a2a_google_adk_client.py +49 -0
- aip_agents/examples/hello_world_a2a_google_adk_client_agent.py +48 -0
- aip_agents/examples/hello_world_a2a_google_adk_client_streaming.py +60 -0
- aip_agents/examples/hello_world_a2a_google_adk_server.py +79 -0
- aip_agents/examples/hello_world_a2a_langchain_client.py +39 -0
- aip_agents/examples/hello_world_a2a_langchain_client_agent.py +39 -0
- aip_agents/examples/hello_world_a2a_langchain_client_lm_invoker.py +37 -0
- aip_agents/examples/hello_world_a2a_langchain_client_streaming.py +41 -0
- aip_agents/examples/hello_world_a2a_langchain_reference_client_streaming.py +60 -0
- aip_agents/examples/hello_world_a2a_langchain_reference_server.py +105 -0
- aip_agents/examples/hello_world_a2a_langchain_server.py +79 -0
- aip_agents/examples/hello_world_a2a_langchain_server_lm_invoker.py +78 -0
- aip_agents/examples/hello_world_a2a_langflow_client.py +83 -0
- aip_agents/examples/hello_world_a2a_langflow_server.py +82 -0
- aip_agents/examples/hello_world_a2a_langgraph_artifact_client.py +73 -0
- aip_agents/examples/hello_world_a2a_langgraph_artifact_client_streaming.py +76 -0
- aip_agents/examples/hello_world_a2a_langgraph_artifact_server.py +92 -0
- aip_agents/examples/hello_world_a2a_langgraph_client.py +54 -0
- aip_agents/examples/hello_world_a2a_langgraph_client_agent.py +54 -0
- aip_agents/examples/hello_world_a2a_langgraph_client_agent_lm_invoker.py +32 -0
- aip_agents/examples/hello_world_a2a_langgraph_client_streaming.py +50 -0
- aip_agents/examples/hello_world_a2a_langgraph_client_streaming_lm_invoker.py +44 -0
- aip_agents/examples/hello_world_a2a_langgraph_client_streaming_tool_streaming.py +92 -0
- aip_agents/examples/hello_world_a2a_langgraph_server.py +84 -0
- aip_agents/examples/hello_world_a2a_langgraph_server_lm_invoker.py +79 -0
- aip_agents/examples/hello_world_a2a_langgraph_server_tool_streaming.py +132 -0
- aip_agents/examples/hello_world_a2a_mcp_langgraph.py +196 -0
- aip_agents/examples/hello_world_a2a_three_level_agent_hierarchy_client.py +244 -0
- aip_agents/examples/hello_world_a2a_three_level_agent_hierarchy_server.py +251 -0
- aip_agents/examples/hello_world_a2a_with_metadata_langchain_client.py +57 -0
- aip_agents/examples/hello_world_a2a_with_metadata_langchain_server_lm_invoker.py +80 -0
- aip_agents/examples/hello_world_google_adk.py +41 -0
- aip_agents/examples/hello_world_google_adk_mcp_http.py +34 -0
- aip_agents/examples/hello_world_google_adk_mcp_http_stream.py +40 -0
- aip_agents/examples/hello_world_google_adk_mcp_sse.py +44 -0
- aip_agents/examples/hello_world_google_adk_mcp_sse_stream.py +48 -0
- aip_agents/examples/hello_world_google_adk_mcp_stdio.py +44 -0
- aip_agents/examples/hello_world_google_adk_mcp_stdio_stream.py +48 -0
- aip_agents/examples/hello_world_google_adk_stream.py +44 -0
- aip_agents/examples/hello_world_langchain.py +28 -0
- aip_agents/examples/hello_world_langchain_lm_invoker.py +15 -0
- aip_agents/examples/hello_world_langchain_mcp_http.py +34 -0
- aip_agents/examples/hello_world_langchain_mcp_http_interactive.py +130 -0
- aip_agents/examples/hello_world_langchain_mcp_http_stream.py +42 -0
- aip_agents/examples/hello_world_langchain_mcp_multi_server.py +155 -0
- aip_agents/examples/hello_world_langchain_mcp_sse.py +34 -0
- aip_agents/examples/hello_world_langchain_mcp_sse_stream.py +40 -0
- aip_agents/examples/hello_world_langchain_mcp_stdio.py +30 -0
- aip_agents/examples/hello_world_langchain_mcp_stdio_stream.py +41 -0
- aip_agents/examples/hello_world_langchain_stream.py +36 -0
- aip_agents/examples/hello_world_langchain_stream_lm_invoker.py +39 -0
- aip_agents/examples/hello_world_langflow_agent.py +163 -0
- aip_agents/examples/hello_world_langgraph.py +39 -0
- aip_agents/examples/hello_world_langgraph_bosa_twitter.py +41 -0
- aip_agents/examples/hello_world_langgraph_mcp_http.py +31 -0
- aip_agents/examples/hello_world_langgraph_mcp_http_stream.py +34 -0
- aip_agents/examples/hello_world_langgraph_mcp_sse.py +35 -0
- aip_agents/examples/hello_world_langgraph_mcp_sse_stream.py +50 -0
- aip_agents/examples/hello_world_langgraph_mcp_stdio.py +35 -0
- aip_agents/examples/hello_world_langgraph_mcp_stdio_stream.py +50 -0
- aip_agents/examples/hello_world_langgraph_stream.py +43 -0
- aip_agents/examples/hello_world_langgraph_stream_lm_invoker.py +37 -0
- aip_agents/examples/hello_world_model_switch_cli.py +210 -0
- aip_agents/examples/hello_world_multi_agent_adk.py +75 -0
- aip_agents/examples/hello_world_multi_agent_langchain.py +54 -0
- aip_agents/examples/hello_world_multi_agent_langgraph.py +66 -0
- aip_agents/examples/hello_world_multi_agent_langgraph_lm_invoker.py +69 -0
- aip_agents/examples/hello_world_pii_logger.py +21 -0
- aip_agents/examples/hello_world_sentry.py +133 -0
- aip_agents/examples/hello_world_step_limits.py +273 -0
- aip_agents/examples/hello_world_stock_a2a_server.py +103 -0
- aip_agents/examples/hello_world_tool_output_client.py +46 -0
- aip_agents/examples/hello_world_tool_output_server.py +114 -0
- aip_agents/examples/hitl_demo.py +724 -0
- aip_agents/examples/mcp_configs/configs.py +63 -0
- aip_agents/examples/mcp_servers/common.py +76 -0
- aip_agents/examples/mcp_servers/mcp_name.py +29 -0
- aip_agents/examples/mcp_servers/mcp_server_http.py +19 -0
- aip_agents/examples/mcp_servers/mcp_server_sse.py +19 -0
- aip_agents/examples/mcp_servers/mcp_server_stdio.py +19 -0
- aip_agents/examples/mcp_servers/mcp_time.py +10 -0
- aip_agents/examples/pii_demo_langgraph_client.py +69 -0
- aip_agents/examples/pii_demo_langgraph_server.py +126 -0
- aip_agents/examples/pii_demo_multi_agent_client.py +80 -0
- aip_agents/examples/pii_demo_multi_agent_server.py +247 -0
- aip_agents/examples/todolist_planning_a2a_langchain_client.py +70 -0
- aip_agents/examples/todolist_planning_a2a_langgraph_server.py +88 -0
- aip_agents/examples/tools/__init__.py +27 -0
- aip_agents/examples/tools/{adk_arithmetic_tools.pyi → adk_arithmetic_tools.py} +12 -0
- aip_agents/examples/tools/adk_weather_tool.py +60 -0
- aip_agents/examples/tools/data_generator_tool.py +103 -0
- aip_agents/examples/tools/data_visualization_tool.py +312 -0
- aip_agents/examples/tools/image_artifact_tool.py +136 -0
- aip_agents/examples/tools/langchain_arithmetic_tools.py +26 -0
- aip_agents/examples/tools/langchain_currency_exchange_tool.py +88 -0
- aip_agents/examples/tools/langchain_graph_artifact_tool.py +172 -0
- aip_agents/examples/tools/langchain_weather_tool.py +48 -0
- aip_agents/examples/tools/langgraph_streaming_tool.py +130 -0
- aip_agents/examples/tools/mock_retrieval_tool.py +56 -0
- aip_agents/examples/tools/pii_demo_tools.py +189 -0
- aip_agents/examples/tools/random_chart_tool.py +142 -0
- aip_agents/examples/tools/serper_tool.py +202 -0
- aip_agents/examples/tools/stock_tools.py +82 -0
- aip_agents/examples/tools/table_generator_tool.py +167 -0
- aip_agents/examples/tools/time_tool.py +82 -0
- aip_agents/examples/tools/weather_forecast_tool.py +38 -0
- aip_agents/executor/agent_executor.py +473 -0
- aip_agents/executor/base.py +48 -0
- aip_agents/mcp/__init__.py +1 -0
- aip_agents/mcp/client/__init__.py +14 -0
- aip_agents/mcp/client/base_mcp_client.py +369 -0
- aip_agents/mcp/client/connection_manager.py +193 -0
- aip_agents/mcp/client/google_adk/__init__.py +11 -0
- aip_agents/mcp/client/google_adk/client.py +381 -0
- aip_agents/mcp/client/langchain/__init__.py +11 -0
- aip_agents/mcp/client/langchain/client.py +265 -0
- aip_agents/mcp/client/persistent_session.py +359 -0
- aip_agents/mcp/client/session_pool.py +351 -0
- aip_agents/mcp/client/transports.py +215 -0
- aip_agents/mcp/utils/__init__.py +7 -0
- aip_agents/mcp/utils/config_validator.py +139 -0
- aip_agents/memory/__init__.py +14 -0
- aip_agents/memory/adapters/__init__.py +10 -0
- aip_agents/memory/adapters/base_adapter.py +717 -0
- aip_agents/memory/adapters/mem0.py +84 -0
- aip_agents/memory/{base.pyi → base.py} +40 -5
- aip_agents/memory/constants.py +49 -0
- aip_agents/memory/factory.py +86 -0
- aip_agents/memory/guidance.py +20 -0
- aip_agents/memory/simple_memory.py +47 -0
- aip_agents/middleware/__init__.py +17 -0
- aip_agents/middleware/base.py +88 -0
- aip_agents/middleware/manager.py +128 -0
- aip_agents/middleware/todolist.py +274 -0
- aip_agents/schema/__init__.py +69 -0
- aip_agents/schema/a2a.py +56 -0
- aip_agents/schema/agent.py +111 -0
- aip_agents/schema/hitl.py +157 -0
- aip_agents/schema/langgraph.py +37 -0
- aip_agents/schema/model_id.py +97 -0
- aip_agents/schema/step_limit.py +108 -0
- aip_agents/schema/storage.py +40 -0
- aip_agents/sentry/__init__.py +11 -0
- aip_agents/sentry/sentry.py +151 -0
- aip_agents/storage/__init__.py +41 -0
- aip_agents/storage/{base.pyi → base.py} +30 -4
- aip_agents/storage/clients/__init__.py +12 -0
- aip_agents/storage/clients/minio_client.py +318 -0
- aip_agents/storage/config.py +62 -0
- aip_agents/storage/providers/__init__.py +15 -0
- aip_agents/storage/providers/{base.pyi → base.py} +23 -6
- aip_agents/storage/providers/{memory.pyi → memory.py} +38 -3
- aip_agents/storage/providers/object_storage.py +214 -0
- aip_agents/tools/__init__.py +6 -0
- aip_agents/tools/bosa_tools.py +105 -0
- aip_agents/tools/browser_use/__init__.py +82 -0
- aip_agents/tools/browser_use/action_parser.py +103 -0
- aip_agents/tools/browser_use/browser_use_tool.py +1112 -0
- aip_agents/tools/browser_use/llm_config.py +120 -0
- aip_agents/tools/browser_use/minio_storage.py +198 -0
- aip_agents/tools/browser_use/schemas.py +119 -0
- aip_agents/tools/browser_use/session.py +76 -0
- aip_agents/tools/browser_use/session_errors.py +132 -0
- aip_agents/tools/browser_use/steel_session_recording.py +317 -0
- aip_agents/tools/browser_use/streaming.py +813 -0
- aip_agents/tools/browser_use/structured_data_parser.py +257 -0
- aip_agents/tools/browser_use/structured_data_recovery.py +204 -0
- aip_agents/tools/browser_use/types.py +78 -0
- aip_agents/tools/code_sandbox/__init__.py +26 -0
- aip_agents/tools/code_sandbox/constant.py +13 -0
- aip_agents/tools/code_sandbox/e2b_cloud_sandbox_extended.py +257 -0
- aip_agents/tools/code_sandbox/e2b_sandbox_tool.py +411 -0
- aip_agents/tools/constants.py +165 -0
- aip_agents/tools/document_loader/__init__.py +37 -0
- aip_agents/tools/document_loader/base_reader.py +262 -0
- aip_agents/tools/document_loader/docx_reader_tool.py +53 -0
- aip_agents/tools/document_loader/excel_reader_tool.py +160 -0
- aip_agents/tools/document_loader/pdf_reader_tool.py +67 -0
- aip_agents/tools/document_loader/pdf_splitter.py +169 -0
- aip_agents/tools/gl_connector/__init__.py +5 -0
- aip_agents/tools/gl_connector/tool.py +351 -0
- aip_agents/tools/memory_search/__init__.py +22 -0
- aip_agents/tools/memory_search/base.py +200 -0
- aip_agents/tools/memory_search/mem0.py +258 -0
- aip_agents/tools/memory_search/schema.py +48 -0
- aip_agents/tools/memory_search_tool.py +26 -0
- aip_agents/tools/tool_config_injector.py +300 -0
- aip_agents/tools/web_search/__init__.py +15 -0
- aip_agents/tools/web_search/serper_tool.py +187 -0
- aip_agents/types/__init__.py +70 -0
- aip_agents/types/a2a_events.py +13 -0
- aip_agents/utils/__init__.py +79 -0
- aip_agents/utils/a2a_connector.py +1757 -0
- aip_agents/utils/artifact_helpers.py +502 -0
- aip_agents/utils/constants.py +22 -0
- aip_agents/utils/datetime/__init__.py +34 -0
- aip_agents/utils/datetime/normalization.py +231 -0
- aip_agents/utils/datetime/timezone.py +206 -0
- aip_agents/utils/env_loader.py +27 -0
- aip_agents/utils/event_handler_registry.py +58 -0
- aip_agents/utils/file_prompt_utils.py +176 -0
- aip_agents/utils/final_response_builder.py +211 -0
- aip_agents/utils/formatter_llm_client.py +231 -0
- aip_agents/utils/langgraph/__init__.py +19 -0
- aip_agents/utils/langgraph/converter.py +128 -0
- aip_agents/utils/langgraph/tool_managers/__init__.py +15 -0
- aip_agents/utils/langgraph/tool_managers/a2a_tool_manager.py +99 -0
- aip_agents/utils/langgraph/tool_managers/{base_tool_manager.pyi → base_tool_manager.py} +25 -8
- aip_agents/utils/langgraph/tool_managers/delegation_tool_manager.py +1071 -0
- aip_agents/utils/langgraph/tool_output_management.py +967 -0
- aip_agents/utils/logger.py +195 -0
- aip_agents/utils/metadata/__init__.py +27 -0
- aip_agents/utils/metadata/activity_metadata_helper.py +407 -0
- aip_agents/utils/metadata/activity_narrative/__init__.py +35 -0
- aip_agents/utils/metadata/activity_narrative/builder.py +817 -0
- aip_agents/utils/metadata/activity_narrative/constants.py +51 -0
- aip_agents/utils/metadata/activity_narrative/context.py +49 -0
- aip_agents/utils/metadata/activity_narrative/formatters.py +230 -0
- aip_agents/utils/metadata/activity_narrative/utils.py +35 -0
- aip_agents/utils/metadata/schemas/__init__.py +16 -0
- aip_agents/utils/metadata/schemas/activity_schema.py +29 -0
- aip_agents/utils/metadata/schemas/thinking_schema.py +31 -0
- aip_agents/utils/metadata/thinking_metadata_helper.py +38 -0
- aip_agents/utils/metadata_helper.py +358 -0
- aip_agents/utils/name_preprocessor/__init__.py +17 -0
- aip_agents/utils/name_preprocessor/{base_name_preprocessor.pyi → base_name_preprocessor.py} +22 -2
- aip_agents/utils/name_preprocessor/google_name_preprocessor.py +100 -0
- aip_agents/utils/name_preprocessor/name_preprocessor.py +87 -0
- aip_agents/utils/name_preprocessor/{openai_name_preprocessor.pyi → openai_name_preprocessor.py} +19 -5
- aip_agents/utils/pii/__init__.py +25 -0
- aip_agents/utils/pii/pii_handler.py +397 -0
- aip_agents/utils/pii/pii_helper.py +207 -0
- aip_agents/utils/pii/uuid_deanonymizer_mapping.py +195 -0
- aip_agents/utils/reference_helper.py +273 -0
- aip_agents/utils/sse_chunk_transformer.py +831 -0
- aip_agents/utils/step_limit_manager.py +265 -0
- aip_agents/utils/token_usage_helper.py +156 -0
- aip_agents_binary-0.5.12.dist-info/METADATA +689 -0
- aip_agents_binary-0.5.12.dist-info/RECORD +279 -0
- {aip_agents_binary-0.0.0b2.dist-info → aip_agents_binary-0.5.12.dist-info}/WHEEL +2 -1
- aip_agents_binary-0.5.12.dist-info/top_level.txt +1 -0
- aip_agents/__init__.pyi +0 -0
- aip_agents/a2a/__init__.pyi +0 -3
- aip_agents/a2a/server/__init__.pyi +0 -4
- aip_agents/a2a/server/base_executor.pyi +0 -63
- aip_agents/a2a/server/google_adk_executor.pyi +0 -51
- aip_agents/a2a/server/langgraph_executor.pyi +0 -48
- aip_agents/a2a/types.pyi +0 -125
- aip_agents/agent/__init__.pyi +0 -7
- aip_agents/agent/base_agent.pyi +0 -205
- aip_agents/agent/base_langgraph_agent.pyi +0 -164
- aip_agents/agent/google_adk_agent.pyi +0 -128
- aip_agents/agent/langgraph_react_agent.pyi +0 -131
- aip_agents/agent/types.pyi +0 -106
- aip_agents/constants.pyi +0 -6
- aip_agents/examples/__init__.pyi +0 -0
- aip_agents/examples/hello_world_a2a_google_adk_client.pyi +0 -9
- aip_agents/examples/hello_world_a2a_google_adk_client_agent.pyi +0 -9
- aip_agents/examples/hello_world_a2a_google_adk_client_streaming.pyi +0 -9
- aip_agents/examples/hello_world_a2a_google_adk_server.pyi +0 -10
- aip_agents/examples/hello_world_a2a_langchain_client.pyi +0 -5
- aip_agents/examples/hello_world_a2a_langchain_client_agent.pyi +0 -5
- aip_agents/examples/hello_world_a2a_langchain_client_lm_invoker.pyi +0 -5
- aip_agents/examples/hello_world_a2a_langchain_client_streaming.pyi +0 -5
- aip_agents/examples/hello_world_a2a_langchain_reference_client_streaming.pyi +0 -5
- aip_agents/examples/hello_world_a2a_langchain_reference_server.pyi +0 -10
- aip_agents/examples/hello_world_a2a_langchain_server.pyi +0 -10
- aip_agents/examples/hello_world_a2a_langchain_server_lm_invoker.pyi +0 -10
- aip_agents/examples/hello_world_a2a_langgraph_artifact_client.pyi +0 -5
- aip_agents/examples/hello_world_a2a_langgraph_artifact_client_streaming.pyi +0 -5
- aip_agents/examples/hello_world_a2a_langgraph_artifact_server.pyi +0 -11
- aip_agents/examples/hello_world_a2a_langgraph_client.pyi +0 -9
- aip_agents/examples/hello_world_a2a_langgraph_client_agent.pyi +0 -9
- aip_agents/examples/hello_world_a2a_langgraph_client_agent_lm_invoker.pyi +0 -2
- aip_agents/examples/hello_world_a2a_langgraph_client_streaming.pyi +0 -9
- aip_agents/examples/hello_world_a2a_langgraph_client_streaming_lm_invoker.pyi +0 -5
- aip_agents/examples/hello_world_a2a_langgraph_client_streaming_tool_streaming.pyi +0 -5
- aip_agents/examples/hello_world_a2a_langgraph_server.pyi +0 -9
- aip_agents/examples/hello_world_a2a_langgraph_server_lm_invoker.pyi +0 -10
- aip_agents/examples/hello_world_a2a_langgraph_server_tool_streaming.pyi +0 -10
- aip_agents/examples/hello_world_a2a_mcp_langgraph.pyi +0 -48
- aip_agents/examples/hello_world_a2a_multi_agent_coordinator_client.pyi +0 -15
- aip_agents/examples/hello_world_a2a_multi_agent_coordinator_client_streaming.pyi +0 -5
- aip_agents/examples/hello_world_a2a_multi_agent_coordinator_server.pyi +0 -11
- aip_agents/examples/hello_world_a2a_three_level_agent_hierarchy_client.pyi +0 -23
- aip_agents/examples/hello_world_a2a_three_level_agent_hierarchy_server.pyi +0 -17
- aip_agents/examples/hello_world_a2a_with_metadata_langchain_client.pyi +0 -5
- aip_agents/examples/hello_world_a2a_with_metadata_langchain_server_lm_invoker.pyi +0 -10
- aip_agents/examples/hello_world_google_adk.pyi +0 -5
- aip_agents/examples/hello_world_google_adk_mcp_http.pyi +0 -5
- aip_agents/examples/hello_world_google_adk_mcp_http_stream.pyi +0 -5
- aip_agents/examples/hello_world_google_adk_mcp_sse.pyi +0 -5
- aip_agents/examples/hello_world_google_adk_mcp_sse_stream.pyi +0 -5
- aip_agents/examples/hello_world_google_adk_mcp_stdio.pyi +0 -5
- aip_agents/examples/hello_world_google_adk_mcp_stdio_stream.pyi +0 -5
- aip_agents/examples/hello_world_google_adk_stream.pyi +0 -5
- aip_agents/examples/hello_world_langchain.pyi +0 -5
- aip_agents/examples/hello_world_langchain_lm_invoker.pyi +0 -2
- aip_agents/examples/hello_world_langchain_mcp_http.pyi +0 -5
- aip_agents/examples/hello_world_langchain_mcp_http_stream.pyi +0 -5
- aip_agents/examples/hello_world_langchain_mcp_sse.pyi +0 -5
- aip_agents/examples/hello_world_langchain_mcp_sse_stream.pyi +0 -5
- aip_agents/examples/hello_world_langchain_mcp_stdio.pyi +0 -5
- aip_agents/examples/hello_world_langchain_mcp_stdio_stream.pyi +0 -5
- aip_agents/examples/hello_world_langchain_stream.pyi +0 -5
- aip_agents/examples/hello_world_langchain_stream_lm_invoker.pyi +0 -5
- aip_agents/examples/hello_world_langgraph.pyi +0 -5
- aip_agents/examples/hello_world_langgraph_bosa.pyi +0 -5
- aip_agents/examples/hello_world_langgraph_bosa_twitter.pyi +0 -5
- aip_agents/examples/hello_world_langgraph_mcp_http.pyi +0 -5
- aip_agents/examples/hello_world_langgraph_mcp_http_stream.pyi +0 -5
- aip_agents/examples/hello_world_langgraph_mcp_sse.pyi +0 -5
- aip_agents/examples/hello_world_langgraph_mcp_sse_stream.pyi +0 -5
- aip_agents/examples/hello_world_langgraph_mcp_stdio.pyi +0 -5
- aip_agents/examples/hello_world_langgraph_mcp_stdio_stream.pyi +0 -5
- aip_agents/examples/hello_world_langgraph_stream.pyi +0 -5
- aip_agents/examples/hello_world_langgraph_stream_lm_invoker.pyi +0 -5
- aip_agents/examples/hello_world_model_switch_cli.pyi +0 -15
- aip_agents/examples/hello_world_multi_agent_adk.pyi +0 -6
- aip_agents/examples/hello_world_multi_agent_langchain.pyi +0 -5
- aip_agents/examples/hello_world_multi_agent_langgraph.pyi +0 -5
- aip_agents/examples/hello_world_multi_agent_langgraph_lm_invoker.pyi +0 -5
- aip_agents/examples/hello_world_pii_logger.pyi +0 -5
- aip_agents/examples/hello_world_sentry.pyi +0 -21
- aip_agents/examples/hello_world_stock_a2a_server.pyi +0 -12
- aip_agents/examples/hello_world_tool_output_client.pyi +0 -5
- aip_agents/examples/hello_world_tool_output_server.pyi +0 -14
- aip_agents/examples/tools/__init__.pyi +0 -7
- aip_agents/examples/tools/adk_weather_tool.pyi +0 -18
- aip_agents/examples/tools/data_generator_tool.pyi +0 -15
- aip_agents/examples/tools/data_visualization_tool.pyi +0 -17
- aip_agents/examples/tools/image_artifact_tool.pyi +0 -24
- aip_agents/examples/tools/langchain_arithmetic_tools.pyi +0 -10
- aip_agents/examples/tools/langchain_currency_exchange_tool.pyi +0 -21
- aip_agents/examples/tools/langchain_graph_artifact_tool.pyi +0 -23
- aip_agents/examples/tools/langchain_weather_tool.pyi +0 -19
- aip_agents/examples/tools/langgraph_streaming_tool.pyi +0 -43
- aip_agents/examples/tools/pr_details_bosa_tool.pyi +0 -26
- aip_agents/examples/tools/serper_tool.pyi +0 -16
- aip_agents/examples/tools/stock_tools.pyi +0 -21
- aip_agents/examples/tools/table_generator_tool.pyi +0 -22
- aip_agents/examples/tools/time_tool.pyi +0 -15
- aip_agents/examples/tools/weather_forecast_tool.pyi +0 -14
- aip_agents/mcp/__init__.pyi +0 -0
- aip_agents/mcp/client/__init__.pyi +0 -4
- aip_agents/mcp/client/google_adk/__init__.pyi +0 -3
- aip_agents/mcp/client/google_adk/client.pyi +0 -38
- aip_agents/mcp/client/langchain/__init__.pyi +0 -3
- aip_agents/mcp/client/langchain/client.pyi +0 -3
- aip_agents/memory/__init__.pyi +0 -0
- aip_agents/memory/simple_memory.pyi +0 -22
- aip_agents/sentry/__init__.pyi +0 -3
- aip_agents/sentry/sentry.pyi +0 -48
- aip_agents/storage/__init__.pyi +0 -8
- aip_agents/storage/clients/__init__.pyi +0 -3
- aip_agents/storage/clients/minio_client.pyi +0 -137
- aip_agents/storage/config.pyi +0 -66
- aip_agents/storage/providers/__init__.pyi +0 -5
- aip_agents/storage/providers/object_storage.pyi +0 -98
- aip_agents/tools/__init__.pyi +0 -3
- aip_agents/tools/base.pyi +0 -44
- aip_agents/tools/base_bosa_tools.pyi +0 -12
- aip_agents/tools/bosa_connector.pyi +0 -30
- aip_agents/tools/bosa_tools.pyi +0 -37
- aip_agents/tools/bosa_tools_interface.pyi +0 -26
- aip_agents/tools/constants.pyi +0 -130
- aip_agents/tools/nested_agent_tool.pyi +0 -45
- aip_agents/tools/tool_config_injector.pyi +0 -26
- aip_agents/types/__init__.pyi +0 -3
- aip_agents/types/a2a_events.pyi +0 -74
- aip_agents/utils/__init__.pyi +0 -5
- aip_agents/utils/a2a_connector.pyi +0 -135
- aip_agents/utils/artifact_helpers.pyi +0 -179
- aip_agents/utils/langgraph/__init__.pyi +0 -3
- aip_agents/utils/langgraph/converter.pyi +0 -49
- aip_agents/utils/langgraph/tool_managers/__init__.pyi +0 -5
- aip_agents/utils/langgraph/tool_managers/a2a_tool_manager.pyi +0 -35
- aip_agents/utils/langgraph/tool_managers/delegation_tool_manager.pyi +0 -50
- aip_agents/utils/langgraph/tool_output_management.pyi +0 -310
- aip_agents/utils/logger_manager.pyi +0 -151
- aip_agents/utils/metadata_helper.pyi +0 -110
- aip_agents/utils/name_preprocessor/__init__.pyi +0 -6
- aip_agents/utils/name_preprocessor/google_name_preprocessor.pyi +0 -38
- aip_agents/utils/name_preprocessor/name_preprocessor.pyi +0 -41
- aip_agents/utils/reference_helper.pyi +0 -49
- aip_agents/utils/token_usage_helper.pyi +0 -60
- aip_agents_binary-0.0.0b2.dist-info/METADATA +0 -277
- aip_agents_binary-0.0.0b2.dist-info/RECORD +0 -157
|
@@ -1,5 +0,0 @@
|
|
|
1
|
-
from aip_agents.agent import LangGraphAgent as LangGraphAgent
|
|
2
|
-
from aip_agents.examples.tools.pr_details_bosa_tool import GitHubPRDetailsTool as GitHubPRDetailsTool
|
|
3
|
-
|
|
4
|
-
async def langgraph_example_bosa() -> None:
|
|
5
|
-
"""Demonstrates the LangGraphAgent's arun method with BOSA tools."""
|
|
@@ -1,5 +0,0 @@
|
|
|
1
|
-
from aip_agents.agent import LangGraphAgent as LangGraphAgent
|
|
2
|
-
from aip_agents.examples.mcp_configs.configs import mcp_config_http as mcp_config_http
|
|
3
|
-
|
|
4
|
-
async def main() -> None:
|
|
5
|
-
"""Demonstrates the LangGraphAgent with MCP tools via Streamable HTTP transport and streaming capabilities."""
|
|
@@ -1,5 +0,0 @@
|
|
|
1
|
-
from aip_agents.agent import LangGraphAgent as LangGraphAgent
|
|
2
|
-
from aip_agents.examples.mcp_configs.configs import mcp_config_sse as mcp_config_sse
|
|
3
|
-
|
|
4
|
-
async def main() -> None:
|
|
5
|
-
"""Demonstrates the LangGraphAgent with MCP tools via SSE transport and streaming capabilities."""
|
|
@@ -1,5 +0,0 @@
|
|
|
1
|
-
from aip_agents.agent import LangGraphAgent as LangGraphAgent
|
|
2
|
-
from aip_agents.examples.mcp_configs.configs import mcp_config_stdio as mcp_config_stdio
|
|
3
|
-
|
|
4
|
-
async def main() -> None:
|
|
5
|
-
"""Demonstrates the LangGraphAgent with MCP tools via stdio transport and streaming."""
|
|
@@ -1,15 +0,0 @@
|
|
|
1
|
-
from _typeshed import Incomplete
|
|
2
|
-
from aip_agents.agent import LangChainAgent as LangChainAgent
|
|
3
|
-
from aip_agents.agent.types import AgentConfig as AgentConfig
|
|
4
|
-
|
|
5
|
-
MODEL_IDS: Incomplete
|
|
6
|
-
MODEL_CONFIGS: Incomplete
|
|
7
|
-
|
|
8
|
-
def make_agent(model_id):
|
|
9
|
-
"""Makes an agent with the given model id."""
|
|
10
|
-
def print_help() -> None:
|
|
11
|
-
"""Prints available commands and their descriptions."""
|
|
12
|
-
def handle_switch_model(current_model):
|
|
13
|
-
"""Handles model switching. Returns (new_model, new_agent)."""
|
|
14
|
-
def main() -> None:
|
|
15
|
-
"""Runs the Hello World Model Switch CLI."""
|
|
@@ -1,6 +0,0 @@
|
|
|
1
|
-
from aip_agents.agent.google_adk_agent import GoogleADKAgent as GoogleADKAgent
|
|
2
|
-
from aip_agents.examples.tools.adk_arithmetic_tools import sum_numbers as sum_numbers
|
|
3
|
-
from aip_agents.examples.tools.adk_weather_tool import get_weather as get_weather
|
|
4
|
-
|
|
5
|
-
async def multi_agent_example() -> None:
|
|
6
|
-
"""Demonstrates multi-agent coordination with GoogleADKAgent."""
|
|
@@ -1,21 +0,0 @@
|
|
|
1
|
-
from _typeshed import Incomplete
|
|
2
|
-
from aip_agents.examples.hello_world_langgraph import langgraph_example as langgraph_example
|
|
3
|
-
from aip_agents.sentry import setup_telemetry as setup_telemetry
|
|
4
|
-
from aip_agents.utils.logger_manager import LoggerManager as LoggerManager
|
|
5
|
-
from fastapi import FastAPI
|
|
6
|
-
|
|
7
|
-
logger: Incomplete
|
|
8
|
-
BASE_URL: str
|
|
9
|
-
SENTRY_ENVIRONMENT: Incomplete
|
|
10
|
-
USE_OPENTELEMETRY: Incomplete
|
|
11
|
-
|
|
12
|
-
def fetch_endpoints() -> None:
|
|
13
|
-
"""Fetch all endpoints from the server."""
|
|
14
|
-
def create_app() -> FastAPI:
|
|
15
|
-
"""Create and configure the FastAPI application.
|
|
16
|
-
|
|
17
|
-
Returns:
|
|
18
|
-
FastAPI: The configured FastAPI application.
|
|
19
|
-
"""
|
|
20
|
-
def run_server() -> None:
|
|
21
|
-
"""Run the FastAPI server."""
|
|
@@ -1,12 +0,0 @@
|
|
|
1
|
-
from _typeshed import Incomplete
|
|
2
|
-
from aip_agents.agent import LangGraphAgent as LangGraphAgent
|
|
3
|
-
from aip_agents.examples.tools.stock_tools import StockNewsInput as StockNewsInput, StockPriceInput as StockPriceInput, get_stock_news as get_stock_news, get_stock_price as get_stock_price
|
|
4
|
-
from aip_agents.utils.logger_manager import LoggerManager as LoggerManager
|
|
5
|
-
from langchain_core.tools import BaseTool as BaseTool
|
|
6
|
-
|
|
7
|
-
logger: Incomplete
|
|
8
|
-
SERVER_AGENT_NAME: str
|
|
9
|
-
stock_tools: list[BaseTool]
|
|
10
|
-
|
|
11
|
-
def main(host: str, port: int):
|
|
12
|
-
"""Runs the StockAgent A2A server."""
|
|
@@ -1,14 +0,0 @@
|
|
|
1
|
-
from _typeshed import Incomplete
|
|
2
|
-
from aip_agents.agent import LangGraphReactAgent as LangGraphReactAgent
|
|
3
|
-
from aip_agents.examples.tools.data_generator_tool import DataGeneratorTool as DataGeneratorTool
|
|
4
|
-
from aip_agents.examples.tools.data_visualization_tool import DataVisualizerTool as DataVisualizerTool
|
|
5
|
-
from aip_agents.storage.clients.minio_client import MinioConfig as MinioConfig, MinioObjectStorage as MinioObjectStorage
|
|
6
|
-
from aip_agents.storage.providers.object_storage import ObjectStorageProvider as ObjectStorageProvider
|
|
7
|
-
from aip_agents.utils.langgraph.tool_output_management import ToolOutputConfig as ToolOutputConfig, ToolOutputManager as ToolOutputManager
|
|
8
|
-
from aip_agents.utils.logger_manager import LoggerManager as LoggerManager
|
|
9
|
-
|
|
10
|
-
logger: Incomplete
|
|
11
|
-
SERVER_AGENT_NAME: str
|
|
12
|
-
|
|
13
|
-
def main(host: str, port: int):
|
|
14
|
-
"""Run the Data Visualization A2A server."""
|
|
@@ -1,7 +0,0 @@
|
|
|
1
|
-
from aip_agents.examples.tools.adk_weather_tool import get_weather as weather_tool_adk
|
|
2
|
-
from aip_agents.examples.tools.langchain_weather_tool import weather_tool as weather_tool_langchain
|
|
3
|
-
from aip_agents.examples.tools.langgraph_streaming_tool import LangGraphStreamingTool as langgraph_streaming_tool
|
|
4
|
-
from aip_agents.examples.tools.serper_tool import MockGoogleSerperTool as google_serper_tool
|
|
5
|
-
from aip_agents.examples.tools.time_tool import TimeTool as time_tool
|
|
6
|
-
|
|
7
|
-
__all__ = ['weather_tool_langchain', 'weather_tool_adk', 'time_tool', 'google_serper_tool', 'langgraph_streaming_tool']
|
|
@@ -1,18 +0,0 @@
|
|
|
1
|
-
from _typeshed import Incomplete
|
|
2
|
-
from aip_agents.utils.logger_manager import LoggerManager as LoggerManager
|
|
3
|
-
|
|
4
|
-
logger: Incomplete
|
|
5
|
-
WEATHER_DATA: Incomplete
|
|
6
|
-
|
|
7
|
-
def get_weather(location: str) -> dict:
|
|
8
|
-
"""Gets detailed weather information for a specified location.
|
|
9
|
-
|
|
10
|
-
Args:
|
|
11
|
-
location: The name of the city to get weather for.
|
|
12
|
-
|
|
13
|
-
Returns:
|
|
14
|
-
A dictionary containing:
|
|
15
|
-
- location: The requested location name
|
|
16
|
-
- weather: A dictionary with temperature, conditions, and humidity
|
|
17
|
-
"""
|
|
18
|
-
weather_tool = get_weather
|
|
@@ -1,15 +0,0 @@
|
|
|
1
|
-
from langchain_core.tools import BaseTool
|
|
2
|
-
from pydantic import BaseModel
|
|
3
|
-
from typing import Literal
|
|
4
|
-
|
|
5
|
-
class DataInput(BaseModel):
|
|
6
|
-
"""Input schema for data generation."""
|
|
7
|
-
data_type: Literal['sales', 'scores', 'growth']
|
|
8
|
-
size: int
|
|
9
|
-
|
|
10
|
-
class DataGeneratorTool(BaseTool):
|
|
11
|
-
"""Tool that generates sample datasets with automatic output storage."""
|
|
12
|
-
name: str
|
|
13
|
-
description: str
|
|
14
|
-
args_schema: type[BaseModel]
|
|
15
|
-
store_final_output: bool
|
|
@@ -1,17 +0,0 @@
|
|
|
1
|
-
from aip_agents.a2a.types import MimeType as MimeType
|
|
2
|
-
from aip_agents.utils.artifact_helpers import create_artifact_command as create_artifact_command
|
|
3
|
-
from langchain_core.tools import BaseTool
|
|
4
|
-
from pydantic import BaseModel
|
|
5
|
-
from typing import Any
|
|
6
|
-
|
|
7
|
-
class ChartInput(BaseModel):
|
|
8
|
-
"""Input schema for chart generation."""
|
|
9
|
-
data_source: Any
|
|
10
|
-
chart_type: str
|
|
11
|
-
title: str
|
|
12
|
-
|
|
13
|
-
class DataVisualizerTool(BaseTool):
|
|
14
|
-
"""Tool that creates visualizations with automatic output storage."""
|
|
15
|
-
name: str
|
|
16
|
-
description: str
|
|
17
|
-
args_schema: type[BaseModel]
|
|
@@ -1,24 +0,0 @@
|
|
|
1
|
-
from aip_agents.a2a.types import MimeType as MimeType
|
|
2
|
-
from aip_agents.utils.artifact_helpers import create_artifact_response as create_artifact_response, create_error_response as create_error_response
|
|
3
|
-
from langchain_core.tools import BaseTool
|
|
4
|
-
from pydantic import BaseModel
|
|
5
|
-
|
|
6
|
-
class ImageArtifactInput(BaseModel):
|
|
7
|
-
"""Input schema for image artifact tool."""
|
|
8
|
-
width: int
|
|
9
|
-
height: int
|
|
10
|
-
color: str
|
|
11
|
-
text: str
|
|
12
|
-
image_name: str
|
|
13
|
-
|
|
14
|
-
class ImageArtifactTool(BaseTool):
|
|
15
|
-
"""Tool that generates simple images with artifact support.
|
|
16
|
-
|
|
17
|
-
This tool demonstrates the standardized artifact format for binary data by:
|
|
18
|
-
1. Generating a simple image using PIL
|
|
19
|
-
2. Providing a confirmation message for the agent
|
|
20
|
-
3. Creating an image file artifact for the user
|
|
21
|
-
"""
|
|
22
|
-
name: str
|
|
23
|
-
description: str
|
|
24
|
-
args_schema: type[BaseModel]
|
|
@@ -1,21 +0,0 @@
|
|
|
1
|
-
from langchain_core.runnables import RunnableConfig as RunnableConfig
|
|
2
|
-
from langchain_core.tools import BaseTool
|
|
3
|
-
from pydantic import BaseModel
|
|
4
|
-
|
|
5
|
-
class CurrencyConfig(BaseModel):
|
|
6
|
-
"""Tool configuration schema."""
|
|
7
|
-
tenant_id: str
|
|
8
|
-
auth_key: str
|
|
9
|
-
|
|
10
|
-
class CurrencyInput(BaseModel):
|
|
11
|
-
"""Input schema for currency exchange."""
|
|
12
|
-
amount: float
|
|
13
|
-
from_currency: str
|
|
14
|
-
to_currency: str
|
|
15
|
-
|
|
16
|
-
class CurrencyExchangeTool(BaseTool):
|
|
17
|
-
"""Simple currency exchange tool with SDK auto-injection configuration."""
|
|
18
|
-
name: str
|
|
19
|
-
description: str
|
|
20
|
-
args_schema: type[BaseModel]
|
|
21
|
-
tool_config_schema: type[BaseModel]
|
|
@@ -1,23 +0,0 @@
|
|
|
1
|
-
from aip_agents.a2a.types import MimeType as MimeType
|
|
2
|
-
from aip_agents.utils.artifact_helpers import create_artifact_command as create_artifact_command
|
|
3
|
-
from langchain_core.tools import BaseTool
|
|
4
|
-
from pydantic import BaseModel
|
|
5
|
-
|
|
6
|
-
class GraphCommandInput(BaseModel):
|
|
7
|
-
"""Input schema for a tiny bar/line chart artifact tool.
|
|
8
|
-
|
|
9
|
-
Note: width/height are clamped to a small size to keep artifacts lightweight.
|
|
10
|
-
"""
|
|
11
|
-
chart_type: str
|
|
12
|
-
labels: list[str]
|
|
13
|
-
values: list[float]
|
|
14
|
-
title: str
|
|
15
|
-
width: int
|
|
16
|
-
height: int
|
|
17
|
-
image_name: str
|
|
18
|
-
|
|
19
|
-
class GraphArtifactCommandTool(BaseTool):
|
|
20
|
-
"""Generate a very small PNG chart and return it as a Command artifact."""
|
|
21
|
-
name: str
|
|
22
|
-
description: str
|
|
23
|
-
args_schema: type[BaseModel]
|
|
@@ -1,19 +0,0 @@
|
|
|
1
|
-
from _typeshed import Incomplete
|
|
2
|
-
from aip_agents.utils.logger_manager import LoggerManager as LoggerManager
|
|
3
|
-
from pydantic import BaseModel
|
|
4
|
-
|
|
5
|
-
logger: Incomplete
|
|
6
|
-
|
|
7
|
-
class WeatherToolInputSchema(BaseModel):
|
|
8
|
-
"""Schema for weather tool input."""
|
|
9
|
-
city: str
|
|
10
|
-
|
|
11
|
-
def weather_tool(city: str) -> str:
|
|
12
|
-
"""Gets the weather for a specified city.
|
|
13
|
-
|
|
14
|
-
Args:
|
|
15
|
-
city: The name of the city to get weather for.
|
|
16
|
-
|
|
17
|
-
Returns:
|
|
18
|
-
A string describing the weather conditions.
|
|
19
|
-
"""
|
|
@@ -1,43 +0,0 @@
|
|
|
1
|
-
from aip_agents.agent.langgraph_react_agent import LangGraphReactAgent as LangGraphReactAgent
|
|
2
|
-
from aip_agents.examples.tools.langchain_weather_tool import weather_tool as weather_tool
|
|
3
|
-
from aip_agents.examples.tools.time_tool import TimeTool as TimeTool
|
|
4
|
-
from langchain_core.runnables import RunnableConfig as RunnableConfig
|
|
5
|
-
from langchain_core.tools import BaseTool
|
|
6
|
-
from pydantic import BaseModel
|
|
7
|
-
from typing import Any
|
|
8
|
-
|
|
9
|
-
class StreamingToolConfig(BaseModel):
|
|
10
|
-
"""Tool configuration schema for the LangGraph streaming tool."""
|
|
11
|
-
time_format: str
|
|
12
|
-
|
|
13
|
-
class LangGraphStreamingToolInput(BaseModel):
|
|
14
|
-
"""Input schema for the LangGraphStreamingTool."""
|
|
15
|
-
query: str
|
|
16
|
-
|
|
17
|
-
class LangGraphStreamingTool(BaseTool):
|
|
18
|
-
"""Tool that wraps a LangGraph agent with time and weather forecast capabilities."""
|
|
19
|
-
name: str
|
|
20
|
-
description: str
|
|
21
|
-
args_schema: type[BaseModel]
|
|
22
|
-
tool_config_schema: type[BaseModel]
|
|
23
|
-
def __init__(self, model: Any = None, **kwargs) -> None:
|
|
24
|
-
"""Initialize the LangGraphStreamingTool.
|
|
25
|
-
|
|
26
|
-
Args:
|
|
27
|
-
model: The model to use for the agent.
|
|
28
|
-
**kwargs: Additional keyword arguments.
|
|
29
|
-
"""
|
|
30
|
-
@property
|
|
31
|
-
def agent(self):
|
|
32
|
-
"""Access the internal agent."""
|
|
33
|
-
async def arun_streaming(self, query: str = None, config: RunnableConfig = None, **kwargs):
|
|
34
|
-
"""Execute the LangGraph agent asynchronously with A2A streaming output.
|
|
35
|
-
|
|
36
|
-
Args:
|
|
37
|
-
query: The query to execute.
|
|
38
|
-
config: Tool configuration containing user context and preferences.
|
|
39
|
-
**kwargs: Additional keyword arguments.
|
|
40
|
-
|
|
41
|
-
Returns:
|
|
42
|
-
An async generator that yields streaming events.
|
|
43
|
-
"""
|
|
@@ -1,26 +0,0 @@
|
|
|
1
|
-
from _typeshed import Incomplete
|
|
2
|
-
from aip_agents.tools.base_bosa_tools import BaseBosaTool as BaseBosaTool
|
|
3
|
-
from aip_agents.tools.constants import Action as Action, ActionEndpointMap as ActionEndpointMap
|
|
4
|
-
from langchain_core.tools import BaseTool as LangchainBaseTool
|
|
5
|
-
from pydantic import BaseModel
|
|
6
|
-
|
|
7
|
-
logger: Incomplete
|
|
8
|
-
|
|
9
|
-
class GitHubPRDetailsSchema(BaseModel):
|
|
10
|
-
"""Schema for retrieving details of specific GitHub Pull Requests."""
|
|
11
|
-
repository: str
|
|
12
|
-
pr_numbers: list[int]
|
|
13
|
-
|
|
14
|
-
class GitHubPRDetailsTool(BaseBosaTool, LangchainBaseTool):
|
|
15
|
-
"""Tool for retrieving details of specific pull requests."""
|
|
16
|
-
name: str
|
|
17
|
-
description: str
|
|
18
|
-
args_schema: type[BaseModel]
|
|
19
|
-
def __init__(self, **kwargs) -> None:
|
|
20
|
-
"""Initialize the GitHubPRDetailsTool.
|
|
21
|
-
|
|
22
|
-
Calls both LangchainBaseTool and BaseBosaTool __init__ methods.
|
|
23
|
-
|
|
24
|
-
Args:
|
|
25
|
-
**kwargs: Additional keyword arguments for initialization.
|
|
26
|
-
"""
|
|
@@ -1,16 +0,0 @@
|
|
|
1
|
-
from _typeshed import Incomplete
|
|
2
|
-
from langchain_core.tools import BaseTool
|
|
3
|
-
from pydantic import BaseModel
|
|
4
|
-
|
|
5
|
-
logger: Incomplete
|
|
6
|
-
|
|
7
|
-
class GoogleSerperInput(BaseModel):
|
|
8
|
-
"""Input schema for the GoogleSerperTool."""
|
|
9
|
-
query: str
|
|
10
|
-
|
|
11
|
-
class MockGoogleSerperTool(BaseTool):
|
|
12
|
-
"""Mock Tool to simulate Google Serper API results for testing."""
|
|
13
|
-
name: str
|
|
14
|
-
description: str
|
|
15
|
-
save_output_history: bool
|
|
16
|
-
args_schema: type[BaseModel]
|
|
@@ -1,21 +0,0 @@
|
|
|
1
|
-
from _typeshed import Incomplete
|
|
2
|
-
from aip_agents.utils.logger_manager import LoggerManager as LoggerManager
|
|
3
|
-
from pydantic import BaseModel
|
|
4
|
-
from typing import Any
|
|
5
|
-
|
|
6
|
-
logger: Incomplete
|
|
7
|
-
|
|
8
|
-
class StockPriceInput(BaseModel):
|
|
9
|
-
"""Input for the stock price tool."""
|
|
10
|
-
symbol: str
|
|
11
|
-
|
|
12
|
-
def get_stock_price(symbol: str) -> dict[str, Any]:
|
|
13
|
-
"""Get current stock price and performance data for a given symbol."""
|
|
14
|
-
|
|
15
|
-
class StockNewsInput(BaseModel):
|
|
16
|
-
"""Input for the stock news tool."""
|
|
17
|
-
symbol: str
|
|
18
|
-
days: int
|
|
19
|
-
|
|
20
|
-
def get_stock_news(symbol: str, days: int = 7) -> dict[str, Any]:
|
|
21
|
-
"""Get recent news for a stock for a specified number of days."""
|
|
@@ -1,22 +0,0 @@
|
|
|
1
|
-
from aip_agents.a2a.types import MimeType as MimeType
|
|
2
|
-
from aip_agents.utils.artifact_helpers import create_artifact_response as create_artifact_response
|
|
3
|
-
from langchain_core.tools import BaseTool
|
|
4
|
-
from pydantic import BaseModel
|
|
5
|
-
|
|
6
|
-
class TableGeneratorInput(BaseModel):
|
|
7
|
-
"""Input schema for table generator tool."""
|
|
8
|
-
rows: int
|
|
9
|
-
columns: list[str]
|
|
10
|
-
table_name: str
|
|
11
|
-
|
|
12
|
-
class TableGeneratorTool(BaseTool):
|
|
13
|
-
"""Tool that generates sample data tables with artifact support.
|
|
14
|
-
|
|
15
|
-
This tool demonstrates the standardized artifact format by:
|
|
16
|
-
1. Generating sample data
|
|
17
|
-
2. Creating a markdown table for the agent's context
|
|
18
|
-
3. Creating a CSV file artifact for the user
|
|
19
|
-
"""
|
|
20
|
-
name: str
|
|
21
|
-
description: str
|
|
22
|
-
args_schema: type[BaseModel]
|
|
@@ -1,15 +0,0 @@
|
|
|
1
|
-
from langchain_core.tools import BaseTool
|
|
2
|
-
from pydantic import BaseModel
|
|
3
|
-
|
|
4
|
-
FORMAT_STRING: str
|
|
5
|
-
|
|
6
|
-
class TimeToolInput(BaseModel):
|
|
7
|
-
"""Input schema for the TimeTool."""
|
|
8
|
-
datetime_format: str
|
|
9
|
-
|
|
10
|
-
class TimeTool(BaseTool):
|
|
11
|
-
"""Tool to get the current time."""
|
|
12
|
-
name: str
|
|
13
|
-
description: str
|
|
14
|
-
save_output_history: bool
|
|
15
|
-
args_schema: type[BaseModel]
|
|
@@ -1,14 +0,0 @@
|
|
|
1
|
-
from _typeshed import Incomplete
|
|
2
|
-
|
|
3
|
-
DAILY_WEATHER: Incomplete
|
|
4
|
-
DayOfWeek: Incomplete
|
|
5
|
-
|
|
6
|
-
def get_weather_forecast(day: DayOfWeek) -> str:
|
|
7
|
-
"""Get the weather forecast for a specific day of the week.
|
|
8
|
-
|
|
9
|
-
Args:
|
|
10
|
-
day: The day of the week (monday, tuesday, wednesday, etc.)
|
|
11
|
-
|
|
12
|
-
Returns:
|
|
13
|
-
str: A weather forecast message including conditions and temperature.
|
|
14
|
-
"""
|
aip_agents/mcp/__init__.pyi
DELETED
|
File without changes
|
|
@@ -1,38 +0,0 @@
|
|
|
1
|
-
from gllm_tools.mcp.client.client import MCPClient
|
|
2
|
-
from gllm_tools.mcp.client.config import MCPConfiguration as MCPConfiguration
|
|
3
|
-
from gllm_tools.mcp.client.resource import MCPResource as MCPResource
|
|
4
|
-
from gllm_tools.mcp.client.tool import MCPTool as MCPTool
|
|
5
|
-
from mcp.types import CallToolResult as CallToolResult, EmbeddedResource, ImageContent
|
|
6
|
-
|
|
7
|
-
NonTextContent = ImageContent | EmbeddedResource
|
|
8
|
-
|
|
9
|
-
class GoogleADKMCPClient(MCPClient):
|
|
10
|
-
'''Google ADK MCP Client.
|
|
11
|
-
|
|
12
|
-
This client is a wrapper around the MCPClient that converts MCP tools and resources
|
|
13
|
-
into Google ADK-compatible FunctionTool instances and resources. It enables seamless
|
|
14
|
-
integration between MCP servers and ADK agents.
|
|
15
|
-
|
|
16
|
-
The client handles:
|
|
17
|
-
- Converting MCP tools to ADK FunctionTool instances
|
|
18
|
-
- Managing MCP server sessions and connections
|
|
19
|
-
- Converting MCP resources to ADK-compatible formats
|
|
20
|
-
- Handling tool execution and response formatting
|
|
21
|
-
|
|
22
|
-
Example:
|
|
23
|
-
```python
|
|
24
|
-
from gllm_tools.mcp.client.google_adk import GoogleADKMCPClient
|
|
25
|
-
from gllm_tools.mcp.client.config import MCPConfiguration
|
|
26
|
-
|
|
27
|
-
servers = {
|
|
28
|
-
"filesystem": MCPConfiguration(
|
|
29
|
-
command="npx",
|
|
30
|
-
args=["-y", "@modelcontextprotocol/server-filesystem", "/path/to/folder"]
|
|
31
|
-
)
|
|
32
|
-
}
|
|
33
|
-
|
|
34
|
-
client = GoogleADKMCPClient(servers)
|
|
35
|
-
tools = await client.get_tools() # Returns list of ADK FunctionTool instances
|
|
36
|
-
```
|
|
37
|
-
'''
|
|
38
|
-
RESOURCE_FETCH_TIMEOUT: int
|
aip_agents/memory/__init__.pyi
DELETED
|
File without changes
|
|
@@ -1,22 +0,0 @@
|
|
|
1
|
-
from aip_agents.memory.base import BaseMemory as BaseMemory, ChatMessage as ChatMessage
|
|
2
|
-
|
|
3
|
-
class SimpleMemory(BaseMemory):
|
|
4
|
-
"""A simple memory implementation that stores messages in a list."""
|
|
5
|
-
messages: list[ChatMessage]
|
|
6
|
-
def __init__(self) -> None:
|
|
7
|
-
"""Initialize the SimpleMemory instance with an empty message list."""
|
|
8
|
-
def add_message(self, message_or_role, content=None) -> None:
|
|
9
|
-
"""Add a message to memory.
|
|
10
|
-
|
|
11
|
-
This method supports two calling patterns for backward compatibility:
|
|
12
|
-
1. add_message(ChatMessage) - Adds a ChatMessage object directly
|
|
13
|
-
2. add_message(role, content) - Creates and adds a ChatMessage with the given role and content
|
|
14
|
-
|
|
15
|
-
Args:
|
|
16
|
-
message_or_role: Either a ChatMessage object or a string role
|
|
17
|
-
content: Optional content string when using the role+content pattern
|
|
18
|
-
"""
|
|
19
|
-
def get_messages(self) -> list[ChatMessage]:
|
|
20
|
-
"""Get all messages from memory."""
|
|
21
|
-
def clear(self) -> None:
|
|
22
|
-
"""Clear all messages from memory."""
|
aip_agents/sentry/__init__.pyi
DELETED