Update chat.py
Browse files
chat.py
CHANGED
|
@@ -13,6 +13,6 @@ def langchainConversation(conversation):
|
|
| 13 |
def converse(conversation,provider,model,key):
|
| 14 |
if(provider=='groq'):
|
| 15 |
chat = ChatGroq(temperature=0, groq_api_key=key, model_name=model)
|
| 16 |
-
elif(provider=='
|
| 17 |
chat = ChatGoogleGenerativeAI(model=model,google_api_key=key)
|
| 18 |
return json.dumps(json.loads(chat.invoke(langchainConversation(conversation)).json()),indent=4)
|
|
|
|
| 13 |
def converse(conversation,provider,model,key):
|
| 14 |
if(provider=='groq'):
|
| 15 |
chat = ChatGroq(temperature=0, groq_api_key=key, model_name=model)
|
| 16 |
+
elif(provider=='gemini'):
|
| 17 |
chat = ChatGoogleGenerativeAI(model=model,google_api_key=key)
|
| 18 |
return json.dumps(json.loads(chat.invoke(langchainConversation(conversation)).json()),indent=4)
|