Update app.py

app.py

@@ -98,7 +98,6 @@ class BasicAgent:
             tokenizer=tok,
             max_new_tokens=512,
             temperature=1.0,
-            do_sample=False
         )
         self.agent = CodeAgent(model=self, tools=[], add_base_tools=True)
         
@@ -138,7 +137,7 @@ class BasicAgent:
         
         # wrap back into a chat message dict
         return {
-            "role": MessageRole.ASSISTANT,
+            "role": 'assistant',
             "content": [{"type": "text", "text": response}],
         }