cast to int and float

pAI-OS · Sep 18, 2024 · 3785c8b · 3785c8b
1 parent de9f870
commit 3785c8b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/backend/managers/MessagesManager.py b/backend/managers/MessagesManager.py
@@ -26,8 +26,8 @@ def __init__(self):
             with self._lock:
                 if not hasattr(self, '_initialized'):
                     self._initialized = True
-                    self.max_tokens = os.environ.get('MAX_TOKENS')
-                    self.temperature = os.environ.get('TEMPERATURE')
+                    self.max_tokens = int(os.environ.get('MAX_TOKENS'))
+                    self.temperature = float(os.environ.get('TEMPERATURE'))
 
     async def __get_llm_name__(self, assistant_id) -> Tuple[Optional[str], Optional[str]]:
         async with db_session_context() as session: