Update custom_llm.py

custom_llm.py

@@ -210,6 +210,7 @@ def out_format(text, llm, df):
 
 
 def custom_dataframe_chain(llm, df):
+
     prompt = PromptTemplate.from_template("""<s><INST>You have access to a pandas dataframe variable named "df". Below are the examples of the dataframe:
 
     {df_example}
@@ -226,6 +227,10 @@ def custom_dataframe_chain(llm, df):
     
     
     Do not import pandas and Do not create or re-assign "df" variable
+
+
+    Below is the unique value of the important categorical columns:
+    1. 
     
     
     The output must follow the following example format:
@@ -239,7 +244,26 @@ def custom_dataframe_chain(llm, df):
 
     
 
+def custom_unique_df_chain(llm, df):
+    prompt = PromptTemplate("""<s><INST>You have access to a pandas dataframe variable named "df". Below are the examples of the dataframe:
 
+    {df_example}
+    
+    
+    Create unique values for the important non-datetime categorical columns with maximum 20 unique values for each columns. Store the unique values in a variable named "response" with the following format of list of python dictionaries: 
+    
+    [{column_name1: [list_of_unique_column1]}, {column_name2: [list_of_unique_values_column2]}]
+
+    
+    The output must follow the following example format:
+    ```python
+    # Generated Code
+    ```
+    
+    </INST></s>""", input_variables=["df_example"])
+
+
+    return prompt | llm | RunnableLambda(lambda x:out_format(x, llm, df))