Update custom_llm.py

custom_llm.py

@@ -1,11 +1,12 @@
+import os
 import cohere
 from langchain.llms.base import LLM
 
-co = cohere.Client('3FZy1Q1sd9Yfs289shdPgwjAt4MEuMxUmQx3oecK')  # This is my trial API key
+co = cohere.Client(os.environ['COHERE_API_KEY'])  # This is my trial API key
 
 # 24222433-2a40-410c-af5c-65a5422282fb-ft
 class CustomLLM(LLM):
-    model: str = '24222433-2a40-410c-af5c-65a5422282fb-ft'   # The custom model we used.
+    model: str = os.environ['CODE_GEN_MODEL_ID']   # The custom model we used.
 
     @property
     def _llm_type(self) -> str: