diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/umapyai/__init__.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/umapyai/__init__.py b/src/umapyai/__init__.py index 4eadd0f..b124579 100644 --- a/src/umapyai/__init__.py +++ b/src/umapyai/__init__.py @@ -44,7 +44,7 @@ def prompt(rag_context, user_query, chat_history): "If the context does not contain enough information to give a complete answer, " "say so, but still provide any relevant information you did find.") history_string = "\n".join( - [f"{msg['role']}: {msg['content']}" for msg in chat_history]) + [f"{message['role']}: {message['content']}" for message in chat_history]) return ( f"{system_prompt}\n\n" |