add index in non streamin chat too

rayonlabs · Jan 7, 2025 · dad8181 · dad8181
1 parent b253de6
commit dad8181
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/validator/organic_node/src/endpoints/text.py b/validator/organic_node/src/endpoints/text.py
@@ -171,6 +171,8 @@ async def _handle_nonstream_response(generator: AsyncGenerator[str, None], chat=
             {
             "choices": [
                 {
+                "index": 0,
+                "finish_reason": "stop",
                 "message": {
                     "content": all_content,
                     "role": role