Debugging shit

app.py

@@ -10,7 +10,6 @@ from typing import Dict, Any
 import autopep8
 import textwrap
 import json
-import asyncio
 
 from datasets import load_dataset
 from fastapi.responses import StreamingResponse
@@ -138,10 +137,7 @@ Here's the complete Python function implementation:
     def generate():
         
         try:
-            # Set a timeout of 2 minutes
-            timeout = asyncio.TimeoutError("LLM generation timed out")
-            with asyncio.TimeoutManager(timeout):
-                for chunk in llm(full_prompt, stream=True, **generation_kwargs):
+            for chunk in llm(full_prompt, stream=True, **generation_kwargs):
                     token = chunk["choices"][0]["text"]
                     generated_text += token
                     logger.info(f"Generated text: {generated_text}")
@@ -150,10 +146,6 @@ Here's the complete Python function implementation:
                     # Optionally send progress updates:
                     progress = len(generated_text) / len(full_prompt)
                     yield {"progress": progress}  # Send a progress update
-        
-        except asyncio.TimeoutError:
-            logger.error("LLM generation timed out")
-            yield {"error": "LLM generation timed out"}
         except Exception as e:
             logger.error(f"Error generating solution: {e}")
             yield {"error": "Error generating solution"}