Update custom_llm.py

I am doing stupid things

custom_llm.py

@@ -20,7 +20,7 @@ class CustomLLM(LLM):
         response = co.chat(message=f'{prompt}', max_tokens=1700, temperature=0.9)
         print(response.text)
 
-        return f' ```python \n {response.generations[0].text} \n ``` '
+        return f' ```python \n {response.text} \n ``` '
 
     @property
     def _identifying_params(self):