From da68926e9c5204940436c4680abe4a6f1ad834a5 Mon Sep 17 00:00:00 2001
From: josc146 <josStorer@outlook.com>
Date: Sun, 13 Aug 2023 21:27:29 +0800
Subject: [PATCH] chore (AddStateBody class)

---
 backend-python/routes/state_cache.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/backend-python/routes/state_cache.py b/backend-python/routes/state_cache.py
index 74d6f65..62d5570 100644
--- a/backend-python/routes/state_cache.py
+++ b/backend-python/routes/state_cache.py
@@ -1,4 +1,4 @@
-from typing import Any, Dict, List
+from typing import Any, Dict, List, Union
 from utils.log import quick_log
 from fastapi import APIRouter, HTTPException, Request, Response, status
 from pydantic import BaseModel
@@ -60,7 +60,7 @@ def enable_state_cache():
 
 class AddStateBody(BaseModel):
     prompt: str
-    tokens: List[str]
+    tokens: List[Union[str, int]]
     state: Any
     logits: Any