Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/JoshuaC215/agent-service-toolkit
/ functions
Functions
316 in github.com/JoshuaC215/agent-service-toolkit
⨍
Functions
316
◇
Types & classes
60
↳
Endpoints
8
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
Function
acall_model
(state: AgentState, config: RunnableConfig)
src/agents/research_assistant.py:70
Function
acall_model
(state: AgentState, config: RunnableConfig)
src/agents/rag_assistant.py:66
Function
acall_model
Generate a response based on the retrieved documents.
src/agents/knowledge_base_agent.py:147
Function
acall_model
(state: AgentState, config: RunnableConfig)
src/agents/bg_task_agent/bg_task_agent.py:28
Function
add
Add two numbers.
src/agents/langgraph_supervisor_agent.py:11
Function
agent_client
Fixture for creating a test client with a clean environment.
tests/client/conftest.py:7
Function
agent_lookup
(agent_id)
tests/service/test_service_e2e.py:99
Function
agent_lookup
(agent_id)
tests/service/test_service.py:48
Function
background
This node is to demonstrate doing work before the interrupt
src/agents/interrupt_agent.py:47
Function
bg_task
(state: AgentState, writer: StreamWriter)
src/agents/bg_task_agent/bg_task_agent.py:37
Method
bind_tools
(self, tools, **kwargs)
tests/service/test_service_message_generator.py:18
Function
block_unsafe_content
(state: AgentState, config: RunnableConfig)
src/agents/research_assistant.py:94
Function
block_unsafe_content
(state: AgentState, config: RunnableConfig)
src/agents/rag_assistant.py:90
Function
calculator_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
Function
chatbot
( inputs: dict[str, list[BaseMessage]], *, previous: dict[str, list[BaseMessage]], config: Run
src/agents/chatbot.py:9
Function
check_safety
(state: AgentState)
src/agents/research_assistant.py:109
Function
check_safety
(state: AgentState)
src/agents/rag_assistant.py:105
Function
create_system_message
(state)
src/agents/knowledge_base_agent.py:50
Function
custom_generate_unique_id
Generate idiomatic operation IDs for OpenAPI client generation.
src/service/service.py:48
Function
database_search_func
Searches chroma_db for information in the company's handbook.
src/agents/tools.py:65
Function
determine_birthdate
This node examines the conversation history to determine user's birthdate, checking store first.
src/agents/interrupt_agent.py:77
Function
fake_get_agent
(agent_key: str)
tests/service/test_service_lifespan.py:48
Function
fake_initialize_database
()
tests/service/test_service_lifespan.py:32
Function
fake_initialize_store
()
tests/service/test_service_lifespan.py:36
Function
fake_load_agent
(agent_key: str)
tests/service/test_service_lifespan.py:44
Function
feedback
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
Method
from_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
Method
from_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
Function
generate_response
Generates the final response based on the user's query and the available birthdate.
src/agents/interrupt_agent.py:190
Method
get_format
Get audio format (MIME type). Returns: MIME type string for generated audio
src/voice/providers/openai_tts.py:118
Function
health_check
Health check endpoint.
src/service/service.py:421
Function
history
Get chat history.
src/service/service.py:402
Function
info
()
src/service/service.py:114
Function
invoke
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
Function
lifespan
Configurable lifespan that initializes the appropriate database checkpointer, store, and agents with async loading - for example for starting
src/service/service.py:67
Function
mock_agent
Fixture to create a mock agent that can be configured for different test scenarios.
tests/service/conftest.py:17
Function
mock_agent_client
Fixture for creating a mock AgentClient with a clean environment.
tests/app/conftest.py:10
Function
mock_astream
(**kwargs)
tests/service/test_service.py:239
Function
mock_audio_file
BytesIO mock audio file for STT tests.
tests/voice/conftest.py:26
Function
mock_database_settings
Fixture to ensure database settings are clean
tests/service/test_service_e2e.py:84
Function
mock_env
Fixture to ensure environment is clean for each test.
tests/conftest.py:26
Function
mock_get
(url: str, **kwargs)
tests/service/conftest.py:46
Function
mock_httpx
Patch httpx.stream and httpx.get to use our test client.
tests/service/conftest.py:36
Function
mock_openai_client
Mock OpenAI client for TTS/STT tests.
tests/voice/conftest.py:10
Function
mock_settings
Fixture to ensure settings are clean for each test.
tests/service/conftest.py:29
Function
mock_stream
(method: str, url: str, **kwargs)
tests/service/conftest.py:41
Method
model_post_init
(self, __context: Any)
src/core/settings.py:155
Function
multi_agent_messages
Fixture providing reusable messages for multi-agent tests
tests/app/test_streamlit_app.py:178
Function
multiply
Multiply two numbers.
src/agents/langgraph_supervisor_agent.py:16
Function
node_a
(state: AgentState)
src/agents/command_agent.py:19
Function
node_b
(state: AgentState)
src/agents/command_agent.py:38
Function
node_c
(state: AgentState)
src/agents/command_agent.py:43
Function
pending_tool_calls
(state: AgentState)
src/agents/research_assistant.py:130
Function
pending_tool_calls
(state: AgentState)
src/agents/rag_assistant.py:126
Function
prepare_augmented_prompt
Prepare a prompt augmented with retrieved document content.
src/agents/knowledge_base_agent.py:124
Method
provider
Get the provider name. Returns: Provider name string
src/voice/stt.py:107
Method
provider
Get the provider name. Returns: Provider name string
src/voice/tts.py:115
Function
pytest_addoption
(parser)
tests/conftest.py:7
Function
pytest_collection_modifyitems
(config, items)
tests/conftest.py:17
Function
pytest_configure
(config)
tests/conftest.py:13
Function
retrieve_documents
Retrieve relevant documents from the knowledge base.
src/agents/knowledge_base_agent.py:85
Function
safeguard_input
(state: AgentState, config: RunnableConfig)
src/agents/research_assistant.py:88
Function
safeguard_input
(state: AgentState, config: RunnableConfig)
src/agents/rag_assistant.py:84
Method
setup
(self)
tests/service/test_service_lifespan.py:19
Method
setup
(self)
tests/service/test_service_lifespan.py:24
Function
static_messages
(state: MessagesState, writer: StreamWriter)
tests/service/test_service_e2e.py:71
Function
stream
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
Function
test_acreate_feedback
Test asynchronous feedback creation.
tests/client/test_client.py:241
Function
test_agent_stream
Test that streaming from our static agent works correctly with token streaming.
tests/service/test_service_e2e.py:90
Function
test_ainvoke
Test asynchronous invocation.
tests/client/test_client.py:85
Function
test_app_feedback
TODO: Can't figure out how to interact with st.feedback
tests/app/test_streamlit_app.py:99
Function
test_app_init_error
Test the app with an error in the agent initialization
tests/app/test_streamlit_app.py:148
Function
test_app_new_chat_btn
(mock_agent_client)
tests/app/test_streamlit_app.py:167
Function
test_app_settings
Test the full app - happy path
tests/app/test_streamlit_app.py:37
Function
test_app_simple_non_streaming
Test the full app - happy path
tests/app/test_streamlit_app.py:12
Function
test_app_streaming
Test the app with streaming enabled - including tool messages
tests/app/test_streamlit_app.py:106
Function
test_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
Function
test_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
Function
test_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
Function
test_app_thread_id_history
Test the thread_id is generated
tests/app/test_streamlit_app.py:75
← previous
next →
101–200 of 316, ranked by callers