hanzoai/python-sdk
2
0
Fork 0
mirror of https://github.com/hanzoai/python-sdk.git synced 2026-10-23 20:23:00 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
v2.1.0
python-sdk/tests/mock_fixtures.py
T
zeekay a99730cbb0 feat: Add CLI tools support to hanzo-mcp with batch execution 
- Implement unified CLI tools module for all AI providers
- Add support for claude/cc, codex, gemini, grok, openhands/oh, hanzo-dev, cline, aider
- Enable parallel batch execution of CLI tools
- Add comprehensive test suite with 16 passing tests
- Fix typing issues with mypy strict mode
- Add GitHub Actions CI/CD workflow with optional integration tests
- Support authentication flow for all providers
- Follow DRY principle with single source of truth
2025-09-05 18:26:03 -05:00

561 lines
23 KiB
Python
Raw Permalink Blame History

# Mock fixtures for integration tests
import json
import respx
import pytest
from httpx import Response
@pytest.fixture
def mock_api(request):
"""Mock API responses for integration tests"""
# Don't set up mocks if the test is using respx_mock directly
if "respx_mock" in request.fixturenames:
yield None
return
with respx.mock(
base_url="http://127.0.0.1:4010", assert_all_called=False
) as respx_mock:
# Audio endpoints
respx_mock.post("/audio/speech").mock(return_value=Response(200, json={}))
respx_mock.post("/audio/transcriptions").mock(
return_value=Response(200, json={})
)
# Batch endpoints
respx_mock.post("/batch/cancel").mock(return_value=Response(200, json={}))
respx_mock.post("/batches").mock(
return_value=Response(200, json={"id": "batch-123"})
)
respx_mock.get("/batches").mock(return_value=Response(200, json={"data": []}))
respx_mock.route(method="GET", path__regex=r"/batches/.*").mock(
return_value=Response(200, json={"id": "batch-123"})
)
respx_mock.route(method="POST", path__regex=r"/batches/.*/cancel").mock(
return_value=Response(200, json={})
)
# Cache endpoints
respx_mock.get("/cache/ping").mock(
return_value=Response(
200,
json={
"status": "healthy",
"cache_type": "redis",
"ping_response": True,
"set_cache_response": None,
"llm_cache_params": None,
"health_check_cache_params": None,
},
)
)
respx_mock.post("/cache/delete").mock(return_value=Response(200, json={}))
respx_mock.post("/cache/flushall").mock(return_value=Response(200, json={}))
respx_mock.get("/cache/redis/ping").mock(
return_value=Response(200, json={"status": "ok"})
)
respx_mock.delete("/cache").mock(return_value=Response(200, json={}))
respx_mock.get("/cache/flush").mock(return_value=Response(200, json={}))
# Chat completions
respx_mock.post("/chat/completions").mock(
return_value=Response(
200,
json={
"id": "chatcmpl-123",
"object": "chat.completion",
"created": 1677652288,
"model": "gpt-3.5-turbo",
"choices": [
{
"index": 0,
"message": {"role": "assistant", "content": "Hello!"},
"finish_reason": "stop",
}
],
},
)
)
# Config endpoints
respx_mock.post("/config/pass_through_endpoints").mock(
return_value=Response(200, json={})
)
respx_mock.post("/config/pass_through_endpoint").mock(
return_value=Response(200, json={})
)
respx_mock.route(
method="PUT", path__regex=r"/config/pass_through_endpoints/.*"
).mock(return_value=Response(200, json={}))
respx_mock.get("/config/pass_through_endpoints").mock(
return_value=Response(200, json={"data": []})
)
respx_mock.get("/config/pass_through_endpoint").mock(
return_value=Response(200, json={"endpoints": []})
)
respx_mock.delete("/config/pass_through_endpoint").mock(
return_value=Response(200, json={"endpoints": []})
)
respx_mock.route(
method="DELETE", path__regex=r"/config/pass_through_endpoints/.*"
).mock(return_value=Response(200, json={}))
# Engine endpoints
respx_mock.route(
method="POST", path__regex=r"/engines/.*/chat/completions"
).mock(return_value=Response(200, json={}))
# Files
respx_mock.route(method="GET", path__regex=r"/files/.*/content").mock(
return_value=Response(200, json="file content")
)
respx_mock.route(method="GET", path__regex=r"/.*/v1/files/.*/content").mock(
return_value=Response(200, json="file content")
)
respx_mock.post("/files").mock(
return_value=Response(200, json={"id": "file-123"})
)
respx_mock.get("/files").mock(return_value=Response(200, json={"data": []}))
respx_mock.route(method="DELETE", path__regex=r"/files/.*").mock(
return_value=Response(200, json={})
)
# Fine-tuning
respx_mock.post("/fine_tuning/jobs").mock(
return_value=Response(200, json={"id": "ft-123"})
)
respx_mock.get("/fine_tuning/jobs").mock(
return_value=Response(200, json={"data": []})
)
respx_mock.route(method="GET", path__regex=r"/fine_tuning/jobs/[^/]+$").mock(
return_value=Response(200, json={"id": "ft-123"})
)
respx_mock.route(
method="POST", path__regex=r"/fine_tuning/jobs/.*/cancel"
).mock(return_value=Response(200, json={}))
# Global spend
respx_mock.get("/global/spend/tags").mock(return_value=Response(200, json=[]))
respx_mock.post("/global/spend/reset").mock(return_value=Response(200, json={}))
respx_mock.get("/global/spend/report").mock(return_value=Response(200, json=[]))
# Guardrails
respx_mock.get("/guardrails/list").mock(
return_value=Response(200, json={"guardrails": []})
)
# Images
respx_mock.post("/images/generations").mock(
return_value=Response(200, json={"data": []})
)
# Models
respx_mock.get("/model/info").mock(
return_value=Response(200, json={"data": []})
)
respx_mock.put("/models/update").mock(return_value=Response(200, json={}))
respx_mock.route(method="PATCH", path__regex=r"/models/.*").mock(
return_value=Response(200, json={})
)
respx_mock.post("/models").mock(return_value=Response(200, json={}))
respx_mock.get("/models").mock(return_value=Response(200, json={"data": []}))
respx_mock.route(method="DELETE", path__regex=r"/models/.*").mock(
return_value=Response(200, json={})
)
# OpenAI deployments
respx_mock.route(
method="POST", path__regex=r"/openai/deployments/.*/chat/completions"
).mock(return_value=Response(200, json={}))
respx_mock.route(
method="POST", path__regex=r"/openai/deployments/.*/completions"
).mock(return_value=Response(200, json={}))
respx_mock.route(
method="POST", path__regex=r"/openai/deployments/.*/embeddings"
).mock(return_value=Response(200, json={}))
# Organization
org_response = {
"budget_id": "budget-123",
"created_at": "2024-01-01T00:00:00Z",
"created_by": "user-123",
"models": ["gpt-3.5-turbo"],
"organization_id": "org-123",
"updated_at": "2024-01-01T00:00:00Z",
"updated_by": "user-123",
}
respx_mock.get("/organization/info").mock(
return_value=Response(
200,
json={
"budget_id": "budget-123",
"created_at": "2024-01-01T00:00:00Z",
"created_by": "user-123",
"models": ["gpt-3.5-turbo"],
"updated_at": "2024-01-01T00:00:00Z",
"updated_by": "user-123",
},
)
)
respx_mock.get("/organization/info/deprecated").mock(
return_value=Response(200, json={})
)
respx_mock.post("/organization/new").mock(
return_value=Response(200, json=org_response)
)
respx_mock.post("/organization").mock(return_value=Response(200, json={}))
respx_mock.get("/organization/list").mock(return_value=Response(200, json=[]))
respx_mock.get("/organization").mock(return_value=Response(200, json=[]))
org_update_response = {
"budget_id": "budget-123",
"created_at": "2024-01-01T00:00:00Z",
"created_by": "user-123",
"models": ["gpt-3.5-turbo"],
"updated_at": "2024-01-01T00:00:00Z",
"updated_by": "user-123",
"organization_id": "org-123",
"members": [],
"teams": [],
}
respx_mock.patch("/organization/update").mock(
return_value=Response(200, json=org_update_response)
)
respx_mock.put("/organization/update").mock(
return_value=Response(200, json=org_update_response)
)
respx_mock.put("/organization").mock(return_value=Response(200, json={}))
respx_mock.delete("/organization/delete").mock(
return_value=Response(200, json=[org_response])
)
respx_mock.delete("/organization").mock(return_value=Response(200, json={}))
org_member_response = {
"organization_id": "org-123",
"updated_organization_memberships": [],
"updated_users": [],
}
respx_mock.post("/organization/member_add").mock(
return_value=Response(200, json=org_member_response)
)
respx_mock.post("/organization/members").mock(
return_value=Response(200, json={})
)
respx_mock.delete("/organization/member_delete").mock(
return_value=Response(200, json=org_response)
)
respx_mock.delete("/organization/members").mock(
return_value=Response(200, json={})
)
org_update_member_response = {
"created_at": "2024-01-01T00:00:00Z",
"organization_id": "org-123",
"updated_at": "2024-01-01T00:00:00Z",
"user_id": "user-123",
}
respx_mock.patch("/organization/member_update").mock(
return_value=Response(200, json=org_update_member_response)
)
respx_mock.put("/organization/member_update").mock(
return_value=Response(200, json=org_update_member_response)
)
respx_mock.put("/organization/members").mock(
return_value=Response(200, json={})
)
# Responses
respx_mock.route(method="GET", path__regex=r"/responses/.*/input_items").mock(
return_value=Response(200, json={"data": []})
)
# Team
team_response = {
"team_id": "team-123",
"models": [],
"members": [],
"admins": [],
}
respx_mock.route(method="GET", path__regex=r"/team/.*/callback").mock(
return_value=Response(200, json={})
)
respx_mock.route(method="POST", path__regex=r"/team/.*/callback").mock(
return_value=Response(200, json={})
)
respx_mock.post("/team/model").mock(return_value=Response(200, json={}))
respx_mock.delete("/team/model").mock(return_value=Response(200, json={}))
respx_mock.post("/team/new").mock(
return_value=Response(200, json=team_response)
)
respx_mock.post("/team").mock(return_value=Response(200, json={}))
respx_mock.get("/team/list").mock(return_value=Response(200, json=[]))
respx_mock.get("/team").mock(return_value=Response(200, json=[]))
respx_mock.get("/team/info").mock(
return_value=Response(200, json=team_response)
)
respx_mock.patch("/team/update").mock(
return_value=Response(200, json=team_response)
)
respx_mock.put("/team/update").mock(
return_value=Response(200, json=team_response)
)
respx_mock.put("/team").mock(return_value=Response(200, json={}))
respx_mock.delete("/team/delete").mock(
return_value=Response(200, json=[team_response])
)
respx_mock.delete("/team").mock(return_value=Response(200, json={}))
team_add_member_response = {
"team_id": "team-123",
"updated_team_memberships": [],
"updated_users": [],
}
team_update_member_response = {"team_id": "team-123", "user_id": "user-123"}
respx_mock.post("/team/member_add").mock(
return_value=Response(200, json=team_add_member_response)
)
respx_mock.post("/team/members").mock(return_value=Response(200, json={}))
respx_mock.delete("/team/member_delete").mock(
return_value=Response(200, json=team_response)
)
respx_mock.delete("/team/members").mock(return_value=Response(200, json={}))
respx_mock.post("/team/member_update").mock(
return_value=Response(200, json=team_update_member_response)
)
respx_mock.patch("/team/member_update").mock(
return_value=Response(200, json=team_update_member_response)
)
respx_mock.put("/team/member_update").mock(
return_value=Response(200, json=team_update_member_response)
)
respx_mock.put("/team/members").mock(return_value=Response(200, json={}))
respx_mock.post("/team/block").mock(
return_value=Response(200, json=team_response)
)
respx_mock.post("/team/unblock").mock(
return_value=Response(200, json=team_response)
)
# Other endpoints
respx_mock.get("/active/callbacks").mock(
return_value=Response(200, json={"data": []})
)
respx_mock.post("/add/allowed_ip").mock(return_value=Response(200, json={}))
respx_mock.post("/delete/allowed_ip").mock(return_value=Response(200, json={}))
# Provider endpoints (anthropic, azure, bedrock, etc.)
for provider in [
"anthropic",
"azure",
"bedrock",
"cohere",
"gemini",
"openai",
"vertex_ai",
"assemblyai",
"eu_assemblyai",
]:
respx_mock.post(f"/{provider}").mock(return_value=Response(200, json={}))
respx_mock.get(f"/{provider}").mock(
return_value=Response(200, json={"data": []})
)
respx_mock.route(method="GET", path__regex=rf"/{provider}/.*").mock(
return_value=Response(200, json={})
)
respx_mock.put(f"/{provider}").mock(return_value=Response(200, json={}))
respx_mock.route(method="PUT", path__regex=rf"/{provider}/.*").mock(
return_value=Response(200, json={})
)
respx_mock.patch(f"/{provider}").mock(return_value=Response(200, json={}))
respx_mock.route(method="PATCH", path__regex=rf"/{provider}/.*").mock(
return_value=Response(200, json={})
)
respx_mock.delete(f"/{provider}").mock(return_value=Response(200, json={}))
respx_mock.route(method="DELETE", path__regex=rf"/{provider}/.*").mock(
return_value=Response(200, json={})
)
respx_mock.post(f"/{provider}/modify").mock(
return_value=Response(200, json={})
)
respx_mock.post(f"/{provider}/call").mock(
return_value=Response(200, json={})
)
# Assistants
respx_mock.post("/assistants").mock(return_value=Response(200, json={}))
respx_mock.get("/assistants").mock(
return_value=Response(200, json={"data": []})
)
respx_mock.route(method="DELETE", path__regex=r"/assistants/.*").mock(
return_value=Response(200, json={})
)
# Budget
respx_mock.post("/budget").mock(return_value=Response(200, json={}))
respx_mock.put("/budget").mock(return_value=Response(200, json={}))
respx_mock.get("/budget").mock(return_value=Response(200, json={"data": []}))
respx_mock.delete("/budget").mock(return_value=Response(200, json={}))
respx_mock.get("/budget/info").mock(return_value=Response(200, json={}))
respx_mock.get("/budget/settings").mock(return_value=Response(200, json={}))
# Completions
respx_mock.post("/completions").mock(return_value=Response(200, json={}))
# Credentials
respx_mock.post("/credentials").mock(return_value=Response(200, json={}))
respx_mock.put("/credentials").mock(return_value=Response(200, json={}))
# Customer
respx_mock.post("/customer").mock(return_value=Response(200, json={}))
respx_mock.get("/customer").mock(return_value=Response(200, json={"data": []}))
respx_mock.get("/customer/list").mock(return_value=Response(200, json=[]))
respx_mock.get("/customer/info").mock(
return_value=Response(200, json={"blocked": False, "user_id": "user-123"})
)
respx_mock.put("/customer").mock(return_value=Response(200, json={}))
respx_mock.delete("/customer").mock(return_value=Response(200, json={}))
respx_mock.post("/customer/block").mock(return_value=Response(200, json={}))
respx_mock.post("/customer/unblock").mock(return_value=Response(200, json={}))
# Embeddings
respx_mock.post("/embeddings").mock(
return_value=Response(200, json={"data": []})
)
# Guardrails
respx_mock.get("/guardrails").mock(
return_value=Response(200, json={"data": []})
)
# Health
respx_mock.get("/health").mock(
return_value=Response(200, json={"status": "healthy"})
)
respx_mock.get("/health/services").mock(return_value=Response(200, json={}))
# Key management
respx_mock.post("/key").mock(return_value=Response(200, json={}))
respx_mock.get("/key").mock(return_value=Response(200, json={"data": []}))
respx_mock.get("/key/info").mock(return_value=Response(200, json={}))
respx_mock.put("/key").mock(return_value=Response(200, json={}))
respx_mock.delete("/key").mock(return_value=Response(200, json={}))
respx_mock.post("/key/generate").mock(
return_value=Response(200, json={"key": "sk-123"})
)
respx_mock.post("/key/regenerate").mock(
return_value=Response(200, json={"key": "sk-456"})
)
respx_mock.post("/key/key/regenerate").mock(
return_value=Response(200, json={"key": "sk-789"})
)
respx_mock.post("/key/block").mock(return_value=Response(200, json={}))
respx_mock.post("/key/unblock").mock(return_value=Response(200, json={}))
respx_mock.get("/key/health").mock(
return_value=Response(200, json={"status": "ok"})
)
# Langfuse
respx_mock.post("/langfuse").mock(return_value=Response(200, json={}))
# Model group
respx_mock.get("/model_group/info").mock(return_value=Response(200, json={}))
# Moderations
respx_mock.post("/moderations").mock(return_value=Response(200, json={}))
# Provider
respx_mock.get("/provider/budgets").mock(
return_value=Response(200, json={"data": []})
)
# Rerank
respx_mock.post("/rerank").mock(return_value=Response(200, json={}))
# Routes
respx_mock.get("/routes").mock(return_value=Response(200, json={"data": []}))
# Settings
respx_mock.get("/settings").mock(return_value=Response(200, json={}))
# Spend
respx_mock.get("/spend/calculate").mock(return_value=Response(200, json={}))
respx_mock.get("/spend/logs").mock(return_value=Response(200, json=[]))
respx_mock.get("/spend/tags").mock(return_value=Response(200, json=[]))
# Test
respx_mock.get("/test").mock(return_value=Response(200, json={}))
# Root endpoint for client tests
respx_mock.get("/").mock(return_value=Response(200, json={}))
# Threads
respx_mock.post("/threads").mock(return_value=Response(200, json={}))
respx_mock.get("/threads").mock(return_value=Response(200, json={"data": []}))
respx_mock.route(method="GET", path__regex=r"/threads/[^/]+$").mock(
return_value=Response(200, json={})
)
respx_mock.route(method="PUT", path__regex=r"/threads/.*").mock(
return_value=Response(200, json={})
)
respx_mock.route(method="DELETE", path__regex=r"/threads/.*").mock(
return_value=Response(200, json={})
)
respx_mock.route(method="POST", path__regex=r"/threads/.*/messages").mock(
return_value=Response(200, json={})
)
respx_mock.route(method="GET", path__regex=r"/threads/.*/messages").mock(
return_value=Response(200, json={"data": []})
)
respx_mock.route(method="POST", path__regex=r"/threads/.*/runs").mock(
return_value=Response(200, json={})
)
respx_mock.route(method="GET", path__regex=r"/threads/.*/runs").mock(
return_value=Response(200, json={"data": []})
)
# User
user_response = {"key": "sk-user-123", "user_id": "user-123"}
respx_mock.post("/user/new").mock(
return_value=Response(200, json=user_response)
)
respx_mock.post("/user").mock(return_value=Response(200, json={}))
respx_mock.get("/user/list").mock(return_value=Response(200, json=[]))
respx_mock.get("/user").mock(return_value=Response(200, json=[]))
respx_mock.get("/user/info").mock(
return_value=Response(200, json=user_response)
)
respx_mock.put("/user/update").mock(
return_value=Response(200, json=user_response)
)
respx_mock.put("/user").mock(return_value=Response(200, json={}))
respx_mock.delete("/user/delete").mock(
return_value=Response(200, json=user_response)
)
respx_mock.delete("/user").mock(return_value=Response(200, json={}))
# Utils
respx_mock.get("/utils/supported_openai_params").mock(
return_value=Response(200, json={})
)
respx_mock.post("/utils/token_counter").mock(
return_value=Response(
200,
json={
"model_used": "gpt-3.5-turbo",
"request_model": "gpt-3.5-turbo",
"tokenizer_type": "cl100k_base",
"total_tokens": 10,
},
)
)
respx_mock.post("/utils/transform_request").mock(
return_value=Response(200, json={})
)
# Home/client
respx_mock.get("/").mock(
return_value=Response(200, json={"message": "Welcome"})
)
# Catch-all for any unmocked endpoints
respx_mock.route().mock(return_value=Response(200, json={}))
yield respx_mock
Reference in New Issue View Git Blame Copy Permalink