Fix the API example with streaming #417

2023-03-19 10:36:57 -03:00
parent c753261338
commit c79fc69e95
1 changed files with 2 additions and 2 deletions
--- a/api-example-stream.py
+++ b/api-example-stream.py
@@ -44,14 +44,14 @@ async def run(context):
                case "send_hash":
                    await websocket.send(json.dumps({
                        "session_hash": session,
-                        "fn_index": 7
+                        "fn_index": 9
                    }))
                case "estimation":
                    pass
                case "send_data":
                    await websocket.send(json.dumps({
                        "session_hash": session,
-                        "fn_index": 7,
+                        "fn_index": 9,
                        "data": [
                            context,
                            params['max_new_tokens'],