chore: Additional load test logging

inferablehq · Dec 2, 2024 · 50ce8a7 · 50ce8a7
1 parent b9bbc54
commit 50ce8a7
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1 deletion.
diff --git a/load-tests/entrypoint.sh b/load-tests/entrypoint.sh
@@ -14,5 +14,6 @@ done
 # Wait for all instances to finish
 for pid in "${PIDS[@]}"; do
   wait $pid
+  echo "Instance $pid finished"
 done
 
diff --git a/load-tests/machine.ts b/load-tests/machine.ts
@@ -8,7 +8,8 @@ const client = new Inferable({
 })
 
 client.default.register({
-  func: () => {
+  func: (_, context) => {
+    console.log("Handling request", context)
     return {
       word: "needle"
     }

diff --git a/load-tests/script.js b/load-tests/script.js
@@ -26,6 +26,7 @@ export default function () {
   // Create a new run
   const postRunResponse = http.post(`${BASE_URL}/clusters/${CLUSTER_ID}/runs`, JSON.stringify({
     initialPrompt: 'Get the special word from the `searchHaystack` function',
+    reasoningTraces: false,
     attachedFunctions: [
       {
         function: "searchHaystack",