Update custom_llm.py

custom_llm.py

@@ -270,7 +270,7 @@ def custom_unique_df_chain(llm, df):
 
 def custom_combined_chain(llm, df_chain, memory_chain):
 
-    prompt = PromptTemplate.from_templates("""<s><INST> Given the following question, classify it as either being more relevant with a dataframe object of ticket submissions' history or several documents of user guide and general knowledge:
+    prompt = PromptTemplate.from_template("""<s><INST> Given the following question, classify it as either being more relevant with a dataframe object of ticket submissions' history or several documents of user guide and general knowledge:
 
     <question>
     {question}
@@ -287,9 +287,9 @@ def custom_combined_chain(llm, df_chain, memory_chain):
     #         return memory_chain
 
     
-    # full_chain = RunnablePassthrough.assign(topic= (llm | prompt)) | RunnableLambda(route)
-
-    return RunnableBranch( (lambda x: "ticket" in x['topic'].lower(), df_chain), (lambda x: "dataframe" in x['topic'].lower(), df_chain), memory_chain )
+    # full_chain = RunnablePassthrough.assign(topic= (prompt | llm)) | RunnableLambda(route)
+    # combined_chain = prompt | llm
+    return RunnablePassthrough.assign(topic=prompt | llm) | RunnableBranch( (lambda x: "ticket" in x['topic'].lower(), df_chain), (lambda x: "dataframe" in x['topic'].lower(), df_chain), memory_chain )
 
 
 class CustomLLM(LLM):