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)
|