aip-agents-binary 0.6.3__py3-none-any.whl → 0.6.5__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.
Potentially problematic release.
This version of aip-agents-binary might be problematic. Click here for more details.
- aip_agents/agent/__init__.py +44 -4
- aip_agents/agent/base_langgraph_agent.py +26 -6
- aip_agents/agent/langgraph_memory_enhancer_agent.py +368 -34
- aip_agents/agent/langgraph_react_agent.py +30 -6
- aip_agents/mcp/client/__init__.py +38 -2
- aip_agents/mcp/client/transports.py +5 -1
- aip_agents/memory/adapters/base_adapter.py +94 -0
- aip_agents/sentry/sentry.py +29 -8
- aip_agents/tools/memory_search/__init__.py +8 -1
- aip_agents/tools/memory_search/mem0.py +108 -1
- aip_agents/tools/memory_search/schema.py +33 -0
- aip_agents/tools/memory_search_tool.py +8 -0
- {aip_agents_binary-0.6.3.dist-info → aip_agents_binary-0.6.5.dist-info}/METADATA +4 -16
- aip_agents_binary-0.6.5.dist-info/RECORD +314 -0
- aip_agents/__init__.pyi +0 -19
- aip_agents/a2a/__init__.pyi +0 -3
- aip_agents/a2a/server/__init__.pyi +0 -4
- aip_agents/a2a/server/base_executor.pyi +0 -73
- aip_agents/a2a/server/google_adk_executor.pyi +0 -51
- aip_agents/a2a/server/langflow_executor.pyi +0 -43
- aip_agents/a2a/server/langgraph_executor.pyi +0 -47
- aip_agents/a2a/types.pyi +0 -132
- aip_agents/agent/__init__.pyi +0 -9
- aip_agents/agent/base_agent.pyi +0 -221
- aip_agents/agent/base_langgraph_agent.pyi +0 -233
- aip_agents/agent/google_adk_agent.pyi +0 -141
- aip_agents/agent/google_adk_constants.pyi +0 -3
- aip_agents/agent/hitl/__init__.pyi +0 -6
- aip_agents/agent/hitl/config.pyi +0 -15
- aip_agents/agent/hitl/langgraph_hitl_mixin.pyi +0 -42
- aip_agents/agent/hitl/manager.pyi +0 -200
- aip_agents/agent/hitl/models.pyi +0 -3
- aip_agents/agent/hitl/prompt/__init__.pyi +0 -4
- aip_agents/agent/hitl/prompt/base.pyi +0 -24
- aip_agents/agent/hitl/prompt/deferred.pyi +0 -30
- aip_agents/agent/hitl/registry.pyi +0 -101
- aip_agents/agent/interface.pyi +0 -81
- aip_agents/agent/interfaces.pyi +0 -44
- aip_agents/agent/langflow_agent.pyi +0 -133
- aip_agents/agent/langgraph_memory_enhancer_agent.pyi +0 -49
- aip_agents/agent/langgraph_react_agent.pyi +0 -170
- aip_agents/agent/system_instruction_context.pyi +0 -13
- aip_agents/clients/__init__.pyi +0 -4
- aip_agents/clients/langflow/__init__.pyi +0 -4
- aip_agents/clients/langflow/client.pyi +0 -140
- aip_agents/clients/langflow/types.pyi +0 -7
- aip_agents/constants.pyi +0 -7
- aip_agents/examples/__init__.pyi +0 -0
- aip_agents/examples/compare_streaming_client.pyi +0 -48
- aip_agents/examples/compare_streaming_server.pyi +0 -18
- aip_agents/examples/demo_memory_recall.py +0 -401
- aip_agents/examples/demo_memory_recall.pyi +0 -58
- 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 -15
- 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 -15
- aip_agents/examples/hello_world_a2a_langchain_server.pyi +0 -15
- aip_agents/examples/hello_world_a2a_langchain_server_lm_invoker.pyi +0 -15
- aip_agents/examples/hello_world_a2a_langflow_client.pyi +0 -9
- aip_agents/examples/hello_world_a2a_langflow_server.pyi +0 -14
- 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 -16
- 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 -14
- aip_agents/examples/hello_world_a2a_langgraph_server_lm_invoker.pyi +0 -15
- aip_agents/examples/hello_world_a2a_langgraph_server_tool_streaming.pyi +0 -15
- aip_agents/examples/hello_world_a2a_mcp_langgraph.pyi +0 -48
- aip_agents/examples/hello_world_a2a_three_level_agent_hierarchy_client.pyi +0 -48
- aip_agents/examples/hello_world_a2a_three_level_agent_hierarchy_server.pyi +0 -45
- 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 -15
- 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_interactive.pyi +0 -16
- aip_agents/examples/hello_world_langchain_mcp_http_stream.pyi +0 -5
- aip_agents/examples/hello_world_langchain_mcp_multi_server.pyi +0 -18
- 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_langflow_agent.pyi +0 -35
- aip_agents/examples/hello_world_langgraph.pyi +0 -5
- aip_agents/examples/hello_world_langgraph_gl_connector_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 -30
- 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_ptc.pyi +0 -5
- aip_agents/examples/hello_world_sentry.pyi +0 -21
- aip_agents/examples/hello_world_step_limits.pyi +0 -17
- aip_agents/examples/hello_world_stock_a2a_server.pyi +0 -17
- aip_agents/examples/hello_world_tool_output_client.pyi +0 -5
- aip_agents/examples/hello_world_tool_output_server.pyi +0 -19
- aip_agents/examples/hitl_demo.pyi +0 -67
- aip_agents/examples/pii_demo_langgraph_client.pyi +0 -5
- aip_agents/examples/pii_demo_langgraph_server.pyi +0 -20
- aip_agents/examples/pii_demo_multi_agent_client.pyi +0 -5
- aip_agents/examples/pii_demo_multi_agent_server.pyi +0 -40
- aip_agents/examples/todolist_planning_a2a_langchain_client.pyi +0 -5
- aip_agents/examples/todolist_planning_a2a_langgraph_server.pyi +0 -19
- aip_agents/examples/tools/__init__.pyi +0 -9
- aip_agents/examples/tools/adk_arithmetic_tools.pyi +0 -24
- 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 -19
- aip_agents/examples/tools/image_artifact_tool.pyi +0 -26
- aip_agents/examples/tools/langchain_arithmetic_tools.pyi +0 -17
- aip_agents/examples/tools/langchain_currency_exchange_tool.pyi +0 -20
- aip_agents/examples/tools/langchain_graph_artifact_tool.pyi +0 -25
- aip_agents/examples/tools/langchain_weather_tool.pyi +0 -19
- aip_agents/examples/tools/langgraph_streaming_tool.pyi +0 -43
- aip_agents/examples/tools/mock_retrieval_tool.pyi +0 -13
- aip_agents/examples/tools/pii_demo_tools.pyi +0 -54
- aip_agents/examples/tools/random_chart_tool.pyi +0 -20
- aip_agents/examples/tools/serper_tool.pyi +0 -16
- aip_agents/examples/tools/stock_tools.pyi +0 -36
- 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/guardrails/__init__.pyi +0 -6
- aip_agents/guardrails/engines/__init__.pyi +0 -4
- aip_agents/guardrails/engines/base.pyi +0 -61
- aip_agents/guardrails/engines/nemo.pyi +0 -46
- aip_agents/guardrails/engines/phrase_matcher.pyi +0 -48
- aip_agents/guardrails/exceptions.pyi +0 -23
- aip_agents/guardrails/manager.pyi +0 -42
- aip_agents/guardrails/middleware.pyi +0 -87
- aip_agents/guardrails/schemas.pyi +0 -43
- aip_agents/guardrails/utils.pyi +0 -19
- aip_agents/mcp/__init__.pyi +0 -0
- aip_agents/mcp/client/__init__.pyi +0 -5
- aip_agents/mcp/client/base_mcp_client.pyi +0 -148
- aip_agents/mcp/client/connection_manager.pyi +0 -51
- aip_agents/mcp/client/google_adk/__init__.pyi +0 -3
- aip_agents/mcp/client/google_adk/client.pyi +0 -75
- aip_agents/mcp/client/langchain/__init__.pyi +0 -3
- aip_agents/mcp/client/langchain/client.pyi +0 -48
- aip_agents/mcp/client/persistent_session.pyi +0 -122
- aip_agents/mcp/client/session_pool.pyi +0 -101
- aip_agents/mcp/client/transports.pyi +0 -132
- aip_agents/mcp/utils/__init__.pyi +0 -0
- aip_agents/mcp/utils/config_validator.pyi +0 -82
- aip_agents/memory/__init__.pyi +0 -5
- aip_agents/memory/adapters/__init__.pyi +0 -4
- aip_agents/memory/adapters/base_adapter.pyi +0 -150
- aip_agents/memory/adapters/mem0.pyi +0 -22
- aip_agents/memory/base.pyi +0 -60
- aip_agents/memory/constants.pyi +0 -25
- aip_agents/memory/factory.pyi +0 -24
- aip_agents/memory/guidance.pyi +0 -3
- aip_agents/memory/simple_memory.pyi +0 -23
- aip_agents/middleware/__init__.pyi +0 -5
- aip_agents/middleware/base.pyi +0 -75
- aip_agents/middleware/manager.pyi +0 -84
- aip_agents/middleware/todolist.pyi +0 -125
- aip_agents/ptc/__init__.pyi +0 -10
- aip_agents/ptc/doc_gen.pyi +0 -40
- aip_agents/ptc/exceptions.pyi +0 -22
- aip_agents/ptc/executor.pyi +0 -73
- aip_agents/ptc/mcp/__init__.pyi +0 -7
- aip_agents/ptc/mcp/sandbox_bridge.pyi +0 -47
- aip_agents/ptc/mcp/templates/__init__.pyi +0 -0
- aip_agents/ptc/naming.pyi +0 -76
- aip_agents/ptc/payload.pyi +0 -15
- aip_agents/ptc/prompt_builder.pyi +0 -55
- aip_agents/ptc/ptc_helper.pyi +0 -1
- aip_agents/ptc/sandbox_bridge.pyi +0 -25
- aip_agents/ptc/template_utils.pyi +0 -13
- aip_agents/ptc/templates/__init__.pyi +0 -0
- aip_agents/sandbox/__init__.pyi +0 -5
- aip_agents/sandbox/defaults.pyi +0 -2
- aip_agents/sandbox/e2b_runtime.pyi +0 -51
- aip_agents/sandbox/template_builder.pyi +0 -36
- aip_agents/sandbox/types.pyi +0 -14
- aip_agents/sandbox/validation.pyi +0 -20
- aip_agents/schema/__init__.pyi +0 -9
- aip_agents/schema/a2a.pyi +0 -40
- aip_agents/schema/agent.pyi +0 -65
- aip_agents/schema/hitl.pyi +0 -89
- aip_agents/schema/langgraph.pyi +0 -28
- aip_agents/schema/model_id.pyi +0 -54
- aip_agents/schema/step_limit.pyi +0 -63
- aip_agents/schema/storage.pyi +0 -21
- aip_agents/sentry/__init__.pyi +0 -3
- aip_agents/sentry/sentry.pyi +0 -48
- aip_agents/storage/__init__.pyi +0 -8
- aip_agents/storage/base.pyi +0 -58
- aip_agents/storage/clients/__init__.pyi +0 -3
- aip_agents/storage/clients/minio_client.pyi +0 -137
- aip_agents/storage/config.pyi +0 -29
- aip_agents/storage/providers/__init__.pyi +0 -5
- aip_agents/storage/providers/base.pyi +0 -88
- aip_agents/storage/providers/memory.pyi +0 -79
- aip_agents/storage/providers/object_storage.pyi +0 -98
- aip_agents/tools/__init__.pyi +0 -11
- aip_agents/tools/browser_use/__init__.pyi +0 -14
- aip_agents/tools/browser_use/action_parser.pyi +0 -18
- aip_agents/tools/browser_use/browser_use_tool.pyi +0 -50
- aip_agents/tools/browser_use/llm_config.pyi +0 -52
- aip_agents/tools/browser_use/minio_storage.pyi +0 -109
- aip_agents/tools/browser_use/schemas.pyi +0 -32
- aip_agents/tools/browser_use/session.pyi +0 -4
- aip_agents/tools/browser_use/session_errors.pyi +0 -53
- aip_agents/tools/browser_use/steel_session_recording.pyi +0 -63
- aip_agents/tools/browser_use/streaming.pyi +0 -81
- aip_agents/tools/browser_use/structured_data_parser.pyi +0 -86
- aip_agents/tools/browser_use/structured_data_recovery.pyi +0 -43
- aip_agents/tools/browser_use/types.pyi +0 -45
- aip_agents/tools/code_sandbox/__init__.pyi +0 -3
- aip_agents/tools/code_sandbox/constant.pyi +0 -4
- aip_agents/tools/code_sandbox/e2b_cloud_sandbox_extended.pyi +0 -102
- aip_agents/tools/code_sandbox/e2b_sandbox_tool.pyi +0 -29
- aip_agents/tools/constants.pyi +0 -138
- aip_agents/tools/date_range_tool.pyi +0 -21
- aip_agents/tools/document_loader/__init__.pyi +0 -7
- aip_agents/tools/document_loader/base_reader.pyi +0 -75
- aip_agents/tools/document_loader/docx_reader_tool.pyi +0 -10
- aip_agents/tools/document_loader/excel_reader_tool.pyi +0 -26
- aip_agents/tools/document_loader/pdf_reader_tool.pyi +0 -11
- aip_agents/tools/document_loader/pdf_splitter.pyi +0 -18
- aip_agents/tools/execute_ptc_code.pyi +0 -90
- aip_agents/tools/gl_connector/__init__.pyi +0 -3
- aip_agents/tools/gl_connector/tool.pyi +0 -74
- aip_agents/tools/gl_connector_tools.pyi +0 -39
- aip_agents/tools/memory_search/__init__.pyi +0 -5
- aip_agents/tools/memory_search/base.pyi +0 -69
- aip_agents/tools/memory_search/mem0.pyi +0 -19
- aip_agents/tools/memory_search/schema.pyi +0 -15
- aip_agents/tools/memory_search_tool.pyi +0 -3
- aip_agents/tools/time_tool.pyi +0 -16
- aip_agents/tools/tool_config_injector.pyi +0 -26
- aip_agents/tools/web_search/__init__.pyi +0 -3
- aip_agents/tools/web_search/serper_tool.pyi +0 -19
- aip_agents/types/__init__.pyi +0 -36
- aip_agents/types/a2a_events.pyi +0 -3
- aip_agents/utils/__init__.pyi +0 -11
- aip_agents/utils/a2a_connector.pyi +0 -146
- aip_agents/utils/artifact_helpers.pyi +0 -203
- aip_agents/utils/constants.pyi +0 -10
- aip_agents/utils/datetime/__init__.pyi +0 -4
- aip_agents/utils/datetime/normalization.pyi +0 -95
- aip_agents/utils/datetime/timezone.pyi +0 -48
- aip_agents/utils/env_loader.pyi +0 -10
- aip_agents/utils/event_handler_registry.pyi +0 -23
- aip_agents/utils/file_prompt_utils.pyi +0 -21
- aip_agents/utils/final_response_builder.pyi +0 -34
- aip_agents/utils/formatter_llm_client.pyi +0 -71
- 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/base_tool_manager.pyi +0 -48
- aip_agents/utils/langgraph/tool_managers/delegation_tool_manager.pyi +0 -56
- aip_agents/utils/langgraph/tool_output_management.pyi +0 -329
- aip_agents/utils/logger.pyi +0 -60
- aip_agents/utils/metadata/__init__.pyi +0 -5
- aip_agents/utils/metadata/activity_metadata_helper.pyi +0 -25
- aip_agents/utils/metadata/activity_narrative/__init__.pyi +0 -7
- aip_agents/utils/metadata/activity_narrative/builder.pyi +0 -35
- aip_agents/utils/metadata/activity_narrative/constants.pyi +0 -10
- aip_agents/utils/metadata/activity_narrative/context.pyi +0 -32
- aip_agents/utils/metadata/activity_narrative/formatters.pyi +0 -48
- aip_agents/utils/metadata/activity_narrative/utils.pyi +0 -12
- aip_agents/utils/metadata/schemas/__init__.pyi +0 -4
- aip_agents/utils/metadata/schemas/activity_schema.pyi +0 -18
- aip_agents/utils/metadata/schemas/thinking_schema.pyi +0 -20
- aip_agents/utils/metadata/thinking_metadata_helper.pyi +0 -4
- aip_agents/utils/metadata_helper.pyi +0 -117
- aip_agents/utils/name_preprocessor/__init__.pyi +0 -6
- aip_agents/utils/name_preprocessor/base_name_preprocessor.pyi +0 -52
- aip_agents/utils/name_preprocessor/google_name_preprocessor.pyi +0 -38
- aip_agents/utils/name_preprocessor/name_preprocessor.pyi +0 -41
- aip_agents/utils/name_preprocessor/openai_name_preprocessor.pyi +0 -34
- aip_agents/utils/pii/__init__.pyi +0 -5
- aip_agents/utils/pii/pii_handler.pyi +0 -96
- aip_agents/utils/pii/pii_helper.pyi +0 -78
- aip_agents/utils/pii/uuid_deanonymizer_mapping.pyi +0 -73
- aip_agents/utils/reference_helper.pyi +0 -81
- aip_agents/utils/sse_chunk_transformer.pyi +0 -166
- aip_agents/utils/step_limit_manager.pyi +0 -112
- aip_agents/utils/token_usage_helper.pyi +0 -60
- aip_agents_binary-0.6.3.dist-info/RECORD +0 -614
- {aip_agents_binary-0.6.3.dist-info → aip_agents_binary-0.6.5.dist-info}/WHEEL +0 -0
- {aip_agents_binary-0.6.3.dist-info → aip_agents_binary-0.6.5.dist-info}/top_level.txt +0 -0
|
@@ -1,9 +0,0 @@
|
|
|
1
|
-
from _typeshed import Incomplete
|
|
2
|
-
from aip_agents.agent import LangGraphAgent as LangGraphAgent
|
|
3
|
-
from aip_agents.schema.agent import A2AClientConfig as A2AClientConfig
|
|
4
|
-
from aip_agents.utils.logger import get_logger as get_logger
|
|
5
|
-
|
|
6
|
-
logger: Incomplete
|
|
7
|
-
|
|
8
|
-
async def main() -> None:
|
|
9
|
-
"""Main function demonstrating the General Assistant agent with streaming A2A capabilities."""
|
|
@@ -1,5 +0,0 @@
|
|
|
1
|
-
from aip_agents.agent import LangChainAgent as LangChainAgent
|
|
2
|
-
from aip_agents.schema.agent import A2AClientConfig as A2AClientConfig
|
|
3
|
-
|
|
4
|
-
async def main() -> None:
|
|
5
|
-
"""Main function demonstrating the General Assistant agent with streaming A2A time/weather capabilities."""
|
|
@@ -1,14 +0,0 @@
|
|
|
1
|
-
from _typeshed import Incomplete
|
|
2
|
-
from aip_agents.agent import LangGraphAgent as LangGraphAgent
|
|
3
|
-
from aip_agents.utils.logger import get_logger as get_logger
|
|
4
|
-
|
|
5
|
-
logger: Incomplete
|
|
6
|
-
SERVER_AGENT_NAME: str
|
|
7
|
-
|
|
8
|
-
def main(host: str, port: int):
|
|
9
|
-
"""Runs the LangGraph Weather A2A server.
|
|
10
|
-
|
|
11
|
-
Args:
|
|
12
|
-
host (str): Host to bind the server to.
|
|
13
|
-
port (int): Port to bind the server to.
|
|
14
|
-
"""
|
|
@@ -1,15 +0,0 @@
|
|
|
1
|
-
from _typeshed import Incomplete
|
|
2
|
-
from aip_agents.agent import LangChainAgent as LangChainAgent
|
|
3
|
-
from aip_agents.examples.tools.langchain_weather_tool import weather_tool as weather_tool
|
|
4
|
-
from aip_agents.utils.logger import get_logger as get_logger
|
|
5
|
-
|
|
6
|
-
logger: Incomplete
|
|
7
|
-
SERVER_AGENT_NAME: str
|
|
8
|
-
|
|
9
|
-
def main(host: str, port: int):
|
|
10
|
-
"""Runs the LangChain Weather A2A server.
|
|
11
|
-
|
|
12
|
-
Args:
|
|
13
|
-
host (str): Host to bind the server to.
|
|
14
|
-
port (int): Port to bind the server to.
|
|
15
|
-
"""
|
|
@@ -1,15 +0,0 @@
|
|
|
1
|
-
from _typeshed import Incomplete
|
|
2
|
-
from aip_agents.agent import LangChainAgent as LangChainAgent
|
|
3
|
-
from aip_agents.examples.tools.langgraph_streaming_tool import LangGraphStreamingTool as LangGraphStreamingTool
|
|
4
|
-
from aip_agents.utils.logger import get_logger as get_logger
|
|
5
|
-
|
|
6
|
-
logger: Incomplete
|
|
7
|
-
SERVER_AGENT_NAME: str
|
|
8
|
-
|
|
9
|
-
def main(host: str, port: int):
|
|
10
|
-
"""Runs the LangGraph Agent Wrapper A2A server.
|
|
11
|
-
|
|
12
|
-
Args:
|
|
13
|
-
host (str): Host to bind the server to.
|
|
14
|
-
port (int): Port to bind the server to.
|
|
15
|
-
"""
|
|
@@ -1,48 +0,0 @@
|
|
|
1
|
-
from aip_agents.agent import LangGraphAgent as LangGraphAgent
|
|
2
|
-
from aip_agents.schema.agent import A2AClientConfig as A2AClientConfig
|
|
3
|
-
|
|
4
|
-
A2A_SERVER_PORT: int
|
|
5
|
-
A2A_AGENT_NAME: str
|
|
6
|
-
MCP_SERVER_PORT: int
|
|
7
|
-
MCP_AGENT_NAME: str
|
|
8
|
-
MODEL_NAME: str
|
|
9
|
-
|
|
10
|
-
async def create_agent() -> tuple[LangGraphAgent, dict[str, bool]]:
|
|
11
|
-
"""Create and configure the LangGraphAgent with MCP and A2A connections.
|
|
12
|
-
|
|
13
|
-
Returns:
|
|
14
|
-
tuple: (agent, connected_servers) where connected_servers is a dict
|
|
15
|
-
showing which servers were successfully connected.
|
|
16
|
-
"""
|
|
17
|
-
async def connect_to_mcp_server(agent: LangGraphAgent) -> bool:
|
|
18
|
-
"""Connect to the MCP server and register tools.
|
|
19
|
-
|
|
20
|
-
Args:
|
|
21
|
-
agent: The LangGraphAgent instance to configure.
|
|
22
|
-
|
|
23
|
-
Returns:
|
|
24
|
-
bool: True if connection was successful, False otherwise.
|
|
25
|
-
"""
|
|
26
|
-
async def connect_to_a2a_server(agent: LangGraphAgent) -> bool:
|
|
27
|
-
"""Connect to the A2A server and register agents.
|
|
28
|
-
|
|
29
|
-
Args:
|
|
30
|
-
agent: The LangGraphAgent instance to configure.
|
|
31
|
-
|
|
32
|
-
Returns:
|
|
33
|
-
bool: True if connection was successful, False otherwise.
|
|
34
|
-
"""
|
|
35
|
-
async def test_mcp_weather_tool(agent: LangGraphAgent) -> None:
|
|
36
|
-
"""Test the MCP weather tool if available.
|
|
37
|
-
|
|
38
|
-
Args:
|
|
39
|
-
agent: The configured LangGraphAgent instance.
|
|
40
|
-
"""
|
|
41
|
-
async def test_stock_agent(agent: LangGraphAgent) -> None:
|
|
42
|
-
"""Test the A2A Stock Agent if available.
|
|
43
|
-
|
|
44
|
-
Args:
|
|
45
|
-
agent: The configured LangGraphAgent instance.
|
|
46
|
-
"""
|
|
47
|
-
async def main() -> None:
|
|
48
|
-
"""Main function to run the A2A and MCP integration example."""
|
|
@@ -1,48 +0,0 @@
|
|
|
1
|
-
from aip_agents.agent import LangGraphAgent as LangGraphAgent
|
|
2
|
-
from aip_agents.schema.agent import A2AClientConfig as A2AClientConfig
|
|
3
|
-
|
|
4
|
-
async def main() -> None:
|
|
5
|
-
"""Main function demonstrating the Three-Level Agent Hierarchy client."""
|
|
6
|
-
async def test_research_task(client_agent, coordinator_card) -> None:
|
|
7
|
-
"""Test a research-only task that flows through research specialists and workers.
|
|
8
|
-
|
|
9
|
-
Args:
|
|
10
|
-
client_agent: The client agent to send the query with.
|
|
11
|
-
coordinator_card: The agent card of the hierarchical coordinator.
|
|
12
|
-
"""
|
|
13
|
-
async def test_content_task(client_agent, coordinator_card) -> None:
|
|
14
|
-
"""Test a content-only task that flows through content specialists and workers.
|
|
15
|
-
|
|
16
|
-
Args:
|
|
17
|
-
client_agent: The client agent to send the query with.
|
|
18
|
-
coordinator_card: The agent card of the hierarchical coordinator.
|
|
19
|
-
"""
|
|
20
|
-
async def test_complex_task(client_agent, coordinator_card) -> None:
|
|
21
|
-
"""Test a complex task requiring both research and content specialists.
|
|
22
|
-
|
|
23
|
-
Args:
|
|
24
|
-
client_agent: The client agent to send the query with.
|
|
25
|
-
coordinator_card: The agent card of the hierarchical coordinator.
|
|
26
|
-
"""
|
|
27
|
-
async def test_artifact_generation(client_agent, coordinator_card) -> None:
|
|
28
|
-
"""Test artifact generation across the hierarchy.
|
|
29
|
-
|
|
30
|
-
Args:
|
|
31
|
-
client_agent: The client agent to send the query with.
|
|
32
|
-
coordinator_card: The agent card of the hierarchical coordinator.
|
|
33
|
-
"""
|
|
34
|
-
def handle_artifacts(result, test_name) -> None:
|
|
35
|
-
"""Handle and display information about artifacts from the result.
|
|
36
|
-
|
|
37
|
-
Args:
|
|
38
|
-
result: The result from the agent containing artifacts.
|
|
39
|
-
test_name: The name of the test for display purposes.
|
|
40
|
-
"""
|
|
41
|
-
def print_artifact_info(artifact: dict) -> None:
|
|
42
|
-
"""Print information about an artifact.
|
|
43
|
-
|
|
44
|
-
Args:
|
|
45
|
-
artifact: Dictionary containing artifact information.
|
|
46
|
-
"""
|
|
47
|
-
def print_hierarchy_info() -> None:
|
|
48
|
-
"""Print information about the 3-level hierarchy being tested."""
|
|
@@ -1,45 +0,0 @@
|
|
|
1
|
-
from _typeshed import Incomplete
|
|
2
|
-
from aip_agents.agent import LangGraphAgent as LangGraphAgent
|
|
3
|
-
from aip_agents.examples.tools.image_artifact_tool import ImageArtifactTool as ImageArtifactTool
|
|
4
|
-
from aip_agents.examples.tools.table_generator_tool import TableGeneratorTool as TableGeneratorTool
|
|
5
|
-
from aip_agents.utils.logger import get_logger as get_logger
|
|
6
|
-
|
|
7
|
-
logger: Incomplete
|
|
8
|
-
SERVER_AGENT_NAME: str
|
|
9
|
-
|
|
10
|
-
def create_worker_agents(llm) -> tuple[LangGraphAgent, LangGraphAgent, LangGraphAgent, LangGraphAgent]:
|
|
11
|
-
"""Create Level 3 worker agents that perform atomic operations.
|
|
12
|
-
|
|
13
|
-
Args:
|
|
14
|
-
llm: The language model to use for the worker agents.
|
|
15
|
-
|
|
16
|
-
Returns:
|
|
17
|
-
tuple[LangGraphAgent, LangGraphAgent, LangGraphAgent, LangGraphAgent]: Tuple of (web_search_worker, data_analysis_worker, writing_worker, formatting_worker).
|
|
18
|
-
"""
|
|
19
|
-
def create_specialist_agents(llm, worker_agents) -> tuple[LangGraphAgent, LangGraphAgent]:
|
|
20
|
-
"""Create Level 2 specialist agents that coordinate worker agents.
|
|
21
|
-
|
|
22
|
-
Args:
|
|
23
|
-
llm: The language model to use for the specialist agents.
|
|
24
|
-
worker_agents: Tuple of worker agents from create_worker_agents function.
|
|
25
|
-
|
|
26
|
-
Returns:
|
|
27
|
-
tuple[LangGraphAgent, LangGraphAgent]: Tuple of (research_specialist, content_specialist).
|
|
28
|
-
"""
|
|
29
|
-
def create_coordinator_agent(llm, specialist_agents) -> LangGraphAgent:
|
|
30
|
-
"""Create Level 1 coordinator agent that orchestrates everything.
|
|
31
|
-
|
|
32
|
-
Args:
|
|
33
|
-
llm: The language model to use for the coordinator agent.
|
|
34
|
-
specialist_agents: Tuple of specialist agents from create_specialist_agents function.
|
|
35
|
-
|
|
36
|
-
Returns:
|
|
37
|
-
LangGraphAgent: The configured coordinator agent.
|
|
38
|
-
"""
|
|
39
|
-
def main(host: str, port: int):
|
|
40
|
-
"""Runs the Three-Level Agent Hierarchy A2A server.
|
|
41
|
-
|
|
42
|
-
Args:
|
|
43
|
-
host (str): Host to bind the server to.
|
|
44
|
-
port (int): Port to bind the server to.
|
|
45
|
-
"""
|
|
@@ -1,15 +0,0 @@
|
|
|
1
|
-
from _typeshed import Incomplete
|
|
2
|
-
from aip_agents.agent import LangGraphReactAgent as LangGraphReactAgent
|
|
3
|
-
from aip_agents.examples.hello_world_a2a_langchain_server import SERVER_AGENT_NAME as SERVER_AGENT_NAME
|
|
4
|
-
from aip_agents.examples.tools.langchain_currency_exchange_tool import CurrencyExchangeTool as CurrencyExchangeTool
|
|
5
|
-
from aip_agents.utils.logger import get_logger as get_logger
|
|
6
|
-
|
|
7
|
-
logger: Incomplete
|
|
8
|
-
|
|
9
|
-
def main(host: str, port: int):
|
|
10
|
-
"""Runs the LangChain Currency Exchange A2A server.
|
|
11
|
-
|
|
12
|
-
Args:
|
|
13
|
-
host (str): Host to bind the server to.
|
|
14
|
-
port (int): Port to bind the server to.
|
|
15
|
-
"""
|
|
@@ -1,5 +0,0 @@
|
|
|
1
|
-
from aip_agents.agent.google_adk_agent import GoogleADKAgent as GoogleADKAgent
|
|
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 GoogleADKAgent with MCP tools via Streamable HTTP transport."""
|
|
@@ -1,5 +0,0 @@
|
|
|
1
|
-
from aip_agents.agent.google_adk_agent import GoogleADKAgent as GoogleADKAgent
|
|
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 GoogleADKAgent with MCP tools via Streamable HTTP transport and streaming."""
|
|
@@ -1,5 +0,0 @@
|
|
|
1
|
-
from aip_agents.agent.google_adk_agent import GoogleADKAgent as GoogleADKAgent
|
|
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 GoogleADKAgent with MCP tools via SSE transport and streaming."""
|
|
@@ -1,5 +0,0 @@
|
|
|
1
|
-
from aip_agents.agent.google_adk_agent import GoogleADKAgent as GoogleADKAgent
|
|
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 GoogleADKAgent with MCP tools via stdio transport."""
|
|
@@ -1,5 +0,0 @@
|
|
|
1
|
-
from aip_agents.agent.google_adk_agent import GoogleADKAgent as GoogleADKAgent
|
|
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 GoogleADKAgent with MCP tools via stdio transport and streaming."""
|
|
@@ -1,5 +0,0 @@
|
|
|
1
|
-
from aip_agents.agent.google_adk_agent import GoogleADKAgent as GoogleADKAgent
|
|
2
|
-
from aip_agents.examples.tools.adk_arithmetic_tools import add_numbers as add_numbers
|
|
3
|
-
|
|
4
|
-
async def google_adk_example_stream() -> None:
|
|
5
|
-
"""Demonstrates the GoogleADKAgent's arun_stream method."""
|
|
@@ -1,16 +0,0 @@
|
|
|
1
|
-
from aip_agents.agent import LangChainAgent as LangChainAgent
|
|
2
|
-
from aip_agents.utils.env_loader import load_local_env as load_local_env
|
|
3
|
-
|
|
4
|
-
def print_help() -> None:
|
|
5
|
-
"""Prints available commands and their descriptions."""
|
|
6
|
-
def build_mcp_config_from_env() -> dict:
|
|
7
|
-
"""Construct MCP configuration from environment variables.
|
|
8
|
-
|
|
9
|
-
Returns:
|
|
10
|
-
dict: MCP configuration dictionary.
|
|
11
|
-
|
|
12
|
-
Raises:
|
|
13
|
-
RuntimeError: If required environment variables are missing.
|
|
14
|
-
"""
|
|
15
|
-
async def main() -> None:
|
|
16
|
-
"""Runs the interactive LangChainAgent with MCP tools via Streamable HTTP transport."""
|
|
@@ -1,5 +0,0 @@
|
|
|
1
|
-
from aip_agents.agent import LangChainAgent as LangChainAgent
|
|
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 LangChainAgent with MCP tools via Streamable HTTP transport and streaming capabilities."""
|
|
@@ -1,18 +0,0 @@
|
|
|
1
|
-
from aip_agents.agent import LangChainAgent as LangChainAgent
|
|
2
|
-
from aip_agents.examples.mcp_configs.configs import mcp_tool_whitelisting_demo as mcp_tool_whitelisting_demo
|
|
3
|
-
from aip_agents.schema.a2a import A2AStreamEventType as A2AStreamEventType
|
|
4
|
-
from aip_agents.utils.logger import logger_manager as logger_manager
|
|
5
|
-
|
|
6
|
-
async def process_query(agent, query: str, is_allowed: bool = True) -> bool:
|
|
7
|
-
"""Process a single query and report if tools were called.
|
|
8
|
-
|
|
9
|
-
Args:
|
|
10
|
-
agent: The LangChain agent to run.
|
|
11
|
-
query: The query to process.
|
|
12
|
-
is_allowed: Whether this query should use allowed tools.
|
|
13
|
-
|
|
14
|
-
Returns:
|
|
15
|
-
True if at least one tool was called while processing the query, False otherwise.
|
|
16
|
-
"""
|
|
17
|
-
async def main() -> None:
|
|
18
|
-
"""Demo allowed_tools filtering with 3 MCP servers (STDIO, SSE, HTTP)."""
|
|
@@ -1,5 +0,0 @@
|
|
|
1
|
-
from aip_agents.agent import LangChainAgent as LangChainAgent
|
|
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 LangChainAgent with MCP tools via SSE transport and streaming capabilities."""
|
|
@@ -1,5 +0,0 @@
|
|
|
1
|
-
from aip_agents.agent import LangChainAgent as LangChainAgent
|
|
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 LangChainAgent with MCP tools via stdio transport and streaming capabilities."""
|
|
@@ -1,5 +0,0 @@
|
|
|
1
|
-
from aip_agents.agent import LangChainAgent as LangChainAgent
|
|
2
|
-
from aip_agents.examples.tools.langchain_arithmetic_tools import add_numbers as add_numbers
|
|
3
|
-
|
|
4
|
-
async def langchain_stream_example() -> None:
|
|
5
|
-
"""Demonstrates the LangChainAgent's arun_stream method with async execution."""
|
|
@@ -1,5 +0,0 @@
|
|
|
1
|
-
from aip_agents.agent import LangChainAgent as LangChainAgent
|
|
2
|
-
from aip_agents.examples.tools.langchain_arithmetic_tools import add_numbers as add_numbers
|
|
3
|
-
|
|
4
|
-
async def langchain_stream_example() -> None:
|
|
5
|
-
"""Demonstrates the LangChainAgent's arun_stream method with async execution."""
|
|
@@ -1,35 +0,0 @@
|
|
|
1
|
-
from aip_agents.agent.langflow_agent import LangflowAgent as LangflowAgent
|
|
2
|
-
from aip_agents.clients.langflow import LangflowApiClient as LangflowApiClient
|
|
3
|
-
|
|
4
|
-
async def fetch_flow_id() -> tuple[str, str]:
|
|
5
|
-
"""Fetch available flows and return the first flow ID and name."""
|
|
6
|
-
async def create_agent(flow_id: str, flow_name: str) -> LangflowAgent:
|
|
7
|
-
"""Create and configure the Langflow agent.
|
|
8
|
-
|
|
9
|
-
Args:
|
|
10
|
-
flow_id (str): The flow ID to use for the agent.
|
|
11
|
-
flow_name (str): The name of the flow.
|
|
12
|
-
|
|
13
|
-
Returns:
|
|
14
|
-
LangflowAgent: The configured Langflow agent.
|
|
15
|
-
"""
|
|
16
|
-
async def demonstrate_regular_execution(agent: LangflowAgent) -> None:
|
|
17
|
-
"""Demonstrate regular execution.
|
|
18
|
-
|
|
19
|
-
Args:
|
|
20
|
-
agent (LangflowAgent): The Langflow agent to demonstrate with.
|
|
21
|
-
"""
|
|
22
|
-
async def demonstrate_streaming(agent: LangflowAgent) -> None:
|
|
23
|
-
"""Demonstrate streaming execution.
|
|
24
|
-
|
|
25
|
-
Args:
|
|
26
|
-
agent (LangflowAgent): The Langflow agent to demonstrate with.
|
|
27
|
-
"""
|
|
28
|
-
async def demonstrate_session_management(agent: LangflowAgent) -> None:
|
|
29
|
-
"""Demonstrate session management.
|
|
30
|
-
|
|
31
|
-
Args:
|
|
32
|
-
agent (LangflowAgent): The Langflow agent to demonstrate with.
|
|
33
|
-
"""
|
|
34
|
-
async def main() -> None:
|
|
35
|
-
"""Demonstrate basic Langflow agent usage."""
|
|
@@ -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,30 +0,0 @@
|
|
|
1
|
-
from _typeshed import Incomplete
|
|
2
|
-
from aip_agents.agent import LangChainAgent as LangChainAgent
|
|
3
|
-
from aip_agents.schema.agent import AgentConfig as AgentConfig
|
|
4
|
-
|
|
5
|
-
DEEPINFRA_URL: str
|
|
6
|
-
MODEL_IDS: Incomplete
|
|
7
|
-
MODEL_CONFIGS: Incomplete
|
|
8
|
-
|
|
9
|
-
def make_agent(model_id):
|
|
10
|
-
"""Makes an agent with the given model id.
|
|
11
|
-
|
|
12
|
-
Args:
|
|
13
|
-
model_id: The model identifier to use for the agent.
|
|
14
|
-
|
|
15
|
-
Returns:
|
|
16
|
-
LangChainAgent: The configured agent instance.
|
|
17
|
-
"""
|
|
18
|
-
def print_help() -> None:
|
|
19
|
-
"""Prints available commands and their descriptions."""
|
|
20
|
-
def handle_switch_model(current_model):
|
|
21
|
-
"""Handles model switching. Returns (new_model, new_agent).
|
|
22
|
-
|
|
23
|
-
Args:
|
|
24
|
-
current_model: The currently active model identifier.
|
|
25
|
-
|
|
26
|
-
Returns:
|
|
27
|
-
tuple: (new_model, new_agent) where new_model is the selected model ID and new_agent is the configured agent instance.
|
|
28
|
-
"""
|
|
29
|
-
def main() -> None:
|
|
30
|
-
"""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 import get_logger as get_logger
|
|
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,17 +0,0 @@
|
|
|
1
|
-
from aip_agents.agent import LangGraphReactAgent as LangGraphReactAgent
|
|
2
|
-
from aip_agents.examples.tools.langchain_arithmetic_tools import add_numbers as add_numbers
|
|
3
|
-
from aip_agents.schema.step_limit import MaxStepsExceededError as MaxStepsExceededError, StepLimitConfig as StepLimitConfig
|
|
4
|
-
from aip_agents.utils.step_limit_manager import StepLimitManager as StepLimitManager
|
|
5
|
-
|
|
6
|
-
def print_config_validation(config: StepLimitConfig | None, context: str):
|
|
7
|
-
"""Print step limit configuration validation."""
|
|
8
|
-
async def test_case_1_success_no_limits() -> None:
|
|
9
|
-
"""Test Case 1: Success - High limits, no warnings."""
|
|
10
|
-
async def test_case_2_parallel_tool_batch_accounting() -> None:
|
|
11
|
-
"""Test Case 2: Demonstrate that parallel tool batches consume multiple steps."""
|
|
12
|
-
async def test_case_3_step_limit_exceeded() -> None:
|
|
13
|
-
"""Test Case 3: Step Limit Exceeded - Only 2 steps allowed."""
|
|
14
|
-
async def test_case_4_delegation_depth_exceeded() -> None:
|
|
15
|
-
"""Test Case 4: Delegation Depth Exceeded - Zero depth."""
|
|
16
|
-
async def main() -> None:
|
|
17
|
-
"""Run all test cases."""
|