MCPcopy Create free account

hub / github.com/JoshuaC215/agent-service-toolkit / functions

Functions316 in github.com/JoshuaC215/agent-service-toolkit

Method__init__
Initialize the client. Args: base_url (str): The base URL of the agent service. agent (str): The name of the
src/client/client.py:26
Method__init__
(self, conn_string: str, db_name: str)
src/memory/mongodb.py:66
Method__init__
(self)
src/memory/sqlite.py:17
Method__init__
(self, responses: list[str])
src/core/llm.py:46
Method__init__
Initialize STT with specified provider. Args: provider: Provider name ("openai", "deepgram", etc.) api_key: API key (
src/voice/stt.py:31
Method__init__
Initialize TTS with specified provider. Args: provider: Provider name ("openai", "elevenlabs", etc.) api_key: API key
src/voice/tts.py:31
Method__init__
Initialize VoiceManager. Args: stt: SpeechToText instance (None to disable STT) tts: TextToSpeech instance (None to d
src/voice/manager.py:35
Method__init__
Initialize OpenAI STT. Args: api_key: OpenAI API key (uses env var if not provided) Raises: Exception: If Op
src/voice/providers/openai_stt.py:14
Method__init__
Initialize OpenAI TTS. Args: api_key: OpenAI API key (uses env var if not provided) voice: Voice name (alloy, echo, f
src/voice/providers/openai_tts.py:21
Method__init__
(self)
src/schema/task_data.py:38
Method__init__
(self)
src/agents/safeguard.py:91
Method__init__
Initialize the agent.
src/agents/lazy_agent.py:12
Method__init__
(self)
src/agents/github_mcp_agent/github_mcp_agent.py:46
Method__init__
(self, task_name: str, writer: StreamWriter | None = None)
src/agents/bg_task_agent/task.py:12
Method__init__
(self, responses: list[BaseMessage])
tests/service/test_service_message_generator.py:15
Method__init__
(self)
tests/agents/test_lazy_agent.py:13
Method_connect
()
src/memory/mongodb.py:72
Method_create_graph
Test graph creation.
tests/agents/test_lazy_agent.py:20
Method_headers
(self)
src/client/client.py:54
Function_sse_response_example
()
src/service/service.py:343
Functionacall_model
(state: AgentState, config: RunnableConfig)
src/agents/research_assistant.py:70
Functionacall_model
(state: AgentState, config: RunnableConfig)
src/agents/rag_assistant.py:66
Functionacall_model
Generate a response based on the retrieved documents.
src/agents/knowledge_base_agent.py:147
Functionacall_model
(state: AgentState, config: RunnableConfig)
src/agents/bg_task_agent/bg_task_agent.py:28
Functionadd
Add two numbers.
src/agents/langgraph_supervisor_agent.py:11
Functionagent_client
Fixture for creating a test client with a clean environment.
tests/client/conftest.py:7
Functionagent_lookup
(agent_id)
tests/service/test_service_e2e.py:99
Functionagent_lookup
(agent_id)
tests/service/test_service.py:48
Functionbackground
This node is to demonstrate doing work before the interrupt
src/agents/interrupt_agent.py:47
Functionbg_task
(state: AgentState, writer: StreamWriter)
src/agents/bg_task_agent/bg_task_agent.py:37
Methodbind_tools
(self, tools, **kwargs)
tests/service/test_service_message_generator.py:18
Functionblock_unsafe_content
(state: AgentState, config: RunnableConfig)
src/agents/research_assistant.py:94
Functionblock_unsafe_content
(state: AgentState, config: RunnableConfig)
src/agents/rag_assistant.py:90
Functioncalculator_func
Calculates a math expression using numexpr. Useful for when you need to answer questions about math using numexpr. This tool is only for math
src/agents/tools.py:10
Functionchatbot
( inputs: dict[str, list[BaseMessage]], *, previous: dict[str, list[BaseMessage]], config: Run
src/agents/chatbot.py:9
Functioncheck_safety
(state: AgentState)
src/agents/research_assistant.py:109
Functioncheck_safety
(state: AgentState)
src/agents/rag_assistant.py:105
Functioncreate_system_message
(state)
src/agents/knowledge_base_agent.py:50
Functioncustom_generate_unique_id
Generate idiomatic operation IDs for OpenAPI client generation.
src/service/service.py:48
Functiondatabase_search_func
Searches chroma_db for information in the company's handbook.
src/agents/tools.py:65
Functiondetermine_birthdate
This node examines the conversation history to determine user's birthdate, checking store first.
src/agents/interrupt_agent.py:77
Functionfake_get_agent
(agent_key: str)
tests/service/test_service_lifespan.py:48
Functionfake_initialize_database
()
tests/service/test_service_lifespan.py:32
Functionfake_initialize_store
()
tests/service/test_service_lifespan.py:36
Functionfake_load_agent
(agent_key: str)
tests/service/test_service_lifespan.py:44
Functionfeedback
Record feedback for a run to LangSmith. This is a simple wrapper for the LangSmith create_feedback API, so the credentials can be stored
src/service/service.py:382
Methodfrom_env
Create TTS from environment variables. Reads VOICE_TTS_PROVIDER env var to determine which provider to use. Returns None if not confi
src/voice/tts.py:124
Methodfrom_env
Create VoiceManager from environment variables. Reads VOICE_STT_PROVIDER and VOICE_TTS_PROVIDER to configure speech-to-text and text-
src/voice/manager.py:51
Functiongenerate_response
Generates the final response based on the user's query and the available birthdate.
src/agents/interrupt_agent.py:190
Methodget_format
Get audio format (MIME type). Returns: MIME type string for generated audio
src/voice/providers/openai_tts.py:118
Functionhealth_check
Health check endpoint.
src/service/service.py:421
Functionhistory
Get chat history.
src/service/service.py:402
Functioninfo
()
src/service/service.py:114
Functioninvoke
Invoke an agent with user input to retrieve a final response. If agent_id is not provided, the default agent will be used. Use thread_id
src/service/service.py:184
Functionlifespan
Configurable lifespan that initializes the appropriate database checkpointer, store, and agents with async loading - for example for starting
src/service/service.py:67
Functionmock_agent
Fixture to create a mock agent that can be configured for different test scenarios.
tests/service/conftest.py:17
Functionmock_agent_client
Fixture for creating a mock AgentClient with a clean environment.
tests/app/conftest.py:10
Functionmock_astream
(**kwargs)
tests/service/test_service.py:239
Functionmock_audio_file
BytesIO mock audio file for STT tests.
tests/voice/conftest.py:26
Functionmock_database_settings
Fixture to ensure database settings are clean
tests/service/test_service_e2e.py:84
Functionmock_env
Fixture to ensure environment is clean for each test.
tests/conftest.py:26
Functionmock_get
(url: str, **kwargs)
tests/service/conftest.py:46
Functionmock_httpx
Patch httpx.stream and httpx.get to use our test client.
tests/service/conftest.py:36
Functionmock_openai_client
Mock OpenAI client for TTS/STT tests.
tests/voice/conftest.py:10
Functionmock_settings
Fixture to ensure settings are clean for each test.
tests/service/conftest.py:29
Functionmock_stream
(method: str, url: str, **kwargs)
tests/service/conftest.py:41
Methodmodel_post_init
(self, __context: Any)
src/core/settings.py:155
Functionmulti_agent_messages
Fixture providing reusable messages for multi-agent tests
tests/app/test_streamlit_app.py:178
Functionmultiply
Multiply two numbers.
src/agents/langgraph_supervisor_agent.py:16
Functionnode_a
(state: AgentState)
src/agents/command_agent.py:19
Functionnode_b
(state: AgentState)
src/agents/command_agent.py:38
Functionnode_c
(state: AgentState)
src/agents/command_agent.py:43
Functionpending_tool_calls
(state: AgentState)
src/agents/research_assistant.py:130
Functionpending_tool_calls
(state: AgentState)
src/agents/rag_assistant.py:126
Functionprepare_augmented_prompt
Prepare a prompt augmented with retrieved document content.
src/agents/knowledge_base_agent.py:124
Methodprovider
Get the provider name. Returns: Provider name string
src/voice/stt.py:107
Methodprovider
Get the provider name. Returns: Provider name string
src/voice/tts.py:115
Functionpytest_addoption
(parser)
tests/conftest.py:7
Functionpytest_collection_modifyitems
(config, items)
tests/conftest.py:17
Functionpytest_configure
(config)
tests/conftest.py:13
Functionretrieve_documents
Retrieve relevant documents from the knowledge base.
src/agents/knowledge_base_agent.py:85
Functionsafeguard_input
(state: AgentState, config: RunnableConfig)
src/agents/research_assistant.py:88
Functionsafeguard_input
(state: AgentState, config: RunnableConfig)
src/agents/rag_assistant.py:84
Methodsetup
(self)
tests/service/test_service_lifespan.py:19
Methodsetup
(self)
tests/service/test_service_lifespan.py:24
Functionstatic_messages
(state: MessagesState, writer: StreamWriter)
tests/service/test_service_e2e.py:71
Functionstream
Stream an agent's response to a user input, including intermediate messages and tokens. If agent_id is not provided, the default agent will
src/service/service.py:364
Functiontest_acreate_feedback
Test asynchronous feedback creation.
tests/client/test_client.py:241
Functiontest_agent_stream
Test that streaming from our static agent works correctly with token streaming.
tests/service/test_service_e2e.py:90
Functiontest_ainvoke
Test asynchronous invocation.
tests/client/test_client.py:85
Functiontest_app_feedback
TODO: Can't figure out how to interact with st.feedback
tests/app/test_streamlit_app.py:99
Functiontest_app_init_error
Test the app with an error in the agent initialization
tests/app/test_streamlit_app.py:148
Functiontest_app_new_chat_btn
(mock_agent_client)
tests/app/test_streamlit_app.py:167
Functiontest_app_settings
Test the full app - happy path
tests/app/test_streamlit_app.py:37
Functiontest_app_simple_non_streaming
Test the full app - happy path
tests/app/test_streamlit_app.py:12
Functiontest_app_streaming
Test the app with streaming enabled - including tool messages
tests/app/test_streamlit_app.py:106
Functiontest_app_streaming_nested_sub_agents
Test nested sub-agents where agent B is a sub-agent of agent A
tests/app/test_streamlit_app.py:477
Functiontest_app_streaming_sequential_sub_agents
Test when the supervisor agent transfers to sub agent A, then back to supervisor, then transfers to sub agent C, and back again
tests/app/test_streamlit_app.py:387
Functiontest_app_streaming_single_sub_agent
Test a single sub-agent with multiple tool calls to verify popover functionality
tests/app/test_streamlit_app.py:317
Functiontest_app_thread_id_history
Test the thread_id is generated
tests/app/test_streamlit_app.py:75
← previousnext →101–200 of 316, ranked by callers