push demo

llm.py

@@ -2,7 +2,9 @@ import os
 
 from orq_ai_sdk import OrqAI
 
-import config
+import logging
+
+LOGGER = logging.getLogger(__name__)
 
 client = OrqAI(api_key=os.environ["ORQ_API_KEY"], environment="develop")
 
@@ -19,10 +21,10 @@ def stream_request(variant: str, secret: str, user_input: str):
         context={"step": variant},  # , "environments": []},
         inputs={"secret": secret, "user_input": user_input},
     )
-    print(stream)
+    LOGGER.info(stream)
 
     for chunk in stream:
-        print(chunk)
+        LOGGER.info(chunk)
         if not chunk.is_final:
             yield chunk.choices[0].message.content
 
@@ -52,7 +54,7 @@ def run_judge(level: int, inputs: dict):
         context={"step": special_checks[level]},
         inputs=inputs,
     )
-    print(generation.choices[0].message.content)
+    LOGGER.info(generation.choices[0].message.content)
     answer = generation.choices[0].message.content.split(" ")[-1]
     return answer.lower() == "yes"