Update custom_llm.py

custom_llm.py

@@ -276,7 +276,7 @@ def custom_combined_chain(llm, df_chain, memory_chain):
     {question}
     </question>
 
-    Do not respond with more than two words.
+    Respond with ONLY one word either "ticket" or "knowledge"
     
     </s></INST>""")
     
@@ -289,7 +289,7 @@ def custom_combined_chain(llm, 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 )
+    return RunnablePassthrough.assign(topic=prompt | llm) | RunnableBranch( (lambda x: "ticket" in x['topic'].lower(), df_chain), memory_chain )
 
 
 class CustomLLM(LLM):