Upload app.py
Browse files
app.py
CHANGED
|
@@ -116,6 +116,9 @@ def llm_query(user_input,company):
|
|
| 116 |
)
|
| 117 |
]
|
| 118 |
|
|
|
|
|
|
|
|
|
|
| 119 |
# Get response from the LLM
|
| 120 |
try:
|
| 121 |
# Call the chat model with the message
|
|
|
|
| 116 |
)
|
| 117 |
]
|
| 118 |
|
| 119 |
+
# Flatten the list of messages
|
| 120 |
+
prompt = [message for sublist in prompt for message in (sublist if isinstance(sublist, list) else [sublist])]
|
| 121 |
+
|
| 122 |
# Get response from the LLM
|
| 123 |
try:
|
| 124 |
# Call the chat model with the message
|