blaxel 0.1.10rc41__tar.gz → 0.1.11__tar.gz

{blaxel-0.1.10rc41 → blaxel-0.1.11}/.github/workflows/dev.yaml

@@ -3,7 +3,7 @@ name: Python SDK dev
 on:
   workflow_dispatch:
   push:
-    branches: ["main"]
+    branches: ["develop"]
 
 jobs:
   pypi-publish:

{blaxel-0.1.10rc41 → blaxel-0.1.11}/PKG-INFO

@@ -1,13 +1,12 @@
 Metadata-Version: 2.4
 Name: blaxel
-Version: 0.1.10rc41
+Version: 0.1.11
 Summary: Add your description here
 Author-email: cploujoux <cploujoux@blaxel.ai>
 License-File: LICENSE
 Requires-Python: >=3.10
 Requires-Dist: attrs>=21.3.0
 Requires-Dist: httpx>=0.27.0
-Requires-Dist: livekit-agents[anthropic,groq,openai]~=1.0
 Requires-Dist: mcp>=1.2.1
 Requires-Dist: opentelemetry-api>=1.28.0
 Requires-Dist: opentelemetry-exporter-otlp>=1.28.0
@@ -40,6 +39,8 @@ Requires-Dist: langchain-xai>=0.2.2; extra == 'langchain'
 Requires-Dist: langgraph<0.3.0,>=0.2.40; extra == 'langchain'
 Requires-Dist: opentelemetry-instrumentation-langchain>=0.35.0; extra == 'langchain'
 Requires-Dist: pillow>=10.0.0; extra == 'langchain'
+Provides-Extra: livekit
+Requires-Dist: livekit-agents[anthropic,groq,openai]~=1.0; extra == 'livekit'
 Provides-Extra: llamaindex
 Requires-Dist: llama-index-llms-anthropic>=0.6.10; extra == 'llamaindex'
 Requires-Dist: llama-index-llms-cerebras>=0.2.2; extra == 'llamaindex'

blaxel-0.1.11/integrationtest/load_test.py

@@ -0,0 +1,53 @@
+import asyncio
+import time
+from collections import Counter
+
+from blaxel.client import client
+from blaxel.common.settings import settings
+
+
+async def request_agent(name):
+    response = await client.get_async_httpx_client().get(
+        f"{settings.run_url}/{settings.workspace}/agents/{name}"
+    )
+    if response.status_code != 200:
+        raise Exception(f"Agent {name} returned status code {response.status_code} with body {response.text}")
+    return response.status_code
+
+async def main():
+    number_of_calls = 500
+    batch_size = 100
+    agent_name = "agent-telemetry-ts"
+
+    total_start_time = time.time()
+    batch_times = []
+
+    for batch_start in range(0, number_of_calls, batch_size):
+        batch_end = min(batch_start + batch_size, number_of_calls)
+        batch_tasks = [request_agent(agent_name) for _ in range(batch_start, batch_end)]
+        print(f"Processing batch {batch_start//batch_size + 1} ({batch_start}-{batch_end-1})")
+
+        batch_start_time = time.time()
+        status_codes = await asyncio.gather(*batch_tasks)
+        batch_time = time.time() - batch_start_time
+        batch_times.append(batch_time)
+
+        status_counts = Counter(status_codes)
+        print(f"Completed batch {batch_start//batch_size + 1}")
+        print(f"Status codes: {dict(status_counts)}")
+        print(f"Batch time: {batch_time:.2f}s")
+        print(f"Mean time per request in batch: {batch_time/batch_size:.3f}s")
+        print("-" * 50)
+
+    total_time = time.time() - total_start_time
+    mean_time_per_request = total_time / number_of_calls
+
+    print("\nSummary:")
+    print(f"Total execution time: {total_time:.2f}s")
+    print(f"Mean time per request: {mean_time_per_request:.3f}s")
+    print(f"Fastest batch: {min(batch_times):.2f}s")
+    print(f"Slowest batch: {max(batch_times):.2f}s")
+    print(f"Average batch time: {sum(batch_times)/len(batch_times):.2f}s")
+
+if __name__ == "__main__":
+    asyncio.run(main())

{blaxel-0.1.10rc41 → blaxel-0.1.11}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "blaxel"
-version = "0.1.10.pre-41"
+version = "0.1.11"
 description = "Add your description here"
 readme = "README.md"
 authors = [{ name = "cploujoux", email = "cploujoux@blaxel.ai" }]
@@ -9,8 +9,6 @@ dependencies = [
     "attrs>=21.3.0",
     "httpx>=0.27.0",
     "mcp>=1.2.1",
-    "livekit-agents[openai,groq,anthropic]~=1.0",
-
     # OpenTelemetry dependencies
     "opentelemetry-api>=1.28.0",
     "opentelemetry-exporter-otlp>=1.28.0",
@@ -66,6 +64,8 @@ pydantic-ai = ["pydantic-ai>=0.0.48"]
 
 google-adk = ["google-adk>=0.2.0", "litellm>=1.63.11"]
 
+livekit = ["livekit-agents[openai,groq,anthropic]~=1.0"]
+
 [dependency-groups]
 dev = [
     "asgi-correlation-id>=4.3.4,<5.0.0",