Update custom_llm.py

update from legacy API to Chat API

custom_llm.py

@@ -17,7 +17,7 @@ class CustomLLM(LLM):
         if stop is not None:
             raise ValueError("stop kwargs are not permitted.")
 
-        response = co.generate(model=self.model, prompt=f'{prompt}', max_tokens=1700, temperature=0.9)
+        response = co.chat(message=f'{prompt}', max_tokens=1700, temperature=0.9)
         print(response.generations[0].text)
 
         return f' ```python \n {response.generations[0].text} \n ``` '