fix: google models do not support system message
Browse files
app.py
CHANGED
|
@@ -47,7 +47,8 @@ class RAGDemo(object):
|
|
| 47 |
elif 'gemini' in model_name:
|
| 48 |
self.chat_model = ChatGoogleGenerativeAI(
|
| 49 |
google_api_key=api_key,
|
| 50 |
-
model='gemini-pro'
|
|
|
|
| 51 |
)
|
| 52 |
|
| 53 |
def _init_embedding(self, embedding_model_name, api_key):
|
|
|
|
| 47 |
elif 'gemini' in model_name:
|
| 48 |
self.chat_model = ChatGoogleGenerativeAI(
|
| 49 |
google_api_key=api_key,
|
| 50 |
+
model='gemini-pro',
|
| 51 |
+
convert_system_message_to_human=True,
|
| 52 |
)
|
| 53 |
|
| 54 |
def _init_embedding(self, embedding_model_name, api_key):
|